Fix arg parsing for plugins. Add missing global args.

Signed-off-by: Artem Mikhalitsin <artemmikhalitsin@gmail.com>

pkg/cmd/load_plugins.go

@@ -159,11 +159,36 @@ func processParent(cmd *cobra.Command, args []string) ([]string, error) {
 // manuallyProcessArgs processes an arg array, removing special args.
 //
 // Returns two sets of args: known and unknown (in that order)
+// Known args are flags that are consumed by the root `helm` command
+// Unknown args are any args that will not be consumed by helm, and that may be
+// processed by the plugin plus
+// The single exception is -h or --help, which is passed to the plugin
 func manuallyProcessArgs(args []string) ([]string, []string) {
 	known := []string{}
 	unknown := []string{}
-	kvargs := []string{"--kube-context", "--namespace", "-n", "--kubeconfig", "--kube-apiserver", "--kube-token", "--kube-as-user", "--kube-as-group", "--kube-ca-file", "--registry-config", "--repository-cache", "--repository-config", "--kube-insecure-skip-tls-verify", "--kube-tls-server-name"}
+	boolArgs := []string{
-	knownArg := func(a string) bool {
+		"--debug",
+		"--kube-insecure-skip-tls-verify",
+	}
+	kvargs := []string{
+		"--burst-limit",
+		"--kube-apiserver",
+		"--kube-as-group",
+		"--kube-as-user",
+		"--kube-ca-file",
+		"--kube-context",
+		"--kube-tls-server-name",
+		"--kube-token",
+		"--kubeconfig",
+		"--namespace", "-n",
+		"--qps",
+		"--registry-config",
+		"--repository-cache",
+		"--repository-config"}
+	kvargs = append(kvargs, boolArgs...) // bool args can also by kv i.e. `--debug=true`
+
+	// detects args with 'arg=val' syntax
+	isKnownKvWithEquals := func(a string) bool {
 		for _, pre := range kvargs {
 			if strings.HasPrefix(a, pre+"=") {
 				return true
@@ -171,29 +196,50 @@ func manuallyProcessArgs(args []string) ([]string, []string) {
 		}
 		return false
 	}
-
+	// detects standalone boolean flags
-	isKnown := func(v string) string {
+	isKnownBoolean := func(v string) bool {
-		if slices.Contains(kvargs, v) {
+		return slices.Contains(boolArgs, v)
-			return v
+	}
+	// detects boolean flags where the next arg can be parsed to boolean
+	isKnownBooleanWithNextValue := func(args []string, i int) bool {
+		if len(args)-i < 2 { // must be at least 2 args remaining
+			return false
 		}
-		return ""
+		a := args[i]
+		for _, known := range boolArgs {
+			if known == a {
+				possibleBool := args[i+1] // safe - checked that at least 2 args left in slice
+				_, err := strconv.ParseBool(possibleBool)
+				if err == nil {
+					return true
+				}
+			}
+		}
+		return false
+	}
+	isKnownKv := func(v string) bool {
+		return slices.Contains(kvargs, v)
 	}
 
 	for i := 0; i < len(args); i++ {
-		switch a := args[i]; a {
+		a := args[i]
-		case "--debug":
+		if isKnownKvWithEquals(a) { // is it an argument using `arg=val` syntax?
 			known = append(known, a)
-		case isKnown(a):
+		} else if isKnownBooleanWithNextValue(args, i) { // is it a boolean flag using `--boolFlag true` syntax?
 			known = append(known, a)
 			i++
 			if i < len(args) {
 				known = append(known, args[i])
 			}
-		default:
+		} else if isKnownBoolean(a) { // is it a known boolean flag by itself? i.e. `--boolFlag`
-			if knownArg(a) {
+			known = append(known, a)
-				known = append(known, a)
+		} else if isKnownKv(a) {
-				continue
+			known = append(known, a) // is it a known kv flag?
+			i++
+			if i < len(args) {
+				known = append(known, args[i])
 			}
+		} else {
 			unknown = append(unknown, a)
 		}
 	}

pkg/cmd/plugin_test.go

@@ -34,7 +34,16 @@ import (
 func TestManuallyProcessArgs(t *testing.T) {
 	input := []string{
 		"--debug",
+		"--debug=true",
+		"--debug", "true",
+		"--debug", "false",
+		"--kube-insecure-skip-tls-verify",
+		"--kube-insecure-skip-tls-verify=true",
+		"--kube-insecure-skip-tls-verify", "true",
+		"--kube-insecure-skip-tls-verify", "false",
 		"--foo", "bar",
+		"--burst-limit", "123",
+		"--burst-limit=123",
 		"--kubeconfig=/home/foo",
 		"--kubeconfig", "/home/foo",
 		"--kube-context=test1",
@@ -46,12 +55,24 @@ func TestManuallyProcessArgs(t *testing.T) {
 		"-n", "test2",
 		"--namespace=test2",
 		"--namespace", "test2",
+		"--qps", "22",
+		"--qps=22",
 		"--home=/tmp",
 		"command",
+		"--debug", // check a boolean flag at the end to test possible out of bounds exception
 	}
 
 	expectKnown := []string{
 		"--debug",
+		"--debug=true",
+		"--debug", "true",
+		"--debug", "false",
+		"--kube-insecure-skip-tls-verify",
+		"--kube-insecure-skip-tls-verify=true",
+		"--kube-insecure-skip-tls-verify", "true",
+		"--kube-insecure-skip-tls-verify", "false",
+		"--burst-limit", "123",
+		"--burst-limit=123",
 		"--kubeconfig=/home/foo",
 		"--kubeconfig", "/home/foo",
 		"--kube-context=test1",
@@ -63,6 +84,9 @@ func TestManuallyProcessArgs(t *testing.T) {
 		"-n", "test2",
 		"--namespace=test2",
 		"--namespace", "test2",
+		"--qps", "22",
+		"--qps=22",
+		"--debug",
 	}
 
 	expectUnknown := []string{
@@ -71,16 +95,8 @@ func TestManuallyProcessArgs(t *testing.T) {
 
 	known, unknown := manuallyProcessArgs(input)
 
-	for i, k := range known {
+	assert.Equal(t, expectKnown, known)
-		if k != expectKnown[i] {
+	assert.Equal(t, expectUnknown, unknown)
-			t.Errorf("expected known flag %d to be %q, got %q", i, expectKnown[i], k)
-		}
-	}
-	for i, k := range unknown {
-		if k != expectUnknown[i] {
-			t.Errorf("expected unknown flag %d to be %q, got %q", i, expectUnknown[i], k)
-		}
-	}
 }
 
 func TestLoadCLIPlugins(t *testing.T) {