feat(cli): Basic project-based shell completion

Author: tony

tmuxp/cli/load.py

@@ -7,17 +7,19 @@
 import importlib
 import logging
 import os
+import pathlib
 import sys
 
 import click
 import kaptan
+from click.shell_completion import CompletionItem
 
 from libtmux.common import has_gte_version, which
 from libtmux.server import Server
 
 from .. import config, exc, log, util
 from ..workspacebuilder import WorkspaceBuilder
-from .utils import ConfigPath, _validate_choices, tmuxp_echo
+from .utils import ConfigPath, _validate_choices, get_config_dir, tmuxp_echo
 
 
 def set_layout_hook(session, hook_name):
@@ -438,9 +440,22 @@ def load_workspace(
     return _setup_plugins(builder)
 
 
+def config_shell_complete(ctx, params, incomplete):
+    config_dir = pathlib.Path(get_config_dir())
+    choices = [config_dir / c for c in config.in_dir(str(config_dir))]
+    choices += ["~" / f.relative_to(pathlib.Path.home()) for f in choices]
+    choices = [str(p) for p in choices]
+    return [CompletionItem(choice) for choice in choices if incomplete in choice]
+
+
 @click.command(name="load", short_help="Load tmuxp workspaces.")
 @click.pass_context
-@click.argument("config", type=ConfigPath(exists=True), nargs=-1)
+@click.argument(
+    "config",
+    type=ConfigPath(exists=True),
+    nargs=-1,
+    shell_complete=config_shell_complete,
+)
 @click.option("-S", "socket_path", help="pass-through for tmux -S")
 @click.option("-L", "socket_name", help="pass-through for tmux -L")
 @click.option("-f", "tmux_config_file", help="pass-through for tmux -f")