feat: Add completion for pixi run --environment for nushell

src/cli/completion.rs

@@ -130,19 +130,17 @@ $2::task"#;
 fn replace_nushell_completion(script: &str) -> Cow<str> {
     // Adds tab completion to the pixi run command.
     // NOTE THIS IS FORMATTED BY HAND
-    let pattern = r#"
-  export extern "pixi run" \[
-    ...task: string"#;
+    let pattern = r#"(#.*\n  export extern "pixi run".*\n.*...task: string)([^\]]*--environment\(-e\): string)"#;
     let replacement = r#"
   def "nu-complete pixi run" [] {
-    let result = (^pixi task list --machine-readable | complete)
-    if $$result.exit_code == 0 {
-      $$result.stderr | split row " "
-    }
+    ^pixi info --json | from json | get environments_info | get tasks | flatten | uniq
   }
 
-  export extern "pixi run" [
-    ...task: string@"nu-complete pixi run""#;
+  def "nu-complete pixi run environment" [] {
+    ^pixi info --json | from json | get environments_info | get name
+  }
+
+  ${1}@"nu-complete pixi run"${2}@"nu-complete pixi run environment""#;
 
     let re = Regex::new(pattern).unwrap();
     re.replace(script, replacement)
@@ -244,6 +242,7 @@ _arguments "${_arguments_options[@]}" \
     pub fn test_nushell_completion() {
         // NOTE THIS IS FORMATTED BY HAND!
         let script = r#"
+  # Runs task in project
   export extern "pixi run" [
     ...task: string           # The pixi task or a task shell command you want to run in the project's environment, which can be an executable in the environment's PATH
     --manifest-path: string   # The path to 'pixi.toml' or 'pyproject.toml'

src/cli/snapshots/pixi__cli__completion__tests__nushell_completion.snap

@@ -2,19 +2,22 @@
 source: src/cli/completion.rs
 expression: result
 ---
+
   def "nu-complete pixi run" [] {
-    let result = (^pixi task list --machine-readable | complete)
-    if $result.exit_code == 0 {
-      $result.stderr | split row " "
-    }
+    ^pixi info --json | from json | get environments_info | get tasks | flatten | uniq
   }
 
+  def "nu-complete pixi run environment" [] {
+    ^pixi info --json | from json | get environments_info | get name
+  }
+
+  # Runs task in project
   export extern "pixi run" [
     ...task: string@"nu-complete pixi run"           # The pixi task or a task shell command you want to run in the project's environment, which can be an executable in the environment's PATH
     --manifest-path: string   # The path to 'pixi.toml' or 'pyproject.toml'
     --frozen                  # Install the environment as defined in the lockfile, doesn't update lockfile if it isn't up-to-date with the manifest file
     --locked                  # Check if lockfile is up-to-date before installing the environment, aborts when lockfile isn't up-to-date with the manifest file
-    --environment(-e): string # The environment to run the task in
+    --environment(-e): string@"nu-complete pixi run environment" # The environment to run the task in
     --tls-no-verify           # Do not verify the TLS certificate of the server
     --auth-file: string       # Path to the file containing the authentication token
     --pypi-keyring-provider: string@"nu-complete pixi run pypi_keyring_provider" # Specifies if we want to use uv keyring provider