Fix handling of aliases

See gohugoio/hugo#11090

Author: bep

simplecobra.go

@@ -233,6 +233,11 @@ func checkArgs(cmd *cobra.Command, args []string) error {
 		return nil
 	}
 
+	// Also check the aliases.
+	if cmd.HasAlias(commandName) {
+		return nil
+	}
+
 	return fmt.Errorf("unknown command %q for %q%s", args[1], cmd.CommandPath(), findSuggestions(cmd, commandName))
 }
 

simplecobra_test.go

@@ -83,6 +83,26 @@ func TestSimpleCobra(t *testing.T) {
 
 }
 
+func TestAliases(t *testing.T) {
+	c := qt.New(t)
+
+	rootCmd := &rootCommand{name: "root",
+		commands: []simplecobra.Commander{
+			&lvl1Command{name: "foo", aliases: []string{"f"},
+				commands: []simplecobra.Commander{
+					&lvl2Command{name: "bar"},
+				},
+			},
+		},
+	}
+
+	x, err := simplecobra.New(rootCmd)
+	c.Assert(err, qt.IsNil)
+	args := []string{"f"}
+	_, err = x.Execute(context.Background(), args)
+	c.Assert(err, qt.IsNil)
+}
+
 func TestInitAncestorsOnly(t *testing.T) {
 	c := qt.New(t)
 
@@ -305,6 +325,8 @@ type lvl1Command struct {
 	name   string
 	isInit bool
 
+	aliases []string
+
 	localFlagName  string
 	localFlagNameC string
 
@@ -348,6 +370,7 @@ func (c *lvl1Command) Init(cd *simplecobra.Commandeer) error {
 	}
 	cmd := cd.CobraCommand
 	cmd.DisableSuggestions = c.disableSuggestions
+	cmd.Aliases = c.aliases
 	localFlags := cmd.Flags()
 	localFlags.StringVar(&c.localFlagName, "localFlagName", "", "set localFlagName for lvl1Command")
 	return nil