rendering.rb in brief-1.2.0

- old
+ new
@@ -1,37 +1,67 @@
 module Brief
   class Document
     module Rendering
       extend ActiveSupport::Concern
 
-      def to_html
-        self.class.renderer.render(content)
+      # Uses a custom Redcarpet::Render::HTML subclass
+      # which simply inserts data attributes on each heading element
+      # so that they can be queried with CSS more deliberately.
+      class HeadingWrapper < ::Redcarpet::Render::HTML
+        def header(text, level)
+          "<h#{level} data-level='#{level}' data-heading='#{ text }'>#{text}</h#{level}>"
+        end
       end
 
-      def css(*args, &block)
-        parser.send(:css, *args, &block)
-      end
-
-      def at(*args, &block)
-        parser.send(:at, *args, &block)
-      end
-
-      def parser
-        @parser ||= Nokogiri::HTML.fragment(to_html)
-      end
-
       module ClassMethods
+        def renderer_class
+          HeadingWrapper
+        end
+
         def renderer
           @renderer ||= begin
-                          r = ::Redcarpet::Render::HTML.new(:tables => true,
+                          r = renderer_class.new(:tables => true,
                                             :autolink => true,
                                             :gh_blockcode => true,
                                             :fenced_code_blocks => true,
                                             :footnotes => true)
 
                           ::Redcarpet::Markdown.new(r)
                         end
         end
+      end
+
+      # Documents can be rendered into HTML.
+      #
+      # They will first be put through a Nokogiri processor pipeline
+      # which allows us to wrap things in section containers, apply data
+      # attributes, and other things to the HTML so that the output HTML retains its
+      # relationship to the underlying data and document structure.
+      def to_html(options={})
+        if options[:wrap] == false
+          unwrapped_html
+        else
+          wrapper = options.fetch(:wrapper, 'div')
+          "<#{ wrapper } data-brief-model='#{ model_class.type_alias }' data-brief-path='#{ relative_path_identifier }'>#{ unwrapped_html }</#{wrapper}>"
+        end
+      end
+
+      def unwrapped_html
+        parser.to_html
+      end
+
+      protected
+
+      def to_raw_html
+        renderer.render(content)
+      end
+
+      def renderer
+        @renderer ||= self.class.renderer
+      end
+
+      def renderer=(value)
+        @renderer = value
       end
     end
   end
 end