gop/printer.SetDebug

README.md

@@ -29,9 +29,11 @@ How do we do this in the Go language?
 ```go
 package main
 
+import "fmt"
+
 func main() {
     a := []float64{1, 2, 3.4}
-    println(a)
+    fmt.Println(a)
 }
 ```
 

cl/class_file.go

@@ -39,7 +39,7 @@ type gmxInfo struct {
 
 var (
 	gmxTypes = map[string]gmxInfo{
-		".gmx": {".spx", []string{"github.com/goplus/spx", "math"}},
+		".gmx": {".spx", []string{"github.com/goplus/spx"}},
 	}
 )
 

cl/compile_test.go

@@ -2353,6 +2353,14 @@ var a = math.Round(1.2)
 `)
 }
 
+func _TestLocalImport(t *testing.T) {
+	gopClTest(t, `import "./internal/spx"
+
+var a = spx.TestIntValue
+`, `
+`)
+}
+
 func TestImportUnused(t *testing.T) {
 	gopClTest(t, `import "fmt"
 

parser/parser.go

@@ -1340,7 +1340,7 @@ func (p *parser) isCommand(x ast.Expr) bool {
 	case token.IDENT, token.RARROW,
 		token.STRING, token.INT, token.FLOAT, token.IMAG, token.CHAR, token.RAT:
 		return true
-	case token.SUB, token.NOT, token.AND, token.MUL, token.ARROW, token.XOR:
+	case token.SUB, token.NOT, token.AND, token.MUL, token.ARROW, token.XOR, token.ADD:
 		if x.End() == p.pos { // x-y
 			return false
 		}

printer/gop_test.go

@@ -29,6 +29,10 @@ import (
 	"github.com/goplus/gop/token"
 )
 
+func init() {
+	printer.SetDebug(printer.DbgFlagAll)
+}
+
 func TestNoPkgDecl(t *testing.T) {
 	var dst bytes.Buffer
 	fset := token.NewFileSet()

printer/nodes.go

@@ -37,6 +37,20 @@ import (
 	"github.com/goplus/gop/token"
 )
 
+const (
+	DbgFlagAll = 1
+)
+
+var (
+	debugFormat bool
+)
+
+func SetDebug(flags int) {
+	if flags != 0 {
+		debugFormat = true
+	}
+}
+
 // Formatting issues:
 // - better comment formatting for /*-style comments at the end of a line (e.g. a declaration)
 //   when the comment spans multiple lines; if such a comment is just two lines, formatting is
@@ -1320,6 +1334,11 @@ func (p *printer) stmt(stmt ast.Stmt, nextIsRBrace bool) {
 		p.stmt(s.Stmt, nextIsRBrace)
 
 	case *ast.ExprStmt:
+		if debugFormat {
+			if e, ok := s.X.(*ast.CallExpr); ok {
+				log.Println("==> ExprStmt", e.Fun)
+			}
+		}
 		const depth = 1
 		p.expr0(s.X, depth)
 
@@ -1335,6 +1354,9 @@ func (p *printer) stmt(stmt ast.Stmt, nextIsRBrace bool) {
 		p.print(s.TokPos, s.Tok)
 
 	case *ast.AssignStmt:
+		if debugFormat {
+			log.Println("==> AssignStmt", s.Lhs)
+		}
 		var depth = 1
 		if len(s.Lhs) > 1 && len(s.Rhs) > 1 {
 			depth++
@@ -1912,6 +1934,9 @@ func (p *printer) distanceFrom(startPos token.Pos, startOutCol int) int {
 }
 
 func (p *printer) funcDecl(d *ast.FuncDecl) {
+	if debugFormat {
+		log.Println("==> Format Func", d.Name.Name)
+	}
 	p.setComment(d.Doc)
 
 	if p.unnamedFuncName == d.Name.Name {