commando.go

package commando

import (
	"fmt"
	"os"
	"strings"

	"github.com/wh64dev/commando/option"
)

type Commando struct {
	args     []string
	internal *CommandoNode
	nodes    []CommandoNode
}

func CommandoNew() *Commando {
	return &Commando{
		internal: nil,
		args:     os.Args,
	}
}

func CommandoArgsGet(c *Commando) []string {
	return c.args
}

func CommandoNodesGet(c *Commando) []CommandoNode {
	return c.nodes
}

func CommandoInternalGet(c *Commando) *CommandoNode {
	return c.internal
}

func CommandoInternalSet(c *Commando, h ArgHandler, options ...option.OptionData) {
	var n = CommandoNodeNew(c, "", "")
	n.options = append(n.options, options...)
	n.handler = h
	c.internal = n
}

func CommandoNodeAppend(c *Commando, n *CommandoNode) {
	c.nodes = append(c.nodes, *n)
}

func handleError(c *Commando) error {
	return fmt.Errorf("%s: error: unknown command", c.args[0])
}

func execute(n *CommandoNode) error {
	return n.handler(n)
}

func executeSubnode(n *CommandoNode, args []string) error {
	if len(args) == 0 {
		if n.handler == nil {
			return handleError(n.Commando)
		}

		return execute(n)
	}

	var next = args[0]
	var remaining = args[1:]

	for _, sn := range n.subnodes {
		if sn.name != next {
			continue
		}

		return executeSubnode(&sn, remaining)
	}

	if n.handler == nil {
		return fmt.Errorf("%s: error: unknown command to '%s'", n.Commando.args[0], next)
	}

	return execute(n)
}

func CommandoExecute(c *Commando) error {
	if len(c.args) == 1 {
		if c.internal == nil {
			return handleError(c)
		}

		return execute(c.internal)
	}

	var current = c.args[1]
	var remaining = c.args[2:]

	var isOpt = strings.HasPrefix(c.args[1], "--") || strings.HasPrefix(c.args[1], "-")
	if isOpt {
		return execute(c.internal)
	}

	for _, n := range c.nodes {
		if n.name != current {
			continue
		}

		return executeSubnode(&n, remaining)
	}

	return handleError(c)
}