lib/gitlab/blame.rb


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

module Gitlab
  class Blame
    attr_accessor :blob, :commit

    def initialize(blob, commit)
      @blob = blob
      @commit = commit
    end

    def groups(highlight: true)
      prev_sha = nil
      groups = []
      current_group = nil

      i = 0
      blame.each do |commit, line|
        commit = Commit.new(commit, project)

        sha = commit.sha
        if prev_sha != sha
          groups << current_group if current_group
          current_group = { commit: commit, lines: [] }
        end

        line = highlighted_lines[i].html_safe if highlight
        current_group[:lines] << line

        prev_sha = sha
        i += 1
      end
      groups << current_group if current_group

      groups
    end

    private

    def blame
      @blame ||= Gitlab::Git::Blame.new(repository, @commit.id, @blob.path)
    end

    def highlighted_lines
      @blob.load_all_data!(repository)
      @highlighted_lines ||=
        Gitlab::Highlight.highlight(@blob.path, @blob.data, repository: repository).lines
    end

    def project
      commit.project
    end

    def repository
      project.repository
    end
  end
end