49 lines
1.6 KiB
Ruby
49 lines
1.6 KiB
Ruby
class Relation < ActiveRecord::Base
|
|
belongs_to :target, polymorphic: true
|
|
belongs_to :actor, polymorphic: true
|
|
|
|
ROLES = %w[reader writer admin]
|
|
validates :role, inclusion: {in: ROLES}
|
|
|
|
# validate { errors.add(:actor, :taken) if Relation.where(actor_type: self.actor_type, actor_id: self.actor_id).present? }
|
|
before_validation :add_default_role
|
|
|
|
scope :by_user_through_groups, lambda {|u| where("actor_type = 'User' AND actor_id = ? OR actor_type = 'Group' AND actor_id IN (?)", u.id, u.group_ids)}
|
|
scope :by_actor, lambda {|obj| where(actor_id: obj.id, actor_type: obj.class.to_s)}
|
|
scope :by_target, lambda {|tar| where(target_id: tar.id, target_type: tar.class.to_s)}
|
|
scope :by_role, lambda {|role| where(role: role)}
|
|
|
|
def self.create_with_role(actor, target, role)
|
|
r = self.new
|
|
r.actor = actor
|
|
r.target = target
|
|
r.role = role
|
|
r.save
|
|
end
|
|
|
|
def self.add_member(member, target, role, relation = :relations)
|
|
if target.send(relation).exists?(actor_id: member.id, actor_type: member.class.to_s) || (target.respond_to?(:owner) && target.owner == member)
|
|
true
|
|
else
|
|
rel = target.send(relation).build(role: role)
|
|
rel.actor = member
|
|
rel.save
|
|
end
|
|
end
|
|
|
|
def self.remove_member(member, target)
|
|
return false if target.respond_to?(:owner) && target.owner == member
|
|
res = Relation.by_actor(member).by_target(target).each{|r| r.destroy}
|
|
if member.is_a?(User) && ['Project', 'Group'].include?(target.class.name)
|
|
member.check_assigned_issues target
|
|
end
|
|
res
|
|
end
|
|
|
|
protected
|
|
|
|
def add_default_role
|
|
self.role = ROLES.first if role.nil? || role.empty?
|
|
end
|
|
end
|