From 1783cf7259156074447aee9a6051c7d993fdd7a0 Mon Sep 17 00:00:00 2001
From: atagen <boss@atagen.co>
Date: Sun, 29 Mar 2026 14:12:08 +1100
Subject: [PATCH] dynamic systemctl unit completions

---
 flake.nix | 69 ++++++++++++++++++++++++++++++-------------------------
 1 file changed, 38 insertions(+), 31 deletions(-)

diff --git a/flake.nix b/flake.nix
index 7c63e3a..67dc0f8 100644
--- a/flake.nix
+++ b/flake.nix
@@ -57,40 +57,47 @@
           imports = [ ./nix/module.nix ];
           programs.inshellah.package = self.packages.${pkgs.stdenv.hostPlatform.system}.default;
           programs.inshellah.snippet = ''
-            let inshellah_complete = { |spans|
-              let completions = inshellah complete ...$spans | from json
-              # special case nix completions, for `nix run nixpkgs#..` type input
-              let additional = if ($completions == null and
-                ($spans | length) > 0 and $spans.0 == "nix") {
-                $env.NIX_GET_COMPLETIONS = ($spans | length) - 1
-                let nix_output = $spans | run-external $in | split row -r '\n' | str trim | skip 1
-                let entries = $nix_output |
-                  if (($in | length) < 6 and
-                    ($spans | last) =~ "[a-zA-Z][a-zA-Z0-9_-]*#[a-zA-Z][a-zA-Z0-9_-]*") { 
-                    hide-env NIX_GET_COMPLETIONS
-                    $env.NIX_ALLOW_UNFREE = 1
-                    $env.NIX_ALLOW_BROKEN = 1
-                    $in | par-each { |e|
-                      try {
-                         { value: $e, description: (^nix eval --impure $e --apply "f: f.meta.description" err> /dev/null) }
-                      } catch {} finally {
-                        { value: $e, description: "" }
+            let inshellah_complete = { |spans| 
+                let completions = (^inshellah complete ...$spans) | from json
+                # dynamic completions
+                let additional = if ($completions == null and
+                  ($spans | length) > 0) {
+                  match $spans.0 {
+                    "nix" => {
+                        $env.NIX_GET_COMPLETIONS = ($spans | length) - 1
+                        let nix_output = $spans | run-external $in | split row -r '\n' | str trim | skip 1
+                        let entries = if (($nix_output | length) < 6 and
+                          ($spans | last) =~ "[a-zA-Z][a-zA-Z0-9_-]*#[a-zA-Z][a-zA-Z0-9_-]*") { 
+                            hide-env NIX_GET_COMPLETIONS
+                            $env.NIX_ALLOW_UNFREE = 1
+                            $env.NIX_ALLOW_BROKEN = 1
+                            $nix_output | par-each { |e|
+                                  try {
+                                     { value: $e, description: (^nix eval --impure $e --apply "f: f.meta.description" err> /dev/null) }
+                                  } catch {} finally {
+                                    { value: $e, description: "" }
+                                  }
+                                }
+                         } else {
+                           $nix_output | each { |e| { value: $e, description: "" } }
+                         }
+                        $entries
+                    }
+                    "systemctl" => {
+                      if ($spans | length) < 2 { null } else {
+                        let kw = $spans | last
+                        ^systemctl list-units --all --no-pager --plain --full $"($kw)*"
+                          | detect columns
+                          | where LOAD != null
+                          | each { |r| {value: $r.UNIT, description: ($r.DESCRIPTION | default "")} }
                       }
                     }
-                } else {
-                  $in | each { |e| { value: $e, description: "" } }
-                }
-                $entries
-              } else {
-                null
-              }
-              $completions | append $additional
-            }
-            $env.config.completions.external = {
-              enable: true
-              max_results: 200
-              completer: $inshellah_complete
+                    _ => { null }
+                  }
+                } else { null }
+                $completions | append $additional
             }
+            $env.config.completions.external = {enable: true, max_results: 200, completer: $inshellah_complete}
           '';
         };
     };