lib/mongoid/tenancy.rb in mongoid-tenant-0.0.5 vs lib/mongoid/tenancy.rb in mongoid-tenant-0.0.7

@@ -1,24 +1,49 @@
 module Mongoid
+  #
+  # Tenancy Module
+  #
+  # Provides #tenant_key and #tenancy!
+  #
   module Tenancy
     extend ActiveSupport::Concern
+    #
+    # Model instance
+    module ClassMethods
+      def tenant_key(key, options = {})
+        field key,    type: Symbol
+        options[:validates] ||= { presence: true }
+        options[:index] ||= {}
 
-    included do
-      field :uri,     type: String
+        validates key, { uniqueness: true }.merge(options[:validates])
 
-      validates :uri, uniqueness: true
+        index({ key => 1 }, {  unique: true }.merge(options[:index]))
 
-      index({ uri: 1 }, unique: true)
+        define_method(:tenant_key) do
+          send(key).to_s
+        end
 
-      def self.tenants
-        all.each do |t|
-          t.tenancy!
-          yield t
+        define_singleton_method(:clear_tenancy!) do
+          Thread.current[:tenancy] = nil
         end
+
+        define_singleton_method(:with_tenants) do |&block|
+          all.each do |t|
+            t.tenancy!
+            block.call(t)
+          end
+          clear_tenancy!
+        end
       end
+
+      def has_tenant(relative)
+        define_method(relative) do
+          tenancy! && relative.to_s.classify.constantize
+        end
+      end
     end
 
     def tenancy!
-      Thread.current[:mongodb] = _id.to_s
+      Thread.current[:tenancy] = tenant_key
     end
   end # Tenancy
 end # Mongoid