91 lines
2.1 KiB
Ruby
91 lines
2.1 KiB
Ruby
module Gitlab
|
|
module GithubImport
|
|
class Client
|
|
GITHUB_SAFE_REMAINING_REQUESTS = 100
|
|
GITHUB_SAFE_SLEEP_TIME = 500
|
|
|
|
attr_reader :client, :api
|
|
|
|
def initialize(access_token)
|
|
@client = ::OAuth2::Client.new(
|
|
config.app_id,
|
|
config.app_secret,
|
|
github_options.merge(ssl: { verify: config['verify_ssl'] })
|
|
)
|
|
|
|
if access_token
|
|
::Octokit.auto_paginate = false
|
|
|
|
@api = ::Octokit::Client.new(
|
|
access_token: access_token,
|
|
api_endpoint: github_options[:site],
|
|
connection_options: {
|
|
ssl: { verify: config['verify_ssl'] }
|
|
}
|
|
)
|
|
end
|
|
end
|
|
|
|
def authorize_url(redirect_uri)
|
|
client.auth_code.authorize_url({
|
|
redirect_uri: redirect_uri,
|
|
scope: "repo, user, user:email"
|
|
})
|
|
end
|
|
|
|
def get_token(code)
|
|
client.auth_code.get_token(code).token
|
|
end
|
|
|
|
def method_missing(method, *args, &block)
|
|
if api.respond_to?(method)
|
|
request { api.send(method, *args, &block) }
|
|
else
|
|
super(method, *args, &block)
|
|
end
|
|
end
|
|
|
|
def respond_to?(method)
|
|
api.respond_to?(method) || super
|
|
end
|
|
|
|
private
|
|
|
|
def config
|
|
Gitlab.config.omniauth.providers.find { |provider| provider.name == "github" }
|
|
end
|
|
|
|
def github_options
|
|
config["args"]["client_options"].deep_symbolize_keys
|
|
end
|
|
|
|
def rate_limit
|
|
api.rate_limit!
|
|
end
|
|
|
|
def rate_limit_exceed?
|
|
rate_limit.remaining <= GITHUB_SAFE_REMAINING_REQUESTS
|
|
end
|
|
|
|
def rate_limit_sleep_time
|
|
rate_limit.resets_in + GITHUB_SAFE_SLEEP_TIME
|
|
end
|
|
|
|
def request
|
|
sleep rate_limit_sleep_time if rate_limit_exceed?
|
|
|
|
data = yield
|
|
|
|
last_response = api.last_response
|
|
|
|
while last_response.rels[:next]
|
|
sleep rate_limit_sleep_time if rate_limit_exceed?
|
|
last_response = last_response.rels[:next].get
|
|
data.concat(last_response.data) if last_response.data.is_a?(Array)
|
|
end
|
|
|
|
data
|
|
end
|
|
end
|
|
end
|
|
end
|