498 lines
10 KiB
Go
498 lines
10 KiB
Go
package pattern
|
|
|
|
import (
|
|
"fmt"
|
|
"go/token"
|
|
"reflect"
|
|
"strings"
|
|
)
|
|
|
|
var (
|
|
_ Node = Ellipsis{}
|
|
_ Node = Binding{}
|
|
_ Node = RangeStmt{}
|
|
_ Node = AssignStmt{}
|
|
_ Node = IndexExpr{}
|
|
_ Node = Ident{}
|
|
_ Node = Builtin{}
|
|
_ Node = String("")
|
|
_ Node = Any{}
|
|
_ Node = ValueSpec{}
|
|
_ Node = List{}
|
|
_ Node = GenDecl{}
|
|
_ Node = BinaryExpr{}
|
|
_ Node = ForStmt{}
|
|
_ Node = ArrayType{}
|
|
_ Node = DeferStmt{}
|
|
_ Node = MapType{}
|
|
_ Node = ReturnStmt{}
|
|
_ Node = SliceExpr{}
|
|
_ Node = StarExpr{}
|
|
_ Node = UnaryExpr{}
|
|
_ Node = SendStmt{}
|
|
_ Node = SelectStmt{}
|
|
_ Node = ImportSpec{}
|
|
_ Node = IfStmt{}
|
|
_ Node = GoStmt{}
|
|
_ Node = Field{}
|
|
_ Node = SelectorExpr{}
|
|
_ Node = StructType{}
|
|
_ Node = KeyValueExpr{}
|
|
_ Node = FuncType{}
|
|
_ Node = FuncLit{}
|
|
_ Node = FuncDecl{}
|
|
_ Node = Token(0)
|
|
_ Node = ChanType{}
|
|
_ Node = CallExpr{}
|
|
_ Node = CaseClause{}
|
|
_ Node = CommClause{}
|
|
_ Node = CompositeLit{}
|
|
_ Node = EmptyStmt{}
|
|
_ Node = SwitchStmt{}
|
|
_ Node = TypeSwitchStmt{}
|
|
_ Node = TypeAssertExpr{}
|
|
_ Node = TypeSpec{}
|
|
_ Node = InterfaceType{}
|
|
_ Node = BranchStmt{}
|
|
_ Node = IncDecStmt{}
|
|
_ Node = BasicLit{}
|
|
_ Node = Nil{}
|
|
_ Node = Object{}
|
|
_ Node = Function{}
|
|
_ Node = Not{}
|
|
_ Node = Or{}
|
|
)
|
|
|
|
type Function struct {
|
|
Name Node
|
|
}
|
|
|
|
type Token token.Token
|
|
|
|
type Nil struct {
|
|
}
|
|
|
|
type Ellipsis struct {
|
|
Elt Node
|
|
}
|
|
|
|
type IncDecStmt struct {
|
|
X Node
|
|
Tok Node
|
|
}
|
|
|
|
type BranchStmt struct {
|
|
Tok Node
|
|
Label Node
|
|
}
|
|
|
|
type InterfaceType struct {
|
|
Methods Node
|
|
}
|
|
|
|
type TypeSpec struct {
|
|
Name Node
|
|
Type Node
|
|
}
|
|
|
|
type TypeAssertExpr struct {
|
|
X Node
|
|
Type Node
|
|
}
|
|
|
|
type TypeSwitchStmt struct {
|
|
Init Node
|
|
Assign Node
|
|
Body Node
|
|
}
|
|
|
|
type SwitchStmt struct {
|
|
Init Node
|
|
Tag Node
|
|
Body Node
|
|
}
|
|
|
|
type EmptyStmt struct {
|
|
}
|
|
|
|
type CompositeLit struct {
|
|
Type Node
|
|
Elts Node
|
|
}
|
|
|
|
type CommClause struct {
|
|
Comm Node
|
|
Body Node
|
|
}
|
|
|
|
type CaseClause struct {
|
|
List Node
|
|
Body Node
|
|
}
|
|
|
|
type CallExpr struct {
|
|
Fun Node
|
|
Args Node
|
|
// XXX handle ellipsis
|
|
}
|
|
|
|
// TODO(dh): add a ChanDir node, and a way of instantiating it.
|
|
|
|
type ChanType struct {
|
|
Dir Node
|
|
Value Node
|
|
}
|
|
|
|
type FuncDecl struct {
|
|
Recv Node
|
|
Name Node
|
|
Type Node
|
|
Body Node
|
|
}
|
|
|
|
type FuncLit struct {
|
|
Type Node
|
|
Body Node
|
|
}
|
|
|
|
type FuncType struct {
|
|
Params Node
|
|
Results Node
|
|
}
|
|
|
|
type KeyValueExpr struct {
|
|
Key Node
|
|
Value Node
|
|
}
|
|
|
|
type StructType struct {
|
|
Fields Node
|
|
}
|
|
|
|
type SelectorExpr struct {
|
|
X Node
|
|
Sel Node
|
|
}
|
|
|
|
type Field struct {
|
|
Names Node
|
|
Type Node
|
|
Tag Node
|
|
}
|
|
|
|
type GoStmt struct {
|
|
Call Node
|
|
}
|
|
|
|
type IfStmt struct {
|
|
Init Node
|
|
Cond Node
|
|
Body Node
|
|
Else Node
|
|
}
|
|
|
|
type ImportSpec struct {
|
|
Name Node
|
|
Path Node
|
|
}
|
|
|
|
type SelectStmt struct {
|
|
Body Node
|
|
}
|
|
|
|
type ArrayType struct {
|
|
Len Node
|
|
Elt Node
|
|
}
|
|
|
|
type DeferStmt struct {
|
|
Call Node
|
|
}
|
|
|
|
type MapType struct {
|
|
Key Node
|
|
Value Node
|
|
}
|
|
|
|
type ReturnStmt struct {
|
|
Results Node
|
|
}
|
|
|
|
type SliceExpr struct {
|
|
X Node
|
|
Low Node
|
|
High Node
|
|
Max Node
|
|
}
|
|
|
|
type StarExpr struct {
|
|
X Node
|
|
}
|
|
|
|
type UnaryExpr struct {
|
|
Op Node
|
|
X Node
|
|
}
|
|
|
|
type SendStmt struct {
|
|
Chan Node
|
|
Value Node
|
|
}
|
|
|
|
type Binding struct {
|
|
Name string
|
|
Node Node
|
|
}
|
|
|
|
type RangeStmt struct {
|
|
Key Node
|
|
Value Node
|
|
Tok Node
|
|
X Node
|
|
Body Node
|
|
}
|
|
|
|
type AssignStmt struct {
|
|
Lhs Node
|
|
Tok Node
|
|
Rhs Node
|
|
}
|
|
|
|
type IndexExpr struct {
|
|
X Node
|
|
Index Node
|
|
}
|
|
|
|
type Node interface {
|
|
String() string
|
|
isNode()
|
|
}
|
|
|
|
type Ident struct {
|
|
Name Node
|
|
}
|
|
|
|
type Object struct {
|
|
Name Node
|
|
}
|
|
|
|
type Builtin struct {
|
|
Name Node
|
|
}
|
|
|
|
type String string
|
|
|
|
type Any struct{}
|
|
|
|
type ValueSpec struct {
|
|
Names Node
|
|
Type Node
|
|
Values Node
|
|
}
|
|
|
|
type List struct {
|
|
Head Node
|
|
Tail Node
|
|
}
|
|
|
|
type GenDecl struct {
|
|
Tok Node
|
|
Specs Node
|
|
}
|
|
|
|
type BasicLit struct {
|
|
Kind Node
|
|
Value Node
|
|
}
|
|
|
|
type BinaryExpr struct {
|
|
X Node
|
|
Op Node
|
|
Y Node
|
|
}
|
|
|
|
type ForStmt struct {
|
|
Init Node
|
|
Cond Node
|
|
Post Node
|
|
Body Node
|
|
}
|
|
|
|
type Or struct {
|
|
Nodes []Node
|
|
}
|
|
|
|
type Not struct {
|
|
Node Node
|
|
}
|
|
|
|
func stringify(n Node) string {
|
|
v := reflect.ValueOf(n)
|
|
var parts []string
|
|
parts = append(parts, v.Type().Name())
|
|
for i := 0; i < v.NumField(); i++ {
|
|
//lint:ignore S1025 false positive in staticcheck 2019.2.3
|
|
parts = append(parts, fmt.Sprintf("%s", v.Field(i)))
|
|
}
|
|
return "(" + strings.Join(parts, " ") + ")"
|
|
}
|
|
|
|
func (stmt AssignStmt) String() string { return stringify(stmt) }
|
|
func (expr IndexExpr) String() string { return stringify(expr) }
|
|
func (id Ident) String() string { return stringify(id) }
|
|
func (spec ValueSpec) String() string { return stringify(spec) }
|
|
func (decl GenDecl) String() string { return stringify(decl) }
|
|
func (lit BasicLit) String() string { return stringify(lit) }
|
|
func (expr BinaryExpr) String() string { return stringify(expr) }
|
|
func (stmt ForStmt) String() string { return stringify(stmt) }
|
|
func (stmt RangeStmt) String() string { return stringify(stmt) }
|
|
func (typ ArrayType) String() string { return stringify(typ) }
|
|
func (stmt DeferStmt) String() string { return stringify(stmt) }
|
|
func (typ MapType) String() string { return stringify(typ) }
|
|
func (stmt ReturnStmt) String() string { return stringify(stmt) }
|
|
func (expr SliceExpr) String() string { return stringify(expr) }
|
|
func (expr StarExpr) String() string { return stringify(expr) }
|
|
func (expr UnaryExpr) String() string { return stringify(expr) }
|
|
func (stmt SendStmt) String() string { return stringify(stmt) }
|
|
func (spec ImportSpec) String() string { return stringify(spec) }
|
|
func (stmt SelectStmt) String() string { return stringify(stmt) }
|
|
func (stmt IfStmt) String() string { return stringify(stmt) }
|
|
func (stmt IncDecStmt) String() string { return stringify(stmt) }
|
|
func (stmt GoStmt) String() string { return stringify(stmt) }
|
|
func (field Field) String() string { return stringify(field) }
|
|
func (expr SelectorExpr) String() string { return stringify(expr) }
|
|
func (typ StructType) String() string { return stringify(typ) }
|
|
func (expr KeyValueExpr) String() string { return stringify(expr) }
|
|
func (typ FuncType) String() string { return stringify(typ) }
|
|
func (lit FuncLit) String() string { return stringify(lit) }
|
|
func (decl FuncDecl) String() string { return stringify(decl) }
|
|
func (stmt BranchStmt) String() string { return stringify(stmt) }
|
|
func (expr CallExpr) String() string { return stringify(expr) }
|
|
func (clause CaseClause) String() string { return stringify(clause) }
|
|
func (typ ChanType) String() string { return stringify(typ) }
|
|
func (clause CommClause) String() string { return stringify(clause) }
|
|
func (lit CompositeLit) String() string { return stringify(lit) }
|
|
func (stmt EmptyStmt) String() string { return stringify(stmt) }
|
|
func (typ InterfaceType) String() string { return stringify(typ) }
|
|
func (stmt SwitchStmt) String() string { return stringify(stmt) }
|
|
func (expr TypeAssertExpr) String() string { return stringify(expr) }
|
|
func (spec TypeSpec) String() string { return stringify(spec) }
|
|
func (stmt TypeSwitchStmt) String() string { return stringify(stmt) }
|
|
func (nil Nil) String() string { return "nil" }
|
|
func (builtin Builtin) String() string { return stringify(builtin) }
|
|
func (obj Object) String() string { return stringify(obj) }
|
|
func (fn Function) String() string { return stringify(fn) }
|
|
func (el Ellipsis) String() string { return stringify(el) }
|
|
func (not Not) String() string { return stringify(not) }
|
|
|
|
func (or Or) String() string {
|
|
s := "(Or"
|
|
for _, node := range or.Nodes {
|
|
s += " "
|
|
s += node.String()
|
|
}
|
|
s += ")"
|
|
return s
|
|
}
|
|
|
|
func isProperList(l List) bool {
|
|
if l.Head == nil && l.Tail == nil {
|
|
return true
|
|
}
|
|
switch tail := l.Tail.(type) {
|
|
case nil:
|
|
return false
|
|
case List:
|
|
return isProperList(tail)
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
func (l List) String() string {
|
|
if l.Head == nil && l.Tail == nil {
|
|
return "[]"
|
|
}
|
|
|
|
if isProperList(l) {
|
|
// pretty-print the list
|
|
var objs []string
|
|
for l.Head != nil {
|
|
objs = append(objs, l.Head.String())
|
|
l = l.Tail.(List)
|
|
}
|
|
return fmt.Sprintf("[%s]", strings.Join(objs, " "))
|
|
}
|
|
|
|
return fmt.Sprintf("%s:%s", l.Head, l.Tail)
|
|
}
|
|
|
|
func (bind Binding) String() string {
|
|
if bind.Node == nil {
|
|
return bind.Name
|
|
}
|
|
return fmt.Sprintf("%s@%s", bind.Name, bind.Node)
|
|
}
|
|
|
|
func (s String) String() string { return fmt.Sprintf("%q", string(s)) }
|
|
|
|
func (tok Token) String() string {
|
|
return fmt.Sprintf("%q", strings.ToUpper(token.Token(tok).String()))
|
|
}
|
|
|
|
func (Any) String() string { return "_" }
|
|
|
|
func (AssignStmt) isNode() {}
|
|
func (IndexExpr) isNode() {}
|
|
func (Ident) isNode() {}
|
|
func (ValueSpec) isNode() {}
|
|
func (GenDecl) isNode() {}
|
|
func (BasicLit) isNode() {}
|
|
func (BinaryExpr) isNode() {}
|
|
func (ForStmt) isNode() {}
|
|
func (RangeStmt) isNode() {}
|
|
func (ArrayType) isNode() {}
|
|
func (DeferStmt) isNode() {}
|
|
func (MapType) isNode() {}
|
|
func (ReturnStmt) isNode() {}
|
|
func (SliceExpr) isNode() {}
|
|
func (StarExpr) isNode() {}
|
|
func (UnaryExpr) isNode() {}
|
|
func (SendStmt) isNode() {}
|
|
func (ImportSpec) isNode() {}
|
|
func (SelectStmt) isNode() {}
|
|
func (IfStmt) isNode() {}
|
|
func (IncDecStmt) isNode() {}
|
|
func (GoStmt) isNode() {}
|
|
func (Field) isNode() {}
|
|
func (SelectorExpr) isNode() {}
|
|
func (StructType) isNode() {}
|
|
func (KeyValueExpr) isNode() {}
|
|
func (FuncType) isNode() {}
|
|
func (FuncLit) isNode() {}
|
|
func (FuncDecl) isNode() {}
|
|
func (BranchStmt) isNode() {}
|
|
func (CallExpr) isNode() {}
|
|
func (CaseClause) isNode() {}
|
|
func (ChanType) isNode() {}
|
|
func (CommClause) isNode() {}
|
|
func (CompositeLit) isNode() {}
|
|
func (EmptyStmt) isNode() {}
|
|
func (InterfaceType) isNode() {}
|
|
func (SwitchStmt) isNode() {}
|
|
func (TypeAssertExpr) isNode() {}
|
|
func (TypeSpec) isNode() {}
|
|
func (TypeSwitchStmt) isNode() {}
|
|
func (Nil) isNode() {}
|
|
func (Builtin) isNode() {}
|
|
func (Object) isNode() {}
|
|
func (Function) isNode() {}
|
|
func (Ellipsis) isNode() {}
|
|
func (Or) isNode() {}
|
|
func (List) isNode() {}
|
|
func (String) isNode() {}
|
|
func (Token) isNode() {}
|
|
func (Any) isNode() {}
|
|
func (Binding) isNode() {}
|
|
func (Not) isNode() {}
|