fix updating latest Zig version to -wip nomination

Signed-off-by: Stephen Gutekanst <[email protected]>

internal/wrench/scripts/install_zig.go

@@ -8,6 +8,7 @@ import (
 	"os/exec"
 	"path/filepath"
 	"runtime"
+	"strings"
 
 	"github.com/hexops/wrench/internal/errors"
 )
@@ -20,7 +21,7 @@ func init() {
 		Execute: func(args ...string) error {
 			force := len(args) == 1 && args[0] == "true"
 
-			wantZigVersion, err := QueryZigVersion("mach-latest")
+			wantZigVersion, err := QueryZigVersion("latest")
 			if err != nil {
 				return errors.Wrap(err, "QueryZigVersion")
 			}
@@ -97,6 +98,16 @@ func QueryZigVersion(name string) (string, error) {
 	if err := json.NewDecoder(resp.Body).Decode(&v); err != nil {
 		return "", err
 	}
+	if name == "latest" {
+		// If we want generic latest it means either we should use e.g. "2024.3.0-mach-wip" if
+		// there is a work-in-progress nomination, otherwise fallback to "mach-latest" tag.
+		for name, version := range v {
+			if strings.HasSuffix(name, "-mach-wip") {
+				return version.Version, nil
+			}
+		}
+		name = "mach-latest"
+	}
 	version, ok := v[name]
 	if !ok {
 		return "", fmt.Errorf("no Zig version %q found in %s", name, indexURL)

internal/wrench/scripts/mach-push-rewrite-zig-version.go

@@ -10,10 +10,14 @@ import (
 func init() {
 	Scripts = append(Scripts, Script{
 		Command:     "mach-push-rewrite-zig-version",
-		Args:        nil,
+		Args:        []string{"new version"},
 		Description: "wrench installs prerequisites (Go), rebuilds itself, and restarts the service",
 		ExecuteResponse: func(args ...string) (*api.ScriptResponse, error) {
-			wantZigVersion, err := QueryZigVersion("mach-latest")
+			if len(args) != 1 {
+				return nil, errors.New("expected [new version] argument (e.g. 'latest', 'mach-latest')")
+			}
+
+			wantZigVersion, err := QueryZigVersion(args[0])
 			if err != nil {
 				return nil, errors.Wrap(err, "QueryZigVersion")
 			}