instiki/vendor/plugins/HTML5lib/lib/html5lib/treebuilders/simpletree.rb

require 'html5lib/treebuilders/base'

module HTML5lib
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[-1].kind_of? TextNode
            childNodes[-1].value += node.value
        else
            childNodes.push 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 and index>0 and 
          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
        return (childNodes.length > 0)
    end
end

class Element < Node
    def to_s
       "<%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
        return 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
        return tree
    end
end

class DocumentType < Node
    def to_s
       "<!DOCTYPE %s>" % name
    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
        "<!-- %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 getFragment
        @document = super
        return @document.childNodes
    end
end

end
end
end