From 114d7daa5e370fb623f696a27a43d6f3a64dff13 Mon Sep 17 00:00:00 2001
From: Cameron Brown <me@cbrxyz.com>
Date: Wed, 1 Jan 2025 14:19:20 -0500
Subject: [PATCH] scripts/setup.bash: Add ccd for moving between packages and
 autocomplete to colcon

---
 scripts/install.sh | 3 ++-
 scripts/setup.bash | 8 ++++++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/scripts/install.sh b/scripts/install.sh
index f551c44..d259dc9 100755
--- a/scripts/install.sh
+++ b/scripts/install.sh
@@ -194,7 +194,8 @@ mil_system_install \
 	ros-jazzy-tf2-sensor-msgs \
 	ros-jazzy-geographic-msgs \
 	ros-jazzy-vision-msgs \
-	ros-jazzy-velodyne
+	ros-jazzy-velodyne \
+	python3-colcon-common-extensions
 
 cat <<EOF
 $(color "$Pur")
diff --git a/scripts/setup.bash b/scripts/setup.bash
index e3bd62f..914761a 100644
--- a/scripts/setup.bash
+++ b/scripts/setup.bash
@@ -19,6 +19,14 @@ _list_complete() {
 # Use Cyclone DDS by default (it's super fast and amazing!)
 export RMW_IMPLEMENTATION=rmw_cyclonedds_cpp
 
+# Setup colcon_cd
+source "/usr/share/colcon_cd/function/colcon_cd.sh"
+export _colcon_cd_root=$MIL_REPO
+alias ccd="colcon_cd"
+
+# Setup colcon autocomplete
+source "/usr/share/colcon_cd/function/colcon_cd-argcomplete.bash"
+
 alias mil='cd $MIL_REPO'
 
 alias srcbrc="source ~/.bashrc"