From 9b49971caef528a116bdf2d7f3922313558984ed Mon Sep 17 00:00:00 2001
From: Kevin Fenzi <kevin@scrye.com>
Date: Wed, 6 May 2020 13:51:16 -0700
Subject: [PATCH] iad2: set prompt in iad2 to avoid confusion

Signed-off-by: Kevin Fenzi <kevin@scrye.com>
---
 roles/base/files/setprodiad2ps1.sh |  3 +++
 roles/base/tasks/main.yml          | 15 +++++++++++++++
 2 files changed, 18 insertions(+)
 create mode 100644 roles/base/files/setprodiad2ps1.sh

diff --git a/roles/base/files/setprodiad2ps1.sh b/roles/base/files/setprodiad2ps1.sh
new file mode 100644
index 0000000000..678b953a7e
--- /dev/null
+++ b/roles/base/files/setprodiad2ps1.sh
@@ -0,0 +1,3 @@
+# Set PS1 based on env
+#
+[[ $- =~ i ]] && PS1="[\u@\h \W]\[$(tput setaf 3)\][PROD-IAD2]\[$(tput sgr0)\]\\$ \[\]"
diff --git a/roles/base/tasks/main.yml b/roles/base/tasks/main.yml
index bc01ad068f..43274a2fd8 100644
--- a/roles/base/tasks/main.yml
+++ b/roles/base/tasks/main.yml
@@ -496,6 +496,21 @@
   - config
   - prompt
 
+#Set PS1 to show prod-iad2 environment at PS1
+#
+- name: set PS1 for prod in /etc/profile.d
+  copy: >
+    src=setprodiad2ps1.sh
+    dest="/etc/profile.d/setprodiad2ps1.sh"
+    owner=root
+    group=root
+    mode=0644
+  when: env == 'production' and datacenter == 'iad2'
+  tags:
+  - base
+  - config
+  - prompt
+
 # Set krb5 conf
 - name: configure krb5
   template: src=krb5.conf.j2 dest=/etc/krb5.conf owner=root group=root mode=0644