module AmazingPrint::MongoMapper

def self.included(base)
  base.send :alias_method, :cast_without_mongo_mapper, :cast
  base.send :alias_method, :cast, :cast_with_mongo_mapper
end

def apply_default_mongo_mapper_options
  @options[:color][:assoc] ||= :greenish
  @options[:mongo_mapper]  ||= {
    show_associations: false, # Display association data for MongoMapper documents and classes.
    inline_embedded: false    # Display embedded associations inline with MongoMapper documents.
  }
end

def awesome_mongo_mapper_association(object)
  return object.inspect unless defined?(::ActiveSupport::OrderedHash)
  return awesome_object(object) if @options[:raw]
  association = object.class.name.split('::').last.titleize.downcase.sub(/ association$/, '')
  association = "embeds #{association}" if object.embeddable?
  class_name = object.class_name
  "#{colorize(association, :assoc)} #{colorize(class_name, :class)}"
end

def awesome_mongo_mapper_bson_id(object)
  object.inspect
end

def awesome_mongo_mapper_class(object)
  if !defined?(::ActiveSupport::OrderedHash) || !object.respond_to?(:keys)
    return object.inspect
  end
  data = object.keys.sort.each_with_object(::ActiveSupport::OrderedHash.new) do |c, hash|
    hash[c.first] = (c.last.type || 'undefined').to_s.underscore.intern
  end
  # Add in associations
  if @options[:mongo_mapper][:show_associations]
    object.associations.each do |name, assoc|
      data[name.to_s] = assoc
    end
  end
  name = "class #{awesome_simple(object.to_s, :class)}"
  base = "< #{awesome_simple(object.superclass.to_s, :class)}"
  [name, base, awesome_hash(data)].join(' ')
end

def awesome_mongo_mapper_instance(object)
  return object.inspect unless defined?(::ActiveSupport::OrderedHash)
  return awesome_object(object) if @options[:raw]
  data = object.keys.keys.sort.each_with_object(::ActiveSupport::OrderedHash.new) do |name, hash|
    hash[name] = object[name]
  end
  # Add in associations
  if @options[:mongo_mapper][:show_associations]
    object.associations.each do |name, assoc|
      data[name.to_s] = if @options[:mongo_mapper][:inline_embedded] && assoc.embeddable?
                          object.send(name)
                        else
                          assoc
                        end
    end
  end
  label = object.to_s
  if object.is_a?(::MongoMapper::EmbeddedDocument)
    label = "#{colorize('embedded', :assoc)} #{label}"
  end
  "#{label} " << awesome_hash(data)
end

def cast_with_mongo_mapper(object, type)
  apply_default_mongo_mapper_options
  cast = cast_without_mongo_mapper(object, type)
  if defined?(::MongoMapper::Document)
    if object.is_a?(Class) && !(object.ancestors & [::MongoMapper::Document, ::MongoMapper::EmbeddedDocument]).empty?
      cast = :mongo_mapper_class
    elsif object.is_a?(::MongoMapper::Document) || object.is_a?(::MongoMapper::EmbeddedDocument)
      cast = :mongo_mapper_instance
    elsif object.is_a?(::MongoMapper::Plugins::Associations::Base)
      cast = :mongo_mapper_association
    elsif object.is_a?(::BSON::ObjectId)
      cast = :mongo_mapper_bson_id
    end
  end
  cast
end