56 lines
1.5 KiB
Ruby
56 lines
1.5 KiB
Ruby
|
module Gitlab
|
||
|
|
module Diff
|
||
|
|
class Highlight
|
||
|
|
def self.process_diff_lines(diff_file)
|
||
|
|
processor = new(diff_file)
|
||
|
|
processor.highlight
|
||
|
|
end
|
||
|
|
|
||
|
|
def initialize(diff_file)
|
||
|
|
text_lines = diff_file.diff_lines.map(&:text)
|
||
|
|
@diff_file = diff_file
|
||
|
|
@diff_lines = diff_file.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].highlighted_text = "#{@diff_line_prefixes[i]}#{line}"
|
||
|
|
end
|
||
|
|
|
||
|
|
@diff_lines
|
||
|
|
end
|
||
|
|
|
||
|
|
def lexer
|
||
|
|
parent = Rouge::Lexer.guess(filename: @diff_file.new_path, 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
|
||
|
|
@formatter ||= Rouge::Formatters::HTMLGitlab.new(
|
||
|
|
nowrap: true,
|
||
|
|
cssclass: 'code highlight',
|
||
|
|
lineanchors: true,
|
||
|
|
lineanchorsid: 'LC'
|
||
|
|
)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|