650f40865e
The `#reload` makes to load all objects into memory, and the main purpose of `#reload` is to drop the association cache. The `#reset` seems to solve exactly that case.
111 lines
2.8 KiB
Ruby
111 lines
2.8 KiB
Ruby
namespace :gitlab do
|
|
desc "GitLab | Update templates"
|
|
task :update_templates do
|
|
TEMPLATE_DATA.each { |template| update(template) }
|
|
end
|
|
|
|
desc "GitLab | Update project templates"
|
|
task :update_project_templates do
|
|
include Gitlab::ImportExport::CommandLineUtil
|
|
|
|
if Rails.env.production?
|
|
puts "This rake task is not meant fo production instances".red
|
|
exit(1)
|
|
end
|
|
|
|
admin = User.find_by(admin: true)
|
|
|
|
unless admin
|
|
puts "No admin user could be found".red
|
|
exit(1)
|
|
end
|
|
|
|
Gitlab::ProjectTemplate.all.each do |template|
|
|
params = {
|
|
import_url: template.clone_url,
|
|
namespace_id: admin.namespace.id,
|
|
path: template.name,
|
|
skip_wiki: true
|
|
}
|
|
|
|
puts "Creating project for #{template.title}"
|
|
project = Projects::CreateService.new(admin, params).execute
|
|
|
|
unless project.persisted?
|
|
puts project.errors.messages
|
|
exit(1)
|
|
end
|
|
|
|
loop do
|
|
if project.finished?
|
|
puts "Import finished for #{template.name}"
|
|
break
|
|
end
|
|
|
|
if project.failed?
|
|
puts "Failed to import from #{project_params[:import_url]}".red
|
|
exit(1)
|
|
end
|
|
|
|
puts "Waiting for the import to finish"
|
|
|
|
sleep(5)
|
|
project.reset
|
|
end
|
|
|
|
Projects::ImportExport::ExportService.new(project, admin).execute
|
|
download_or_copy_upload(project.export_file, template.archive_path)
|
|
Projects::DestroyService.new(admin, project).execute
|
|
puts "Exported #{template.name}".green
|
|
end
|
|
puts "Done".green
|
|
end
|
|
|
|
def update(template)
|
|
sub_dir = template.repo_url.match(/([A-Za-z-]+)\.git\z/)[1]
|
|
dir = File.join(vendor_directory, sub_dir)
|
|
|
|
unless clone_repository(template.repo_url, dir)
|
|
puts "Cloning the #{sub_dir} templates failed".red
|
|
return
|
|
end
|
|
|
|
remove_unneeded_files(dir, template.cleanup_regex)
|
|
puts "Done".green
|
|
end
|
|
|
|
def clone_repository(url, directory)
|
|
FileUtils.rm_rf(directory) if Dir.exist?(directory)
|
|
|
|
system("git clone #{url} --depth=1 --branch=master #{directory}")
|
|
end
|
|
|
|
# Retain only certain files:
|
|
# - The LICENSE, because we have to
|
|
# - The sub dirs so we can organise the file by category
|
|
# - The templates themself
|
|
# - Dir.entries returns also the entries '.' and '..'
|
|
def remove_unneeded_files(directory, regex)
|
|
Dir.foreach(directory) do |file|
|
|
FileUtils.rm_rf(File.join(directory, file)) unless file =~ regex
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
Template = Struct.new(:repo_url, :cleanup_regex)
|
|
TEMPLATE_DATA = [
|
|
Template.new(
|
|
"https://github.com/github/gitignore.git",
|
|
/(\.{1,2}|LICENSE|Global|\.gitignore)\z/
|
|
),
|
|
Template.new(
|
|
"https://gitlab.com/gitlab-org/Dockerfile.git",
|
|
/(\.{1,2}|LICENSE|CONTRIBUTING.md|\.Dockerfile)\z/
|
|
)
|
|
].freeze
|
|
|
|
def vendor_directory
|
|
Rails.root.join('vendor')
|
|
end
|
|
end
|