feat(machine): support vm-max-memory for run and update

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.1
+	github.com/superfly/fly-go v0.4.2
 	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.1 h1:l29ohaVaeQ55VdMJJsxCiIPUHrROXvwDswE8LAttXgk=
-github.com/superfly/fly-go v0.4.1/go.mod h1:9ZPc5Yagx2y4gQJ0yP1+xxHyWR9U9lA2QSKjb5h4944=
+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/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/flag/machines.go

@@ -58,6 +58,19 @@ func GetMachineGuest(ctx context.Context, guest *fly.MachineGuest) (*fly.Machine
 		}
 	}
 
+	if IsSpecified(ctx, "vm-max-memory") {
+		rawValue := GetString(ctx, "vm-max-memory")
+		maxMemoryMB, err := helpers.ParseSize(rawValue, units.RAMInBytes, units.MiB)
+		switch {
+		case err != nil:
+			return nil, err
+		case maxMemoryMB == 0:
+			return nil, fmt.Errorf("--vm-max-memory cannot be zero")
+		default:
+			guest.MaxMemoryMB = maxMemoryMB
+		}
+	}
+
 	if IsSpecified(ctx, "vm-cpu-kind") {
 		guest.CPUKind = GetString(ctx, "vm-cpu-kind")
 		if k := guest.CPUKind; k != "shared" && k != "performance" {
@@ -121,6 +134,11 @@ var VMSizeFlags = Set{
 		Description: "Memory (in megabytes) to attribute to the VM",
 		Aliases:     []string{"memory"},
 	},
+	String{
+		Name:        "vm-max-memory",
+		Description: "Maximum memory (in megabytes) to allow for the VM",
+		Aliases:     []string{"max-memory"},
+	},
 	Int{
 		Name:        "vm-gpus",
 		Description: "Number of GPUs. Must also choose the GPU model with --vm-gpu-kind flag",