debian-mirror-gitlab/lib/tasks/gitlab/seed.rake

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

93 lines
2.9 KiB
Ruby
Raw Normal View History

2021-03-11 19:13:27 +05:30
# frozen_string_literal: true
2019-07-31 22:56:46 +05:30
namespace :gitlab do
namespace :seed do
2023-01-13 00:05:48 +05:30
def projects_from_args(args)
full_path = args.project_full_path
2019-09-30 21:07:59 +05:30
2023-01-13 00:05:48 +05:30
if full_path
project = Project.find_by_full_path(full_path)
2019-07-31 22:56:46 +05:30
2023-01-13 00:05:48 +05:30
unless project
error_message = "Project '#{full_path}' does not exist!"
potential_projects = Project.search(full_path)
2019-07-31 22:56:46 +05:30
2023-01-13 00:05:48 +05:30
if potential_projects.present?
error_message += " Did you mean '#{potential_projects.first.full_path}'?"
2019-07-31 22:56:46 +05:30
end
2023-01-13 00:05:48 +05:30
puts error_message.color(:red)
exit 1
2019-07-31 22:56:46 +05:30
end
2023-01-13 00:05:48 +05:30
[project]
else
scope = Project.respond_to?(:not_mass_generated) ? Project.not_mass_generated : Project
scope.find_each
end
end
desc "GitLab | Seed | Seeds issues"
task :issues, [:project_full_path, :backfill_weeks, :average_issues_per_week] => :environment do |t, args|
args.with_defaults(backfill_weeks: 5, average_issues_per_week: 2)
projects = projects_from_args(args)
2019-07-31 22:56:46 +05:30
projects.each do |project|
puts "\nSeeding issues for the '#{project.full_path}' project"
seeder = Quality::Seeders::Issues.new(project: project)
2019-09-30 21:07:59 +05:30
issues_created = seeder.seed(backfill_weeks: args.backfill_weeks.to_i,
average_issues_per_week: args.average_issues_per_week.to_i)
2019-07-31 22:56:46 +05:30
puts "\n#{issues_created} issues created!"
end
end
2022-11-25 23:54:43 +05:30
task :epics, [:group_full_path, :backfill_weeks, :average_issues_per_week] => :environment do |t, args|
args.with_defaults(backfill_weeks: 5, average_issues_per_week: 2)
groups =
if args.group_full_path
group = Group.find_by_full_path(args.group_full_path)
unless group
error_message = "Group '#{args.group_full_path}' does not exist!"
potential_groups = Group.search(args.group_full_path)
if potential_groups.present?
error_message += " Did you mean '#{potential_groups.first.full_path}'?"
end
puts error_message.color(:red)
exit 1
end
[group]
else
Group.not_mass_generated.find_each
end
groups.each do |group|
puts "\nSeeding epics for the '#{group.full_path}' group"
seeder = Quality::Seeders::Epics.new(group: group)
epics = seeder.seed(
backfill_weeks: args.backfill_weeks.to_i,
average_issues_per_week: args.average_issues_per_week.to_i
)
puts "\n#{epics} epics created!"
end
end
2023-01-13 00:05:48 +05:30
desc "GitLab | Seed | Seed a project with vulnerabilities"
task :vulnerabilities, [:project_full_path] => :environment do |t, args|
projects = projects_from_args(args)
projects.each do |project|
puts "\nSeeding vulnerabilities for the '#{project.full_path}' project"
seeder = Quality::Seeders::Vulnerabilities.new(project)
seeder.seed!
puts "\nDone."
end
end
2019-07-31 22:56:46 +05:30
end
end