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

90 lines
2.5 KiB
Ruby
Raw Normal View History

module Gitlab
module Diff
class Parser
include Enumerable
def parse(lines)
return [] if lines.blank?
2015-02-02 23:58:28 -05:00
@lines = lines
line_obj_index = 0
line_old = 1
line_new = 1
type = nil
context = nil
2016-03-03 12:38:44 -05:00
# By returning an Enumerator we make it possible to search for a single line (with #find)
# without having to instantiate all the others that come after it.
Enumerator.new do |yielder|
@lines.each do |line|
# We're expecting a filename parameter only in a meta-part of the diff content
# when type is defined then we're already in a content-part
next if filename?(line) && type.nil?
2016-05-31 18:33:46 -04:00
full_line = line.delete("\n")
2016-05-31 18:33:46 -04:00
2017-02-21 19:16:48 -05:00
if line =~ /^@@ -/
2016-03-03 12:38:44 -05:00
type = "match"
2016-05-31 18:33:46 -04:00
2016-03-03 12:38:44 -05:00
line_old = line.match(/\-[0-9]*/)[0].to_i.abs rescue 0
line_new = line.match(/\+[0-9]*/)[0].to_i.abs rescue 0
2016-05-31 18:33:46 -04:00
next if line_old <= 1 && line_new <= 1 # top of file
2016-03-03 12:38:44 -05:00
yielder << Gitlab::Diff::Line.new(full_line, type, line_obj_index, line_old, line_new)
line_obj_index += 1
next
elsif line[0] == '\\'
type = "#{context}-nonewline"
2016-03-03 12:38:44 -05:00
yielder << Gitlab::Diff::Line.new(full_line, type, line_obj_index, line_old, line_new)
line_obj_index += 1
else
type = identification_type(line)
yielder << Gitlab::Diff::Line.new(full_line, type, line_obj_index, line_old, line_new)
line_obj_index += 1
end
2016-05-31 18:33:46 -04:00
2016-03-03 12:38:44 -05:00
case line[0]
when "+"
line_new += 1
context = :new
2016-03-03 12:38:44 -05:00
when "-"
line_old += 1
context = :old
when "\\" # rubocop:disable Lint/EmptyWhen
2016-03-03 12:38:44 -05:00
# No increment
else
line_new += 1
line_old += 1
end
end
end
end
def empty?
@lines.empty?
end
private
def filename?(line)
line.start_with?( '--- /dev/null', '+++ /dev/null', '--- a', '+++ b',
'+++ a', # The line will start with `+++ a` in the reverse diff of an orphan commit
'--- /tmp/diffy', '+++ /tmp/diffy')
end
def identification_type(line)
case line[0]
when "+"
"new"
when "-"
"old"
else
nil
end
end
end
end
end