From 050723abc48aa95a843e54382ce20e6863adf4d0 Mon Sep 17 00:00:00 2001
From: Umang Mistry <umistry@aarete.com>
Date: Thu, 23 May 2024 15:33:19 -0500
Subject: [PATCH] testing ssm commands with new dev instance

---
 bitbucket-pipelines.yml  | 24 ++++++++++++++++++++----
 streamlit/constants.py   |  8 +++++++-
 streamlit/interface_0.py |  1 -
 3 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/bitbucket-pipelines.yml b/bitbucket-pipelines.yml
index 2ea362a..e1de89b 100644
--- a/bitbucket-pipelines.yml
+++ b/bitbucket-pipelines.yml
@@ -16,7 +16,15 @@ definitions:
         export AWS_SECRET_ACCESS_KEY=$(echo $STS_OUTPUT | jq -r '.Credentials.SecretAccessKey');
         export AWS_SECURITY_TOKEN=$(echo $STS_OUTPUT | jq -r '.Credentials.SessionToken');
         export AWS_REGION='us-east-2';
-  steps:
+    - script: &ssm-send-command
+        aws ssm send-command --document-name "AWS-RunShellScript" --instance-ids $DEV_INSTANCE_ID_TF --region $AWS_DEFAULT_REGION --parameters commands='[
+          "sudo -u ubuntu -i <<EOF",
+          "export HOME=/home/ubuntu",
+          "git config --global --add safe.directory /home/ubuntu/doczy.ai",
+          "if [ -d /home/ubuntu/doczy.ai ]; then cd /home/ubuntu/doczy.ai && git fetch && git pull; else cd /home/ubuntu && git clone git@bitbucket.org:aarete/doczy.ai.git; fi",
+          "EOF"
+          ]'
+
     - step: &snowflake_deploy
         name: "Deploy Snowflake Changes"
         image: python:3.7
@@ -103,12 +111,20 @@ pipelines:
               includePaths:
               - snowflake/**/*
       - step:
-          <<: *streamlit_deploy
+          image: python:3.8
+          name: "Deploy streamlit to EC2"
+          oidc: true
+          script:
+            - python -m pip install --upgrade pip 
+            - apt-get update && apt-get install -y jq git
+            - pip install awscli
+            - *aws-context
+            - *ssm-send-command
           condition:
             changesets:
               includePaths:
-              - "streamlit/*"
-              - "streamlit/**/*"
+              - streamlit/*
+              - streamlit/**/*
       - step:
           <<: *airflow_dags_deploy
           condition:
diff --git a/streamlit/constants.py b/streamlit/constants.py
index 813e454..e7b9fc8 100644
--- a/streamlit/constants.py
+++ b/streamlit/constants.py
@@ -211,4 +211,10 @@ SNOWFLAKE_ACCOUNT_LOCATOR="aarete-doczyai",
 DEV_DB_ROLE = "DEVADMIN",
 DEV_WH="DEV_XS",
 DEV_DB="DOCZY_DEV",
-DEV_STAGING_SCHEMA="STG"
\ No newline at end of file
+DEV_STAGING_SCHEMA="STG"
+
+# SNOWFLAKE UAT DATABASE
+UAT_DB_ROLE = "UATADMIN",
+UAT_WH="UAT_XS",
+UAT_DB="DOCZY_UAT",
+UAT_STAGING_SCHEMA="STG"
\ No newline at end of file
diff --git a/streamlit/interface_0.py b/streamlit/interface_0.py
index cd41ed1..de2f0b5 100644
--- a/streamlit/interface_0.py
+++ b/streamlit/interface_0.py
@@ -172,4 +172,3 @@ else:
     st.write("Access Denied")
 
 
-