nvim: add FPP copyright snippet

[dotfiles.git] / zsh / functions / rosinit

diff --git a/zsh/functions/rosinit b/zsh/functions/rosinit
index 245cd3fc83d13b68bf26fcc00e33ad98125b2092..5924b3b4b6c3c885fe1dbb7f3e778486fce3f885 100644 (file)
--- a/zsh/functions/rosinit
+++ b/zsh/functions/rosinit
@@ -1,15 +1,27 @@
 #!/bin/zsh
 
 function rosinit() {
-    local ROS_DISTRO=hydro
     local WORKSPACE=~/src/ros
-    if [[ -f "/opt/ros/$ROS_DISTRO/setup.zsh" ]]; then
+    local -a ROS_DISTROS
+    ROS_DISTROS=("hydro" "indigo" "jade")
+
+    local distro=$1
+    if [[ -z "$distro" ]]; then
+        for d in "${ROS_DISTROS[@]}"; do
+            [[ -f "/opt/ros/$d/setup.zsh" ]] && distro=$d
+        done
+    fi
+    if [[ -f "/opt/ros/$distro/setup.zsh" ]]; then
+        export ROS_DISTRO=$distro
         export ROS_OS_OVERRIDE=arch
         source /opt/ros/$ROS_DISTRO/setup.zsh
         source $WORKSPACE/devel/setup.zsh
         export ROS_WORKSPACE=$WORKSPACE
+        export ROS_HOME=$XDG_CACHE_HOME/ros
+        export ROS_LOG_DIR=$XDG_CACHE_HOME/ros/log
+        # export ROS_TEST_RESULTS_DIR=$HOME/.ros
         roscd
     else
-        echo "ROS $ROS_DISTRO doesn't seem to be installed." >&2
+        echo "ROS $distro doesn't seem to be installed." >&2
     fi
 }