mmozuras/grit-ext

module Grit
  class Diff
    attr_reader :repo

    def hunks
      header_lines.each_with_index
                  .map do |header, index|
        next_header = header_lines[index + 1]
        next_line_number = next_header ? next_header[1] : lines.count
        DiffHunk.new(header[0], lines[header[1], next_line_number - 1])
      end
    end

    def full_a_path
      full(a_path)
    end

    def full_b_path
      full(b_path)
    end

    def added
      hunks_lines { |h| h.added }
    end

    def removed
      hunks_lines { |h| h.removed }
    end

    protected

    def hunks_lines
      hunks.map { |h| yield(h) }.flatten
    end

    def full(path)
      File.join(repo.working_dir, path)
    end

    def lines
      @lines ||= diff.split("\n")
    end

    def header_lines
      @header_lines ||= lines.each_with_index
           .select { |line, _index| line.start_with?('@@') }
           .map do |line, index|
             header = DiffHeader.new(line)
             [header, index]
           end
    end
  end
end