feat: support multiple shells in the setup script (#216)

RobotecAI · Sep 11, 2024 · 5e8360f · 5e8360f
1 parent 3d3b7fa
commit 5e8360f
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/setup_shell.sh b/setup_shell.sh
@@ -6,7 +6,27 @@
 
 # Suppress ShellCheck warning about not following external file
 # shellcheck disable=SC1091
-. install/setup.bash
+
+case "$SHELL" in
+    *bash)
+        . install/setup.bash
+        echo "Sourced bash install"
+        ;;
+    *zsh)
+        . install/setup.zsh
+        echo "Sourced zsh install"
+        ;;
+    *fish)
+        echo "fish is not supported"
+        ;;
+    *sh)
+        . install/setup.sh
+        echo "Sourced sh install."
+        ;;
+    *)
+        echo "Unknown shell: $0"
+        ;;
+esac
 
 export PYTHONPATH
 PYTHONPATH="$(dirname "$(dirname "$(poetry run which python)")")/lib/python$(poetry run python --version | awk '{print $2}' | cut -d. -f1,2)/site-packages:$PYTHONPATH"