module YARD::Templates::Helpers::HtmlSyntaxHighlightHelper

def clean_token_object(token_obj)
  return unless token_obj
  if token_obj == object
    token_obj = nil
  elsif token_obj.is_a?(CodeObjects::MethodObject)
    token_obj = prune_method_listing([token_obj], false).first
  else
    token_obj = run_verifier([token_obj]).first
  end
  token_obj
end

def html_syntax_highlight_ruby(source)
  if Parser::SourceParser.parser_type == :ruby
    html_syntax_highlight_ruby_ripper(source)
  else
    html_syntax_highlight_ruby_legacy(source)
  end
end

def html_syntax_highlight_ruby_legacy(source)
  tokenlist = Parser::Ruby::Legacy::TokenList.new(source)
  tokenlist.map do |s|
    prettyclass = s.class.class_name.sub(/^Tk/, '').downcase
    prettysuper = s.class.superclass.class_name.sub(/^Tk/, '').downcase
    case s
    when Parser::Ruby::Legacy::RubyToken::TkWhitespace, Parser::Ruby::Legacy::RubyToken::TkUnknownChar
      h s.text
    when Parser::Ruby::Legacy::RubyToken::TkId
      prettyval = h(s.text)
      "<span class='rubyid_#{prettyval} #{prettyclass} #{prettysuper}'>#{prettyval}</span>"
    else
      "<span class='#{prettyclass} #{prettysuper}'>#{h s.text}</span>"
    end
  end.join
end

def html_syntax_highlight_ruby_ripper(source)
  resolver = Parser::Ruby::TokenResolver.new(source, object)
  output = String.new("")
  resolver.each do |s, token_obj|
    token_obj = clean_token_object(token_obj)
    output << "<span class='tstring'>" if [:tstring_beg, :regexp_beg].include?(s[0])
    case s.first
    when :nl, :ignored_nl, :sp
      output << h(s[1])
    when :ident, :const
      klass = s.first == :ident ? "id identifier rubyid_#{h(s[1])}" : s.first
      val = token_obj ? link_object(token_obj, s[1]) : h(s[1])
      output << "<span class='#{klass}'>#{val}</span>"
    else
      output << "<span class='#{s.first}'>#{h(s[1])}</span>"
    end
    output << "</span>" if [:tstring_end, :regexp_end].include?(s[0])
  end
  output
rescue Parser::ParserSyntaxError
  source =~ /^<span\s+class=/ ? source : h(source)
end