debian-mirror-gitlab/lib/gitlab/harbor/query.rb
2022-10-11 01:57:18 +05:30

126 lines
3.2 KiB
Ruby

# frozen_string_literal: true
module Gitlab
module Harbor
class Query
include ActiveModel::Validations
attr_reader :client, :repository_id, :artifact_id, :search, :limit, :sort, :page
DEFAULT_LIMIT = 10
SORT_REGEX = %r{\A(creation_time|update_time|name) (asc|desc)\z}.freeze
validates :page, numericality: { greater_than: 0, integer: true }, allow_blank: true
validates :limit, numericality: { greater_than: 0, less_than_or_equal_to: 25, integer: true }, allow_blank: true
validates :repository_id, format: {
with: /\A[a-zA-Z0-9\_\.\-$]+\z/,
message: 'Id invalid'
}, allow_blank: true
validates :artifact_id, format: {
with: /\A[a-zA-Z0-9\_\.\-$:]+\z/,
message: 'Id invalid'
}, allow_blank: true
validates :sort, format: {
with: SORT_REGEX,
message: 'params invalid'
}, allow_blank: true
validates :search, format: {
with: /\A([a-z\_]*=[a-zA-Z0-9\- :]*,*)*\z/,
message: 'params invalid'
}, allow_blank: true
def initialize(integration, params)
@client = Client.new(integration)
@repository_id = params[:repository_id]
@artifact_id = params[:artifact_id]
@search = params[:search]
@limit = params[:limit]
@sort = params[:sort]
@page = params[:page]
validate
end
def repositories
result = @client.get_repositories(query_options)
return [] if result[:total_count] == 0
Kaminari.paginate_array(
result[:body],
limit: query_page_size,
total_count: result[:total_count]
)
end
def artifacts
result = @client.get_artifacts(query_artifacts_options)
return [] if result[:total_count] == 0
Kaminari.paginate_array(
result[:body],
limit: query_page_size,
total_count: result[:total_count]
)
end
def tags
result = @client.get_tags(query_tags_options)
return [] if result[:total_count] == 0
Kaminari.paginate_array(
result[:body],
limit: query_page_size,
total_count: result[:total_count]
)
end
private
def query_artifacts_options
options = query_options
options[:repository_name] = repository_id
options[:with_tag] = true
options
end
def query_options
options = {
page: query_page,
page_size: query_page_size
}
options[:q] = query_search if search.present?
options[:sort] = query_sort if sort.present?
options
end
def query_tags_options
options = query_options
options[:repository_name] = repository_id
options[:artifact_name] = artifact_id
options
end
def query_page
page.presence || 1
end
def query_page_size
(limit.presence || DEFAULT_LIMIT).to_i
end
def query_search
search.gsub('=', '=~')
end
def query_sort
match = sort.match(SORT_REGEX)
order = (match[2] == 'asc' ? '' : '-')
"#{order}#{match[1]}"
end
end
end
end