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