chore: add fmt cmd for outputs

cmd/llpyg/llpyg.go

@@ -63,6 +63,9 @@ func main() {
 	// tidy go module
 	goModTidy(args.OutputDir)
 
+	// format go code
+	codeFormat(args.OutputDir)
+
 	fmt.Printf("LLGo bindings generated successfully in %s\n", args.OutputDir)
 }
 

cmd/llpyg/llpyg_test.go

@@ -106,3 +106,37 @@ func assertArgsEqual(t *testing.T, got, want Args) {
 		t.Errorf("unexpected Kwarg: got %q, want %q", got.Kwarg, want.Kwarg)
 	}
 }
+
+
+func TestGoModuleUtils(t *testing.T) {
+	tempDir := t.TempDir()
+	goFile := filepath.Join(tempDir, "test.go")
+	goContent := `package main
+
+	import (
+		"fmt"
+		"github.com/goplus/lib/py"
+	)
+
+	func main() {
+	a := py.Object{}
+	fmt.Printf("hello %v", a)
+	}
+	`
+	err := os.WriteFile(goFile, []byte(goContent), 0644)
+	if err != nil {
+		t.Fatalf("Failed to create test Go file: %v", err)
+	}
+	err = initGoModule("test", tempDir)
+	if err != nil {
+		t.Fatal(err)
+	}
+	err = goModTidy(tempDir)
+	if err != nil {
+		t.Log(err)
+	}
+	err = codeFormat(tempDir)
+	if err != nil {
+		t.Log(err)
+	}
+}

cmd/llpyg/utils.go

@@ -32,36 +32,43 @@ func writeConfig(cfg Config, outDir string) error {
 }
 
 func initGoModule(modName string, outDir string) error {
-	if err := os.Chdir(outDir); err != nil {
-		return fmt.Errorf("error: failed to change directory: %w", err)
-	}
 	// init go module
 	cmd := exec.Command("go", "mod", "init", modName)
+	cmd.Dir = outDir
 	cmd.Stdout = os.Stdout
 	cmd.Stderr = os.Stderr
 	if err := cmd.Run(); err != nil {
 		return fmt.Errorf("error: failed to initialize Go module: %w", err)
 	}
 
-	getCmd := exec.Command("go", "get", "github.com/goplus/lib/py")
-	getCmd.Stdout = os.Stdout
-	getCmd.Stderr = os.Stderr
-	if err := getCmd.Run(); err != nil {
+	cmd = exec.Command("go", "get", "github.com/goplus/lib/py")
+	cmd.Dir = outDir
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	if err := cmd.Run(); err != nil {
 		return fmt.Errorf("error: failed to get github.com/goplus/lib/py: %w", err)
 	}
 	return nil
 }
 
 func goModTidy(outDir string) error {
-	if err := os.Chdir(outDir); err != nil {
-		return fmt.Errorf("error: failed to change directory: %w", err)
-	}
-	tidyCmd := exec.Command("go", "mod", "tidy")
-	tidyCmd.Stdout = os.Stdout
-	tidyCmd.Stderr = os.Stderr
-	if err := tidyCmd.Run(); err != nil {
+	cmd := exec.Command("go", "mod", "tidy")
+	cmd.Dir = outDir
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	if err := cmd.Run(); err != nil {
 		return fmt.Errorf("error: failed to tidy Go module: %w", err)
 	}
 	return nil
 }
 
+func codeFormat(outDir string) error {
+	cmd := exec.Command("go", "fmt", "./...")
+	cmd.Dir = outDir
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	if err := cmd.Run(); err != nil {
+		return fmt.Errorf("error: failed to format Go code: %w", err)
+	}
+	return nil
+}