machine: add --cachedrive-size flag to run and update (#4805)

Author: dangra

go.mod

@@ -73,7 +73,7 @@ require (
 	github.com/spf13/pflag v1.0.10
 	github.com/spf13/viper v1.20.1
 	github.com/stretchr/testify v1.11.1
-	github.com/superfly/fly-go v0.4.2
+	github.com/superfly/fly-go v0.4.3
 	github.com/superfly/graphql v0.2.6
 	github.com/superfly/lfsc-go v0.1.1
 	github.com/superfly/macaroon v0.3.0

go.sum

@@ -653,8 +653,8 @@ github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu
 github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
 github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8=
 github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU=
-github.com/superfly/fly-go v0.4.2 h1:QDAKDUSjR0tzYHoSqWTaHJXkECTwNl/F6RUFAaWikgQ=
-github.com/superfly/fly-go v0.4.2/go.mod h1:9ZPc5Yagx2y4gQJ0yP1+xxHyWR9U9lA2QSKjb5h4944=
+github.com/superfly/fly-go v0.4.3 h1:vJd7t3gY0VfXTJacKYFIqZJ1WnMjJE5gHZxQbgKUe6g=
+github.com/superfly/fly-go v0.4.3/go.mod h1:9ZPc5Yagx2y4gQJ0yP1+xxHyWR9U9lA2QSKjb5h4944=
 github.com/superfly/graphql v0.2.6 h1:zppbodNerWecoXEdjkhrqaNaSjGqobhXNlViHFuZzb4=
 github.com/superfly/graphql v0.2.6/go.mod h1:CVfDl31srm8HnJ9udwLu6hFNUW/P6GUM2dKcG1YQ8jc=
 github.com/superfly/lfsc-go v0.1.1 h1:dGjLgt81D09cG+aR9lJZIdmonjZSR5zYCi7s54+ZU2Q=

internal/command/machine/run.go

@@ -163,6 +163,10 @@ var sharedFlags = flag.Set{
 		Name:        "swap-size",
 		Description: "Swap size in MB. Accepts a plain number (in MB) or a human-readable size (e.g. 512mb, 1gb).",
 	},
+	flag.String{
+		Name:        "cachedrive-size",
+		Description: "Cache drive size in MB. Accepts a plain number (in MB) or a human-readable size (e.g. 512mb, 10gb). Set to 0 to disable.",
+	},
 }
 
 var runOrCreateFlags = flag.Set{
@@ -758,6 +762,21 @@ func determineMachineConfig(
 		machineConf.Init.SwapSizeMB = new(sizeMB)
 	}
 
+	if flag.IsSpecified(ctx, "cachedrive-size") {
+		sizeMB, err := helpers.ParseSize(flag.GetString(ctx, "cachedrive-size"), units.RAMInBytes, units.MiB)
+		if err != nil {
+			return machineConf, fmt.Errorf("invalid cachedrive size: %w", err)
+		}
+		if sizeMB < 0 {
+			return machineConf, fmt.Errorf("--cachedrive-size must not be negative")
+		}
+		if sizeMB == 0 {
+			machineConf.CacheDrive = nil
+		} else {
+			machineConf.CacheDrive = &fly.MachineCacheDrive{SizeMB: uint64(sizeMB)}
+		}
+	}
+
 	parsedEnv, err := parseKVFlag(ctx, "env", machineConf.Env)
 	if err != nil {
 		return machineConf, err