class Tilt::NokogiriTemplate

def evaluate(scope, locals)
  if @data.respond_to?(:to_str)
    if block_given?
      super(scope, locals){yield.sub(DOCUMENT_HEADER, "")}
    else
      super
    end
  else
    ::Nokogiri::XML::Builder.new(&@data).to_xml
  end
end

def precompiled_postamble(locals)
  "xml.to_xml"
end

def precompiled_preamble(locals)
  return super if locals.include? :xml
  "xml = ::Nokogiri::XML::Builder.new { |xml| }\n#{super}"
end

def precompiled_template(locals)
  @data.to_str
end