Boundary check for optional arg

urfave · Feb 11, 2024 · f7deded · f7deded
1 parent 00e6067
commit f7deded
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/args.go b/args.go
@@ -132,7 +132,7 @@ func (a *ArgumentBase[T, C, VC]) Parse(s []string) ([]string, error) {
 		*a.Values = values
 	}
 
-	if a.Max == 1 && a.Destination != nil {
+	if a.Max == 1 && a.Destination != nil && len(values) > 0 {
 		*a.Destination = values[0]
 	}
 	return s[count:], nil

diff --git a/args_test.go b/args_test.go
@@ -151,3 +151,23 @@ func TestArgsUsage(t *testing.T) {
 		})
 	}
 }
+
+func TestSingleOptionalArg(t *testing.T) {
+	cmd := buildMinimalTestCommand()
+	var s1 string
+	arg := &StringArg{
+		Min:         0,
+		Max:         1,
+		Destination: &s1,
+	}
+	cmd.Arguments = []Argument{
+		arg,
+	}
+
+	require.NoError(t, cmd.Run(context.Background(), []string{"foo"}))
+	require.Equal(t, "", s1)
+
+	arg.Value = "bar"
+	require.NoError(t, cmd.Run(context.Background(), []string{"foo", "bar"}))
+	require.Equal(t, "bar", s1)
+}