From 3bdf911685555445212b9a43aa036423f2b36b2a Mon Sep 17 00:00:00 2001
From: Jules <guillaume.j.gautier@gmail.com>
Date: Tue, 9 Jul 2024 22:14:49 +0200
Subject: [PATCH] Add ROS prompt

---
 bullet-train.zsh-theme | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/bullet-train.zsh-theme b/bullet-train.zsh-theme
index d72c4f5..2ec90aa 100644
--- a/bullet-train.zsh-theme
+++ b/bullet-train.zsh-theme
@@ -26,6 +26,7 @@ if [ ! -n "${BULLETTRAIN_PROMPT_ORDER+1}" ]; then
     screen
     perl
     ruby
+    ros
     virtualenv
     nvm
     aws
@@ -96,6 +97,17 @@ if [ ! -n "${BULLETTRAIN_VIRTUALENV_PREFIX+1}" ]; then
   BULLETTRAIN_VIRTUALENV_PREFIX=🐍
 fi
 
+# ROS
+if [ ! -n "${BULLETTRAIN_ROS_BG+1}" ]; then
+  BULLETTRAIN_ROS_BG=green
+fi
+if [ ! -n "${BULLETTRAIN_ROS_FG+1}" ]; then
+  BULLETTRAIN_ROS_FG=black
+fi
+if [ ! -n "${BULLETTRAIN_ROS_PREFIX+1}" ]; then
+  BULLETTRAIN_ROS_PREFIX=🤖
+fi
+
 # NVM
 if [ ! -n "${BULLETTRAIN_NVM_BG+1}" ]; then
   BULLETTRAIN_NVM_BG=green
@@ -584,6 +596,14 @@ prompt_virtualenv() {
   fi
 }
 
+prompt_ros() {
+  local ros_distro="$ROS_DISTRO"
+  if [[ -n $ros_distro ]]; then
+    prompt_segment $BULLETTRAIN_ROS_BG $BULLETTRAIN_ROS_FG $BULLETTRAIN_ROS_PREFIX" $(basename $ros_distro)"
+  fi
+}
+
+
 # NVM: Node version manager
 prompt_nvm() {
   local nvm_prompt