From d4080e434ea2133b2156cba0035429b25f6b6f56 Mon Sep 17 00:00:00 2001
From: Dor Munis <5811812+dormunis@users.noreply.github.com>
Date: Fri, 12 May 2023 19:02:51 +0300
Subject: [PATCH] Added terraform workspace support

---
 scripts/dracula.sh   |  5 +++++
 scripts/terraform.sh | 30 ++++++++++++++++++++++++++++++
 2 files changed, 35 insertions(+)
 create mode 100755 scripts/terraform.sh

diff --git a/scripts/dracula.sh b/scripts/dracula.sh
index a2bff10..af0b222 100755
--- a/scripts/dracula.sh
+++ b/scripts/dracula.sh
@@ -8,6 +8,7 @@ source $current_dir/utils.sh
 main()
 {
   # set configuration option variables
+  terraform_label=$(get_tmux_option "@dracula-terraform-label" "")
   show_fahrenheit=$(get_tmux_option "@dracula-show-fahrenheit" true)
   show_location=$(get_tmux_option "@dracula-show-location" true)
   fixed_location=$(get_tmux_option "@dracula-fixed-location")
@@ -197,6 +198,10 @@ main()
       IFS=' ' read -r -a colors <<<$(get_tmux_option "@dracula-kubernetes-context-colors" "cyan dark_gray")
       script="#($current_dir/kubernetes_context.sh $show_kubernetes_context_label)"
 
+    elif [ $plugin = "terraform" ]; then
+      IFS=' ' read -r -a colors <<<$(get_tmux_option "@dracula-terraform-colors" "light_purple dark_gray")
+      script="#($current_dir/terraform.sh $terraform_label)"
+
     elif [ $plugin = "weather" ]; then
       IFS=' ' read -r -a colors <<< $(get_tmux_option "@dracula-weather-colors" "orange dark_gray")
       script="#($current_dir/weather_wrapper.sh $show_fahrenheit $show_location $fixed_location)"
diff --git a/scripts/terraform.sh b/scripts/terraform.sh
new file mode 100755
index 0000000..72e144c
--- /dev/null
+++ b/scripts/terraform.sh
@@ -0,0 +1,30 @@
+#!/usr/bin/env bash
+# setting the locale, some users have issues with different locales, this forces the correct one
+export LC_ALL=en_US.UTF-8
+
+label=$1
+
+current_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+source $current_dir/utils.sh
+
+main() {
+  # storing the refresh rate in the variable RATE, default is 5
+  RATE=$(get_tmux_option "@dracula-refresh-rate" 5)
+  OUTPUT_STRING="N/A" 
+  terraform_dir="$(tmux display-message -p '#{pane_current_path}')/.terraform"
+  if [ -d $terraform_dir ]; then
+    current_workspace=$(terraform workspace show 2>/dev/null)
+    OUTPUT_STRING="${current_workspace}"
+  fi
+  if [ "$label" = "" ]
+  then
+      echo "⚙️ ${OUTPUT_STRING}"
+  else
+      echo "⚙️ ${label} ${OUTPUT_STRING}"
+  fi
+
+  sleep $RATE
+}
+
+# run the main driver
+main