package ast

import (
	"fmt"
	"strings"

	"github.com/open-policy-agent/opa/v1/util"
)

// Dump returns a string representation of the tree structure rooted at this node.
func (n *TreeNode) Dump() string {
	var sb strings.Builder
	n.dumpRecursive(&sb, "", "")
	return sb.String()
}

func (n *TreeNode) dumpRecursive(sb *strings.Builder, prefix, childPrefix string) {
	sb.WriteString(prefix)
	fmt.Fprintf(sb, "%v", n.Key)

	if n.Hide {
		sb.WriteString(" [hidden]")
	}
	if n.External != nil {
		fmt.Fprintf(sb, " ext:%v", n.External.Ref)
	}
	if len(n.Values) > 0 {
		sb.WriteString(" rules:")
		util.WriteInt(sb, len(n.Values))
	}
	sb.WriteByte('\n')

	if len(n.Children) == 0 {
		return
	}

	keys := util.SortedFunc(util.Keys(n.Children), Value.Compare)

	for i, key := range keys {
		child := n.Children[key]
		isLast := i == len(keys)-1
		var newPrefix, newChildPrefix string
		if isLast {
			newPrefix = childPrefix + "└── "
			newChildPrefix = childPrefix + "    "
		} else {
			newPrefix = childPrefix + "├── "
			newChildPrefix = childPrefix + "│   "
		}
		child.dumpRecursive(sb, newPrefix, newChildPrefix)
	}
}