qll库模块生成路径支持自定义

cmd/codemillx/main.go

@@ -9,9 +9,13 @@ import (
 )
 
 var moduleName string
+var customizeCodeQLAction bool
+var output string
 
 func init() {
-	flag.StringVar(&moduleName, "module", "", "module name")
+	flag.StringVar(&moduleName, "module", "Customizations", "module name")
+	flag.BoolVar(&customizeCodeQLAction, "customizeCodeQLAction", false, "customize CodeQL Action")
+	flag.StringVar(&output, "output", "Customizations.qll", "output file name")
 }
 
 func main() {
@@ -25,10 +29,14 @@ func main() {
 
 	codeqlModuleSpec := codemillx.ExtractCodeqlModuleSpec(moduleName, pkgs)
 
-	ouputPath, err := codemill.GenerateCodeQL(codeqlModuleSpec)
-	if err != nil {
+	if customizeCodeQLAction {
+		if customizeFile := codemillx.FindCustomizationsQllFile(); len(customizeFile) > 0 {
+			output = customizeFile
+		}
+	}
+
+	if err := codemill.GenerateCodeQL(codeqlModuleSpec, output); err != nil {
 		fmt.Println("GenerateCodeQL with error:", err)
 		return
 	}
-	fmt.Println(ouputPath)
 }

codemill/codeql.go

@@ -8,7 +8,7 @@ import (
 	"github.com/gagliardetto/utilz"
 )
 
-func GenerateCodeQL(module CodeqlModuleSpec) (string, error) {
+func GenerateCodeQL(module CodeqlModuleSpec, output string) error {
 	cqlFile := cqljen.NewFile()
 	cqlFile.Import("go")
 	cqlFile.Private().Module().Id(utilz.ToCamel(module.ModuleName)).BlockFunc(func(moduleGroup *cqljen.Group) {
@@ -39,12 +39,12 @@ func GenerateCodeQL(module CodeqlModuleSpec) (string, error) {
 		}
 	})
 
-	codeqlFile, err := os.Create(module.ModuleName + ".qll")
+	codeqlFile, err := os.Create(output)
 	if err != nil {
-		return "", err
+		return err
 	}
 	defer codeqlFile.Close()
-	return codeqlFile.Name(), cqlFile.Render(codeqlFile)
+	return cqlFile.Render(codeqlFile)
 }
 
 func GetFuncQualifierCodeElements(qual *FuncQualifier) []cqljen.Code {

util.go

@@ -0,0 +1,29 @@
+package codemillx
+
+import (
+	"os"
+	"path/filepath"
+)
+
+func FindCustomizationsQllFile() string {
+	matches, err := filepath.Glob("/opt/hostedtoolcache/CodeQL/*/x64/codeql/qlpacks/codeql/go-all/*")
+	if err != nil || len(matches) == 0 {
+		return ""
+	}
+	for _, v := range matches {
+		f := filepath.Join(v, "Customizations.qll")
+		if FileExists(f) {
+			return f
+		}
+	}
+	return ""
+}
+
+func FileExists(name string) bool {
+	if _, err := os.Stat(name); err != nil {
+		if os.IsNotExist(err) {
+			return false
+		}
+	}
+	return true
+}