commando/node.go at master · wh64dev/commando · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package commando

import (
	"strings"

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

type ArgHandler func(n *CommandoNode) error

type CommandoNode struct {
	name        string
	description string
	subnodes    []CommandoNode
	options     []option.OptionData
	handler     ArgHandler
	Commando    *Commando
}

func CommandoNodeNew(c *Commando, name, description string) *CommandoNode {
	return &CommandoNode{
		name:        name,
		description: description,
		subnodes:    make([]CommandoNode, 0),
		options:     make([]option.OptionData, 0),
		handler:     nil,
		Commando:    c,
	}
}

func CommandoNodeNameGet(n *CommandoNode) string {
	return n.name
}

func CommandoNodeNameSet(n *CommandoNode, name string) {
	n.name = name
}

func CommandoNodeDescriptionGet(n *CommandoNode) string {
	return n.description
}

func CommandoNodeDescriptionSet(n *CommandoNode, description string) {
	n.description = description
}

func CommandoNodeSubnodesGet(n *CommandoNode) []CommandoNode {
	return n.subnodes
}

func CommandoNodeSubnodeAppend(n *CommandoNode, subnode *CommandoNode) {
	n.subnodes = append(n.subnodes, *subnode)
}

func CommandoNodeOptionsGet(n *CommandoNode) []option.OptionData {
	return n.options
}

func CommandoNodeOptionFind(n *CommandoNode, name string) *option.OptionData {
	var option *option.OptionData = nil
	for _, opt := range n.options {
		var origin, _ = strings.CutPrefix(opt.Name, "--")
		origin, _ = strings.CutPrefix(origin, "-")

		if origin != name {
			continue
		}

		option = &opt
	}

	return option
}

func CommandoNodeOptionAppend(n *CommandoNode, o *option.OptionData) {
	n.options = append(n.options, *o)
}

func CommandoNodeOptionsAppend(n *CommandoNode, o ...option.OptionData) {
	n.options = append(n.options, o...)
}

func CommandoNodeHandlerGet(n *CommandoNode) ArgHandler {
	return n.handler
}

func CommandoNodeHandlerSet(n *CommandoNode, h ArgHandler) {
	n.handler = h
}