79 lines
2.5 KiB
Ruby
79 lines
2.5 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] = create(:group)
|
|
params[:namespace].add_owner(user)
|
|
end
|
|
|
|
namespace = params[:namespace]
|
|
create_repository = params.delete(:repository)
|
|
|
|
unless params[:target_project] || params[:using_service]
|
|
target_level = [project.visibility_level, namespace.visibility_level].min
|
|
visibility_level = Gitlab::VisibilityLevel.closest_allowed_level(target_level)
|
|
|
|
params[:target_project] =
|
|
create(:project,
|
|
(:repository if create_repository),
|
|
visibility_level: visibility_level, creator: user, namespace: namespace)
|
|
end
|
|
|
|
service = Projects::ForkService.new(project, user, params)
|
|
|
|
# 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.expire_content_cache
|
|
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.expire_content_cache
|
|
|
|
forked_project
|
|
end
|
|
end
|
|
end
|