Contents

# frozen_string_literal: true
require "nokogumbo"
require "forwardable"

module ThemeCheck
  class HtmlVisitor
    attr_reader :checks

    def initialize(checks)
      @checks = checks
    end

    def visit_template(template)
      doc = parse(template)
      visit(HtmlNode.new(doc, template))
    rescue ArgumentError => e
      call_checks(:on_parse_error, e, template)
    end

    private

    def parse(template)
      Nokogiri::HTML5.fragment(template.source, max_tree_depth: 400, max_attributes: 400)
    end

    def visit(node)
      call_checks(:on_element, node) if node.element?
      call_checks(:"on_#{node.name}", node)
      node.children.each { |child| visit(child) }
      unless node.literal?
        call_checks(:"after_#{node.name}", node)
        call_checks(:after_element, node) if node.element?
      end
    end

    def call_checks(method, *args)
      checks.call(method, *args)
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
theme-check-1.2.0	lib/theme_check/html_visitor.rb
theme-check-1.1.0	lib/theme_check/html_visitor.rb
theme-check-1.0.0	lib/theme_check/html_visitor.rb
theme-check-0.10.2	lib/theme_check/html_visitor.rb