2017-02-16 02:08:30 +00:00
|
|
|
module Gitlab
|
|
|
|
module GitalyClient
|
|
|
|
class Ref
|
|
|
|
attr_accessor :stub
|
|
|
|
|
2017-04-07 09:19:20 +00:00
|
|
|
# 'repository' is a Gitlab::Git::Repository
|
|
|
|
def initialize(repository)
|
|
|
|
@gitaly_repo = repository.gitaly_repository
|
|
|
|
@stub = Gitaly::Ref::Stub.new(nil, nil, channel_override: repository.gitaly_channel)
|
2017-02-16 02:08:30 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def default_branch_name
|
2017-04-07 09:19:20 +00:00
|
|
|
request = Gitaly::FindDefaultBranchNameRequest.new(repository: @gitaly_repo)
|
2017-02-16 02:08:30 +00:00
|
|
|
stub.find_default_branch_name(request).name.gsub(/^refs\/heads\//, '')
|
|
|
|
end
|
|
|
|
|
|
|
|
def branch_names
|
2017-04-07 09:19:20 +00:00
|
|
|
request = Gitaly::FindAllBranchNamesRequest.new(repository: @gitaly_repo)
|
2017-02-16 02:08:30 +00:00
|
|
|
consume_refs_response(stub.find_all_branch_names(request), prefix: 'refs/heads/')
|
|
|
|
end
|
|
|
|
|
|
|
|
def tag_names
|
2017-04-07 09:19:20 +00:00
|
|
|
request = Gitaly::FindAllTagNamesRequest.new(repository: @gitaly_repo)
|
2017-02-16 02:08:30 +00:00
|
|
|
consume_refs_response(stub.find_all_tag_names(request), prefix: 'refs/tags/')
|
|
|
|
end
|
|
|
|
|
2017-03-28 17:14:48 +00:00
|
|
|
def find_ref_name(commit_id, ref_prefix)
|
2017-04-06 16:47:35 +00:00
|
|
|
request = Gitaly::FindRefNameRequest.new(
|
2017-03-28 17:14:48 +00:00
|
|
|
repository: @repository,
|
|
|
|
commit_id: commit_id,
|
|
|
|
prefix: ref_prefix
|
|
|
|
)
|
|
|
|
|
|
|
|
stub.find_ref_name(request).name
|
|
|
|
end
|
|
|
|
|
2017-02-16 02:08:30 +00:00
|
|
|
private
|
|
|
|
|
|
|
|
def consume_refs_response(response, prefix:)
|
|
|
|
response.flat_map do |r|
|
|
|
|
r.names.map { |name| name.sub(/\A#{Regexp.escape(prefix)}/, '') }
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|