From 833fd260919ff7d12ae5d63f944bed2e0beb8b4c Mon Sep 17 00:00:00 2001
From: Harvey Tindall <hrfee@protonmail.ch>
Date: Tue, 9 Dec 2025 15:44:47 +0000
Subject: [PATCH] scripts/variants: add parallel flag

doesn't result in much gain on my machine or my CI server so it's not
the default. On local machine (5800x), went from ~5ms to ~4ms, on my CI
(hetzner ampere altra with 8 vCPUs) went from ~15ms to ~10ms.
---
 scripts/variants/main.go | 47 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 45 insertions(+), 2 deletions(-)

diff --git a/scripts/variants/main.go b/scripts/variants/main.go
index 0962079..df3611d 100644
--- a/scripts/variants/main.go
+++ b/scripts/variants/main.go
@@ -13,6 +13,7 @@ import (
 	"os"
 	"path/filepath"
 	"regexp"
+	"sync"
 )
 
 var VERBOSE = false
@@ -71,6 +72,42 @@ func ParseDir(in, out string) error {
 	return err
 }
 
+func ParseDirParallel(in, out string) error {
+	var wg sync.WaitGroup
+	err := filepath.WalkDir(in, func(path string, d fs.DirEntry, err error) error {
+		if err != nil {
+			return err
+		}
+		info, err := d.Info()
+		if err != nil {
+			return err
+		}
+		perm := info.Mode()
+		rel, err := filepath.Rel(in, path)
+		outFile := filepath.Join(out, rel)
+		if d.IsDir() {
+			return os.MkdirAll(outFile, perm)
+		}
+		if VERBOSE {
+			log.Printf("\"%s\" => \"%s\"\n", path, outFile)
+		}
+		if err != nil {
+			return err
+		}
+		wg.Go(func() {
+			if err := ParseFile(path, outFile, &perm); err != nil {
+				panic(err)
+			}
+		})
+		return nil
+	})
+	if err != nil {
+		return err
+	}
+	wg.Wait()
+	return err
+}
+
 func ParseFile(in, out string, perm *fs.FileMode) error {
 	file, err := os.ReadFile(in)
 	if err != nil {
@@ -133,10 +170,12 @@ func ParseFile(in, out string, perm *fs.FileMode) error {
 
 func main() {
 	var inFile, inDir, out string
+	var parallel bool
 	flag.StringVar(&inFile, "file", "", "Input of an individual file.")
 	flag.StringVar(&inDir, "dir", "", "Input of a whole directory.")
 	flag.StringVar(&out, "out", "", "Output filepath/directory, depending on if -file or -dir passed.")
-	flag.BoolVar(&VERBOSE, "v", false, "prints information about files and replacements as they are made")
+	flag.BoolVar(&VERBOSE, "v", false, "Prints information about files and replacements as they are made")
+	flag.BoolVar(&parallel, "p", false, "Run a goroutine per file. Probably won't speed things up.")
 	flag.Parse()
 
 	if out == "" {
@@ -147,7 +186,11 @@ func main() {
 	if inFile != "" {
 		err = ParseFile(inFile, out, nil)
 	} else if inDir != "" {
-		err = ParseDir(inDir, out)
+		if parallel {
+			err = ParseDirParallel(inDir, out)
+		} else {
+			err = ParseDir(inDir, out)
+		}
 	} else {
 		flag.PrintDefaults()
 		os.Exit(1)