Fix:(issue_2209) Add mutex flags to completion

Author: dearchap

completion_test.go

@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"context"
 	"fmt"
+	"os"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -181,6 +182,51 @@ func TestCompletionSubcommand(t *testing.T) {
 	}
 }
 
+func TestMutuallyExclusiveFlagsCompletion(t *testing.T) {
+	osArgsBak := os.Args
+	defer func() {
+		os.Args = osArgsBak
+	}()
+	out := &bytes.Buffer{}
+
+	cmd := &Command{
+		EnableShellCompletion: true,
+		Writer:                out,
+		Flags: []Flag{
+			&StringFlag{
+				Name: "l1",
+			},
+		},
+		MutuallyExclusiveFlags: []MutuallyExclusiveFlags{
+			{
+				Flags: [][]Flag{
+					{
+						&StringFlag{
+							Name: "m1",
+						},
+						&StringFlag{
+							Name: "m2",
+						},
+					},
+					{
+						&StringFlag{
+							Name: "l2",
+						},
+					},
+				},
+			},
+		},
+	}
+
+	os.Args = []string{"foo", "-", completionFlag}
+	err := cmd.Run(buildTestContext(t), os.Args)
+	assert.NoError(t, err, "Expected no error for completion")
+	assert.Contains(t, out.String(), "l1", "Expected output to contain flag l1")
+	assert.Contains(t, out.String(), "m1", "Expected output to contain flag m1")
+	assert.Contains(t, out.String(), "m2", "Expected output to contain flag m2")
+	assert.Contains(t, out.String(), "l2", "Expected output to contain flag l2")
+}
+
 type mockWriter struct {
 	err error
 }

help.go

@@ -271,7 +271,7 @@ func DefaultCompleteWithFlags(ctx context.Context, cmd *Command) {
 
 	if strings.HasPrefix(lastArg, "-") {
 		tracef("printing flag suggestion for flag[%v] on command %[1]q", lastArg, cmd.Name)
-		printFlagSuggestions(lastArg, cmd.Flags, cmd.Root().Writer)
+		printFlagSuggestions(lastArg, cmd.appliedFlags, cmd.Root().Writer)
 		return
 	}
 

help_test.go

@@ -1371,6 +1371,7 @@ func TestDefaultCompleteWithFlags(t *testing.T) {
 			tc.cmd.parsedArgs = &stringSliceArgs{
 				tc.argv[1:],
 			}
+			tc.cmd.appliedFlags = tc.cmd.Flags
 			f := DefaultCompleteWithFlags
 			f(buildTestContext(ct), tc.cmd)