summaryrefslogtreecommitdiff
path: root/lib/banzai/filter/merge_request_reference_filter.rb
blob: 0eab865ac04802340b932a4c2a02a6379e6e781a (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
module Banzai
  module Filter
    # HTML filter that replaces merge request references with links. References
    # to merge requests that do not exist are ignored.
    #
    # This filter supports cross-project references.
    class MergeRequestReferenceFilter < AbstractReferenceFilter
      self.reference_type = :merge_request

      def self.object_class
        MergeRequest
      end

      def find_object(project, iid)
        merge_requests_per_project[project][iid]
      end

      def url_for_object(mr, project)
        h = Gitlab::Routing.url_helpers
        h.project_merge_request_url(project, mr,
                                            only_path: context[:only_path])
      end

      def project_from_ref(ref)
        projects_per_reference[ref || current_project_path]
      end

      # Returns a Hash containing the merge_requests per Project instance.
      def merge_requests_per_project
        @merge_requests_per_project ||= begin
          hash = Hash.new { |h, k| h[k] = {} }

          projects_per_reference.each do |path, project|
            merge_request_ids = references_per_project[path]

            merge_requests = project.merge_requests
              .where(iid: merge_request_ids.to_a)
              .includes(target_project: :namespace)

            merge_requests.each do |merge_request|
              hash[project][merge_request.iid.to_i] = merge_request
            end
          end

          hash
        end
      end

      def object_link_text_extras(object, matches)
        extras = super

        path = matches[:path] if matches.names.include?("path")
        case path
        when '/diffs'
          extras.unshift "diffs"
        when '/commits'
          extras.unshift "commits"
        when '/builds'
          extras.unshift "builds"
        end

        extras
      end
    end
  end
end