just-completion: "export def" -> "export extern"

custom-completions/just/just-completions.nu

@@ -1,44 +1,143 @@
-def "nu-complete just recipes" [] {
-    ^just --unsorted --dump --dump-format json
-        | from json
-        | get recipes
-        | transpose k v
-        | each {|x|
-            {
-                value: $x.k,
-                description: ( $x.v.parameters
-                             | each {|y|
-                                    if ($y.default|is-empty) {
-                                        $y.name
-                                    } else {
-                                        $'($y.name)="($y.default)"'
-                                    }
-                                }
-                             | str join ' '
-                             )
+
+# official completion from `just --completion nushell`
+def get-recipes [] {
+    (^just --unstable --dump --dump-format json | from json).recipes
+    | transpose recipe data
+    | flatten
+    | where {|recipe| not $recipe.private }
+}
+
+def "nu-complete just" [] {
+    get-recipes
+        | select recipe doc parameters 
+        | each {|recipe| 
+            let name = $recipe.recipe
+            mut desc = $recipe.doc | default " "
+            for $p in $recipe.parameters {
+              if ($p.default | is-empty) {
+                  $desc += $" [(ansi blue)($p.name)(ansi reset)]"
+              } else {
+                  $desc += $" [(ansi blue)($p.name)='($p.default)'(ansi reset)]"
+              }
             }
+            { value: $name description: ($desc | str trim) }
         }
 }
 
-def "nu-complete just args" [context: string, offset: int] {
+def "nu-complete args" [context:string,offset:int] {
     let r = ($context | split row ' ')
-    ^just -u --dump --dump-format json
-        | from json
-        | get recipes
-        | get ($r.1)
-        | get body
-        | each {|x| {description: ($x | get 0) }}
-        | prepend ''
+    let name = $r.1
+    # First two are 'just' and name, minus one because `get-just-args` is zero-indexed
+    let position = ($r | length) - 3
 
+    get-just-arg $name $position
 }
 
-export def just [
-    recipes?: string@"nu-complete just recipes"
-    ...args: any@"nu-complete just args"
-] {
-    if ($recipes | is-empty) {
-        ^just
+# Provides argument of recipe $name at position $positions. 
+def get-just-arg [name:string,position:int] {
+    let recipes = (
+        get-recipes
+            | filter {|r| $r.name == $name }
+            | get parameters
+            | flatten)
+    let len = $recipes | length
+    if $position > ($recipes | length) {
+        [[value description]; ["" "No more args"]]
     } else {
-        ^just $recipes ...$args
+        $recipes
+            | get $position
+            | each {|recipe| 
+                let name = $recipe.name
+                if not ($recipe.default | is-empty) {
+                    [[value description];
+                        [$recipe.default $"'($name)' (ansi green)\(default\)(ansi reset)"]
+                        ["" $name]]
+                } else {
+                    # Need two empty different values
+                    # Otherwise they get filtered and empty value gets inserted automatically
+                    [[value description];
+                        [null $"'($name)'"]
+                        [' '  $"'($name)'"]]
+                }
+          }
     }
 }
+
+# Veriables and Assignments are apparently the same thing...
+def "nu-complete just assignments" [] {
+    ^just --dump --dump-format json
+        | from json
+        | get assignments
+        | transpose name value
+        | each {|x| {
+            value: $x.name,
+            description: $"Current value: '($x.value.value)'"
+        }}
+}
+
+export extern "just" [
+    recipe?:string@"nu-complete just"               # Recipe to execute
+    ...args:any@"nu-complete args"                  # Arguments
+    --changelog                                     # Print changelog
+    --check                                         # Run `--fmt` in 'check' mode. Exits with 0 if justfile is formatted correctly. Exits with 1 and prints a diff if formatting is required.
+    --choose                                        # Select one or more recipes to run using a binary chooser. If `--chooser` is not passed the chooser defaults to the value of $JUST_CHOOSER, falling back to `fzf`
+    --clear-shell-args                              # Clear shell arguments
+    --dry-run(-n)                                   # Print what just would do without doing it
+    --dump                                          # Print justfile
+    --edit(-e)                                      # Edit justfile with editor given by $VISUAL or $EDITOR, falling back to `vim`
+    --evaluate                                      # Evaluate and print all variables. If a variable name is given as an argument, only print that variable's value.
+    --fmt                                           # Format and overwrite justfile
+    --highlight                                     # Highlight echoed recipe lines in bold
+    --init                                          # Initialize new justfile in project root
+    --list(-l)                                      # List available recipes and their arguments
+    --no-aliases                                    # Don't show aliases in list
+    --no-deps                                       # Don't run recipe dependencies
+    --no-dotenv                                     # Don't load `.env` file
+    --no-highlight                                  # Don't highlight echoed recipe lines in bold
+    --quiet(-q)                                     # Suppress all output
+    --shell-command                                 # Invoke <COMMAND> with the shell used to run recipe lines and backticks
+    --summary                                       # List names of available recipes
+    --unsorted(-u)                                  # Return list and summary entries in source order
+    --unstable                                      # Enable unstable features
+    --variables                                     # List names of variables
+    --verbose(-v)                                   # Use verbose output
+    --yes                                           # Automatically confirm all recipes.
+    --help(-h)                                      # Print help information
+    --version(-V)                                   # Print version information
+
+    --chooser                                       # Override binary invoked by `--choose`
+    --color:string@"nu-complete just color-when"    # Print colorful output [default: auto]  [possible values: auto, always, never]
+    --command(-c)                                   # Run an arbitrary command with the working directory, `.env`, overrides, and exports set
+    --command-color:string@"nu-complete just color" # Echo recipe lines in <COMMAND-COLOR> [possible values: black, blue, cyan, green, purple, red, yellow]
+    --completions:string@"nu-complete just shells"  # Print shell completion script for <SHELL> [possible values: zsh, bash, fish, powershell, elvish]
+    --dotenv-filename                               # Search for environment file named <DOTENV-FILENAME> instead of `.env`
+    --dotenv-path(-E)                               # Load <DOTENV-PATH> as environment file instead of searching for one
+    --dump-format:string@"nu-complete just format"  # Dump justfile as <FORMAT> [default: just]  [possible values: just, json]
+    --justfile(-f)                                  # Use <JUSTFILE> as justfile
+    --list-heading                                  # Print <TEXT> before list
+    --list-prefix                                   # Print <TEXT> before each list item
+    --set:string@"nu-complete just assignments"     # Override <VARIABLE> with <VALUE>
+    --shell:string@"nu-complete just shells"        # Invoke <SHELL> to run recipes
+    --shell-arg                                     # Invoke shell with <SHELL-ARG> as an argument
+    --show(-s):string@"nu-complete just"            # Show information about <RECIPE>
+    --working-directory(-d)                         # Use <WORKING-DIRECTORY> as working directory. --justfile must also be set
+] 
+
+def "nu-complete just format" [] {
+    [ "just", "json" ]
+}
+
+# Shells as listed in `--completions`
+def "nu-complete just shells" [] {
+    [ "zsh", "bash", "fish", "powershell", "elvish" ]
+}
+
+# Color outputs as listed in `--color`
+def "nu-complete just color-when" [] {
+    [ "auto", "always", "never" ]
+}
+
+# Colors as listed in `--command-color`
+def "nu-complete just color" [] {
+    [ "black", "blue", "cyan", "green", "purple", "red", "yellow" ]
+}

custom-completions/mask/mask-completions.nu

@@ -44,13 +44,7 @@ def "nu-complete mask args" [context: string, offset: int] {
     $rt
 }
 
-export def main [
+export extern "mask" [
     recipes?: string@"nu-complete mask recipes"
     ...args: any@"nu-complete mask args"
-] {
-    if ($recipes | is-empty) {
-        ^mask --help
-    } else {
-        ^mask $recipes $args
-    }
-}
+]