lib/lookbook/parser.rb in lookbook-1.0.0.beta.6 vs lib/lookbook/parser.rb in lookbook-1.0.0.beta.7

@@ -22,16 +22,19 @@
       registry.load!(registry_path)
       registry.get(path)
     end
 
     class << self
-      def define_tags
+      def define_tags(custom = {})
         YARD::Tags::Library.define_tag("Hidden status", :hidden)
         YARD::Tags::Library.define_tag("Label", :label)
         YARD::Tags::Library.define_tag("Display", :display)
         YARD::Tags::Library.define_tag("Position", :position)
         YARD::Tags::Library.define_tag("ID", :id)
         YARD::Tags::Library.define_tag("Component", :component)
+        custom.each do |name, opts|
+          YARD::Tags::Library.define_tag(name.to_s.titleize, name)
+        end
       end
     end
   end
 end