BigW Consortium Gitlab

parser.rb 1.9 KB
Newer Older
1 2 3 4 5
module Gitlab
  module Diff
    class Parser
      include Enumerable

6
      def parse(lines)
7
        @lines = lines
8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
        lines_obj = []
        line_obj_index = 0
        line_old = 1
        line_new = 1
        type = nil

        lines_arr = ::Gitlab::InlineDiff.processing lines

        lines_arr.each do |line|
          next if filename?(line)

          full_line = html_escape(line.gsub(/\n/, ''))
          full_line = ::Gitlab::InlineDiff.replace_markers full_line

          if line.match(/^@@ -/)
            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

28
            next if line_old <= 1 && line_new <= 1 #top of file
29 30 31 32 33
            lines_obj << Gitlab::Diff::Line.new(full_line, type, line_obj_index, line_old, line_new)
            line_obj_index += 1
            next
          else
            type = identification_type(line)
34
            lines_obj << Gitlab::Diff::Line.new(full_line, type, line_obj_index, line_old, line_new)
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
            line_obj_index += 1
          end


          if line[0] == "+"
            line_new += 1
          elsif line[0] == "-"
            line_old += 1
          else
            line_new += 1
            line_old += 1
          end
        end

        lines_obj
      end

      def empty?
        @lines.empty?
      end

      private

      def filename?(line)
        line.start_with?('--- /dev/null', '+++ /dev/null', '--- a', '+++ b',
                         '--- /tmp/diffy', '+++ /tmp/diffy')
      end

      def identification_type(line)
        if line[0] == "+"
          "new"
        elsif line[0] == "-"
          "old"
        else
          nil
        end
      end

73
      def html_escape(str)
74
        replacements = { '&' => '&amp;', '>' => '&gt;', '<' => '&lt;', '"' => '&quot;', "'" => '&#39;' }
75
        str.gsub(/[&"'><]/, replacements)
76 77 78 79
      end
    end
  end
end