require 'html5/treebuilders/base' module HTML5 module TreeBuilders module SimpleTree class Node < Base::Node # Node representing an item in the tree. # name - The tag name associated with the node attr_accessor :name # The value of the current node (applies to text nodes and # comments attr_accessor :value # a dict holding name, value pairs for attributes of the node attr_accessor :attributes def initialize name super @name = name @value = nil @attributes = {} end def appendChild node if node.kind_of? TextNode and childNodes.length > 0 and childNodes.last.kind_of? TextNode childNodes.last.value += node.value else childNodes << node end node.parent = self end def removeChild node childNodes.delete node node.parent = nil end def cloneNode newNode = self.class.new name attributes.each {|name,value| newNode.attributes[name] = value} newNode.value = value newNode end def insertText data, before=nil if before insertBefore TextNode.new(data), before else appendChild TextNode.new(data) end end def insertBefore node, refNode index = childNodes.index(refNode) if node.kind_of?(TextNode) && index > 0 && childNodes[index-1].kind_of?(TextNode) childNodes[index-1].value += node.value else childNodes.insert index, node end end def printTree indent=0 tree = "\n|%s%s" % [' '* indent, self.to_s] for child in childNodes tree += child.printTree(indent + 2) end return tree end def hasContent childNodes.length > 0 end end class Element < Node def to_s "<#{name}>" end def printTree indent=0 tree = "\n|%s%s" % [' '* indent, self.to_s] indent += 2 for name, value in attributes tree += "\n|%s%s=\"%s\"" % [' ' * indent, name, value] end for child in childNodes tree += child.printTree(indent) end tree end end class Document < Node def to_s "#document" end def initialize super nil end def printTree indent=0 tree = to_s for child in childNodes tree += child.printTree(indent + 2) end tree end end class DocumentType < Node attr_accessor :public_id, :system_id def to_s "" end def initialize name super name @public_id = nil @system_id = nil end end class DocumentFragment < Element def initialize super nil end def printTree indent=0 tree = "" for child in childNodes tree += child.printTree(indent+2) end return tree end end class TextNode < Node def initialize value super nil @value = value end def to_s '"%s"' % value end end class CommentNode < Node def initialize value super nil @value = value end def to_s "" % value end end class TreeBuilder < Base::TreeBuilder def initialize @documentClass = Document @doctypeClass = DocumentType @elementClass = Element @commentClass = CommentNode @fragmentClass = DocumentFragment end def testSerializer node node.printTree end def get_fragment @document = super @document.childNodes end end end end end