scinote-web/app/models/repository.rb

# frozen_string_literal: true

class Repository < RepositoryBase
  include SearchableModel
  include SearchableByNameModel
  include Assignable
  include PermissionCheckableModel
  include RepositoryImportParser
  include ArchivableModel

  ID_PREFIX = 'IN'
  include PrefixedIdModel

  enum permission_level: Extends::SHARED_OBJECTS_PERMISSION_LEVELS

  belongs_to :archived_by,
             foreign_key: :archived_by_id,
             class_name: 'User',
             inverse_of: :archived_repositories,
             optional: true
  belongs_to :restored_by,
             foreign_key: :restored_by_id,
             class_name: 'User',
             inverse_of: :restored_repositories,
             optional: true
  has_many :team_shared_objects, as: :shared_object, dependent: :destroy
  has_many :teams_shared_with, through: :team_shared_objects, source: :team, dependent: :destroy
  has_many :repository_snapshots,
           class_name: 'RepositorySnapshot',
           foreign_key: :parent_id,
           inverse_of: :original_repository
  has_many :repository_ledger_records, as: :reference, dependent: :nullify
  has_many :repository_table_filters, dependent: :destroy

  before_save :sync_name_with_snapshots, if: :name_changed?
  before_destroy :refresh_report_references_on_destroy, prepend: true
  after_save :assign_globally_shared_inventories, if: -> { saved_change_to_permission_level? && globally_shared? }
  after_save :unassign_globally_shared_inventories, if: -> { saved_change_to_permission_level? && !globally_shared? }
  after_save :unassign_unshared_items, if: :saved_change_to_permission_level
  after_save :unlink_unshared_items, if: -> { saved_change_to_permission_level? && !globally_shared? }

  validates :name,
            presence: true,
            uniqueness: { scope: :team_id, case_sensitive: false },
            length: { maximum: Constants::NAME_MAX_LENGTH }

  scope :active, -> { where(archived: false) }
  scope :archived, -> { where(archived: true) }
  scope :globally_shared, -> { where(permission_level: %i(shared_read shared_write)) }

  scope :accessible_by_teams, lambda { |teams|
    accessible_repositories = left_outer_joins(:team_shared_objects)
    accessible_repositories =
      accessible_repositories
      .where(team: teams)
      .or(accessible_repositories.where(team_shared_objects: { team: teams }))
      .or(accessible_repositories
            .where(permission_level: [Extends::SHARED_OBJECTS_PERMISSION_LEVELS[:shared_read],
                                      Extends::SHARED_OBJECTS_PERMISSION_LEVELS[:shared_write]]))
    accessible_repositories.distinct
  }

  scope :assigned_to_project, lambda { |project|
    accessible_by_teams(project.team)
      .joins(repository_rows: { my_module_repository_rows: { my_module: { experiment: :project } } })
      .where(repository_rows: { my_module_repository_rows: { my_module: { experiments: { project: project } } } })
  }

  def self.within_global_limits?
    return true unless Rails.configuration.x.global_repositories_limit.positive?

    count < Rails.configuration.x.global_repositories_limit
  end

  def self.within_team_limits?(team)
    return true unless Rails.configuration.x.team_repositories_limit.positive?

    team.repositories.count < Rails.configuration.x.team_repositories_limit
  end

  def self.search(
    user,
    query = nil,
    page = 1,
    repository = nil,
    options = {}
  )
    serchable_row_fields = [RepositoryRow::PREFIXED_ID_SQL, 'repository_rows.name', 'users.full_name']

    repositories = repository&.id || Repository.accessible_by_teams(user.teams).pluck(:id)

    readable_rows = RepositoryRow.joins(:repository, :created_by).where(repository_id: repositories)

    repository_rows = readable_rows.where_attributes_like(serchable_row_fields, query, options)

    Extends::REPOSITORY_EXTRA_SEARCH_ATTR.each do |_data_type, config|
      custom_cell_matches = readable_rows.joins(config[:includes])
                                         .where_attributes_like(config[:field], query, options)
      repository_rows = repository_rows.or(readable_rows.where(id: custom_cell_matches))
    end

    # Show all results if needed
    if page == Constants::SEARCH_NO_LIMIT
      repository_rows.select('repositories.id AS id, COUNT(DISTINCT repository_rows.id) AS counter')
                     .group('repositories.id')
    else
      repository_rows.limit(Constants::SEARCH_LIMIT).offset((page - 1) * Constants::SEARCH_LIMIT)
    end
  end

  def self.filter_by_teams(teams = [])
    teams.blank? ? self : where(team: teams)
  end

  def shareable_write?
    true
  end

  def permission_parent
    team
  end

  def default_table_state
    Constants::REPOSITORY_TABLE_DEFAULT_STATE
  end

  def default_sortable_columns
    [
      'assigned',
      'repository_rows.id',
      'repository_rows.name',
      'relationships',
      'repository_rows.created_at',
      'users.full_name',
      'repository_rows.archived_on',
      'archived_bies_repository_rows.full_name'
    ]
  end

  def default_search_fileds
    ['repository_rows.name', RepositoryRow::PREFIXED_ID_SQL, 'users.full_name']
  end

  def i_shared?(team)
    shared_with_anybody? && self.team == team
  end

  def globally_shared?
    shared_read? || shared_write?
  end

  def shared_with_anybody?
    (!not_shared? || team_shared_objects.any?)
  end

  def shared_with?(team)
    return false if self.team == team

    !not_shared? || private_shared_with?(team)
  end

  def shared_with_write?(team)
    return false if self.team == team

    shared_write? || private_shared_with_write?(team)
  end

  def shared_with_read?(team)
    return false if self.team == team

    shared_read? || team_shared_objects.where(team: team, permission_level: :shared_read).any?
  end

  def private_shared_with?(team)
    team_shared_objects.where(team: team).any?
  end

  def private_shared_with_write?(team)
    team_shared_objects.where(team: team, permission_level: :shared_write).any?
  end

  def self.viewable_by_user(_user, teams)
    accessible_by_teams(teams)
  end

  def self.name_like(query)
    where('repositories.name ILIKE ?', "%#{query}%")
  end

  def importable_repository_fields
    fields = {}
    # First and foremost add record name
    fields['-1'] = I18n.t('repositories.default_column')
    # Add all other custom columns
    repository_columns.order(:created_at).each do |rc|
      next unless rc.importable?

      fields[rc.id] = rc.name
    end
    fields
  end

  def copy(created_by, new_name)
    new_repo = nil

    begin
      Repository.transaction do
        # Clone the repository object
        new_repo = Repository.create!(name: new_name, team:, created_by:)

        # Clone columns (only if new_repo was saved)
        repository_columns.find_each do |col|
          new_col = col.dup
          new_col.repository = new_repo
          new_col.created_by = created_by
          new_col.save!
        end
      end
    rescue ActiveRecord::RecordInvalid
      return false
    end

    # If everything is okay, return new_repo
    Activities::CreateActivityService
      .call(activity_type: :copy_inventory,
            owner: created_by,
            subject: new_repo,
            team: new_repo.team,
            message_items: { repository_new: new_repo.id, repository_original: id })

    new_repo
  end

  def import_records(sheet, mappings, user, can_edit_existing_items, should_overwrite_with_empty_cells)
    importer = RepositoryImportParser::Importer.new(sheet, mappings, user, self)
    importer.run(can_edit_existing_items, should_overwrite_with_empty_cells)
  end

  def assigned_rows(my_module)
    repository_rows.joins(:my_module_repository_rows).where(my_module_repository_rows: { my_module_id: my_module.id })
  end

  def unassign_unshared_items
    return if shared_read? || shared_write?

    MyModuleRepositoryRow.joins(my_module: { experiment: { project: :team } })
                         .joins(repository_row: :repository)
                         .where(repository_rows: { repository: self })
                         .where.not(my_module: { experiment: { projects: { team: team } } })
                         .where.not(my_module: { experiment: { projects: { team: teams_shared_with } } })
                         .destroy_all
  end

  def unlink_unshared_items
    repository_rows_ids = repository_rows.select(:id)
    rows_to_unlink = RepositoryRow.joins("LEFT JOIN repository_row_connections \
                                         ON repository_rows.id = repository_row_connections.parent_id \
                                         OR repository_rows.id = repository_row_connections.child_id")
                                  .where("repository_row_connections.parent_id IN (?) \
                                         OR repository_row_connections.child_id IN (?)",
                                         repository_rows_ids,
                                         repository_rows_ids)
                                  .joins(:repository)
                                  .where.not(repositories: self)
                                  .where.not(repositories: { team: team })
                                  .distinct

    RepositoryRowConnection.where(parent: repository_rows_ids, child: rows_to_unlink)
                           .destroy_all
    RepositoryRowConnection.where(child: repository_rows_ids, parent: rows_to_unlink)
                           .destroy_all
  end

  def archived_branch?
    archived?
  end

  private

  def sync_name_with_snapshots
    repository_snapshots.update(name: name)
  end

  def assign_globally_shared_inventories
    viewer_role = UserRole.find_by(name: UserRole.public_send('viewer_role').name)
    normal_user_role = UserRole.find_by(name: UserRole.public_send('normal_user_role').name)

    team_shared_objects.find_each(&:destroy!)

    Team.where.not(id: team.id).find_each do |team|
      team.users.find_each do |user|
        team.repository_sharing_user_assignments.find_or_initialize_by(
          user: user,
          assignable: self
        ).update!(user_role: shared_write? ? normal_user_role : viewer_role)
      end
    end
  end

  def unassign_globally_shared_inventories
    user_assignments.where.not(team: team).find_each(&:destroy!)
  end

  def refresh_report_references_on_destroy
    report_elements.find_each do |report_element|
      repository_snapshot = report_element.my_module
                                          .repository_snapshots
                                          .where(original_repository: self)
                                          .order(:selected, created_at: :desc)
                                          .first
      report_element.update(repository: repository_snapshot) if repository_snapshot
    end
  end
end