# 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