module Nokogiri::Hpricot

def HTML(string)
  doc = Nokogiri::HTML::Document.parse(string)
  add_decorators(doc)
end

def XML(string)
  doc = Nokogiri::XML::Document.parse(string)
  add_decorators(doc)
end

def add_decorators(doc)
  doc.decorators(XML::Node) << Decorators::Hpricot::Node
  doc.decorators(XML::NodeSet) << Decorators::Hpricot::NodeSet
  doc.decorate!
  doc
end

def make string
  doc = XML::Document.new
  ns = XML::NodeSet.new(doc)
  ns << XML::Text.new(string, doc)
  ns
end

def parse(*args)
  doc = Nokogiri.parse(*args)
  add_decorators(doc)
end