From d82f4021b6242d144443e09cdbcc7ec30e5615a1 Mon Sep 17 00:00:00 2001
From: Rob Lugton <rob.lugton@gmail.com>
Date: Thu, 30 May 2019 12:27:46 +1000
Subject: [PATCH] add terrafrom workspace label to prompt

---
 themes/agnoster.zsh-theme | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/themes/agnoster.zsh-theme b/themes/agnoster.zsh-theme
index 518a14a37..cfa4f0dd1 100644
--- a/themes/agnoster.zsh-theme
+++ b/themes/agnoster.zsh-theme
@@ -236,12 +236,19 @@ prompt_aws() {
   esac
 }
 
+prompt_terraform() {
+  local terraform_info=$(tf_prompt_info)
+  [[ -z "$terraform_info" ]] && return
+  prompt_segment magenta yellow "TF: $terraform_info"
+}
+
 ## Main prompt
 build_prompt() {
   RETVAL=$?
   prompt_status
   prompt_virtualenv
   prompt_aws
+  prompt_terraform
   prompt_context
   prompt_dir
   prompt_git