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