app/finders/branches_finder.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
57
58
59
60
61
62
63
64

# frozen_string_literal: true

class BranchesFinder < GitRefsFinder
  def initialize(repository, params = {})
    super(repository, params)
  end

  def execute(gitaly_pagination: false)
    if gitaly_pagination && names.blank? && search.blank? && regex.blank?
      repository.branches_sorted_by(sort, pagination_params)
    else
      branches = repository.branches_sorted_by(sort)
      branches = by_search(branches)
      branches = by_regex(branches)
      by_names(branches)
    end
  end

  def total
    repository.branch_count
  end

  private

  def names
    @params[:names].presence
  end

  def per_page
    @params[:per_page].presence
  end

  def regex
    @params[:regex].to_s.presence
  end
  strong_memoize_attr :regex

  def page_token
    "#{Gitlab::Git::BRANCH_REF_PREFIX}#{@params[:page_token]}" if @params[:page_token]
  end

  def pagination_params
    { limit: per_page, page_token: page_token }
  end

  def by_names(branches)
    return branches unless names

    branch_names = names.to_set
    branches.select do |branch|
      branch_names.include?(branch.name)
    end
  end

  def by_regex(branches)
    return branches unless regex

    branch_filter = ::Gitlab::UntrustedRegexp.new(regex)

    branches.select do |branch|
      branch_filter.match?(branch.name)
    end
  end
end