rosa-build/app/models/relation.rb

# -*- encoding : utf-8 -*-
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
    Relation.by_actor(member).by_target(target).each{|r| r.destroy}
  end

  protected

  def add_default_role
    self.role = ROLES.first if role.nil? || role.empty?
  end
end