Module: Amistad::FriendModel

Defined in:
lib/amistad/friend_model.rb

Class Method Summary collapse

Class Method Details

.included(receiver) ⇒ Object



3
4
5
6
7
8
9
10
11
12
13
14
15
# File 'lib/amistad/friend_model.rb', line 3

def self.included(receiver)
  if receiver.ancestors.map(&:to_s).include?("ActiveRecord::Base")
    receiver.class_exec do
      include Amistad::ActiveRecord::FriendModel
    end
  elsif receiver.ancestors.map(&:to_s).include?("Mongoid::Document")
    receiver.class_exec do
      include Amistad::Mongoid::FriendModel
    end
  else
    raise "Amistad only supports ActiveRecord and Mongoid"
  end
end