require 'set'

class Array
  def self.to_mongo(value)
    value = value.respond_to?(:lines) ? value.lines : value
    value.to_a
  end
  
  def self.from_mongo(value)
    value || []
  end
end

class Binary
  def self.to_mongo(value)
    if value.is_a?(ByteBuffer)
      value
    else
      value.nil? ? nil : ByteBuffer.new(value)
    end
  end

  def self.from_mongo(value)
    value
  end
end

class Boolean
  def self.to_mongo(value)
    if value.is_a?(Boolean)
      value
    else
      ['true', 't', '1'].include?(value.to_s.downcase)
    end
  end

  def self.from_mongo(value)
    !!value
  end
end

class Date
  def self.to_mongo(value)
    date = Date.parse(value.to_s)
    Time.utc(date.year, date.month, date.day)
  rescue
    nil
  end
  
  def self.from_mongo(value)
    value.to_date if value.present?
  end
end

class Float
  def self.to_mongo(value)
    value.to_f
  end
end

class Hash
  def self.from_mongo(value)
    HashWithIndifferentAccess.new(value || {})
  end
  
  def to_mongo
    self
  end
end

class Integer
  def self.to_mongo(value)
    value_to_i = value.to_i
    if value_to_i == 0
      value.to_s =~ /^(0x|0b)?0+/ ? 0 : nil
    else
      value_to_i
    end
  end
end

class NilClass
  def to_mongo(value)
    value
  end
  
  def from_mongo(value)
    value
  end
end

class Object
  # The hidden singleton lurks behind everyone
  def metaclass
    class << self; self end
  end

  def meta_eval(&blk)
    metaclass.instance_eval(&blk)
  end

  # Adds methods to a metaclass
  def meta_def(name, &blk)
    meta_eval { define_method(name, &blk) }
  end

  # Defines an instance method within a class
  def class_def(name, &blk)
    class_eval { define_method(name, &blk) }
  end
  
  def self.to_mongo(value)
    value
  end
  
  def self.from_mongo(value)
    value
  end
end

class ObjectId
  def self.to_mongo(value)
    if value.blank?
      nil
    elsif value.is_a?(Mongo::ObjectID)
      value
    else
      Mongo::ObjectID.from_string(value.to_s)
    end
  end
  
  def self.from_mongo(value)
    value
  end
end

class Set
  def self.to_mongo(value)
    value.to_a
  end
  
  def self.from_mongo(value)
    Set.new(value || [])
  end
end

class String
  def self.to_mongo(value)
    value.nil? ? nil : value.to_s
  end
  
  def self.from_mongo(value)
    value.nil? ? nil : value.to_s
  end
end

class Symbol
  %w{gt lt gte lte ne in nin mod size where exists}.each do |operator|
    define_method operator do
      MongoMapper::FinderOperator.new(self, "$#{operator}")
    end
  end
  
  def asc;  MongoMapper::OrderOperator.new(self, 'asc') end
  def desc; MongoMapper::OrderOperator.new(self, 'desc') end
end

class Time
  def self.to_mongo(value)
    if value.nil? || value == ''
      nil
    else
      time = MongoMapper.time_class.parse(value.to_s)
      time && time.utc
    end
  end
  
  def self.from_mongo(value)
    if MongoMapper.use_time_zone? && value.present?
      value.in_time_zone(Time.zone)
    else
      value
    end
  end
end

# TODO: Remove when patch accepted into driver
class Mongo::ObjectID
  def to_json(options = nil)
    %Q("#{to_s}")
  end
end