Merge pull request #213 from dormunis/feature/terraform-workspace

Added terraform workspace support

Author: ethancedwards8

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)"

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