Merge pull request #334 from raballew/004-shell-completion

feat: add jmp completion command for bash/zsh/fish

Author: mangelajo

python/packages/jumpstarter-cli/jumpstarter_cli/completion.py

@@ -0,0 +1,13 @@
+import click
+from click.shell_completion import get_completion_class
+
+
+@click.command("completion")
+@click.argument("shell", type=click.Choice(["bash", "zsh", "fish"]))
+def completion(shell: str):
+    """Generate shell completion script."""
+    from jumpstarter_cli.jmp import jmp
+
+    comp_cls = get_completion_class(shell)
+    comp = comp_cls(jmp, {}, "jmp", "_JMP_COMPLETE")
+    click.echo(comp.source())

python/packages/jumpstarter-cli/jumpstarter_cli/completion_test.py

@@ -0,0 +1,43 @@
+from click.testing import CliRunner
+
+from .jmp import jmp
+
+
+def test_completion_bash():
+    runner = CliRunner()
+    result = runner.invoke(jmp, ["completion", "bash"])
+    assert result.exit_code == 0
+    assert len(result.output) > 0
+    assert "complete" in result.output.lower()
+    assert "jmp" in result.output.lower()
+
+
+def test_completion_zsh():
+    runner = CliRunner()
+    result = runner.invoke(jmp, ["completion", "zsh"])
+    assert result.exit_code == 0
+    assert len(result.output) > 0
+    assert "compdef" in result.output.lower()
+
+
+def test_completion_fish():
+    runner = CliRunner()
+    result = runner.invoke(jmp, ["completion", "fish"])
+    assert result.exit_code == 0
+    assert len(result.output) > 0
+    assert "complete" in result.output.lower()
+    assert "--command jmp" in result.output.lower()
+
+
+def test_completion_no_args():
+    runner = CliRunner()
+    result = runner.invoke(jmp, ["completion"])
+    assert result.exit_code == 2
+    assert "Missing argument" in result.output or "bash" in result.output
+
+
+def test_completion_unsupported_shell():
+    runner = CliRunner()
+    result = runner.invoke(jmp, ["completion", "powershell"])
+    assert result.exit_code == 2
+    assert "Invalid value" in result.output or "powershell" in result.output

python/packages/jumpstarter-cli/jumpstarter_cli/jmp.py

@@ -6,6 +6,7 @@
 from jumpstarter_cli_driver import driver
 
 from .auth import auth
+from .completion import completion
 from .config import config
 from .create import create
 from .delete import delete
@@ -23,6 +24,7 @@ def jmp():
 
 
 jmp.add_command(auth)
+jmp.add_command(completion)
 jmp.add_command(create)
 jmp.add_command(delete)
 jmp.add_command(update)