Validate number of CPUs

Signed-off-by: Veda Kutagula <[email protected]>

Author: k-veda

cmd/limactl/editflags/editflags.go

@@ -191,7 +191,24 @@ func YQExpressions(flags *flag.FlagSet, newInstance bool) ([]string, error) {
 	}
 	d := defaultExprFunc
 	defs := []def{
-		{"cpus", d(".cpus = %s"), false, false},
+		{
+			"cpus",
+			func(_*flag.Flag) ([]string, error) {
+
+				num_cpus, err := flags.GetInt("cpus")
+				if err != nil {
+					return nil, err
+				}
+
+				if num_cpus < 0 {
+					return nil, fmt.Errorf("invalid value for number of cpus, must be >= 0")
+				} else {
+					return []string{fmt.Sprintf(".cpus = %d", num_cpus)}, nil
+				}
+			},
+			false,
+			false,
+		},
 		{
 			"dns",
 			func(_ *flag.Flag) ([]string, error) {