gitlab-org--gitlab-foss/lib/gitlab/harbor/client.rb

73 lines
2.1 KiB
Ruby

# frozen_string_literal: true
module Gitlab
module Harbor
class Client
Error = Class.new(StandardError)
ConfigError = Class.new(Error)
attr_reader :integration
def initialize(integration)
raise ConfigError, 'Please check your integration configuration.' unless integration
@integration = integration
end
def ping
options = { headers: headers.merge!('Accept': 'text/plain') }
response = Gitlab::HTTP.get(url('ping'), options)
{ success: response.success? }
end
def get_repositories(params)
get(url("projects/#{integration.project_name}/repositories"), params)
end
def get_artifacts(params)
repository_name = params.delete(:repository_name)
get(url("projects/#{integration.project_name}/repositories/#{repository_name}/artifacts"), params)
end
def get_tags(params)
repository_name = params.delete(:repository_name)
artifact_name = params.delete(:artifact_name)
get(
url("projects/#{integration.project_name}/repositories/#{repository_name}/artifacts/#{artifact_name}/tags"),
params
)
end
private
def get(path, params = {})
options = { headers: headers, query: params }
response = Gitlab::HTTP.get(path, options)
raise Gitlab::Harbor::Client::Error, 'request error' unless response.success?
{
body: Gitlab::Json.parse(response.body),
total_count: response.headers['x-total-count'].to_i
}
rescue JSON::ParserError
raise Gitlab::Harbor::Client::Error, 'invalid response format'
end
# url must be used within get method otherwise this would avoid validation by GitLab::HTTP
def url(path)
base_url = Gitlab::Utils.append_path(integration.url, '/api/v2.0/')
Gitlab::Utils.append_path(base_url, path)
end
def headers
auth = Base64.strict_encode64("#{integration.username}:#{integration.password}")
{
'Content-Type': 'application/json',
'Authorization': "Basic #{auth}"
}
end
end
end
end