133 lines
3.5 KiB
Ruby
133 lines
3.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'benchmark'
|
|
require 'rainbow/ext/string'
|
|
|
|
class GithubImport
|
|
def self.run!(...)
|
|
new(...).run!
|
|
end
|
|
|
|
def initialize(token, gitlab_username, project_path, extras)
|
|
@options = { token: token }
|
|
@project_path = project_path
|
|
@current_user = UserFinder.new(gitlab_username).find_by_username
|
|
|
|
raise "GitLab user #{gitlab_username} not found. Please specify a valid username." unless @current_user
|
|
|
|
@github_repo = extras.empty? ? nil : extras.first
|
|
end
|
|
|
|
def run!
|
|
@repo = GithubRepos
|
|
.new(@options[:token], @current_user, @github_repo)
|
|
.choose_one!
|
|
|
|
raise 'No repo found!' unless @repo
|
|
|
|
show_warning!
|
|
|
|
@project = Project.find_by_full_path(@project_path) || new_project
|
|
|
|
import!
|
|
end
|
|
|
|
private
|
|
|
|
def show_warning!
|
|
puts "This will import GitHub #{@repo[:full_name].bright} into GitLab #{@project_path.bright} as #{@current_user.name}"
|
|
puts 'Permission checks are ignored. Press any key to continue.'.color(:red)
|
|
|
|
$stdin.getch
|
|
|
|
puts 'Starting the import (this could take a while)'.color(:green)
|
|
end
|
|
|
|
def import!
|
|
@project.import_state.force_start
|
|
|
|
timings = Benchmark.measure do
|
|
Gitlab::GithubImport::SequentialImporter
|
|
.new(@project, token: @options[:token])
|
|
.execute
|
|
end
|
|
|
|
@project.after_import
|
|
puts "Import finished. Timings: #{timings}".color(:green)
|
|
end
|
|
|
|
def new_project
|
|
namespace_path, _sep, project_name = @project_path.rpartition('/')
|
|
target_namespace = Namespace.find_by_full_path(namespace_path)
|
|
|
|
raise 'Namespace or group to import repository into does not exist.' unless target_namespace
|
|
|
|
Project.transaction do
|
|
project = Projects::CreateService.new(
|
|
@current_user,
|
|
name: project_name,
|
|
path: project_name,
|
|
description: @repo[:description],
|
|
namespace_id: target_namespace.id,
|
|
visibility_level: visibility_level,
|
|
skip_wiki: @repo[:has_wiki]
|
|
).execute
|
|
|
|
project.update!(
|
|
import_type: 'github',
|
|
import_source: @repo[:full_name],
|
|
import_url: @repo[:clone_url].sub('://', "://#{@options[:token]}@")
|
|
)
|
|
|
|
project
|
|
end
|
|
end
|
|
|
|
def visibility_level
|
|
@repo[:private] ? Gitlab::VisibilityLevel::PRIVATE : Gitlab::CurrentSettings.current_application_settings.default_project_visibility
|
|
end
|
|
end
|
|
|
|
class GithubRepos
|
|
def initialize(token, current_user, github_repo)
|
|
@client = Gitlab::GithubImport::Client.new(token)
|
|
@client.octokit.auto_paginate = true
|
|
|
|
@current_user = current_user
|
|
@github_repo = github_repo
|
|
end
|
|
|
|
def choose_one!
|
|
return found_github_repo if @github_repo
|
|
|
|
repos.each do |repo|
|
|
print "ID: #{repo[:id].to_s.bright}".color(:green)
|
|
print "\tName: #{repo[:full_name]}\n".color(:green)
|
|
end
|
|
|
|
print 'ID? '.bright
|
|
|
|
repos.find { |repo| repo[:id] == repo_id }
|
|
end
|
|
|
|
def found_github_repo
|
|
repos.find { |repo| repo[:full_name] == @github_repo }
|
|
end
|
|
|
|
def repo_id
|
|
@repo_id ||= $stdin.gets.chomp.to_i
|
|
end
|
|
|
|
def repos
|
|
@repos ||= @client.repos
|
|
end
|
|
end
|
|
|
|
namespace :import do
|
|
desc 'GitLab | Import | Import a GitHub project - Example: import:github[ToKeN,root,root/blah,my/github_repo] (optional my/github_repo)'
|
|
task :github, [:token, :gitlab_username, :project_path] => :environment do |_t, args|
|
|
abort 'Project path must be: namespace(s)/project_name'.color(:red) unless args.project_path.include?('/')
|
|
|
|
GithubImport.run!(args.token, args.gitlab_username, args.project_path, args.extras)
|
|
end
|
|
end
|