Contents

require 'loofah'
require 'loofah/helpers'

module JekyllSearch
  class HtmlProcessor
    def self.strip_html(input)
      strip_pre = Loofah::Scrubber.new do |node|
        if node.name == 'pre'
          node.remove
          Loofah::Scrubber::STOP
        end
      end

      Loofah.fragment(input).
          scrub!(:prune).
          scrub!(strip_pre).
          to_text.
          gsub(/([\r\n\t\s]+)/, ' ').strip
    end

    def self.detect_sections(input)
      result = []

      current = { :id => nil, :title => nil, :content => '' }
      Loofah.fragment(input).children.each { |node|
        if node.name =~ /^h\d$/
          result << current
          current = { :id => nil, :title => nil, :content => '' }
          current[:id] = extract_headline_id(node)
          current[:title] = node.text
        else
          current[:content] += node.to_html
        end
      }
      if current[:title] != nil or current[:content] != ''
        result << current
      end

      result
    end

    def self.extract_headline_id(node)
      if node.has_attribute?('id')
        node.attribute('id').value
      else
        children = []
        node.traverse { |n| children << n }
        children_with_id = children.select { |n| n['id'] }

        if children_with_id.count == 1
          children_with_id.first['id']
        else
          nil
        end
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
jekyll_search-0.0.8	lib/jekyll_search/html_processor.rb
jekyll_search-0.0.7	lib/jekyll_search/html_processor.rb
jekyll_search-0.0.6	lib/jekyll_search/html_processor.rb
jekyll_search-0.0.5	lib/jekyll_search/html_processor.rb