patch args

complete.go

@@ -5,6 +5,7 @@ import (
 
 	"github.com/carapace-sh/carapace/internal/config"
 	"github.com/carapace-sh/carapace/internal/shell/bash"
+	"github.com/carapace-sh/carapace/internal/shell/cmd_clink"
 	"github.com/carapace-sh/carapace/internal/shell/nushell"
 	"github.com/carapace-sh/carapace/pkg/ps"
 	"github.com/spf13/cobra"
@@ -37,6 +38,14 @@ func complete(cmd *cobra.Command, args []string) (string, error) {
 				}
 				return ActionMessage(err.Error()).Invoke(context).value(args[0], args[len(args)-1]), nil
 			}
+		case "cmd":
+			var err error
+			args, err = cmd_clink.Patch(args)
+			LOG.Printf("patching args to %#v", args)
+			if err != nil {
+				context := NewContext(args...)
+				return ActionMessage(err.Error()).Invoke(context).value(args[0], args[len(args)-1]), nil
+			}
 		}
 
 		action, context := traverse(cmd, args[2:])

internal/shell/cmd_clink/patch.go

@@ -0,0 +1,26 @@
+package cmd_clink
+
+import (
+	"os"
+
+	shlex "github.com/carapace-sh/carapace-shlex"
+)
+
+func Patch(args []string) ([]string, error) {
+	compline, ok := os.LookupEnv("CARAPACE_COMPLINE")
+	if !ok {
+		return args, nil
+	}
+	os.Unsetenv("CARAPACE_COMPLINE")
+
+	if compline == "" {
+		return args, nil
+	}
+
+	tokens, err := shlex.Split(compline)
+	if err != nil {
+		return nil, err
+	}
+	args = append(args[:1], tokens.CurrentPipeline().FilterRedirects().Words().Strings()...)
+	return args, nil
+}

pkg/ps/ps.go

@@ -26,6 +26,8 @@ func DetermineShell() string {
 				return "bash-ble"
 			}
 			return "bash"
+		case "cmd":
+			return "cmd"
 		case "elvish":
 			return "elvish"
 		case "fish":