move asthelper functions to separate package (#78)
lu4p
5 years ago
committed by
GitHub
parent
846ddb4097
commit
5cbbac56f3
@ -0,0 +1,98 @@
|
||||
package asthelper
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"go/ast"
|
||||
"go/token"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
// Ident an identifier
|
||||
func Ident(name string) *ast.Ident {
|
||||
return &ast.Ident{Name: name}
|
||||
}
|
||||
|
||||
// StringLit returns an ast.BasicLit of kind STRING
|
||||
func StringLit(value string) *ast.BasicLit {
|
||||
return &ast.BasicLit{
|
||||
Kind: token.STRING,
|
||||
Value: fmt.Sprintf("%q", value),
|
||||
}
|
||||
}
|
||||
|
||||
// IntLit returns an ast.BasicLit of kind INT
|
||||
func IntLit(value int) *ast.BasicLit {
|
||||
return &ast.BasicLit{
|
||||
Kind: token.INT,
|
||||
Value: strconv.Itoa(value),
|
||||
}
|
||||
}
|
||||
|
||||
// IndexExpr "name[index]"
|
||||
func IndexExpr(name string, index ast.Expr) *ast.IndexExpr {
|
||||
return &ast.IndexExpr{
|
||||
X: Ident(name),
|
||||
Index: index,
|
||||
}
|
||||
}
|
||||
|
||||
// CallExpr "fun(arg)"
|
||||
func CallExpr(fun ast.Expr, args ...ast.Expr) *ast.CallExpr {
|
||||
return &ast.CallExpr{
|
||||
Fun: fun,
|
||||
Args: args,
|
||||
}
|
||||
}
|
||||
|
||||
// LambdaCall "func() resultType {block}()"
|
||||
func LambdaCall(resultType ast.Expr, block *ast.BlockStmt) *ast.CallExpr {
|
||||
funcLit := &ast.FuncLit{
|
||||
Type: &ast.FuncType{
|
||||
Params: &ast.FieldList{},
|
||||
Results: &ast.FieldList{
|
||||
List: []*ast.Field{
|
||||
{Type: resultType},
|
||||
},
|
||||
},
|
||||
},
|
||||
Body: block,
|
||||
}
|
||||
return CallExpr(funcLit)
|
||||
}
|
||||
|
||||
// ReturnStmt "return result"
|
||||
func ReturnStmt(results ...ast.Expr) *ast.ReturnStmt {
|
||||
return &ast.ReturnStmt{
|
||||
Results: results,
|
||||
}
|
||||
}
|
||||
|
||||
// BoundsCheck "_ = name[pos]"
|
||||
func BoundsCheck(name string, pos int) *ast.AssignStmt {
|
||||
return &ast.AssignStmt{
|
||||
Lhs: []ast.Expr{Ident("_")},
|
||||
Tok: token.ASSIGN,
|
||||
Rhs: []ast.Expr{IndexExpr("data", IntLit(pos))},
|
||||
}
|
||||
}
|
||||
|
||||
// BlockStmt a block of multiple statments e.g. a function body
|
||||
func BlockStmt(stmts ...ast.Stmt) *ast.BlockStmt {
|
||||
return &ast.BlockStmt{List: stmts}
|
||||
}
|
||||
|
||||
// ExprStmt convert an ast.Expr to an ast.Stmt
|
||||
func ExprStmt(expr ast.Expr) *ast.ExprStmt {
|
||||
return &ast.ExprStmt{X: expr}
|
||||
}
|
||||
|
||||
// DataToByteSlice turns a byte slice like []byte{1, 2, 3} into an AST
|
||||
// expression
|
||||
func DataToByteSlice(data []byte) *ast.CallExpr {
|
||||
return &ast.CallExpr{
|
||||
Fun: &ast.ArrayType{
|
||||
Elt: &ast.Ident{Name: "byte"},
|
||||
},
|
||||
Args: []ast.Expr{StringLit(string(data))},
|
||||
}
|
||||
}
|
||||
@ -1,71 +0,0 @@
|
||||
package literals
|
||||
|
||||
import (
|
||||
"go/ast"
|
||||
"go/token"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
func ident(name string) *ast.Ident {
|
||||
return &ast.Ident{Name: name}
|
||||
}
|
||||
|
||||
func intLiteral(value int) *ast.BasicLit {
|
||||
return &ast.BasicLit{
|
||||
Kind: token.INT,
|
||||
Value: strconv.Itoa(value),
|
||||
}
|
||||
}
|
||||
|
||||
// name[index]
|
||||
func indexExpr(name string, index ast.Expr) *ast.IndexExpr {
|
||||
return &ast.IndexExpr{
|
||||
X: ident(name),
|
||||
Index: index,
|
||||
}
|
||||
}
|
||||
|
||||
// fun(arg)
|
||||
func callExpr(fun ast.Expr, arg ast.Expr) *ast.CallExpr {
|
||||
var args []ast.Expr
|
||||
if arg != nil {
|
||||
args = []ast.Expr{arg}
|
||||
}
|
||||
|
||||
return &ast.CallExpr{
|
||||
Fun: fun,
|
||||
Args: args,
|
||||
}
|
||||
}
|
||||
|
||||
// func() resultType {block}()
|
||||
func lambdaCall(resultType ast.Expr, block *ast.BlockStmt) *ast.CallExpr {
|
||||
funcLit := &ast.FuncLit{
|
||||
Type: &ast.FuncType{
|
||||
Params: &ast.FieldList{},
|
||||
Results: &ast.FieldList{
|
||||
List: []*ast.Field{
|
||||
{Type: resultType},
|
||||
},
|
||||
},
|
||||
},
|
||||
Body: block,
|
||||
}
|
||||
return callExpr(funcLit, nil)
|
||||
}
|
||||
|
||||
// return result
|
||||
func returnStmt(results ...ast.Expr) *ast.ReturnStmt {
|
||||
return &ast.ReturnStmt{
|
||||
Results: results,
|
||||
}
|
||||
}
|
||||
|
||||
// _ = data[pos]
|
||||
func boundsCheckData(pos int) *ast.AssignStmt {
|
||||
return &ast.AssignStmt{
|
||||
Lhs: []ast.Expr{ident("_")},
|
||||
Tok: token.ASSIGN,
|
||||
Rhs: []ast.Expr{indexExpr("data", intLiteral(pos))},
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue