chore/litgen: safely refresh continuous checks by default

chore/litgen/litgen.go

@@ -21,12 +21,18 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"sync"
 
+	llgoenv "github.com/goplus/llgo/internal/env"
 	"github.com/goplus/llgo/internal/littest"
 	"github.com/goplus/llgo/xtool/env/llvm"
 )
 
+var force = flag.Bool("force", false, "replace all CHECK directives with fully regenerated IR checks")
+var setupLLGoRootOnce sync.Once
+
 func main() {
+	setupLLGoRoot()
 	llvm.SetupPath()
 	flag.Usage = func() {
 		fmt.Fprintf(flag.CommandLine.Output(), "Usage: %s <file-or-dir> [<file-or-dir>...]\n", filepath.Base(os.Args[0]))
@@ -42,7 +48,22 @@ func main() {
 	}
 }
 
+func setupLLGoRoot() {
+	setupLLGoRootOnce.Do(func() {
+		if os.Getenv("LLGO_ROOT") != "" {
+			return
+		}
+		if root := llgoenv.LLGoROOT(); root != "" {
+			_ = os.Setenv("LLGO_ROOT", root)
+		}
+	})
+}
+
 func processPath(path string) error {
+	return processPathWithForce(path, *force)
+}
+
+func processPathWithForce(path string, force bool) error {
 	abs, err := filepath.Abs(path)
 	if err != nil {
 		return err
@@ -52,7 +73,7 @@ func processPath(path string) error {
 		return err
 	}
 	if fi.IsDir() {
-		return processTree(abs)
+		return processTree(abs, force)
 	}
 	if filepath.Ext(abs) != ".go" {
 		return fmt.Errorf("%s: expected .go file or directory", abs)
@@ -69,10 +90,10 @@ func processPath(path string) error {
 		return err
 	}
 	fmt.Fprintln(os.Stderr, "litgen", target.sourceFile)
-	return generateFile(target)
+	return generateFile(target, force)
 }
 
-func processTree(root string) error {
+func processTree(root string, force bool) error {
 	var targets []resolvedTarget
 	err := filepath.WalkDir(root, func(path string, d os.DirEntry, err error) error {
 		if err != nil {
@@ -106,7 +127,7 @@ func processTree(root string) error {
 	}
 	for _, target := range targets {
 		fmt.Fprintln(os.Stderr, "litgen", target.sourceFile)
-		if err := generateFile(target); err != nil {
+		if err := generateFile(target, force); err != nil {
 			return err
 		}
 	}

chore/litgen/litgen_test.go

@@ -27,7 +27,7 @@ func TestProcessPath_SingleFileUsesContainingDir(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	if err := processPath(sourceFile); err != nil {
+	if err := processPathWithForce(sourceFile, true); err != nil {
 		t.Fatal(err)
 	}
 
@@ -68,7 +68,7 @@ func TestProcessPath_UsesFlagsFileTarget(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	if err := processPath(sourceFile); err != nil {
+	if err := processPathWithForce(sourceFile, true); err != nil {
 		t.Fatal(err)
 	}