138 lines
3.8 KiB
Ruby
138 lines
3.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Import
|
|
class GithubService < Import::BaseService
|
|
include ActiveSupport::NumberHelper
|
|
include Gitlab::Utils::StrongMemoize
|
|
|
|
attr_accessor :client
|
|
attr_reader :params, :current_user
|
|
|
|
def execute(access_params, provider)
|
|
context_error = validate_context
|
|
return context_error if context_error
|
|
|
|
project = create_project(access_params, provider)
|
|
|
|
if project.persisted?
|
|
store_import_settings(project)
|
|
success(project)
|
|
elsif project.errors[:import_source_disabled].present?
|
|
error(project.errors[:import_source_disabled], :forbidden)
|
|
else
|
|
error(project_save_error(project), :unprocessable_entity)
|
|
end
|
|
rescue Octokit::Error => e
|
|
log_error(e)
|
|
end
|
|
|
|
def create_project(access_params, provider)
|
|
Gitlab::LegacyGithubImport::ProjectCreator.new(
|
|
repo,
|
|
project_name,
|
|
target_namespace,
|
|
current_user,
|
|
type: provider,
|
|
**access_params
|
|
).execute(extra_project_attrs)
|
|
end
|
|
|
|
def repo
|
|
@repo ||= client.repository(params[:repo_id].to_i)
|
|
end
|
|
|
|
def project_name
|
|
@project_name ||= params[:new_name].presence || repo[:name]
|
|
end
|
|
|
|
def namespace_path
|
|
@namespace_path ||= params[:target_namespace].presence || current_user.namespace_path
|
|
end
|
|
|
|
def target_namespace
|
|
@target_namespace ||= find_or_create_namespace(namespace_path, current_user.namespace_path)
|
|
end
|
|
|
|
def extra_project_attrs
|
|
{}
|
|
end
|
|
|
|
def oversized?
|
|
repository_size_limit > 0 && repo[:size] > repository_size_limit
|
|
end
|
|
|
|
def oversize_error_message
|
|
_('"%{repository_name}" size (%{repository_size}) is larger than the limit of %{limit}.') % {
|
|
repository_name: repo[:name],
|
|
repository_size: number_to_human_size(repo[:size]),
|
|
limit: number_to_human_size(repository_size_limit)
|
|
}
|
|
end
|
|
|
|
def repository_size_limit
|
|
strong_memoize :repository_size_limit do
|
|
namespace_limit = target_namespace.repository_size_limit.to_i
|
|
|
|
if namespace_limit > 0
|
|
namespace_limit
|
|
else
|
|
Gitlab::CurrentSettings.repository_size_limit.to_i
|
|
end
|
|
end
|
|
end
|
|
|
|
def url
|
|
@url ||= params[:github_hostname]
|
|
end
|
|
|
|
def allow_local_requests?
|
|
Gitlab::CurrentSettings.allow_local_requests_from_web_hooks_and_services?
|
|
end
|
|
|
|
def blocked_url?
|
|
Gitlab::UrlBlocker.blocked_url?(
|
|
url,
|
|
allow_localhost: allow_local_requests?,
|
|
allow_local_network: allow_local_requests?,
|
|
schemes: %w(http https)
|
|
)
|
|
end
|
|
|
|
private
|
|
|
|
def validate_context
|
|
if blocked_url?
|
|
log_and_return_error("Invalid URL: #{url}", _("Invalid URL: %{url}") % { url: url }, :bad_request)
|
|
elsif !authorized?
|
|
error(_('This namespace has already been taken. Choose a different one.'), :unprocessable_entity)
|
|
elsif oversized?
|
|
error(oversize_error_message, :unprocessable_entity)
|
|
end
|
|
end
|
|
|
|
def log_error(exception)
|
|
Gitlab::GithubImport::Logger.error(
|
|
message: 'Import failed due to a GitHub error',
|
|
status: exception.response_status,
|
|
error: exception.response_body
|
|
)
|
|
|
|
error(_('Import failed due to a GitHub error: %{original} (HTTP %{code})') % { original: exception.response_body, code: exception.response_status }, :unprocessable_entity)
|
|
end
|
|
|
|
def log_and_return_error(message, translated_message, http_status)
|
|
Gitlab::GithubImport::Logger.error(
|
|
message: 'Error while attempting to import from GitHub',
|
|
error: message
|
|
)
|
|
|
|
error(translated_message, http_status)
|
|
end
|
|
|
|
def store_import_settings(project)
|
|
Gitlab::GithubImport::Settings.new(project).write(params[:optional_stages])
|
|
end
|
|
end
|
|
end
|
|
|
|
Import::GithubService.prepend_mod_with('Import::GithubService')
|