class Nokogiri::XML::DocumentFragment

def coerce data
  return super unless String === data
  document.fragment(data).children
end

def css *args
  if children.any?
    children.css(*args)
  else
    NodeSet.new(document)
  end
end

def initialize document, tags = nil, ctx = nil
  return self unless tags
  children = if ctx
               # Fix for issue#490
               if Nokogiri.jruby?
                 # fix for issue #770
                 ctx.parse("<root #{namespace_declarations(ctx)}>#{tags}</root>").children
               else
                 ctx.parse(tags)
               end
             else
               XML::Document.parse("<root>#{tags}</root>") \
                 .xpath("/root/node()")
             end
  children.each { |child| child.parent = self }
end

def name
  '#document-fragment'
end

def namespace_declarations ctx
  ctx.namespace_scopes.map do |namespace|
    prefix = namespace.prefix.nil? ? "" : ":#{namespace.prefix}"
    %Q{xmlns#{prefix}="#{namespace.href}"}
  end.join ' '
end

def parse tags
  self.new(XML::Document.new, tags)
end

def to_html *args
  if Nokogiri.jruby?
    options = args.first.is_a?(Hash) ? args.shift : {}
    if !options[:save_with]
      options[:save_with] = Node::SaveOptions::NO_DECLARATION | Node::SaveOptions::NO_EMPTY_TAGS | Node::SaveOptions::AS_HTML
    end
    args.insert(0, options)
  end
  children.to_html(*args)
end

def to_s
  children.to_s
end

def to_xhtml *args
  if Nokogiri.jruby?
    options = args.first.is_a?(Hash) ? args.shift : {}
    if !options[:save_with]
      options[:save_with] = Node::SaveOptions::NO_DECLARATION | Node::SaveOptions::NO_EMPTY_TAGS | Node::SaveOptions::AS_XHTML
    end
    args.insert(0, options)
  end
  children.to_xhtml(*args)
end

def to_xml *args
  children.to_xml(*args)
end