add autocomplete to window command

Author: obfuscatedgenerated

src/programs/pkg/index.ts

@@ -457,9 +457,10 @@ export default {
     completion: async (data) => {
         const arg_index = data.raw_parts.length - 1;
 
+        // TODO: smarter completion that understands flags for subcommands
         switch (arg_index) {
             case 1:
-                return helper_completion_options(["add", "remove", "list", "info", "read", "browse", "clean"])(data);
+                return helper_completion_options(["add", "remove", "list", "info", "read", "browse", "clean", "-h"])(data);
             case 2:
                 if (["info", "read", "remove"].includes(data.args[0])) {
                     // complete with installed package names
@@ -474,7 +475,7 @@ export default {
                     }
 
                     const pkgs = Object.keys(local_graph);
-                    return pkgs.filter((pkg) => pkg.startsWith(data.current_partial));
+                    return helper_completion_options(pkgs)(data);
                 }
                 break;
         }

src/programs/window/index.ts

@@ -7,6 +7,7 @@ import {show_subcommand} from "./show";
 import {hide_subcommand} from "./hide";
 import {close_subcommand} from "./close";
 import {center_subcommand} from "./center";
+import {helper_completion_options} from "../../tab_completion";
 
 // extract from ANSI to make code less verbose
 const {STYLE, PREFABS} = ANSI;
@@ -37,6 +38,32 @@ export default {
             }
         }
     },
+    completion: async (data) => {
+        const arg_index = data.raw_parts.length - 1;
+
+        switch (arg_index) {
+            case 1:
+                return helper_completion_options(["info", "list", "show", "hide", "close", "center"])(data);
+            case 2:
+                // completing first argument of subcommand
+                if (data.raw_parts[1] === "list") {
+                    return helper_completion_options(["-v", "-i"])(data);
+                } else if (["show", "hide", "close", "center"].includes(data.raw_parts[1])) {
+                    // complete window ids
+                    const term = data.term;
+                    const wm = term.get_window_manager();
+                    if (!wm) {
+                        return [];
+                    }
+
+                    const window_ids = wm.get_all_windows().map((win) => win.id.toString());
+                    return helper_completion_options(window_ids)(data);
+                }
+                return [];
+            default:
+                return [];
+        }
+    },
     main: async (data) => {
         // extract from data to make code less verbose
         const {args, term} = data;