87 lines
2.3 KiB
Ruby
87 lines
2.3 KiB
Ruby
module Gitlab
|
|
module GitlabImport
|
|
class Client
|
|
attr_reader :client, :api
|
|
|
|
PER_PAGE = 100
|
|
|
|
def initialize(access_token)
|
|
@client = ::OAuth2::Client.new(
|
|
config.app_id,
|
|
config.app_secret,
|
|
gitlab_options
|
|
)
|
|
|
|
if access_token
|
|
@api = OAuth2::AccessToken.from_hash(@client, access_token: access_token)
|
|
end
|
|
end
|
|
|
|
def authorize_url(redirect_uri)
|
|
client.auth_code.authorize_url({
|
|
redirect_uri: redirect_uri,
|
|
scope: "api"
|
|
})
|
|
end
|
|
|
|
def get_token(code, redirect_uri)
|
|
client.auth_code.get_token(code, redirect_uri: redirect_uri).token
|
|
end
|
|
|
|
def user
|
|
api.get("/api/v4/user").parsed
|
|
end
|
|
|
|
def issues(project_identifier, **kwargs)
|
|
lazy_page_iterator(**kwargs) do |page, per_page|
|
|
api.get("/api/v4/projects/#{project_identifier}/issues?per_page=#{per_page}&page=#{page}").parsed
|
|
end
|
|
end
|
|
|
|
def issue_comments(project_identifier, issue_id, **kwargs)
|
|
lazy_page_iterator(**kwargs) do |page, per_page|
|
|
api.get("/api/v4/projects/#{project_identifier}/issues/#{issue_id}/notes?per_page=#{per_page}&page=#{page}").parsed
|
|
end
|
|
end
|
|
|
|
def project(id)
|
|
api.get("/api/v4/projects/#{id}").parsed
|
|
end
|
|
|
|
def projects(**kwargs)
|
|
lazy_page_iterator(**kwargs) do |page, per_page|
|
|
api.get("/api/v4/projects?per_page=#{per_page}&page=#{page}&simple=true&membership=true").parsed
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def lazy_page_iterator(starting_page: 1, page_limit: nil, per_page: PER_PAGE)
|
|
Enumerator.new do |y|
|
|
page = starting_page
|
|
page_limit = (starting_page - 1) + page_limit if page_limit
|
|
|
|
loop do
|
|
items = yield(page, per_page)
|
|
|
|
items.each do |item|
|
|
y << item
|
|
end
|
|
|
|
break if items.empty? || items.size < per_page || (page_limit && page >= page_limit)
|
|
|
|
page += 1
|
|
end
|
|
end
|
|
end
|
|
|
|
def config
|
|
Gitlab::Auth::OAuth::Provider.config_for('gitlab')
|
|
end
|
|
|
|
def gitlab_options
|
|
OmniAuth::Strategies::GitLab.default_options[:client_options].to_h.symbolize_keys
|
|
end
|
|
end
|
|
end
|
|
end
|