gitlab-org--gitlab-foss/lib/gitlab/diff/highlight.rb

56 lines
1.4 KiB
Ruby
Raw Normal View History

module Gitlab
module Diff
class Highlight
2015-12-31 02:23:50 +00:00
def self.process_diff_lines(file_name, diff_lines)
processor = new(file_name, diff_lines)
processor.highlight
end
2015-12-31 02:23:50 +00:00
def initialize(file_name, diff_lines)
text_lines = diff_lines.map(&:text)
@file_name = file_name
@diff_lines = diff_lines
@diff_line_prefixes = text_lines.map { |line| line.sub!(/\A((\+|\-)\s*)/, '');$1 }
@raw_lines = text_lines.join("\n")
end
def highlight
@code = unescape_html(@raw_lines)
@highlighted_code = formatter.format(lexer.lex(@code))
update_diff_lines
end
private
def update_diff_lines
@highlighted_code.lines.each_with_index do |line, i|
@diff_lines[i].text = "#{@diff_line_prefixes[i]}#{line}"
end
@diff_lines
end
def lexer
2015-12-31 02:23:50 +00:00
parent = Rouge::Lexer.guess(filename: @file_name, source: @code).new rescue Rouge::Lexers::PlainText.new
Rouge::Lexers::GitlabDiff.new(parent_lexer: parent)
end
def unescape_html(content)
text = CGI.unescapeHTML(content)
text.gsub!(' ', ' ')
text
end
def formatter
2015-12-31 02:23:50 +00:00
Rouge::Formatters::HTMLGitlab.new(
2015-12-30 18:10:28 +00:00
nowrap: true,
cssclass: 'code highlight',
lineanchors: true,
lineanchorsid: 'LC'
)
end
end
end
end