Compare commits
34 Commits
d2405f75d8
...
arghing
Author | SHA1 | Date | |
---|---|---|---|
3de96b813f
|
|||
a73acedc6d
|
|||
92e3d6fe5b
|
|||
395006cdb8
|
|||
622d47071a
|
|||
1452d544bb
|
|||
56c6a8cf09
|
|||
cc29386cee
|
|||
95453bf197
|
|||
a6526af6ff
|
|||
3ea30a997a
|
|||
03edacc8ec
|
|||
d1ffbe25a3
|
|||
dc3a40b19d
|
|||
f2e0de1b66
|
|||
58842504c4
|
|||
1080737931
|
|||
de6e907c60
|
|||
9989801e62
|
|||
8c280c303e
|
|||
b2e61cd0d2
|
|||
c15bafe55d
|
|||
af7d5c6e14
|
|||
512eddc1ab
|
|||
2bc9788441
|
|||
c52d9b9563
|
|||
fbb04df86d
|
|||
e30cbe2aca
|
|||
116ad347db
|
|||
1493deac96
|
|||
a7e45b8add
|
|||
f1cc614836
|
|||
a2661f4369
|
|||
b020618d57
|
1
.gitignore
vendored
1
.gitignore
vendored
@@ -1,5 +1,6 @@
|
|||||||
*.hex
|
*.hex
|
||||||
*.log
|
*.log
|
||||||
|
*.out
|
||||||
*env
|
*env
|
||||||
.dep
|
.dep
|
||||||
**/target/
|
**/target/
|
||||||
|
2
LICENSE
2
LICENSE
@@ -1,4 +1,4 @@
|
|||||||
Copyright (C) 2021 Dan Buch
|
Copyright (C) 2022 Dan Buch
|
||||||
|
|
||||||
MIT License
|
MIT License
|
||||||
|
|
||||||
|
4
argh/README.md
Normal file
4
argh/README.md
Normal file
@@ -0,0 +1,4 @@
|
|||||||
|
# argh command line parser
|
||||||
|
|
||||||
|
> NOTE: much of this is lifted from
|
||||||
|
> https://blog.gopheracademy.com/advent-2014/parsers-lexers/
|
34
argh/argh.go
Normal file
34
argh/argh.go
Normal file
@@ -0,0 +1,34 @@
|
|||||||
|
package argh
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"runtime"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
tracingEnabled = os.Getenv("ARGH_TRACING") == "enabled"
|
||||||
|
traceLogger *log.Logger
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
if !tracingEnabled {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
traceLogger = log.New(os.Stderr, "ARGH TRACING: ", 0)
|
||||||
|
}
|
||||||
|
|
||||||
|
func tracef(format string, v ...any) {
|
||||||
|
if !tracingEnabled {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, file, line, ok := runtime.Caller(1); ok {
|
||||||
|
format = fmt.Sprintf("%v:%v ", filepath.Base(file), line) + format
|
||||||
|
}
|
||||||
|
|
||||||
|
traceLogger.Printf(format, v...)
|
||||||
|
}
|
46
argh/cmd/argh-example/main.go
Normal file
46
argh/cmd/argh-example/main.go
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"git.meatballhat.com/x/box-o-sand/argh"
|
||||||
|
"github.com/davecgh/go-spew/spew"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
asJSON := os.Getenv("ARGH_OUTPUT_JSON") == "enabled"
|
||||||
|
|
||||||
|
log.SetFlags(0)
|
||||||
|
|
||||||
|
pt, err := argh.ParseArgs(os.Args, argh.NewParserConfig(
|
||||||
|
&argh.CommandConfig{
|
||||||
|
NValue: argh.OneOrMoreValue,
|
||||||
|
ValueNames: []string{"topping"},
|
||||||
|
Flags: &argh.Flags{
|
||||||
|
Automatic: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
nil,
|
||||||
|
))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ast := argh.NewQuerier(pt.Nodes).AST()
|
||||||
|
|
||||||
|
if asJSON {
|
||||||
|
b, err := json.MarshalIndent(ast, "", " ")
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Println(string(b))
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
spew.Dump(ast)
|
||||||
|
}
|
12
argh/go.mod
Normal file
12
argh/go.mod
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
module git.meatballhat.com/x/box-o-sand/argh
|
||||||
|
|
||||||
|
go 1.18
|
||||||
|
|
||||||
|
require github.com/pkg/errors v0.9.1
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/davecgh/go-spew v1.1.0 // indirect
|
||||||
|
github.com/pmezard/go-difflib v1.0.0 // indirect
|
||||||
|
github.com/stretchr/testify v1.7.1 // indirect
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c // indirect
|
||||||
|
)
|
12
argh/go.sum
Normal file
12
argh/go.sum
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
|
||||||
|
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
|
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
|
||||||
|
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
github.com/stretchr/testify v1.7.1 h1:5TQK59W5E3v0r2duFAb7P95B6hEeOyEnHRa8MjYSMTY=
|
||||||
|
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
|
||||||
|
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
46
argh/node.go
Normal file
46
argh/node.go
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
package argh
|
||||||
|
|
||||||
|
type Node interface{}
|
||||||
|
|
||||||
|
type TypedNode struct {
|
||||||
|
Type string
|
||||||
|
Node Node
|
||||||
|
}
|
||||||
|
|
||||||
|
type PassthroughArgs struct {
|
||||||
|
Nodes []Node
|
||||||
|
}
|
||||||
|
|
||||||
|
type CompoundShortFlag struct {
|
||||||
|
Nodes []Node
|
||||||
|
}
|
||||||
|
|
||||||
|
type Ident struct {
|
||||||
|
Literal string
|
||||||
|
}
|
||||||
|
|
||||||
|
type BadArg struct {
|
||||||
|
Literal string
|
||||||
|
From Pos
|
||||||
|
To Pos
|
||||||
|
}
|
||||||
|
|
||||||
|
type Command struct {
|
||||||
|
Name string
|
||||||
|
Values map[string]string
|
||||||
|
Nodes []Node
|
||||||
|
}
|
||||||
|
|
||||||
|
type Flag struct {
|
||||||
|
Name string
|
||||||
|
Values map[string]string
|
||||||
|
Nodes []Node
|
||||||
|
}
|
||||||
|
|
||||||
|
type StdinFlag struct{}
|
||||||
|
|
||||||
|
type StopFlag struct{}
|
||||||
|
|
||||||
|
type ArgDelimiter struct{}
|
||||||
|
|
||||||
|
type Assign struct{}
|
26
argh/nvalue_string.go
Normal file
26
argh/nvalue_string.go
Normal file
@@ -0,0 +1,26 @@
|
|||||||
|
// Code generated by "stringer -type NValue"; DO NOT EDIT.
|
||||||
|
|
||||||
|
package argh
|
||||||
|
|
||||||
|
import "strconv"
|
||||||
|
|
||||||
|
func _() {
|
||||||
|
// An "invalid array index" compiler error signifies that the constant values have changed.
|
||||||
|
// Re-run the stringer command to generate them again.
|
||||||
|
var x [1]struct{}
|
||||||
|
_ = x[OneOrMoreValue - -2]
|
||||||
|
_ = x[ZeroOrMoreValue - -1]
|
||||||
|
_ = x[ZeroValue-0]
|
||||||
|
}
|
||||||
|
|
||||||
|
const _NValue_name = "OneOrMoreValueZeroOrMoreValueZeroValue"
|
||||||
|
|
||||||
|
var _NValue_index = [...]uint8{0, 14, 29, 38}
|
||||||
|
|
||||||
|
func (i NValue) String() string {
|
||||||
|
i -= -2
|
||||||
|
if i < 0 || i >= NValue(len(_NValue_index)-1) {
|
||||||
|
return "NValue(" + strconv.FormatInt(int64(i+-2), 10) + ")"
|
||||||
|
}
|
||||||
|
return _NValue_name[_NValue_index[i]:_NValue_index[i+1]]
|
||||||
|
}
|
346
argh/parser.go
Normal file
346
argh/parser.go
Normal file
@@ -0,0 +1,346 @@
|
|||||||
|
package argh
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type parser struct {
|
||||||
|
s *Scanner
|
||||||
|
|
||||||
|
cfg *ParserConfig
|
||||||
|
|
||||||
|
errors ParserErrorList
|
||||||
|
|
||||||
|
tok Token
|
||||||
|
lit string
|
||||||
|
pos Pos
|
||||||
|
|
||||||
|
buffered bool
|
||||||
|
}
|
||||||
|
|
||||||
|
type ParseTree struct {
|
||||||
|
Nodes []Node `json:"nodes"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func ParseArgs(args []string, pCfg *ParserConfig) (*ParseTree, error) {
|
||||||
|
p := &parser{}
|
||||||
|
p.init(
|
||||||
|
strings.NewReader(strings.Join(args, string(nul))),
|
||||||
|
pCfg,
|
||||||
|
)
|
||||||
|
|
||||||
|
tracef("ParseArgs(...) parser=%+#v", p)
|
||||||
|
|
||||||
|
return p.parseArgs()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) addError(msg string) {
|
||||||
|
p.errors.Add(Position{Column: int(p.pos)}, msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) init(r io.Reader, pCfg *ParserConfig) {
|
||||||
|
p.errors = ParserErrorList{}
|
||||||
|
|
||||||
|
if pCfg == nil {
|
||||||
|
pCfg = POSIXyParserConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
p.cfg = pCfg
|
||||||
|
|
||||||
|
p.s = NewScanner(r, pCfg.ScannerConfig)
|
||||||
|
|
||||||
|
p.next()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) parseArgs() (*ParseTree, error) {
|
||||||
|
if p.errors.Len() != 0 {
|
||||||
|
tracef("parseArgs() bailing due to initial error")
|
||||||
|
return nil, p.errors.Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
tracef("parseArgs() parsing %q as program command; cfg=%+#v", p.lit, p.cfg.Prog)
|
||||||
|
prog := p.parseCommand(&p.cfg.Prog)
|
||||||
|
|
||||||
|
tracef("parseArgs() top level node is %T", prog)
|
||||||
|
|
||||||
|
nodes := []Node{prog}
|
||||||
|
if v := p.parsePassthrough(); v != nil {
|
||||||
|
tracef("parseArgs() appending passthrough argument %v", v)
|
||||||
|
nodes = append(nodes, v)
|
||||||
|
}
|
||||||
|
|
||||||
|
tracef("parseArgs() returning ParseTree")
|
||||||
|
|
||||||
|
return &ParseTree{Nodes: nodes}, p.errors.Err()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) next() {
|
||||||
|
tracef("next() before scan: %v %q %v", p.tok, p.lit, p.pos)
|
||||||
|
|
||||||
|
p.tok, p.lit, p.pos = p.s.Scan()
|
||||||
|
|
||||||
|
tracef("next() after scan: %v %q %v", p.tok, p.lit, p.pos)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) parseCommand(cCfg *CommandConfig) Node {
|
||||||
|
tracef("parseCommand(%+#v)", cCfg)
|
||||||
|
|
||||||
|
node := &Command{
|
||||||
|
Name: p.lit,
|
||||||
|
}
|
||||||
|
values := map[string]string{}
|
||||||
|
nodes := []Node{}
|
||||||
|
|
||||||
|
identIndex := 0
|
||||||
|
|
||||||
|
for i := 0; p.tok != EOL; i++ {
|
||||||
|
if !p.buffered {
|
||||||
|
tracef("parseCommand(...) buffered=false; scanning next")
|
||||||
|
p.next()
|
||||||
|
}
|
||||||
|
|
||||||
|
p.buffered = false
|
||||||
|
|
||||||
|
tracef("parseCommand(...) for=%d values=%+#v", i, values)
|
||||||
|
tracef("parseCommand(...) for=%d nodes=%+#v", i, nodes)
|
||||||
|
tracef("parseCommand(...) for=%d tok=%s lit=%q pos=%v", i, p.tok, p.lit, p.pos)
|
||||||
|
|
||||||
|
tracef("parseCommand(...) cCfg=%+#v", cCfg)
|
||||||
|
|
||||||
|
if subCfg, ok := cCfg.GetCommandConfig(p.lit); ok {
|
||||||
|
subCommand := p.lit
|
||||||
|
|
||||||
|
nodes = append(nodes, p.parseCommand(&subCfg))
|
||||||
|
|
||||||
|
tracef("parseCommand(...) breaking after sub-command=%v", subCommand)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
switch p.tok {
|
||||||
|
case ARG_DELIMITER:
|
||||||
|
tracef("parseCommand(...) handling %s", p.tok)
|
||||||
|
|
||||||
|
nodes = append(nodes, &ArgDelimiter{})
|
||||||
|
|
||||||
|
continue
|
||||||
|
case IDENT, STDIN_FLAG:
|
||||||
|
tracef("parseCommand(...) handling %s", p.tok)
|
||||||
|
|
||||||
|
if cCfg.NValue.Contains(identIndex) {
|
||||||
|
name := fmt.Sprintf("%d", identIndex)
|
||||||
|
|
||||||
|
tracef("parseCommand(...) checking for name of identIndex=%d", identIndex)
|
||||||
|
|
||||||
|
if len(cCfg.ValueNames) > identIndex {
|
||||||
|
name = cCfg.ValueNames[identIndex]
|
||||||
|
tracef("parseCommand(...) setting name=%s from config value names", name)
|
||||||
|
} else if len(cCfg.ValueNames) == 1 && (cCfg.NValue == OneOrMoreValue || cCfg.NValue == ZeroOrMoreValue) {
|
||||||
|
name = fmt.Sprintf("%s.%d", cCfg.ValueNames[0], identIndex)
|
||||||
|
tracef("parseCommand(...) setting name=%s from repeating value name", name)
|
||||||
|
}
|
||||||
|
|
||||||
|
values[name] = p.lit
|
||||||
|
}
|
||||||
|
|
||||||
|
if p.tok == STDIN_FLAG {
|
||||||
|
nodes = append(nodes, &StdinFlag{})
|
||||||
|
} else {
|
||||||
|
nodes = append(nodes, &Ident{Literal: p.lit})
|
||||||
|
}
|
||||||
|
|
||||||
|
identIndex++
|
||||||
|
case LONG_FLAG, SHORT_FLAG, COMPOUND_SHORT_FLAG:
|
||||||
|
tok := p.tok
|
||||||
|
|
||||||
|
flagNode := p.parseFlag(cCfg.Flags)
|
||||||
|
|
||||||
|
tracef("parseCommand(...) appending %s node=%+#v", tok, flagNode)
|
||||||
|
|
||||||
|
nodes = append(nodes, flagNode)
|
||||||
|
case ASSIGN:
|
||||||
|
tracef("parseCommand(...) error on bare %s", p.tok)
|
||||||
|
|
||||||
|
p.addError("invalid bare assignment")
|
||||||
|
|
||||||
|
break
|
||||||
|
default:
|
||||||
|
tracef("parseCommand(...) breaking on %s", p.tok)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(nodes) > 0 {
|
||||||
|
node.Nodes = nodes
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(values) > 0 {
|
||||||
|
node.Values = values
|
||||||
|
}
|
||||||
|
|
||||||
|
tracef("parseCommand(...) returning node=%+#v", node)
|
||||||
|
return node
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) parseIdent() Node {
|
||||||
|
node := &Ident{Literal: p.lit}
|
||||||
|
return node
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) parseFlag(flags *Flags) Node {
|
||||||
|
switch p.tok {
|
||||||
|
case SHORT_FLAG:
|
||||||
|
tracef("parseFlag(...) parsing short flag with config=%+#v", flags)
|
||||||
|
return p.parseShortFlag(flags)
|
||||||
|
case LONG_FLAG:
|
||||||
|
tracef("parseFlag(...) parsing long flag with config=%+#v", flags)
|
||||||
|
return p.parseLongFlag(flags)
|
||||||
|
case COMPOUND_SHORT_FLAG:
|
||||||
|
tracef("parseFlag(...) parsing compound short flag with config=%+#v", flags)
|
||||||
|
return p.parseCompoundShortFlag(flags)
|
||||||
|
}
|
||||||
|
|
||||||
|
panic(fmt.Sprintf("token %v cannot be parsed as flag", p.tok))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) parseShortFlag(flags *Flags) Node {
|
||||||
|
node := &Flag{Name: string(p.lit[1])}
|
||||||
|
|
||||||
|
flCfg, ok := flags.Get(node.Name)
|
||||||
|
if !ok {
|
||||||
|
p.addError(fmt.Sprintf("unknown flag %q", node.Name))
|
||||||
|
|
||||||
|
return node
|
||||||
|
}
|
||||||
|
|
||||||
|
return p.parseConfiguredFlag(node, flCfg)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) parseLongFlag(flags *Flags) Node {
|
||||||
|
node := &Flag{Name: string(p.lit[2:])}
|
||||||
|
|
||||||
|
flCfg, ok := flags.Get(node.Name)
|
||||||
|
if !ok {
|
||||||
|
p.addError(fmt.Sprintf("unknown flag %q", node.Name))
|
||||||
|
|
||||||
|
return node
|
||||||
|
}
|
||||||
|
|
||||||
|
return p.parseConfiguredFlag(node, flCfg)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) parseCompoundShortFlag(flags *Flags) Node {
|
||||||
|
flagNodes := []Node{}
|
||||||
|
|
||||||
|
withoutFlagPrefix := p.lit[1:]
|
||||||
|
|
||||||
|
for i, r := range withoutFlagPrefix {
|
||||||
|
node := &Flag{Name: string(r)}
|
||||||
|
|
||||||
|
if i == len(withoutFlagPrefix)-1 {
|
||||||
|
flCfg, ok := flags.Get(node.Name)
|
||||||
|
if !ok {
|
||||||
|
p.addError(fmt.Sprintf("unknown flag %q", node.Name))
|
||||||
|
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
flagNodes = append(flagNodes, p.parseConfiguredFlag(node, flCfg))
|
||||||
|
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
flagNodes = append(flagNodes, node)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &CompoundShortFlag{Nodes: flagNodes}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) parseConfiguredFlag(node *Flag, flCfg FlagConfig) Node {
|
||||||
|
values := map[string]string{}
|
||||||
|
nodes := []Node{}
|
||||||
|
|
||||||
|
identIndex := 0
|
||||||
|
|
||||||
|
for i := 0; p.tok != EOL; i++ {
|
||||||
|
if !flCfg.NValue.Contains(identIndex) {
|
||||||
|
tracef("parseConfiguredFlag(...) identIndex=%d exceeds expected=%v; breaking", identIndex, flCfg.NValue)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
p.next()
|
||||||
|
|
||||||
|
switch p.tok {
|
||||||
|
case ARG_DELIMITER:
|
||||||
|
nodes = append(nodes, &ArgDelimiter{})
|
||||||
|
|
||||||
|
continue
|
||||||
|
case ASSIGN:
|
||||||
|
nodes = append(nodes, &Assign{})
|
||||||
|
|
||||||
|
continue
|
||||||
|
case IDENT, STDIN_FLAG:
|
||||||
|
name := fmt.Sprintf("%d", identIndex)
|
||||||
|
|
||||||
|
tracef("parseConfiguredFlag(...) checking for name of identIndex=%d", identIndex)
|
||||||
|
|
||||||
|
if len(flCfg.ValueNames) > identIndex {
|
||||||
|
name = flCfg.ValueNames[identIndex]
|
||||||
|
tracef("parseConfiguredFlag(...) setting name=%s from config value names", name)
|
||||||
|
} else if len(flCfg.ValueNames) == 1 && (flCfg.NValue == OneOrMoreValue || flCfg.NValue == ZeroOrMoreValue) {
|
||||||
|
name = fmt.Sprintf("%s.%d", flCfg.ValueNames[0], identIndex)
|
||||||
|
tracef("parseConfiguredFlag(...) setting name=%s from repeating value name", name)
|
||||||
|
} else {
|
||||||
|
tracef("parseConfiguredFlag(...) setting name=%s", name)
|
||||||
|
}
|
||||||
|
|
||||||
|
values[name] = p.lit
|
||||||
|
|
||||||
|
if p.tok == STDIN_FLAG {
|
||||||
|
nodes = append(nodes, &StdinFlag{})
|
||||||
|
} else {
|
||||||
|
nodes = append(nodes, &Ident{Literal: p.lit})
|
||||||
|
}
|
||||||
|
|
||||||
|
identIndex++
|
||||||
|
default:
|
||||||
|
tracef("parseConfiguredFlag(...) breaking on %s %q %v; setting buffered=true", p.tok, p.lit, p.pos)
|
||||||
|
p.buffered = true
|
||||||
|
|
||||||
|
if len(nodes) > 0 {
|
||||||
|
node.Nodes = nodes
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(values) > 0 {
|
||||||
|
node.Values = values
|
||||||
|
}
|
||||||
|
|
||||||
|
return node
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(nodes) > 0 {
|
||||||
|
node.Nodes = nodes
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(values) > 0 {
|
||||||
|
node.Values = values
|
||||||
|
}
|
||||||
|
|
||||||
|
return node
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *parser) parsePassthrough() Node {
|
||||||
|
nodes := []Node{}
|
||||||
|
|
||||||
|
for ; p.tok != EOL; p.next() {
|
||||||
|
nodes = append(nodes, &Ident{Literal: p.lit})
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(nodes) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return &PassthroughArgs{Nodes: nodes}
|
||||||
|
}
|
146
argh/parser_config.go
Normal file
146
argh/parser_config.go
Normal file
@@ -0,0 +1,146 @@
|
|||||||
|
package argh
|
||||||
|
|
||||||
|
const (
|
||||||
|
OneOrMoreValue NValue = -2
|
||||||
|
ZeroOrMoreValue NValue = -1
|
||||||
|
ZeroValue NValue = 0
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
POSIXyParserConfig = NewParserConfig(
|
||||||
|
nil,
|
||||||
|
POSIXyScannerConfig,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
type NValue int
|
||||||
|
|
||||||
|
func (nv NValue) Contains(i int) bool {
|
||||||
|
tracef("NValue.Contains(%v)", i)
|
||||||
|
|
||||||
|
if i < int(ZeroValue) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
if nv == OneOrMoreValue || nv == ZeroOrMoreValue {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
return int(nv) > i
|
||||||
|
}
|
||||||
|
|
||||||
|
type ParserConfig struct {
|
||||||
|
Prog CommandConfig
|
||||||
|
|
||||||
|
ScannerConfig *ScannerConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewParserConfig(prog *CommandConfig, sCfg *ScannerConfig) *ParserConfig {
|
||||||
|
if sCfg == nil {
|
||||||
|
sCfg = POSIXyScannerConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
if prog == nil {
|
||||||
|
prog = &CommandConfig{}
|
||||||
|
}
|
||||||
|
|
||||||
|
prog.init()
|
||||||
|
|
||||||
|
pCfg := &ParserConfig{
|
||||||
|
Prog: *prog,
|
||||||
|
ScannerConfig: sCfg,
|
||||||
|
}
|
||||||
|
|
||||||
|
return pCfg
|
||||||
|
}
|
||||||
|
|
||||||
|
type CommandConfig struct {
|
||||||
|
NValue NValue
|
||||||
|
ValueNames []string
|
||||||
|
Flags *Flags
|
||||||
|
Commands *Commands
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cCfg *CommandConfig) init() {
|
||||||
|
if cCfg.ValueNames == nil {
|
||||||
|
cCfg.ValueNames = []string{}
|
||||||
|
}
|
||||||
|
|
||||||
|
if cCfg.Flags == nil {
|
||||||
|
cCfg.Flags = &Flags{}
|
||||||
|
}
|
||||||
|
|
||||||
|
if cCfg.Commands == nil {
|
||||||
|
cCfg.Commands = &Commands{}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cCfg *CommandConfig) GetCommandConfig(name string) (CommandConfig, bool) {
|
||||||
|
tracef("CommandConfig.GetCommandConfig(%q)", name)
|
||||||
|
|
||||||
|
if cCfg.Commands == nil {
|
||||||
|
cCfg.Commands = &Commands{Map: map[string]CommandConfig{}}
|
||||||
|
}
|
||||||
|
|
||||||
|
return cCfg.Commands.Get(name)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cCfg *CommandConfig) GetFlagConfig(name string) (FlagConfig, bool) {
|
||||||
|
tracef("CommandConfig.GetFlagConfig(%q)", name)
|
||||||
|
|
||||||
|
if cCfg.Flags == nil {
|
||||||
|
cCfg.Flags = &Flags{Map: map[string]FlagConfig{}}
|
||||||
|
}
|
||||||
|
|
||||||
|
return cCfg.Flags.Get(name)
|
||||||
|
}
|
||||||
|
|
||||||
|
type FlagConfig struct {
|
||||||
|
NValue NValue
|
||||||
|
Persist bool
|
||||||
|
ValueNames []string
|
||||||
|
}
|
||||||
|
|
||||||
|
type Flags struct {
|
||||||
|
Parent *Flags
|
||||||
|
Map map[string]FlagConfig
|
||||||
|
|
||||||
|
Automatic bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (fl *Flags) Get(name string) (FlagConfig, bool) {
|
||||||
|
tracef("Flags.Get(%q)", name)
|
||||||
|
|
||||||
|
if fl.Map == nil {
|
||||||
|
fl.Map = map[string]FlagConfig{}
|
||||||
|
}
|
||||||
|
|
||||||
|
flCfg, ok := fl.Map[name]
|
||||||
|
if !ok {
|
||||||
|
if fl.Automatic {
|
||||||
|
return FlagConfig{}, true
|
||||||
|
}
|
||||||
|
|
||||||
|
if fl.Parent != nil {
|
||||||
|
flCfg, ok = fl.Parent.Get(name)
|
||||||
|
return flCfg, ok && flCfg.Persist
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return flCfg, ok
|
||||||
|
}
|
||||||
|
|
||||||
|
type Commands struct {
|
||||||
|
Map map[string]CommandConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cmd *Commands) Get(name string) (CommandConfig, bool) {
|
||||||
|
tracef("Commands.Get(%q)", name)
|
||||||
|
|
||||||
|
if cmd.Map == nil {
|
||||||
|
cmd.Map = map[string]CommandConfig{}
|
||||||
|
}
|
||||||
|
|
||||||
|
cmdCfg, ok := cmd.Map[name]
|
||||||
|
return cmdCfg, ok
|
||||||
|
}
|
88
argh/parser_error.go
Normal file
88
argh/parser_error.go
Normal file
@@ -0,0 +1,88 @@
|
|||||||
|
package argh
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"sort"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ParserError is largely borrowed from go/scanner.Error
|
||||||
|
type ParserError struct {
|
||||||
|
Pos Position
|
||||||
|
Msg string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e ParserError) Error() string {
|
||||||
|
if e.Pos.IsValid() {
|
||||||
|
return e.Pos.String() + ":" + e.Msg
|
||||||
|
}
|
||||||
|
|
||||||
|
return e.Msg
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParserErrorList is largely borrowed from go/scanner.ErrorList
|
||||||
|
type ParserErrorList []*ParserError
|
||||||
|
|
||||||
|
func (el *ParserErrorList) Add(pos Position, msg string) {
|
||||||
|
*el = append(*el, &ParserError{Pos: pos, Msg: msg})
|
||||||
|
}
|
||||||
|
|
||||||
|
func (el *ParserErrorList) Reset() { *el = (*el)[0:0] }
|
||||||
|
|
||||||
|
func (el ParserErrorList) Len() int { return len(el) }
|
||||||
|
|
||||||
|
func (el ParserErrorList) Swap(i, j int) { el[i], el[j] = el[j], el[i] }
|
||||||
|
|
||||||
|
func (el ParserErrorList) Less(i, j int) bool {
|
||||||
|
e := &el[i].Pos
|
||||||
|
f := &el[j].Pos
|
||||||
|
|
||||||
|
if e.Column != f.Column {
|
||||||
|
return e.Column < f.Column
|
||||||
|
}
|
||||||
|
|
||||||
|
return el[i].Msg < el[j].Msg
|
||||||
|
}
|
||||||
|
|
||||||
|
func (el ParserErrorList) Sort() {
|
||||||
|
sort.Sort(el)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (el ParserErrorList) Error() string {
|
||||||
|
switch len(el) {
|
||||||
|
case 0:
|
||||||
|
return "no errors"
|
||||||
|
case 1:
|
||||||
|
return el[0].Error()
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%s (and %d more errors)", el[0], len(el)-1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (el ParserErrorList) Err() error {
|
||||||
|
if len(el) == 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return el
|
||||||
|
}
|
||||||
|
|
||||||
|
func (el ParserErrorList) Is(other error) bool {
|
||||||
|
if _, ok := other.(ParserErrorList); ok {
|
||||||
|
return el.Error() == other.Error()
|
||||||
|
}
|
||||||
|
|
||||||
|
if v, ok := other.(*ParserErrorList); ok {
|
||||||
|
return el.Error() == (*v).Error()
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func PrintParserError(w io.Writer, err error) {
|
||||||
|
if list, ok := err.(ParserErrorList); ok {
|
||||||
|
for _, e := range list {
|
||||||
|
fmt.Fprintf(w, "%s\n", e)
|
||||||
|
}
|
||||||
|
} else if err != nil {
|
||||||
|
fmt.Fprintf(w, "%s\n", err)
|
||||||
|
}
|
||||||
|
}
|
1028
argh/parser_test.go
Normal file
1028
argh/parser_test.go
Normal file
File diff suppressed because it is too large
Load Diff
94
argh/querier.go
Normal file
94
argh/querier.go
Normal file
@@ -0,0 +1,94 @@
|
|||||||
|
package argh
|
||||||
|
|
||||||
|
type Querier interface {
|
||||||
|
Program() (*Command, bool)
|
||||||
|
AST() []Node
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewQuerier(nodes []Node) Querier {
|
||||||
|
return &defaultQuerier{nodes: nodes}
|
||||||
|
}
|
||||||
|
|
||||||
|
type defaultQuerier struct {
|
||||||
|
nodes []Node
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dq *defaultQuerier) Program() (*Command, bool) {
|
||||||
|
if len(dq.nodes) == 0 {
|
||||||
|
tracef("Program nodes are empty")
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
|
||||||
|
tracef("Program node[0] is %T", dq.nodes[0])
|
||||||
|
|
||||||
|
v, ok := dq.nodes[0].(*Command)
|
||||||
|
if ok && v.Name == "" {
|
||||||
|
return v, false
|
||||||
|
}
|
||||||
|
|
||||||
|
return v, ok
|
||||||
|
}
|
||||||
|
|
||||||
|
func (dq *defaultQuerier) AST() []Node {
|
||||||
|
ret := []Node{}
|
||||||
|
|
||||||
|
for i, node := range dq.nodes {
|
||||||
|
tracef("AST i=%d node type=%T", i, node)
|
||||||
|
|
||||||
|
if _, ok := node.(*ArgDelimiter); ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, ok := node.(*StopFlag); ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if v, ok := node.(*CompoundShortFlag); ok {
|
||||||
|
if v.Nodes != nil {
|
||||||
|
ret = append(ret, NewQuerier(v.Nodes).AST()...)
|
||||||
|
}
|
||||||
|
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if v, ok := node.(*Command); ok {
|
||||||
|
astNodes := NewQuerier(v.Nodes).AST()
|
||||||
|
|
||||||
|
if len(astNodes) == 0 {
|
||||||
|
astNodes = nil
|
||||||
|
}
|
||||||
|
|
||||||
|
ret = append(
|
||||||
|
ret,
|
||||||
|
&Command{
|
||||||
|
Name: v.Name,
|
||||||
|
Values: v.Values,
|
||||||
|
Nodes: astNodes,
|
||||||
|
})
|
||||||
|
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if v, ok := node.(*Flag); ok {
|
||||||
|
astNodes := NewQuerier(v.Nodes).AST()
|
||||||
|
|
||||||
|
if len(astNodes) == 0 {
|
||||||
|
astNodes = nil
|
||||||
|
}
|
||||||
|
|
||||||
|
ret = append(
|
||||||
|
ret,
|
||||||
|
&Flag{
|
||||||
|
Name: v.Name,
|
||||||
|
Values: v.Values,
|
||||||
|
Nodes: astNodes,
|
||||||
|
})
|
||||||
|
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
ret = append(ret, node)
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret
|
||||||
|
}
|
60
argh/querier_test.go
Normal file
60
argh/querier_test.go
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
package argh_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"git.meatballhat.com/x/box-o-sand/argh"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestQuerier_Program(t *testing.T) {
|
||||||
|
for _, tc := range []struct {
|
||||||
|
name string
|
||||||
|
args []string
|
||||||
|
cfg *argh.ParserConfig
|
||||||
|
exp string
|
||||||
|
expOK bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "typical",
|
||||||
|
args: []string{"pizzas", "ahoy", "--treatsa", "fun"},
|
||||||
|
cfg: &argh.ParserConfig{
|
||||||
|
Prog: argh.CommandConfig{
|
||||||
|
Commands: &argh.Commands{
|
||||||
|
Map: map[string]argh.CommandConfig{
|
||||||
|
"ahoy": argh.CommandConfig{
|
||||||
|
Flags: &argh.Flags{
|
||||||
|
Map: map[string]argh.FlagConfig{
|
||||||
|
"treatsa": argh.FlagConfig{NValue: 1},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
exp: "pizzas",
|
||||||
|
expOK: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "minimal",
|
||||||
|
args: []string{"pizzas"},
|
||||||
|
exp: "pizzas",
|
||||||
|
expOK: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "invalid",
|
||||||
|
args: []string{},
|
||||||
|
expOK: false,
|
||||||
|
},
|
||||||
|
} {
|
||||||
|
t.Run(tc.name, func(ct *testing.T) {
|
||||||
|
pt, err := argh.ParseArgs(tc.args, tc.cfg)
|
||||||
|
require.Nil(ct, err)
|
||||||
|
|
||||||
|
prog, ok := argh.NewQuerier(pt.Nodes).Program()
|
||||||
|
require.Equal(ct, tc.expOK, ok)
|
||||||
|
require.Equal(ct, tc.exp, prog.Name)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
159
argh/scanner.go
Normal file
159
argh/scanner.go
Normal file
@@ -0,0 +1,159 @@
|
|||||||
|
package argh
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"bytes"
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"unicode"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Scanner struct {
|
||||||
|
r *bufio.Reader
|
||||||
|
i int
|
||||||
|
cfg *ScannerConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewScanner(r io.Reader, cfg *ScannerConfig) *Scanner {
|
||||||
|
if cfg == nil {
|
||||||
|
cfg = POSIXyScannerConfig
|
||||||
|
}
|
||||||
|
|
||||||
|
return &Scanner{
|
||||||
|
r: bufio.NewReader(r),
|
||||||
|
cfg: cfg,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Scanner) Scan() (Token, string, Pos) {
|
||||||
|
ch, pos := s.read()
|
||||||
|
|
||||||
|
if s.cfg.IsBlankspace(ch) {
|
||||||
|
_ = s.unread()
|
||||||
|
return s.scanBlankspace()
|
||||||
|
}
|
||||||
|
|
||||||
|
if s.cfg.IsAssignmentOperator(ch) {
|
||||||
|
return ASSIGN, string(ch), pos
|
||||||
|
}
|
||||||
|
|
||||||
|
if s.cfg.IsMultiValueDelim(ch) {
|
||||||
|
return MULTI_VALUE_DELIMITER, string(ch), pos
|
||||||
|
}
|
||||||
|
|
||||||
|
if ch == eol {
|
||||||
|
return EOL, "", pos
|
||||||
|
}
|
||||||
|
|
||||||
|
if ch == nul {
|
||||||
|
return ARG_DELIMITER, string(ch), pos
|
||||||
|
}
|
||||||
|
|
||||||
|
if unicode.IsGraphic(ch) {
|
||||||
|
_ = s.unread()
|
||||||
|
return s.scanArg()
|
||||||
|
}
|
||||||
|
|
||||||
|
return ILLEGAL, string(ch), pos
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Scanner) read() (rune, Pos) {
|
||||||
|
ch, _, err := s.r.ReadRune()
|
||||||
|
s.i++
|
||||||
|
|
||||||
|
if errors.Is(err, io.EOF) {
|
||||||
|
return eol, Pos(s.i)
|
||||||
|
} else if err != nil {
|
||||||
|
log.Printf("unknown scanner error=%+v", err)
|
||||||
|
return eol, Pos(s.i)
|
||||||
|
}
|
||||||
|
|
||||||
|
return ch, Pos(s.i)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Scanner) unread() Pos {
|
||||||
|
_ = s.r.UnreadRune()
|
||||||
|
s.i--
|
||||||
|
return Pos(s.i)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Scanner) scanBlankspace() (Token, string, Pos) {
|
||||||
|
buf := &bytes.Buffer{}
|
||||||
|
ch, pos := s.read()
|
||||||
|
buf.WriteRune(ch)
|
||||||
|
|
||||||
|
for {
|
||||||
|
ch, pos = s.read()
|
||||||
|
|
||||||
|
if ch == eol {
|
||||||
|
break
|
||||||
|
} else if !s.cfg.IsBlankspace(ch) {
|
||||||
|
pos = s.unread()
|
||||||
|
break
|
||||||
|
} else {
|
||||||
|
_, _ = buf.WriteRune(ch)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return BS, buf.String(), pos
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Scanner) scanArg() (Token, string, Pos) {
|
||||||
|
buf := &bytes.Buffer{}
|
||||||
|
ch, pos := s.read()
|
||||||
|
buf.WriteRune(ch)
|
||||||
|
|
||||||
|
for {
|
||||||
|
ch, pos = s.read()
|
||||||
|
|
||||||
|
if ch == eol || ch == nul || s.cfg.IsAssignmentOperator(ch) || s.cfg.IsMultiValueDelim(ch) {
|
||||||
|
pos = s.unread()
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
_, _ = buf.WriteRune(ch)
|
||||||
|
}
|
||||||
|
|
||||||
|
str := buf.String()
|
||||||
|
|
||||||
|
if len(str) == 0 {
|
||||||
|
return EMPTY, str, pos
|
||||||
|
}
|
||||||
|
|
||||||
|
ch0 := rune(str[0])
|
||||||
|
|
||||||
|
if len(str) == 1 {
|
||||||
|
if s.cfg.IsFlagPrefix(ch0) {
|
||||||
|
return STDIN_FLAG, str, pos
|
||||||
|
}
|
||||||
|
|
||||||
|
if s.cfg.IsAssignmentOperator(ch0) {
|
||||||
|
return ASSIGN, str, pos
|
||||||
|
}
|
||||||
|
|
||||||
|
return IDENT, str, pos
|
||||||
|
}
|
||||||
|
|
||||||
|
ch1 := rune(str[1])
|
||||||
|
|
||||||
|
if len(str) == 2 {
|
||||||
|
if s.cfg.IsFlagPrefix(ch0) && s.cfg.IsFlagPrefix(ch1) {
|
||||||
|
return STOP_FLAG, str, pos
|
||||||
|
}
|
||||||
|
|
||||||
|
if s.cfg.IsFlagPrefix(ch0) {
|
||||||
|
return SHORT_FLAG, str, pos
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if s.cfg.IsFlagPrefix(ch0) {
|
||||||
|
if s.cfg.IsFlagPrefix(ch1) {
|
||||||
|
return LONG_FLAG, str, pos
|
||||||
|
}
|
||||||
|
|
||||||
|
return COMPOUND_SHORT_FLAG, str, pos
|
||||||
|
}
|
||||||
|
|
||||||
|
return IDENT, str, pos
|
||||||
|
}
|
39
argh/scanner_config.go
Normal file
39
argh/scanner_config.go
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
package argh
|
||||||
|
|
||||||
|
var (
|
||||||
|
// POSIXyScannerConfig defines a scanner config that uses '-'
|
||||||
|
// as the flag prefix, which also means that "--" is the "long
|
||||||
|
// flag" prefix, a bare "--" is considered STOP_FLAG, and a
|
||||||
|
// bare "-" is considered STDIN_FLAG.
|
||||||
|
POSIXyScannerConfig = &ScannerConfig{
|
||||||
|
AssignmentOperator: '=',
|
||||||
|
FlagPrefix: '-',
|
||||||
|
MultiValueDelim: ',',
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
type ScannerConfig struct {
|
||||||
|
AssignmentOperator rune
|
||||||
|
FlagPrefix rune
|
||||||
|
MultiValueDelim rune
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cfg *ScannerConfig) IsFlagPrefix(ch rune) bool {
|
||||||
|
return ch == cfg.FlagPrefix
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cfg *ScannerConfig) IsMultiValueDelim(ch rune) bool {
|
||||||
|
return ch == cfg.MultiValueDelim
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cfg *ScannerConfig) IsAssignmentOperator(ch rune) bool {
|
||||||
|
return ch == cfg.AssignmentOperator
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cfg *ScannerConfig) IsBlankspace(ch rune) bool {
|
||||||
|
return ch == ' ' || ch == '\t' || ch == '\n' || ch == '\r'
|
||||||
|
}
|
||||||
|
|
||||||
|
func (cfg *ScannerConfig) IsUnderscore(ch rune) bool {
|
||||||
|
return ch == '_'
|
||||||
|
}
|
53
argh/token.go
Normal file
53
argh/token.go
Normal file
@@ -0,0 +1,53 @@
|
|||||||
|
//go:generate stringer -type Token
|
||||||
|
|
||||||
|
package argh
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
const (
|
||||||
|
ILLEGAL Token = iota
|
||||||
|
EOL
|
||||||
|
EMPTY // ''
|
||||||
|
BS // ' ' '\t' '\n'
|
||||||
|
IDENT // char group without flag prefix: 'some' 'words'
|
||||||
|
ARG_DELIMITER // rune(0)
|
||||||
|
ASSIGN // '='
|
||||||
|
MULTI_VALUE_DELIMITER // ','
|
||||||
|
LONG_FLAG // char group with double flag prefix: '--flag'
|
||||||
|
SHORT_FLAG // single char with single flag prefix: '-f'
|
||||||
|
COMPOUND_SHORT_FLAG // char group with single flag prefix: '-flag'
|
||||||
|
STDIN_FLAG // '-'
|
||||||
|
STOP_FLAG // '--'
|
||||||
|
|
||||||
|
nul = rune(0)
|
||||||
|
eol = rune(-1)
|
||||||
|
)
|
||||||
|
|
||||||
|
type Token int
|
||||||
|
|
||||||
|
// Position is adapted from go/token.Position
|
||||||
|
type Position struct {
|
||||||
|
Column int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Position) IsValid() bool { return p.Column > 0 }
|
||||||
|
|
||||||
|
func (p Position) String() string {
|
||||||
|
s := ""
|
||||||
|
if p.IsValid() {
|
||||||
|
s = fmt.Sprintf("%d", p.Column)
|
||||||
|
}
|
||||||
|
if s == "" {
|
||||||
|
s = "-"
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
// Pos is borrowed from go/token.Pos
|
||||||
|
type Pos int
|
||||||
|
|
||||||
|
const NoPos Pos = 0
|
||||||
|
|
||||||
|
func (p Pos) IsValid() bool {
|
||||||
|
return p != NoPos
|
||||||
|
}
|
35
argh/token_string.go
Normal file
35
argh/token_string.go
Normal file
@@ -0,0 +1,35 @@
|
|||||||
|
// Code generated by "stringer -type Token"; DO NOT EDIT.
|
||||||
|
|
||||||
|
package argh
|
||||||
|
|
||||||
|
import "strconv"
|
||||||
|
|
||||||
|
func _() {
|
||||||
|
// An "invalid array index" compiler error signifies that the constant values have changed.
|
||||||
|
// Re-run the stringer command to generate them again.
|
||||||
|
var x [1]struct{}
|
||||||
|
_ = x[ILLEGAL-0]
|
||||||
|
_ = x[EOL-1]
|
||||||
|
_ = x[EMPTY-2]
|
||||||
|
_ = x[BS-3]
|
||||||
|
_ = x[IDENT-4]
|
||||||
|
_ = x[ARG_DELIMITER-5]
|
||||||
|
_ = x[ASSIGN-6]
|
||||||
|
_ = x[MULTI_VALUE_DELIMITER-7]
|
||||||
|
_ = x[LONG_FLAG-8]
|
||||||
|
_ = x[SHORT_FLAG-9]
|
||||||
|
_ = x[COMPOUND_SHORT_FLAG-10]
|
||||||
|
_ = x[STDIN_FLAG-11]
|
||||||
|
_ = x[STOP_FLAG-12]
|
||||||
|
}
|
||||||
|
|
||||||
|
const _Token_name = "ILLEGALEOLEMPTYBSIDENTARG_DELIMITERASSIGNMULTI_VALUE_DELIMITERLONG_FLAGSHORT_FLAGCOMPOUND_SHORT_FLAGSTDIN_FLAGSTOP_FLAG"
|
||||||
|
|
||||||
|
var _Token_index = [...]uint8{0, 7, 10, 15, 17, 22, 35, 41, 62, 71, 81, 100, 110, 119}
|
||||||
|
|
||||||
|
func (i Token) String() string {
|
||||||
|
if i < 0 || i >= Token(len(_Token_index)-1) {
|
||||||
|
return "Token(" + strconv.FormatInt(int64(i), 10) + ")"
|
||||||
|
}
|
||||||
|
return _Token_name[_Token_index[i]:_Token_index[i+1]]
|
||||||
|
}
|
207
hyrule/README.md
Normal file
207
hyrule/README.md
Normal file
@@ -0,0 +1,207 @@
|
|||||||
|
# Hyrule
|
||||||
|
|
||||||
|
A card game for 2-7 players.
|
||||||
|
|
||||||
|
## requirements
|
||||||
|
|
||||||
|
Use a standard card playing deck, keeping Jokers. Shuffle however
|
||||||
|
you like.
|
||||||
|
|
||||||
|
## dealing
|
||||||
|
|
||||||
|
Each player is dealt 7 cards, kept secret from other players.
|
||||||
|
|
||||||
|
## turns
|
||||||
|
|
||||||
|
There are two variations for taking turns. "Chance rules" is
|
||||||
|
strictly a matter of chance, and is recommended for impatient
|
||||||
|
players because it tends to be faster-paced and has more of a
|
||||||
|
*surprise!* element. "Strategy rules" introduces an element of
|
||||||
|
strategy and is better suited for players with more _patience_.
|
||||||
|
|
||||||
|
### chance rules
|
||||||
|
|
||||||
|
Each player selects one card and places it face down in the middle
|
||||||
|
playing area between themselves and other players.
|
||||||
|
|
||||||
|
On the count of 3, all said together in rhythm, each player flips
|
||||||
|
over their own card. Precedence applies and the winning player
|
||||||
|
takes all cards.
|
||||||
|
|
||||||
|
### strategy rules
|
||||||
|
|
||||||
|
On the first turn or when an equal number of cards have been
|
||||||
|
captured by each player, the tallest player goes first. On each
|
||||||
|
subsequent turn, the player with the most captured cards plays
|
||||||
|
first. The play order then proceeds counter-clockwise.
|
||||||
|
|
||||||
|
When taking one's turn, the player may choose to either *play* or
|
||||||
|
*swap*.
|
||||||
|
|
||||||
|
#### strategy *play*
|
||||||
|
|
||||||
|
When choosing to *play*, the player selects a card and places it
|
||||||
|
face-up in the center play area. The remaining players take turns
|
||||||
|
counter-clockwise, each placing a card face-up in the center play
|
||||||
|
area. Precedence rules apply and the winning player takes all
|
||||||
|
cards.
|
||||||
|
|
||||||
|
#### strategy *swap*
|
||||||
|
|
||||||
|
When choosing to *swap*, the player discards one card face-up in
|
||||||
|
the discard pile (next to the stock pile) and then draws a card
|
||||||
|
from the stock, which should be laying face-down. The goal here may
|
||||||
|
be to take a chance at getting a better card than one is
|
||||||
|
discarding, or to force the next player in the rotation to play, or
|
||||||
|
both.
|
||||||
|
|
||||||
|
> **NOTE**: a player may only choose to *swap* if the player on the
|
||||||
|
> previous turn _did not swap_.
|
||||||
|
|
||||||
|
## precedence
|
||||||
|
|
||||||
|
The following rules apply across suits with the exception of
|
||||||
|
*jokers and fives* (explained below). Cards within a single suit
|
||||||
|
are compared with higher-value cards winning. Cards are counted
|
||||||
|
from ace (1) through king (13).
|
||||||
|
|
||||||
|
Ways to think about these rules could include:
|
||||||
|
|
||||||
|
- "rupee buys bomb" / money buys weapon
|
||||||
|
- "rupee buys sword" / money buys weapon
|
||||||
|
- "bomb blows up sword" / range weapon beats melee weapon
|
||||||
|
- "bomb blows up heart" / range weapon beats unarmed
|
||||||
|
- "sword cuts through heart" / melee weapon beats unarmed
|
||||||
|
- "heart is stronger than rupee" / love conquers money
|
||||||
|
|
||||||
|
**NOTE:** In the case of a turn that involves 3 or more cards, the
|
||||||
|
presence of both diamonds *and* hearts will result in the
|
||||||
|
highest-value heart winning the turn.
|
||||||
|
|
||||||
|
### diamonds ("rupees")
|
||||||
|
|
||||||
|
Beats:
|
||||||
|
|
||||||
|
- clubs ("bombs")
|
||||||
|
- spades ("swords")
|
||||||
|
|
||||||
|
### clubs ("bombs")
|
||||||
|
|
||||||
|
Beats:
|
||||||
|
|
||||||
|
- spades ("swords")
|
||||||
|
- hearts
|
||||||
|
|
||||||
|
### spades ("swords")
|
||||||
|
|
||||||
|
Beats:
|
||||||
|
|
||||||
|
- hearts
|
||||||
|
|
||||||
|
### hearts
|
||||||
|
|
||||||
|
Beats:
|
||||||
|
|
||||||
|
- diamonds ("rupees")
|
||||||
|
|
||||||
|
### jokers ("tingles")
|
||||||
|
|
||||||
|
Beats everything except *fives*
|
||||||
|
|
||||||
|
### fives
|
||||||
|
|
||||||
|
The 5 of a given suit will be granted special status *only* when
|
||||||
|
played against a joker ("tingle") and will win with the following
|
||||||
|
sub-precedence, which is roughly the inverse of the main
|
||||||
|
precedence:
|
||||||
|
|
||||||
|
- 5 of hearts
|
||||||
|
- 5 of spades / "master sword"
|
||||||
|
- 5 of clubs / "bomb cluster"
|
||||||
|
- 5 of diamonds / "blue rupee"
|
||||||
|
|
||||||
|
## scoring
|
||||||
|
|
||||||
|
At the conclusion of a round of 7 cards, the player with the most
|
||||||
|
cards is the winner. The face values of the cards are not
|
||||||
|
considered at scoring time. A draw may be handled in a "run-off
|
||||||
|
game" or ignored as you like.
|
||||||
|
|
||||||
|
## examples
|
||||||
|
|
||||||
|
In the following examples, the **winning card** is highlighted at
|
||||||
|
the top of each list, followed by an explanation for the outcome.
|
||||||
|
|
||||||
|
### 2-player
|
||||||
|
|
||||||
|
- **2 of clubs**
|
||||||
|
- 2 of spades
|
||||||
|
|
||||||
|
Clubs are higher-value than spades, or "bomb beats sword".
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
- **2 of clubs**
|
||||||
|
- ace of clubs
|
||||||
|
|
||||||
|
Cards within the same suit are compared at face value with aces
|
||||||
|
being *1*.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
- **ace of hearts**
|
||||||
|
- king of diamonds
|
||||||
|
|
||||||
|
Hearts take precedence over diamonds, or "love is stronger than
|
||||||
|
money".
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
- **joker**
|
||||||
|
- king of hearts
|
||||||
|
|
||||||
|
The opposing card is not a *five*, or "tingle takes _(thing)_".
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
- **5 of spades**
|
||||||
|
- joker
|
||||||
|
|
||||||
|
The 5 of any suit will beat a joker given its special item status
|
||||||
|
in that scenario, or "tingle is distracted by the beauty of
|
||||||
|
_(thing)_".
|
||||||
|
|
||||||
|
### 3-player
|
||||||
|
|
||||||
|
- **2 of diamonds**
|
||||||
|
- 2 of clubs
|
||||||
|
- 2 of spades
|
||||||
|
|
||||||
|
Diamonds have higher precedence than clubs and spades, and there
|
||||||
|
are no hearts present.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
- **6 of clubs**
|
||||||
|
- 6 of spades
|
||||||
|
- 6 of hearts
|
||||||
|
|
||||||
|
Clubs have higher precedence than spades and hearts, and there are
|
||||||
|
no diamonds present.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
- **8 of hearts**
|
||||||
|
- 8 of diamonds
|
||||||
|
- 8 of spades
|
||||||
|
|
||||||
|
When both diamonds and hearts are present, hearts is highest
|
||||||
|
precedence.
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
- **king of diamonds**
|
||||||
|
- jack of diamonds
|
||||||
|
- queen of diamonds
|
||||||
|
|
||||||
|
Within the same suit, cards are compared by face value.
|
BIN
hyrule/README.pdf
Normal file
BIN
hyrule/README.pdf
Normal file
Binary file not shown.
86
hyrule/cards.py
Normal file
86
hyrule/cards.py
Normal file
@@ -0,0 +1,86 @@
|
|||||||
|
import dataclasses
|
||||||
|
import enum
|
||||||
|
import random
|
||||||
|
|
||||||
|
Suit = enum.Enum(
|
||||||
|
"Suit",
|
||||||
|
"""
|
||||||
|
CLUBS
|
||||||
|
DIAMONDS
|
||||||
|
HEARTS
|
||||||
|
SPADES
|
||||||
|
""",
|
||||||
|
)
|
||||||
|
|
||||||
|
FaceValue = enum.Enum(
|
||||||
|
"FaceValue",
|
||||||
|
"""
|
||||||
|
ACE
|
||||||
|
TWO
|
||||||
|
THREE
|
||||||
|
FOUR
|
||||||
|
FIVE
|
||||||
|
SIX
|
||||||
|
SEVEN
|
||||||
|
EIGHT
|
||||||
|
NINE
|
||||||
|
TEN
|
||||||
|
JACK
|
||||||
|
QUEEN
|
||||||
|
KING
|
||||||
|
""",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@dataclasses.dataclass
|
||||||
|
class Card:
|
||||||
|
suit: Suit
|
||||||
|
face_value: FaceValue
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return f"{self.face_value} of {self.suit}"
|
||||||
|
|
||||||
|
|
||||||
|
class Joker:
|
||||||
|
def __str__(self):
|
||||||
|
return "JOKER"
|
||||||
|
|
||||||
|
|
||||||
|
class Deck:
|
||||||
|
def __init__(self, n_jokers=2):
|
||||||
|
self._cards = list(Deck.generate(n_jokers=n_jokers))
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return f"<Deck len={len(self)}>"
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
as_string = []
|
||||||
|
for card in self._cards:
|
||||||
|
as_string.append(str(card))
|
||||||
|
return "\n".join(as_string)
|
||||||
|
|
||||||
|
def __len__(self):
|
||||||
|
return len(self._cards)
|
||||||
|
|
||||||
|
def draw(self):
|
||||||
|
if len(self._cards) > 0:
|
||||||
|
return self._cards.pop()
|
||||||
|
return None
|
||||||
|
|
||||||
|
def shuffle(self):
|
||||||
|
random.shuffle(self._cards)
|
||||||
|
return self
|
||||||
|
|
||||||
|
def cut(self):
|
||||||
|
cut_point = random.randint(0, len(self))
|
||||||
|
self._cards = self._cards[cut_point:] + self._cards[:cut_point]
|
||||||
|
return self
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def generate(n_jokers=2):
|
||||||
|
for _ in range(n_jokers):
|
||||||
|
yield Joker()
|
||||||
|
|
||||||
|
for suit in Suit:
|
||||||
|
for face_value in FaceValue:
|
||||||
|
yield Card(suit=suit, face_value=face_value)
|
66
hyrule/hyrule.py
Normal file
66
hyrule/hyrule.py
Normal file
@@ -0,0 +1,66 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
import argparse
|
||||||
|
import dataclasses
|
||||||
|
import pprint
|
||||||
|
import sys
|
||||||
|
import typing
|
||||||
|
|
||||||
|
import cards
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
parser = argparse.ArgumentParser(
|
||||||
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--player-count", "-c", default=2, type=int, help="The number of players"
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"--turn-count",
|
||||||
|
"-n",
|
||||||
|
default=5,
|
||||||
|
type=int,
|
||||||
|
help="The number of turns to play",
|
||||||
|
)
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
Hyrule(player_count=args.player_count, turn_count=args.turn_count).play()
|
||||||
|
return 0
|
||||||
|
|
||||||
|
|
||||||
|
@dataclasses.dataclass
|
||||||
|
class Player:
|
||||||
|
index: int
|
||||||
|
hand: typing.Set[cards.Card]
|
||||||
|
captured: typing.Set[cards.Card]
|
||||||
|
|
||||||
|
@property
|
||||||
|
def score(self):
|
||||||
|
return len(self.captured)
|
||||||
|
|
||||||
|
|
||||||
|
class Hyrule:
|
||||||
|
def __init__(self, player_count=2, turn_count=5):
|
||||||
|
self._players = [
|
||||||
|
Player(index=i, hand=set(), captured=set()) for i in range(player_count)
|
||||||
|
]
|
||||||
|
self._turn_count = turn_count
|
||||||
|
|
||||||
|
def play(self):
|
||||||
|
for turn in self._each_turn():
|
||||||
|
self._show_turn(turn)
|
||||||
|
|
||||||
|
def _each_turn(self):
|
||||||
|
for turn_number in range(self._turn_count):
|
||||||
|
yield self._simulate_turn(turn_number)
|
||||||
|
|
||||||
|
def _simulate_turn(self, turn_number):
|
||||||
|
...
|
||||||
|
|
||||||
|
def _show_turn(self, turn):
|
||||||
|
print(turn)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
sys.exit(main())
|
2
hyrule/justfile
Normal file
2
hyrule/justfile
Normal file
@@ -0,0 +1,2 @@
|
|||||||
|
build:
|
||||||
|
pandoc -r markdown -w pdf -o README.pdf README.md
|
Reference in New Issue
Block a user