From 694e8a1a8c7e5c39792825954b6678953529acd4 Mon Sep 17 00:00:00 2001
From: Bradley <bg@pcmsystems.co.uk>
Date: Thu, 8 Aug 2024 15:34:25 +0100
Subject: [PATCH] Test adding colour parameter to override object colour

---
 src/Twig/Component/Badge.php | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/Twig/Component/Badge.php b/src/Twig/Component/Badge.php
index 633eceb..3baf22a 100644
--- a/src/Twig/Component/Badge.php
+++ b/src/Twig/Component/Badge.php
@@ -4,6 +4,7 @@ declare(strict_types=1);
 
 namespace Pcm\BadgeBundle\Twig\Component;
 
+use Pcm\BadgeBundle\Enum\Badge as EnumBadge;
 use Pcm\BadgeBundle\Interface\BadgeableInterface;
 use Symfony\UX\TwigComponent\Attribute\AsTwigComponent;
 use TailwindMerge\TailwindMerge;
@@ -11,7 +12,9 @@ use TailwindMerge\TailwindMerge;
 #[AsTwigComponent(name: 'Pcm:Badge', template: "@PcmBadge/Badge.html.twig")]
 final class Badge
 {
-    public string $finalClasses = '';
+    public string $finalClasses;
+
+    private ?string $colour;
 
     public function __construct(private string $baseClasses)
     {
@@ -21,9 +24,17 @@ final class Badge
      * @param BadgeableInterface $object The object to be converted into a badge
      * @param string $class              Extra classes you want to add to the badge template
      */
-    public function mount(BadgeableInterface $object, string $class = ''): void
+    public function mount(BadgeableInterface $object, string $class = '', string $colour = null): void
     {
-        $palette = $object->getBadgeColour()->getPalette();
+        if (null !== $colour) {
+
+            if (!in_array($colour, EnumBadge::cases())) {
+                throw new \RuntimeException(sprintf('"%s" is not a valid Badge colour. Available options are: %s', $colour, implode(", ", EnumBadge::cases())));
+            }
+
+        } else {
+            $palette = $object->getBadgeColour()->getPalette();
+        }
 
         $merger = TailwindMerge::instance();