feat: add improved bash completion script

pyproject.toml

@@ -21,7 +21,7 @@ classifiers = [
 dependencies = [
     "craft-application~=4.6.0",
     "craft-archives>=2.0.0",
-    "craft-cli",
+    "craft-cli>=2.15.0",
     "craft-parts~=2.1.4",
     "craft-platforms~=0.3",
     "craft-providers>=2.0.4",

rockcraft/cli.py

@@ -1,6 +1,6 @@
 # -*- Mode:Python; indent-tabs-mode:nil; tab-width:4 -*-
 #
-# Copyright 2021-2022 Canonical Ltd.
+# Copyright 2021-2025 Canonical Ltd.
 #
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License version 3 as
@@ -16,9 +16,10 @@
 
 """Command-line application entry point."""
 
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 from craft_application import commands as appcommands
+from craft_cli import Dispatcher
 
 from . import commands
 from .services import RockcraftServiceFactory
@@ -59,3 +60,11 @@ def _create_app() -> "Rockcraft":
     app.add_command_group("Lifecycle", [appcommands.RemoteBuild])
 
     return app
+
+
+def get_app_info() -> tuple[Dispatcher, dict[str, Any]]:
+    """Retrieve application info. Used by craft-cli's completion module."""
+    app = _create_app()
+    dispatcher = app._create_dispatcher()
+
+    return dispatcher, app.app_config

rockcraft/pebble.py

@@ -137,9 +137,7 @@ class Pebble:
         "stage-snaps": ["pebble/latest/stable"],
         # We need this because "services" is Optional, but the directory must exist
         "override-prime": str(
-            "craftctl default\n"
-            f"mkdir -p {PEBBLE_LAYERS_PATH}\n"
-            f"chmod 777 {PEBBLE_PATH}"
+            f"craftctl default\nmkdir -p {PEBBLE_LAYERS_PATH}\nchmod 777 {PEBBLE_PATH}"
         ),
     }
     PEBBLE_PART_SPEC = {

snap/snapcraft.yaml

@@ -25,6 +25,7 @@ platforms:
 apps:
   rockcraft:
     command: bin/python $SNAP/bin/rockcraft
+    completer: completion.sh
   skopeo:
     command: bin/skopeo
 
@@ -126,6 +127,10 @@ parts:
       uv export --no-dev --no-emit-workspace --no-emit-package pywin32 --output-file uv-requirements.txt
       ${SNAP}/libexec/snapcraft/craftctl default
 
+      PYTHONPATH=/root/parts/$CRAFT_PART_NAME/install/lib/python3.12/site-packages \
+        python3 -m craft_cli.completion $CRAFT_PROJECT_NAME rockcraft.cli:get_app_info \
+        > $CRAFT_PART_INSTALL/completion.sh
+
       version="$("${CRAFT_STAGE}/usr/bin/python3" -c "import rockcraft;print(rockcraft.__version__)")"
       ${SNAP}/libexec/snapcraft/craftctl set version="$version"
       [ -n "$(echo $version | grep "post")" ] && grade=devel || grade=stable