67 lines
2.1 KiB
Ruby
67 lines
2.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module ProjectForksHelper
|
|
def fork_project(project, user = nil, params = {})
|
|
Gitlab::GitalyClient.allow_n_plus_1_calls do
|
|
fork_project_direct(project, user, params)
|
|
end
|
|
end
|
|
|
|
def fork_project_direct(project, user = nil, params = {})
|
|
# Load the `fork_network` for the project to fork as there might be one that
|
|
# wasn't loaded yet.
|
|
project.reload unless project.fork_network
|
|
|
|
unless user
|
|
user = create(:user)
|
|
project.add_developer(user)
|
|
end
|
|
|
|
unless params[:namespace] || params[:namespace_id]
|
|
params[:namespace] = create(:group)
|
|
params[:namespace].add_owner(user)
|
|
end
|
|
|
|
service = Projects::ForkService.new(project, user, params)
|
|
|
|
create_repository = params.delete(:repository)
|
|
# Avoid creating a repository
|
|
unless create_repository
|
|
allow(RepositoryForkWorker).to receive(:perform_async).and_return(true)
|
|
shell = double('gitlab_shell', fork_repository: true)
|
|
allow(service).to receive(:gitlab_shell).and_return(shell)
|
|
end
|
|
|
|
forked_project = service.execute(params[:target_project])
|
|
|
|
# Reload the both projects so they know about their newly created fork_network
|
|
if forked_project.persisted?
|
|
project.reload
|
|
forked_project.reload
|
|
end
|
|
|
|
if create_repository
|
|
# The call to project.repository.after_import in RepositoryForkWorker does
|
|
# not reset the @exists variable of this forked_project.repository
|
|
# so we have to explicitly call this method to clear the @exists variable.
|
|
# of the instance we're returning here.
|
|
forked_project.repository.after_import
|
|
end
|
|
|
|
forked_project
|
|
end
|
|
|
|
def fork_project_with_submodules(project, user = nil, params = {})
|
|
Gitlab::GitalyClient.allow_n_plus_1_calls do
|
|
forked_project = fork_project_direct(project, user, params)
|
|
TestEnv.copy_repo(
|
|
forked_project,
|
|
bare_repo: TestEnv.forked_repo_path_bare,
|
|
refs: TestEnv::FORKED_BRANCH_SHA
|
|
)
|
|
forked_project.repository.after_import
|
|
|
|
forked_project
|
|
end
|
|
end
|
|
end
|