82 lines
2.6 KiB
Ruby
82 lines
2.6 KiB
Ruby
module Gitlab
|
|
module GitalyClient
|
|
class Ref
|
|
include Gitlab::EncodingHelper
|
|
|
|
# 'repository' is a Gitlab::Git::Repository
|
|
def initialize(repository)
|
|
@repository = repository
|
|
@gitaly_repo = repository.gitaly_repository
|
|
@storage = repository.storage
|
|
end
|
|
|
|
def default_branch_name
|
|
request = Gitaly::FindDefaultBranchNameRequest.new(repository: @gitaly_repo)
|
|
response = GitalyClient.call(@storage, :ref, :find_default_branch_name, request)
|
|
Gitlab::Git.branch_name(response.name)
|
|
end
|
|
|
|
def branch_names
|
|
request = Gitaly::FindAllBranchNamesRequest.new(repository: @gitaly_repo)
|
|
response = GitalyClient.call(@storage, :ref, :find_all_branch_names, request)
|
|
consume_refs_response(response) { |name| Gitlab::Git.branch_name(name) }
|
|
end
|
|
|
|
def tag_names
|
|
request = Gitaly::FindAllTagNamesRequest.new(repository: @gitaly_repo)
|
|
response = GitalyClient.call(@storage, :ref, :find_all_tag_names, request)
|
|
consume_refs_response(response) { |name| Gitlab::Git.tag_name(name) }
|
|
end
|
|
|
|
def find_ref_name(commit_id, ref_prefix)
|
|
request = Gitaly::FindRefNameRequest.new(
|
|
repository: @gitaly_repo,
|
|
commit_id: commit_id,
|
|
prefix: ref_prefix
|
|
)
|
|
encode!(GitalyClient.call(@storage, :ref, :find_ref_name, request).name.dup)
|
|
end
|
|
|
|
def count_tag_names
|
|
tag_names.count
|
|
end
|
|
|
|
def count_branch_names
|
|
branch_names.count
|
|
end
|
|
|
|
def local_branches(sort_by: nil)
|
|
request = Gitaly::FindLocalBranchesRequest.new(repository: @gitaly_repo)
|
|
request.sort_by = sort_by_param(sort_by) if sort_by
|
|
response = GitalyClient.call(@storage, :ref, :find_local_branches, request)
|
|
consume_branches_response(response)
|
|
end
|
|
|
|
private
|
|
|
|
def consume_refs_response(response)
|
|
response.flat_map { |message| message.names.map { |name| yield(name) } }
|
|
end
|
|
|
|
def sort_by_param(sort_by)
|
|
sort_by = 'name' if sort_by == 'name_asc'
|
|
|
|
enum_value = Gitaly::FindLocalBranchesRequest::SortBy.resolve(sort_by.upcase.to_sym)
|
|
raise ArgumentError, "Invalid sort_by key `#{sort_by}`" unless enum_value
|
|
enum_value
|
|
end
|
|
|
|
def consume_branches_response(response)
|
|
response.flat_map do |message|
|
|
message.branches.map do |gitaly_branch|
|
|
Gitlab::Git::Branch.new(
|
|
@repository,
|
|
encode!(gitaly_branch.name.dup),
|
|
gitaly_branch.commit_id
|
|
)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|