onthebeach/daodalus

module Daodalus
  module DAO
    include DSL

    def save(*args)
      collection.save(*args)
    end

    def find_one(*args)
      collection.find_one(*args)
    end

    def find(*args)
      collection.find(*args)
    end

    def insert(*args)
      collection.insert(*args)
    end

    def update(*args)
      collection.update(*args)
    end

    def remove(*args)
      collection.remove(*args)
    end

    def remove_all
      collection.remove
    end

    def count(*args)
      collection.find(*args).count
    end

    def find_and_modify(*args)
      collection.find_and_modify(*args)
    end

    def aggregate(*args)
      collection.aggregate(*args)
    end


    private

    def connection_name
      @connection_name || self.class.connection_name
    end

    def collection_name
      @collection_name || self.class.collection_name
    end

    def db
      Daodalus::Pool.instance[connection_name].db
    end

    def collection
      db[collection_name]
    end

    module Targetable
      def target(conn, coll)
        @connection_name = conn.to_s
        @collection_name = coll.to_s
      end
      attr_reader :collection_name
      attr_reader :connection_name
    end

    def self.included(base)
      base.extend(Targetable)
    end
    include Targetable

  end
end