summaryrefslogtreecommitdiff
path: root/lib/gitlab/diff/parser.rb
blob: 742f989c50b40f35c6fc630dcd88d0626556ed3d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
module Gitlab
  module Diff
    class Parser
      include Enumerable

      def parse(lines)
        return [] if lines.blank?

        @lines = lines
        line_obj_index = 0
        line_old = 1
        line_new = 1
        type = nil
        context = nil

        # 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|
            next if filename?(line)

            full_line = line.delete("\n")

            if line =~ /^@@ -/
              type = "match"

              line_old = line.match(/\-[0-9]*/)[0].to_i.abs rescue 0
              line_new = line.match(/\+[0-9]*/)[0].to_i.abs rescue 0

              next if line_old <= 1 && line_new <= 1 # top of file
              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"

              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

            case line[0]
            when "+"
              line_new += 1
              context = :new
            when "-"
              line_old += 1
              context = :old
            when "\\" # rubocop:disable Lint/EmptyWhen
              # 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