Added Nokogiri plugin

ryansch · Nov 30, 2011 · 89b8fc9 · 89b8fc9
1 parent f24462a
commit 89b8fc9
Show file tree

Hide file tree

Showing 3 changed files with 103 additions and 0 deletions.
diff --git a/lib/awesome_print/ext/mongoid.rb b/lib/awesome_print/ext/mongoid.rb
@@ -1,3 +1,8 @@
+# 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 AwesomePrint
   module Mongoid
 

diff --git a/lib/awesome_print/ext/nokogiri.rb b/lib/awesome_print/ext/nokogiri.rb
@@ -0,0 +1,45 @@
+# 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 AwesomePrint
+  module Nokogiri
+
+    def self.included(base)
+      base.send :alias_method, :cast_without_nokogiri, :cast
+      base.send :alias_method, :cast, :cast_with_nokogiri
+    end
+
+    # Add Nokogiri XML Node and NodeSet names to the dispatcher pipeline.
+    #------------------------------------------------------------------------------
+    def cast_with_nokogiri(object, type)
+      cast = cast_without_nokogiri(object, type)
+      if (defined?(::Nokogiri::XML::Node) && object.is_a?(::Nokogiri::XML::Node)) || 
+         (defined?(::Nokogiri::XML::NodeSet) && object.is_a?(::Nokogiri::XML::NodeSet))
+        cast = :nokogiri_xml_node
+      end
+      cast
+    end
+
+    #------------------------------------------------------------------------------
+    def awesome_nokogiri_xml_node(object)
+      if object.is_a?(::Nokogiri::XML::NodeSet) && object.empty?
+        return "[]"
+      end
+      xml = object.to_xml(:indent => 2)
+      #
+      # Colorize tag, id/class name, and contents.
+      #
+      xml.gsub!(/(<)(\/?[A-Za-z1-9]+)/) { |tag| "#{$1}#{colorize($2, :keyword)}" }
+      xml.gsub!(/(id|class)="[^"]+"/i) { |id| colorize(id, :class) }
+      xml.gsub!(/>([^<]+)</) do |contents|
+        contents = colorize($1, :trueclass) if contents && !contents.empty?
+        ">#{contents}<"
+      end
+      xml
+    end
+  end
+end
+
+AwesomePrint::Formatter.send(:include, AwesomePrint::Nokogiri)
diff --git a/spec/ext/nokogiri_spec.rb b/spec/ext/nokogiri_spec.rb
@@ -0,0 +1,53 @@
+require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
+
+begin
+  require "nokogiri"
+  require "awesome_print/ext/nokogiri"
+
+  describe "AwesomePrint/Nokogiri" do
+    before do
+      stub_dotfile!
+    end
+
+    it "should colorize tags" do
+      xml = Nokogiri::XML('<html><body><h1></h1></body></html>')
+      xml.ai.should ==  <<-EOS
+<?xml version=\"1.0\"?>\e[1;32m
+\e[0m<\e[1;36mhtml\e[0m>\e[1;32m
+  \e[0m<\e[1;36mbody\e[0m>\e[1;32m
+    \e[0m<\e[1;36mh1\e[0m/>\e[1;32m
+  \e[0m<\e[1;36m/body\e[0m>\e[1;32m
+\e[0m<\e[1;36m/html\e[0m>
+EOS
+    end
+
+    it "should colorize contents" do
+      xml = Nokogiri::XML('<html><body><h1>Hello</h1></body></html>')
+      xml.ai.should ==  <<-EOS
+<?xml version=\"1.0\"?>\e[1;32m
+\e[0m<\e[1;36mhtml\e[0m>\e[1;32m
+  \e[0m<\e[1;36mbody\e[0m>\e[1;32m
+    \e[0m<\e[1;36mh1\e[0m>\e[1;32mHello\e[0m<\e[1;36m/h1\e[0m>\e[1;32m
+  \e[0m<\e[1;36m/body\e[0m>\e[1;32m
+\e[0m<\e[1;36m/html\e[0m>
+EOS
+    end
+
+    it "should colorize class and id" do
+      xml = Nokogiri::XML('<html><body><h1><span id="hello" class="world"></span></h1></body></html>')
+      xml.ai.should ==  <<-EOS
+<?xml version=\"1.0\"?>\e[1;32m
+\e[0m<\e[1;36mhtml\e[0m>\e[1;32m
+  \e[0m<\e[1;36mbody\e[0m>\e[1;32m
+    \e[0m<\e[1;36mh1\e[0m>\e[1;32m
+      \e[0m<\e[1;36mspan\e[0m \e[1;33mid=\"hello\"\e[0m \e[1;33mclass=\"world\"\e[0m/>\e[1;32m
+    \e[0m<\e[1;36m/h1\e[0m>\e[1;32m
+  \e[0m<\e[1;36m/body\e[0m>\e[1;32m
+\e[0m<\e[1;36m/html\e[0m>
+EOS
+    end
+  end
+
+rescue LoadError => error
+  puts "Skipping Nokogiri specs: #{error}"
+end