tree.go

package qjson

// JSONTree represent full json
type JSONTree struct {
	Root *Node
}

/* tree methods */

// IsNull tell node is null or not
func (tree *JSONTree) IsNull() bool {
	return tree.Root == nil || tree.Root.IsNull()
}

// MarshalJSON json marshaller
func (tree *JSONTree) MarshalJSON() ([]byte, error) {
	return tree.Root.MarshalJSON()
}

// UnmarshalJSON json unmarshaller
func (tree *JSONTree) UnmarshalJSON(data []byte) (err error) {
	var tree1 *JSONTree
	if tree1, err = Decode(data); err != nil {
		return err
	}
	*tree = *tree1
	return
}

// JSONString tree to string
func (tree *JSONTree) JSONString() string {
	return string(JSONMarshalWithPanic(tree))
}

// JSONIndentString tree to string with indent
func (tree *JSONTree) JSONIndentString() string {
	return string(JSONIndentMarshalWithPanic(tree))
}

// ColorfulMarshal print json with color
func (tree *JSONTree) ColorfulMarshal() []byte {
	return new(Formatter).Format(tree)
}

// ColorfulMarshalWithIndent print json with indent
func (tree *JSONTree) ColorfulMarshalWithIndent() []byte {
	return NewFormatter().Format(tree)
}

/* tree generator */
func makeNewTree() *JSONTree {
	return &JSONTree{Root: CreateNode()}
}

// Find json node/nodes by path selector
func (tree *JSONTree) Find(path string) *Node {
	p, ok := makeStPath(path)
	if !ok {
		return nil
	}
	return findNode(tree.Root, p)
}

// Remove json node
func (tree *JSONTree) Remove(path string) {
	paths, ok := makeStPath(path)
	if !ok {
		return
	}
	if len(paths) == 0 {
		return
	}
	if node := findNode(tree.Root, paths[:len(paths)-1]); node != nil {
		lastKey := paths[len(paths)-1]
		switch node.Type {
		case Object:
			node.RemoveObjectElemByKey(lastKey.Name)
		case Array:
			if lastKey.isInteger() {
				node.RemoveArrayElemByIndex(lastKey.asInteger())
			} else if lastKey.isArrayElemSelector() {
				node.clearArray()
			}
		}
	}
}

// Equal two json tree
func (tree *JSONTree) Equal(t2 *JSONTree) bool {
	return tree.Root.Equal(t2.Root)
}