Alec Thomas
123 lines
2.2 KiB
Go
123 lines
2.2 KiB
Go
package kong
|
|
|
|
import (
|
|
"strconv"
|
|
)
|
|
|
|
//go:generate stringer -type=TokenType
|
|
|
|
type TokenType int
|
|
|
|
const (
|
|
UntypedToken TokenType = iota
|
|
EOLToken
|
|
FlagToken // --<flag>
|
|
FlagValueToken // =<value>
|
|
ShortFlagToken // -<short>[<tail]
|
|
ShortFlagTailToken // <tail>
|
|
PositionalArgumentToken // <arg>
|
|
)
|
|
|
|
type Token struct {
|
|
Value string
|
|
Type TokenType
|
|
}
|
|
|
|
func (t Token) String() string {
|
|
switch t.Type {
|
|
case FlagToken:
|
|
return "--" + t.Value
|
|
|
|
case ShortFlagToken:
|
|
return "-" + t.Value
|
|
|
|
case EOLToken:
|
|
return "EOL"
|
|
|
|
default:
|
|
return strconv.Quote(t.Value)
|
|
}
|
|
}
|
|
|
|
func (t Token) IsAny(types ...TokenType) bool {
|
|
for _, typ := range types {
|
|
if t.Type == typ {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (t Token) IsValue() bool {
|
|
return t.IsAny(FlagValueToken, ShortFlagTailToken, PositionalArgumentToken, UntypedToken)
|
|
}
|
|
|
|
type Scanner struct {
|
|
args []Token
|
|
}
|
|
|
|
func Scan(args ...string) *Scanner {
|
|
s := &Scanner{}
|
|
for _, arg := range args {
|
|
s.args = append(s.args, Token{Value: arg})
|
|
}
|
|
return s
|
|
}
|
|
|
|
func (s *Scanner) Len() int {
|
|
return len(s.args)
|
|
}
|
|
|
|
func (s *Scanner) Pop() Token {
|
|
if len(s.args) == 0 {
|
|
return Token{Type: EOLToken}
|
|
}
|
|
arg := s.args[0]
|
|
s.args = s.args[1:]
|
|
return arg
|
|
}
|
|
|
|
// PopValue or panic with Error.
|
|
func (s *Scanner) PopValue(context string) string {
|
|
t := s.Pop()
|
|
if !t.IsValue() {
|
|
fail("expected %s value but got %s", context, t)
|
|
}
|
|
return t.Value
|
|
}
|
|
|
|
// PopWhile predicate returns true.
|
|
func (s *Scanner) PopWhile(predicate func(Token) bool) (values []string) {
|
|
for predicate(s.Peek()) {
|
|
values = append(values, s.Pop().Value)
|
|
}
|
|
return
|
|
}
|
|
|
|
// PopUntil predicate returns true.
|
|
func (s *Scanner) PopUntil(predicate func(Token) bool) (values []string) {
|
|
for !predicate(s.Peek()) {
|
|
values = append(values, s.Pop().Value)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (s *Scanner) Peek() Token {
|
|
if len(s.args) == 0 {
|
|
return Token{Type: EOLToken}
|
|
}
|
|
return s.args[0]
|
|
}
|
|
|
|
func (s *Scanner) Push(arg string) {
|
|
s.PushToken(Token{Value: arg})
|
|
}
|
|
|
|
func (s *Scanner) PushTyped(arg string, typ TokenType) {
|
|
s.PushToken(Token{Value: arg, Type: typ})
|
|
}
|
|
|
|
func (s *Scanner) PushToken(token Token) {
|
|
s.args = append([]Token{token}, s.args...)
|
|
}
|