mirror of
https://github.com/scinote-eln/scinote-web.git
synced 2025-03-06 20:56:42 +08:00
102 lines
3.8 KiB
Ruby
102 lines
3.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class MigrateToNewUserRoles < ActiveRecord::Migration[6.1]
|
|
def change
|
|
reversible do |dir|
|
|
dir.up do
|
|
owner_role = UserRole.owner_role
|
|
owner_role.save!
|
|
normal_user_role = UserRole.normal_user_role
|
|
normal_user_role.save!
|
|
technician_role = UserRole.technician_role
|
|
technician_role.save!
|
|
viewer_role = UserRole.viewer_role
|
|
viewer_role.save!
|
|
|
|
create_user_assignments(UserProject.owner, owner_role)
|
|
create_user_assignments(UserProject.normal_user, normal_user_role)
|
|
create_user_assignments(UserProject.technician, technician_role)
|
|
create_user_assignments(UserProject.viewer, viewer_role)
|
|
end
|
|
dir.down do
|
|
project_assignments = UserAssignment.joins(:user_role).where(assignable_type: 'Project')
|
|
create_user_project(
|
|
project_assignments.where(user_role: { name: I18n.t('user_roles.predefined.owner'), predefined: true }),
|
|
'owner'
|
|
)
|
|
create_user_project(
|
|
project_assignments.where(user_role: { name: I18n.t('user_roles.predefined.normal_user'), predefined: true }),
|
|
'normal_user'
|
|
)
|
|
create_user_project(
|
|
project_assignments.where(user_role: { name: I18n.t('user_roles.predefined.technician'), predefined: true }),
|
|
'technician'
|
|
)
|
|
create_user_project(
|
|
project_assignments.where(user_role: { name: I18n.t('user_roles.predefined.viewer'), predefined: true }),
|
|
'viewer'
|
|
)
|
|
|
|
UserAssignment.joins(:user_role).where(user_role: { predefined: true }).delete_all
|
|
UserRole.where(predefined: true).delete_all
|
|
end
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def new_user_assignment(user, assignable, user_role)
|
|
UserAssignment.new(
|
|
user: user,
|
|
assignable: assignable,
|
|
assigned: :automatically,
|
|
user_role: user_role
|
|
)
|
|
end
|
|
|
|
def create_team_admin_assignments
|
|
user_assignments = []
|
|
owner_role = UserRole.find_by(name: I18n.t('user_roles.predefined.owner'))
|
|
admin_user_teams =
|
|
UserTeam.includes(:user, :team).where({ role: UserTeam.roles[:admin] })
|
|
|
|
admin_user_teams.find_each do |user_team|
|
|
user_team.team.projects.find_each do |project|
|
|
user_assignments << new_user_assignment(user_team.user, project, owner_role)
|
|
project.experiments.find_each do |experiment|
|
|
user_assignments << new_user_assignment(user_team.user, experiment, owner_role)
|
|
experiment.my_modules.find_each do |my_module|
|
|
user_assignments << new_user_assignment(user_team.user, my_module, owner_role)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
UserAssignment.import(user_assignments)
|
|
end
|
|
|
|
def create_user_assignments(user_projects, user_role)
|
|
user_projects.includes(:user, :project).find_in_batches(batch_size: 100) do |user_project_batch|
|
|
user_assignments = []
|
|
user_project_batch.each do |user_project|
|
|
user_assignments << new_user_assignment(user_project.user, user_project.project, user_role)
|
|
user_project.project.experiments.each do |experiment|
|
|
user_assignments << new_user_assignment(user_project.user, experiment, user_role)
|
|
experiment.my_modules.each do |my_module|
|
|
user_assignments << new_user_assignment(user_project.user, my_module, user_role)
|
|
end
|
|
end
|
|
end
|
|
UserAssignment.import(user_assignments)
|
|
end
|
|
end
|
|
|
|
def create_user_project(user_roles, role)
|
|
user_roles.includes(:user, :assignable).find_in_batches(batch_size: 100) do |user_role_batch|
|
|
user_projects = []
|
|
user_role_batch.each do |user_role|
|
|
user_projects << UserProject.new(user: user_role.user, project: user_role.assignable, role: role)
|
|
end
|
|
UserProject.import(user_projects)
|
|
end
|
|
end
|
|
end
|