workspace env

Author: Akol125

azure/azure-pr-teardown-pipeline.yml

@@ -47,16 +47,28 @@ jobs:
           export AWS_PROFILE=apim-dev
           account_id="$(aws sts get-caller-identity --query Account --output text)"
 
-          export ENVIRONMENT="internal-dev"
-          export BUCKET_NAME="immunisation-internal-dev-terraform-state-files"
+          export environment="internal-dev"
+          export sub_environment="$(WORKSPACE)"
+
+          echo "Checking if bucket immunisation-${sub_environment}-terraform-state-files exists..."
+          BUCKET_NAME="immunisation-${sub_environment}-terraform-state-files"
+          FALLBACK_BUCKET_NAME="immunisation-internal-dev-terraform-state-files"
+
+          if ! aws s3api head-bucket --bucket "$BUCKET_NAME" 2>/dev/null; then
+            echo "Bucket $BUCKET_NAME does NOT exist. Falling back to internal-dev for init only..."
+            export ENVIRONMENT="internal-dev"
+            export BUCKET_NAME="$FALLBACK_BUCKET_NAME"
+          else
+            export ENVIRONMENT="$environment"
+            export BUCKET_NAME="$BUCKET_NAME"
+          cd terraform
           make init
 
-         
-          export ENVIRONMENT="pr-703"
-          export BUCKET_NAME="immunisation-pr-703-terraform-state-files"
+          export ENVIRONMENT="$sub_environment"
+          export BUCKET_NAME="immunisation-${sub_environment}-terraform-state-files"
 
           cd terraform
-          terraform workspace select $(WORKSPACE)
-          make destroy aws_account_no=${account_id} environment=$(WORKSPACE)
+          make workspace sub_environment=$sub_environment
+          make destroy sub_environment=$sub_environment aws_account_no=${account_id}
         displayName: Destroy terraform PR workspace and linked resources
         retryCountOnTaskFailure: 2

terraform/Makefile

@@ -20,9 +20,6 @@ lock-provider:
 	$(tf_cmd) providers lock -platform=darwin_arm64 -platform=darwin_amd64 -platform=linux_amd64 -platform=windows_amd64
 
 workspace:
-ifndef sub_environment
-	$(error sub_environment is not set. Use 'make workspace sub_environment=your-env')
-endif
 	$(tf_cmd) workspace new $(sub_environment) || $(tf_cmd) workspace select $(sub_environment) && echo "Switched to workspace/environment: $(sub_environment)"
 
 init: