zsh: rosinit now supports multiple distros + default to indigo

[dotfiles.git] / zsh / functions / rosinit

diff --git a/zsh/functions/rosinit b/zsh/functions/rosinit
index 245cd3fc83d13b68bf26fcc00e33ad98125b2092..4b297d818a846d6c73d5d19905c5bd01a16e6362 100644 (file)
--- a/zsh/functions/rosinit
+++ b/zsh/functions/rosinit
@@ -1,8 +1,16 @@
 #!/bin/zsh
 
 function rosinit() {
-    local ROS_DISTRO=hydro
     local WORKSPACE=~/src/ros
+    local -a ROS_DISTROS
+    ROS_DISTROS=("hydro" "indigo")
+
+    local ROS_DISTRO=$1
+    if [[ -z "$ROS_DISTRO" ]]; then
+        for d in "${ROS_DISTROS[@]}"; do
+            [[ -f "/opt/ros/$d/setup.zsh" ]] && ROS_DISTRO=$d
+        done
+    fi
     if [[ -f "/opt/ros/$ROS_DISTRO/setup.zsh" ]]; then
         export ROS_OS_OVERRIDE=arch
         source /opt/ros/$ROS_DISTRO/setup.zsh