83 lines
1.2 KiB
Go
83 lines
1.2 KiB
Go
|
package argh
|
||
|
|
|
||
|
|
import "fmt"
|
||
|
|
|
||
|
|
type ParseTree struct {
|
||
|
|
Nodes []Node `json:"nodes"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (pt *ParseTree) toAST() []TypedNode {
|
||
|
|
ret := []TypedNode{}
|
||
|
|
|
||
|
|
for _, node := range pt.Nodes {
|
||
|
|
if _, ok := node.(ArgDelimiter); ok {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, ok := node.(StopFlag); ok {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
ret = append(
|
||
|
|
ret,
|
||
|
|
TypedNode{
|
||
|
|
Type: fmt.Sprintf("%T", node),
|
||
|
|
Node: node,
|
||
|
|
},
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
|
||
|
|
type Node interface{}
|
||
|
|
|
||
|
|
type TypedNode struct {
|
||
|
|
Type string `json:"type"`
|
||
|
|
Node Node `json:"node"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Args struct {
|
||
|
|
Pos int `json:"pos"`
|
||
|
|
Nodes []Node `json:"nodes"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Statement struct {
|
||
|
|
Pos int `json:"pos"`
|
||
|
|
Nodes []Node `json:"nodes"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Program struct {
|
||
|
|
Pos int `json:"pos"`
|
||
|
|
Name string `json:"name"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Ident struct {
|
||
|
|
Pos int `json:"pos"`
|
||
|
|
Literal string `json:"literal"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Command struct {
|
||
|
|
Pos int `json:"pos"`
|
||
|
|
Name string `json:"name"`
|
||
|
|
Nodes []Node `json:"nodes"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type Flag struct {
|
||
|
|
Pos int `json:"pos"`
|
||
|
|
Name string `json:"name"`
|
||
|
|
Value *string `json:"value,omitempty"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type StdinFlag struct {
|
||
|
|
Pos int `json:"pos"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type StopFlag struct {
|
||
|
|
Pos int `json:"pos"`
|
||
|
|
}
|
||
|
|
|
||
|
|
type ArgDelimiter struct {
|
||
|
|
Pos int `json:"pos"`
|
||
|
|
}
|