chore(pipeline): select workspace in python

terraform/azure-pipelines.yml

@@ -9,11 +9,18 @@ stages:
       vmImage: ubuntu-latest
     jobs:
       - job: plan
-        # https://learn.microsoft.com/en-us/azure/devops/pipelines/process/expressions?view=azure-devops#conditional-insertion
-        variables:
-          reason: $[variables['Build.Reason']]
-          target: $[variables['System.PullRequest.TargetBranch']]
         steps:
+          - bash: |
+              echo $REASON
+              echo $SOURCE
+              echo $TARGET
+
+              WORKSPACE=$(python terraform/workspace.py)
+              echo "##vso[task.setvariable variable=one]$(WORKSPACE)"
+            env:
+              REASON: $(Build.Reason)
+              SOURCE: $(Build.SourceBranchName)
+              TARGET: $(System.PullRequest.TargetBranch)
           - task: Bash@3
             inputs:
               targetType: "inline"

terraform/workspace.py

@@ -0,0 +1,13 @@
+import os
+
+REASON = os.environ["REASON"]
+SOURCE = os.environ["SOURCE"]
+TARGET = os.environ["TARGET"]
+
+# on pull requests, base on target branch; otherwise base on the source branch
+if REASON == "PullRequest" and TARGET in ["test", "prod"]:
+    print(TARGET)
+elif SOURCE in ["test", "prod"]:
+    print(SOURCE)
+else:
+    print("dev")