module Saasable::Mongoid::ScopedDocument
  @scoped_documents = []

  def self.included klass
    @scoped_documents << klass unless @scoped_documents.include? klass

    klass.extend ClassMethods
    klass.class_eval do
      # Fields
      field :saas_id, :type => BSON::ObjectId

      # Indexes
      index({saas_id: 1})
      index({saad_id: 1, _id: 1})

      class << self
        alias_method_chain :index, :saasable
      end
    end
  end

  def self.scoped_documents
    @scoped_documents
  end

  def saas= a_saas
    self.saas_id = a_saas._id
  end

  def saas
    @saas ||= Saasable::Mongoid::SaasDocument.saas_document.find(self.saas_id)
  end

  module ClassMethods
    def validates_uniqueness_of(*args)
      attributes = _merge_attributes(args)
      attributes[:scope] ||= []
      attributes[:scope] << :saas_id unless attributes[:scope].include?(:saas_id)

      validates_with(Mongoid::Validatable::UniquenessValidator, attributes)
    end

    def index_with_saasable(spec, options = nil)
      index_without_saasable(spec, options.except(:unique)) # Never create this one with unique since it's only unique in the saas scope.
      index_without_saasable({saas_id: 1}.merge(spec), options) unless spec.include?(:saas_id)
    end
  end
end