riir

nix/module.nix

@@ -10,7 +10,7 @@
 #
 # Usage:
 #   { pkgs, ... }: {
-#     imports = [ ./path/to/inshellah/nix/module.nix ];
+#     imports = [ ./path/to/inshellah-rs/nix/module.nix ];
 #     programs.inshellah.enable = true;
 #   }
 
@@ -23,6 +23,7 @@
 
 let
   cfg = config.programs.inshellah;
+  completerSnippet = ./inshellah-completer.nu;
 in
 {
   options.programs.inshellah = {
@@ -72,9 +73,33 @@ in
       '';
     };
 
+    timeoutMs = lib.mkOption {
+      type = lib.types.nullOr lib.types.int;
+      default = null;
+      example = 200;
+      description = ''
+        per-subprocess timeout in milliseconds. when null the binary's
+        compiled-in default is used (currently 200ms).
+      '';
+    };
+
+    workers = lib.mkOption {
+      type = lib.types.nullOr lib.types.int;
+      default = null;
+      example = 8;
+      description = ''
+        worker thread count for the parallel scrape pool. when null,
+        `std::thread::available_parallelism` is used.
+      '';
+    };
+
     snippet = lib.mkOption {
       type = lib.types.str;
       readOnly = true;
+      default = builtins.readFile completerSnippet;
+      description = ''
+        nushell external completer snippet installed by the module.
+      '';
     };
   };
 
@@ -98,7 +123,10 @@ in
         (lib.hiPrio wrapped)
         cfg.package
       ];
-    environment.pathsToLink = [ "/share/nushell/autoload" ];
+    environment.pathsToLink = [
+      "/share/nushell/autoload"
+      "/share/nushell/vendor/autoload"
+    ];
     environment.extraSetup =
       let
         inshellah = "${cfg.package}/bin/inshellah";
@@ -109,30 +137,24 @@ in
           lib.concatStringsSep "\n" cfg.helpOnlyCommands
         );
         helpOnlyFlag = lib.optionalString (cfg.helpOnlyCommands != [ ]) " --help-only ${helpOnlyFile}";
+        timeoutFlag = lib.optionalString (cfg.timeoutMs != null) " --timeout-ms ${toString cfg.timeoutMs}";
+        workersFlag = lib.optionalString (cfg.workers != null) " --workers ${toString cfg.workers}";
+        snippetFile = pkgs.writeText "inshellah-completer.nu" cfg.snippet;
       in
       ''
         mkdir -p ${destDir}
 
         if [ -d "$out/bin" ] && [ -d "$out/share/man" ]; then
-          ${inshellah} index "$out" --dir ${destDir}${ignoreFlag}${helpOnlyFlag} \
+          ${inshellah} index "$out" --dir ${destDir}${ignoreFlag}${helpOnlyFlag}${timeoutFlag}${workersFlag} \
             2>/dev/null || true
         fi
 
         find ${destDir} -maxdepth 1 -empty -delete
 
-        # nushell hardcodes sudo and doas to bypass the external completer,
-        # returning command-name completion instead of calling inshellah.
-        # these @complete external stubs override that so inshellah handles
-        # their flags and elevation stripping. placed in the nushell autoload
-        # dir so they are sourced automatically at shell startup.
+        # Install the full nushell completer plus sudo/doas wrapped commands.
+        # Nushell otherwise hardcodes sudo/doas to bypass external completers.
         mkdir -p $out/share/nushell/vendor/autoload
-        cat > $out/share/nushell/vendor/autoload/inshellah-elevation.nu << 'NUSHELL'
-        @complete external
-        extern "sudo" []
-
-        @complete external
-        extern "doas" []
-        NUSHELL
+        cp ${snippetFile} $out/share/nushell/vendor/autoload/inshellah.nu
       '';
   };
 }