mirror of
https://github.com/scinote-eln/scinote-web.git
synced 2024-11-16 06:06:56 +08:00
75 lines
3.3 KiB
Ruby
75 lines
3.3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class TeamSharedObject < ApplicationRecord
|
|
enum permission_level: Extends::SHARED_OBJECTS_PERMISSION_LEVELS.except(:not_shared)
|
|
|
|
after_create :assign_shared_inventories, if: -> { shared_object.is_a?(Repository) }
|
|
before_destroy :unlink_unshared_items, if: -> { shared_object.is_a?(Repository) }
|
|
before_destroy :unassign_unshared_inventories, if: -> { shared_object.is_a?(Repository) }
|
|
|
|
belongs_to :team
|
|
belongs_to :shared_object, polymorphic: true, inverse_of: :team_shared_objects
|
|
belongs_to :shared_repository,
|
|
(lambda do |team_shared_object|
|
|
team_shared_object.shared_object_type == 'RepositoryBase' ? self : none
|
|
end),
|
|
optional: true,
|
|
class_name: 'RepositoryBase',
|
|
foreign_key: :shared_object_id
|
|
|
|
validates :permission_level, presence: true
|
|
validates :shared_object_type, uniqueness: { scope: %i(shared_object_id team_id) }
|
|
validate :team_cannot_be_the_same
|
|
validate :not_globally_shared, if: -> { shared_object.is_a?(Repository) }
|
|
|
|
private
|
|
|
|
def team_cannot_be_the_same
|
|
errors.add(:team_id, :same_team) if shared_object.team.id == team_id
|
|
end
|
|
|
|
def not_globally_shared
|
|
errors.add(:shared_object_id, :is_globally_shared) if shared_object.globally_shared?
|
|
end
|
|
|
|
def assign_shared_inventories
|
|
team.user_assignments.find_each do |user_assignment|
|
|
shared_object.user_assignments.create!(
|
|
user: user_assignment.user,
|
|
user_role: user_assignment.user_role,
|
|
team: team
|
|
)
|
|
end
|
|
end
|
|
|
|
def unassign_unshared_inventories
|
|
team.repository_sharing_user_assignments.where(assignable: shared_object).find_each(&:destroy!)
|
|
end
|
|
|
|
def unlink_unshared_items
|
|
# We keep all the other teams shared with and the repository's own team
|
|
teams_ids = shared_object.teams_shared_with.where.not(id: team).pluck(:id)
|
|
teams_ids << shared_object.team_id
|
|
repository_rows_ids = shared_object.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: { team: teams_ids })
|
|
.select(:id)
|
|
|
|
RepositoryRowConnection.where("(repository_row_connections.parent_id IN (?) \
|
|
AND repository_row_connections.child_id IN (?)) \
|
|
OR (repository_row_connections.parent_id IN (?) \
|
|
AND repository_row_connections.child_id IN (?))",
|
|
repository_rows_ids,
|
|
rows_to_unlink,
|
|
rows_to_unlink,
|
|
repository_rows_ids)
|
|
.destroy_all
|
|
end
|
|
end
|