module AwesomePrint::Nokogiri

def self.included(base)
  base.send :alias_method, :cast_without_nokogiri, :cast
  base.send :alias_method, :cast, :cast_with_nokogiri
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

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