require 'mongo_mapper'
require 'active_record'
require 'set'

require 'bigamy/proxy'
require 'bigamy/mongo'
require 'bigamy/ar'

module Bigamy
  class NewRecordAssignment < StandardError; end

  def self.setup *args
    args.each do |klass|
      case
      when klass.ancestors.include?(::ActiveRecord::Base)
        Bigamy::ActiveRecord.configure klass
      when (klass.included_modules & [::MongoMapper::Document, ::MongoMapper::EmbeddedDocument]).present?
        klass.plugin Bigamy::Mongo
      else
        raise "NO #{klass}"
      end
    end
  end

  module Base
    def self.configure(model)
      class << model
        def bigamy_associations
          @bigamy_associations ||= {}
        end
      end
    end
  end

  module Mongo

    def self.configure(model)
      ::Bigamy::Base.configure(model)
    end

    module ClassMethods
      def divorce_everyone
        self.bigamy_associations.each {|k,v| v.divorce_everyone }
        self.bigamy_associations.clear
      end

      def belongs_to_ar name, options = {}
        bigamy_associations[name] = MongoBelongsTo.new(self, name, options)
      end

      def has_one_ar name, options = {}
        bigamy_associations[name] = MongoHasOne.new(self, name, options)
      end

      def has_many_ar name, options = {}
        bigamy_associations[name] = MongoHasMany.new(self, name, options)
      end
    end

    module InstanceMethods
      def set_value c, val
        write_key c, val
      end

      def read_val c
        read_key c
      end

      def export_id_val i
        i.to_s
      end

      def import_id_val i
        i
      end
    end

    module PluginAddition
      def self.included(model)
        model.plugin Bigamy::Mongo
      end
    end
  end

  module ActiveRecord
    def self.configure(model)
      ::Bigamy::Base.configure(model)

      model.extend ClassMethods
      model.send :include, InstanceMethods
    end

    module ClassMethods
      def divorce_everyone
        self.bigamy_associations.each {|k,v| v.divorce_everyone }
        self.bigamy_associations.clear
      end

      def belongs_to_mm name, options = {}
        bigamy_associations[name] = ARBelongsTo.new(self, name, options)
      end

      def has_one_mm name, options = {}
        bigamy_associations[name] = ARHasOne.new(self, name, options)
      end

      def has_many_mm name, options = {}
        bigamy_associations[name] = ARHasMany.new(self, name, options)
      end
    end

    module InstanceMethods
      def set_value c, val
        self[c] = val
      end

      def read_val c
        read_attribute c
      end

      def export_id_val i
        i
      end

      def import_id_val i
        BSON::ObjectID.from_string(i)
      end

      def reset_current_relationships f_key, f_id
        collection.update({f_key => f_id}, {f_key => nil}, :multi => true, :upsert => false)
      end
    end
  end

end

MongoMapper::Document.append_inclusions(Bigamy::Mongo::PluginAddition)
Bigamy::ActiveRecord.configure(ActiveRecord::Base)