incus: Fix alias arguments handling

Remove the limitations stated in 50373fe

Signed-off-by: montag451 <[email protected]>

Author: montag451

cmd/incus/main.go

@@ -91,14 +91,7 @@ func aliases() []string {
 	return aliases
 }
 
-func main() {
-	// Process aliases
-	err := execIfAliases()
-	if err != nil {
-		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
-		os.Exit(1)
-	}
-
+func createApp() (*cobra.Command, *cmdGlobal) {
 	// Setup the parser
 	app := &cobra.Command{}
 	app.Use = "incus"
@@ -305,8 +298,16 @@ Custom commands can be defined through aliases, use "incus alias" to control tho
 	app.Flags().BoolVar(&globalCmd.flagHelpAll, "all", false, i18n.G("Show less common commands"))
 	help.Flags().BoolVar(&globalCmd.flagHelpAll, "all", false, i18n.G("Show less common commands"))
 
+	return app, &globalCmd
+}
+
+func main() {
+	app, globalCmd := createApp()
+
+	// Parse flags to deal with --all flag and --sub-commands flag and to process aliases
+	err := app.ParseFlags(os.Args[1:])
+
 	// Deal with --all flag and --sub-commands flag
-	err = app.ParseFlags(os.Args[1:])
 	if err == nil {
 		if globalCmd.flagHelpAll {
 			// Show all commands
@@ -324,6 +325,13 @@ Custom commands can be defined through aliases, use "incus alias" to control tho
 		}
 	}
 
+	// Process aliases
+	err = execIfAliases(app)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+		os.Exit(1)
+	}
+
 	// Run the main command and handle errors
 	err = app.Execute()
 	if err != nil {

cmd/incus/main_aliases.go

@@ -10,6 +10,7 @@ import (
 	"strings"
 
 	"github.com/kballard/go-shellquote"
+	"github.com/spf13/cobra"
 
 	"github.com/lxc/incus/v6/internal/i18n"
 	config "github.com/lxc/incus/v6/shared/cliconfig"
@@ -76,26 +77,24 @@ func findAlias(aliases map[string]string, origArgs []string) ([]string, []string
 	return aliasKey, aliasValue, foundAlias
 }
 
-func expandAlias(conf *config.Config, args []string) ([]string, bool, error) {
-	var completion = false
-	var completionFrament string
-	var newArgs []string
-	var origArgs []string
-
-	for _, arg := range args[1:] {
-		if !strings.HasPrefix(arg, "-") {
-			break
-		}
-
-		newArgs = append(newArgs, arg)
+func expandAlias(conf *config.Config, args []string, app *cobra.Command) ([]string, bool, error) {
+	fset := app.Flags()
+	if fset.NArg() == 0 {
+		return nil, false, nil
 	}
 
-	origArgs = append([]string{args[0]}, args[len(newArgs)+1:]...)
+	// newArgs contains all the flags before the first positional argument
+	newArgs := args[1:slices.Index(args, fset.Arg(0))]
+
+	// origArgs contains everything except the flags in newArgs
+	origArgs := append([]string{args[0]}, args[len(newArgs)+1:]...)
 
 	// strip out completion subcommand and fragment from end
+	completion := false
+	completionFragment := ""
 	if len(origArgs) >= 3 && origArgs[1] == "__complete" {
 		completion = true
-		completionFrament = origArgs[len(origArgs)-1]
+		completionFragment = origArgs[len(origArgs)-1]
 		origArgs = append(origArgs[:1], origArgs[2:len(origArgs)-1]...)
 	}
 
@@ -189,7 +188,7 @@ func expandAlias(conf *config.Config, args []string) ([]string, bool, error) {
 	// add back in completion if it was stripped before
 	if completion {
 		newArgs = append([]string{newArgs[0], "__complete"}, newArgs[1:]...)
-		newArgs = append(newArgs, completionFrament)
+		newArgs = append(newArgs, completionFragment)
 	}
 
 	// Add the rest of the arguments only if @ARGS@ wasn't used.
@@ -200,9 +199,7 @@ func expandAlias(conf *config.Config, args []string) ([]string, bool, error) {
 	return newArgs, true, nil
 }
 
-func execIfAliases() error {
-	args := os.Args
-
+func execIfAliases(app *cobra.Command) error {
 	// Avoid loops
 	if os.Getenv("INCUS_ALIASES") == "1" {
 		return nil
@@ -214,7 +211,7 @@ func execIfAliases() error {
 	}
 
 	// Expand the aliases
-	newArgs, expanded, err := expandAlias(conf, args)
+	newArgs, expanded, err := expandAlias(conf, os.Args, app)
 	if err != nil {
 		return err
 	} else if !expanded {

cmd/incus/main_test.go

@@ -89,7 +89,9 @@ func TestExpandAliases(t *testing.T) {
 	conf := &config.Config{Aliases: aliases}
 
 	for _, tc := range testcases {
-		result, expanded, err := expandAlias(conf, tc.input)
+		app, _ := createApp()
+		_ = app.ParseFlags(tc.input[1:])
+		result, expanded, err := expandAlias(conf, tc.input, app)
 		if tc.expectErr {
 			assert.Error(t, err)
 			continue