module Nokogiri::Hpricot

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

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

def add_decorators(doc)
  doc.decorators['node'] << Decorators::Hpricot::Node
  doc.decorators['element'] << Decorators::Hpricot::Node
  doc.decorators['document'] << Decorators::Hpricot::Node
  doc.decorators['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