Add classes class methods

Author: koppen

app/components/flowbite/badge.rb

@@ -28,6 +28,14 @@ class Badge < ViewComponent::Base
     }.freeze
 
     class << self
+      def classes(size: :default, state: :default, style: :brand)
+        styles.fetch(style).fetch(state) + sizes.fetch(size)
+      end
+
+      def sizes
+        SIZES
+      end
+
       # rubocop:disable Layout/LineLength
       def styles
         Flowbite::Styles.from_hash({
@@ -89,16 +97,12 @@ def link?
     private
 
     def classes
-      result = self.class.styles.fetch(@style).fetch(:default) + size_classes
+      result = self.class.classes(size: @size, state: :default, style: @style)
       result += BORDER_CLASSES.fetch(@style) if bordered?
       result += ["inline-flex", "items-center"] if dot?
       result + @class
     end
 
-    def size_classes
-      SIZES.fetch(@size)
-    end
-
     def tag_name
       link? ? :a : :span
     end

app/components/flowbite/badge/dot.rb

@@ -16,14 +16,20 @@ class Dot < ViewComponent::Base
         warning: ["bg-fg-warning", "h-1.5", "me-1", "rounded-full", "w-1.5"]
       }.freeze
 
+      class << self
+        def classes(style: :brand)
+          CLASSES.fetch(style)
+        end
+      end
+
       attr_reader :style
 
       def initialize(style: :brand)
         @style = style
       end
 
       def call
-        content_tag(:span, nil, class: CLASSES.fetch(style))
+        content_tag(:span, nil, class: self.class.classes(style: style))
       end
     end
   end