Chore: Add go workspace scripts (#91707)

scripts/go-workspace/main.go

@@ -0,0 +1,93 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+	"strings"
+
+	"golang.org/x/mod/modfile"
+)
+
+func main() {
+	if len(os.Args) < 2 {
+		printUsage()
+		os.Exit(1)
+	}
+
+	var err error
+	switch os.Args[1] {
+	case "list-submodules":
+		err = listSubmodules()
+	default:
+		printUsage()
+	}
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+		os.Exit(1)
+	}
+}
+
+func printUsage() {
+	println("Usage: go-workspace <command> [args]")
+	println("Commands:")
+	println("  list-submodules - List submodules in go.work")
+}
+
+func listSubmodules() error {
+	fs := flag.NewFlagSet("list-submodules", flag.ExitOnError)
+	workPath := fs.String("path", "go.work", "Path to go.work")
+	delimiter := fs.String("delimiter", "\n", "Delimiter to use when printing paths")
+	skip := fs.String("skip", "", "Skip submodules with this comment tag")
+	help := fs.Bool("help", false, "Print help message")
+	fs.Parse(os.Args[2:])
+
+	if *help {
+		fs.Usage()
+		return nil
+	}
+
+	workfile, err := parseGoWork(*workPath)
+	if err != nil {
+		return err
+	}
+
+	paths := getSubmodulePaths(workfile, *skip)
+	for _, p := range paths {
+		fmt.Printf("%s%s", p, *delimiter)
+	}
+
+	return nil
+}
+
+func getSubmodulePaths(wf *modfile.WorkFile, skip string) []string {
+	var paths []string
+	for _, d := range wf.Use {
+		if hasSkipTag(d, skip) {
+			continue
+		}
+		paths = append(paths, d.Path)
+	}
+	return paths
+}
+
+func hasSkipTag(d *modfile.Use, tag string) bool {
+	if tag == "" {
+		return false
+	}
+	for _, c := range d.Syntax.Comments.Suffix {
+		if strings.Contains(c.Token, fmt.Sprintf("skip:%s", tag)) {
+			return true
+		}
+	}
+	return false
+}
+
+func parseGoWork(p string) (*modfile.WorkFile, error) {
+	contents, err := os.ReadFile(p)
+	if err != nil {
+		return nil, err
+	}
+
+	return modfile.ParseWork(p, contents, nil)
+}