add support for mongo_mapper

lib/ap/mixin/mongo_mapper.rb

@@ -0,0 +1,57 @@
+# Copyright (c) 2010-2011 Michael Dvorkin
+#
+# Awesome Print is freely distributable under the terms of MIT license.
+# See LICENSE file or http://www.opensource.org/licenses/mit-license.php
+#------------------------------------------------------------------------------
+module AwesomePrintMongoMapper
+
+  def self.included(base)
+    base.send :alias_method, :printable_without_mongo_mapper, :printable
+    base.send :alias_method, :printable, :printable_with_mongo_mapper
+  end
+
+  # Add MongoMapper class names to the dispatcher pipeline.
+  #------------------------------------------------------------------------------
+  def printable_with_mongo_mapper(object)
+    printable = printable_without_mongo_mapper(object)
+    return printable if !defined?(MongoMapper::Document)
+    
+    is_mongo = object.class.include?(MongoMapper::Document) ||
+               object.class.include?(MongoMapper::EmbeddedDocument)
+
+    if printable == :self
+      if is_mongo
+        printable = :mongo_mapper_instance
+      end
+    elsif printable == :class and is_mongo
+      printable = :mongo_mapper_class
+    end
+    printable
+  end
+
+  # Format MongoMapper instance object.
+  #------------------------------------------------------------------------------
+  def awesome_mongo_mapper_instance(object)
+    return object.inspect if !defined?(ActiveSupport::OrderedHash)
+
+    data = object.keys.keys.inject(ActiveSupport::OrderedHash.new) do |hash, name|
+      hash[name] = object[name]
+      hash
+    end
+    "#{object} " + awesome_hash(data)
+  end
+
+  # Format MongoMapper class object.
+  #------------------------------------------------------------------------------
+  def awesome_mongo_mapper_class(object)
+    return object.inspect if !defined?(ActiveSupport::OrderedHash) || !object.respond_to?(:columns)
+
+    data = object.keys.inject(ActiveSupport::OrderedHash.new) do |hash, c|
+      hash[c.first] = c.last.class
+      hash
+    end
+    "class #{object} < #{object.superclass} " << awesome_hash(data)
+  end
+end
+
+AwesomePrint.send(:include, AwesomePrintMongoMapper)

lib/awesome_print.rb

@@ -22,3 +22,4 @@ require File.dirname(__FILE__) + "/ap/mixin/action_view" if defined?(ActionView)
 # console when required from ~/.irbrc.
 require File.dirname(__FILE__) + "/ap/mixin/active_record"  if defined?(ActiveRecord)  || (defined?(IRB) && ENV['RAILS_ENV'])
 require File.dirname(__FILE__) + "/ap/mixin/active_support" if defined?(ActiveSupport) || (defined?(IRB) && ENV['RAILS_ENV'])
+require File.dirname(__FILE__) + "/ap/mixin/mongo_mapper" if defined?(MongoMapper)