add flag to machine run, use IsSpecified to set opts

superfly · Nov 15, 2024 · 8f0f5e5 · 8f0f5e5
1 parent eddb4ed
commit 8f0f5e5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/internal/command/command_run.go b/internal/command/command_run.go
@@ -76,9 +76,10 @@ func DetermineImage(ctx context.Context, appName string, imageOrPath string) (im
 		if cfg != nil && cfg.Experimental != nil {
 			opts.UseZstd = cfg.Experimental.UseZstd
 		}
+
 		// use-zstd passed through flags takes precedence over the one set in config
-		if useZstd := flag.GetBool(ctx, "use-zstd"); useZstd {
-			opts.UseZstd = useZstd
+		if flag.IsSpecified(ctx, "use-zstd") {
+			opts.UseZstd = flag.GetBool(ctx, "use-zstd")
 		}
 
 		img, err = resolver.BuildImage(ctx, io, opts)

diff --git a/internal/command/machine/run.go b/internal/command/machine/run.go
@@ -180,6 +180,10 @@ var runOrCreateFlags = flag.Set{
 		Description: "Enable LSVD for this machine",
 		Hidden:      true,
 	},
+	flag.Bool{
+		Name:        "use-zstd",
+		Description: "Enable zstd compression for the image",
+	},
 }
 
 func soManyErrors(args ...interface{}) error {