bloeys/cogo
1
0
Fork 0
mirror of https://github.com/bloeys/cogo.git synced 2025-12-29 08:58:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Comments

Browse Source
This commit is contained in:
bloeys
2022-10-24 06:30:07 +04:00
parent a648550835
commit 30e95f401a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
inliner/main.go
View File

@ -52,9 +52,6 @@ func processPkg(pkg *packages.Package) {
for i, synFile := range pkg.Syntax {
pkg.Syntax[i] = astutil.Apply(synFile, processDeclNode, nil).(*ast.File)
// for _, dec := range synFile.Decls {
// ast.Inspect(dec, processDeclNode)
// }
}
// f, err := os.Create("main_gen.go")
@ -80,12 +77,14 @@ func processDeclNode(c *astutil.Cursor) bool {
return true
}
// Only operate on func called 'test'
if funcDecl.Name.Name != "test" {
return false
}
for _, stmt := range funcDecl.Body.List {
// Find functions calls in the style of 'cogo.ABC123()'
exprStmt, ok := stmt.(*ast.ExprStmt)
if !ok {
continue
@ -107,6 +106,7 @@ func processDeclNode(c *astutil.Cursor) bool {
}
fmt.Printf("Found: %+v\n", pkgFuncCallExpr)
// Now that we found a call to cogo decide what to do
if pkgFuncCallExpr.Sel.Name == "Yield" {
exprStmt.X = &ast.CallExpr{