Adds ability to include module for an icon helper

README.md

@@ -27,6 +27,23 @@ require "protos-icon"
 render Protos::Icon.heroicon("academic-cap", variant: :mini)
 ```
 
+Or you can include the module in a class and render them as an element:
+
+```ruby
+class MyComponent < Phlex::HTML
+  include Protos::Icon
+
+  def template
+    div(class: "w-6 h-6") do
+      icon("academic-cap", variant: :mini)
+    end
+  end
+end
+```
+
+This will call render on the icon component and output it directly to the
+component.
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run

lib/protos/icon.rb

@@ -48,5 +48,9 @@ def self.lookup(name, variant: :solid)
     def self.heroicon(name, variant: :solid)
       Heroicons.new(name, variant:)
     end
+
+    def icon(...)
+      render Protos::Icon.heroicon(...)
+    end
   end
 end

spec/protos/icon_spec.rb

@@ -1,5 +1,13 @@
 # frozen_string_literal: true
 
+TestComponent = Class.new(Phlex::HTML) do
+  include Protos::Icon
+
+  def template
+    icon("academic-cap", variant: :mini)
+  end
+end
+
 RSpec.describe Protos::Icon do
   it "has a version number" do
     expect(Protos::Icon::VERSION).not_to be nil
@@ -12,4 +20,11 @@
     output = component.call
     expect(output).to eq(svg)
   end
+
+  it "can be included in a component" do
+    component = TestComponent.new
+    output = component.call
+
+    expect(output.start_with?("<svg")).to be true
+  end
 end