feat: command id generator support

Signed-off-by: chenk <[email protected]>

cmd/command__id/main.go

@@ -0,0 +1,65 @@
+package main
+
+import (
+	"fmt"
+	"path/filepath"
+	"strconv"
+	"strings"
+
+	trivy_checks "github.com/aquasecurity/trivy-checks"
+	"gopkg.in/yaml.v2"
+)
+
+const (
+	commandPath = "commands/kubernetes"
+)
+
+func main() {
+	ids, err := GetCommandIDRange()
+	if err != nil {
+		panic(err)
+	}
+	for i := 0; i < len(ids); i++ {
+		if !ids[i] {
+			println(fmt.Sprintf("%s-%04d", "CMD", i+1))
+		}
+	}
+}
+func GetCommandIDRange() ([]bool, error) {
+	commandsIds := make([]bool, 9999)
+	entries, err := trivy_checks.EmbeddedK8sCommandsFileSystem.ReadDir(commandPath)
+	if err != nil {
+		panic(err)
+	}
+	for _, entry := range entries {
+		if entry.IsDir() {
+			continue
+		}
+		fContent, err := trivy_checks.EmbeddedK8sCommandsFileSystem.ReadFile(filepath.Join(commandPath, entry.Name()))
+		if err != nil {
+			return nil, err
+		}
+		var fileCommand any
+		err = yaml.Unmarshal(fContent, &fileCommand)
+		if err != nil {
+			panic(err)
+		}
+
+		if commandArr, ok := fileCommand.([]interface{}); ok {
+			if commandMap, ok := commandArr[0].(map[any]any); ok {
+				if id, ok := commandMap["id"]; ok {
+					idStr := id.(string)
+					idWithPrefix := strings.TrimPrefix(idStr, "CMD-")
+					idNum, err := strconv.Atoi(idWithPrefix)
+					if err != nil {
+						return nil, err
+					}
+					if idNum > 0 && idNum <= 9999 {
+						commandsIds[idNum-1] = true
+					}
+				}
+			}
+		}
+	}
+	return commandsIds, nil
+}

embed.go

@@ -10,8 +10,8 @@ var EmbeddedPolicyFileSystem embed.FS
 //go:embed lib/*
 var EmbeddedLibraryFileSystem embed.FS
 
-//go:embed commands/kubernetes/*
+//go:embed commands/kubernetes
 var EmbeddedK8sCommandsFileSystem embed.FS
 
-//go:embed commands/config/*
+//go:embed commands/config
 var EmbeddedConfigCommandsFileSystem embed.FS