From 82dbdb25c6c78f2f3d3897cc1caae6c28591acd2 Mon Sep 17 00:00:00 2001
From: "Pavel.Vervenko" <pavel.vervenko@coople.com>
Date: Thu, 1 Sep 2022 00:00:23 +0200
Subject: [PATCH] Possibility to provide extra env variables to terraform

---
 .../main/java/com/microsoft/terraform/TerraformClient.java  | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/terraform-client/src/main/java/com/microsoft/terraform/TerraformClient.java b/terraform-client/src/main/java/com/microsoft/terraform/TerraformClient.java
index 96d59d0..e8f2870 100644
--- a/terraform-client/src/main/java/com/microsoft/terraform/TerraformClient.java
+++ b/terraform-client/src/main/java/com/microsoft/terraform/TerraformClient.java
@@ -22,6 +22,7 @@ public class TerraformClient implements AutoCloseable {
     private File workingDirectory;
     private boolean inheritIO;
     private Consumer<String> outputListener, errorListener;
+    private Map<String, String> extraEnvVars = Collections.emptyMap();
 
     public TerraformClient() {
         this(new TerraformOptions());
@@ -68,6 +69,10 @@ public void setInheritIO(boolean inheritIO) {
         this.inheritIO = inheritIO;
     }
 
+    public void setExtraEnvVars(Map<String, String> envVars) {
+        this.extraEnvVars = envVars;
+    }
+
     public CompletableFuture<String> version() throws IOException {
         ProcessLauncher launcher = this.getTerraformLauncher(VERSION_COMMAND);
         StringBuilder version = new StringBuilder();
@@ -130,6 +135,7 @@ private ProcessLauncher getTerraformLauncher(String command) throws IOException
         launcher.setEnvironmentVariable(CLIENT_ID_ENV_NAME, this.options.getArmClientId());
         launcher.setEnvironmentVariable(SECRET_ENV_NAME, this.options.getArmClientSecret());
         launcher.setEnvironmentVariable(TENANT_ID_ENV_NAME, this.options.getArmTenantId());
+        extraEnvVars.forEach(launcher::setEnvironmentVariable);
         launcher.appendCommands(NON_INTERACTIVE_COMMAND_MAP.get(command));
         launcher.setOutputListener(this.getOutputListener());
         launcher.setErrorListener(this.getErrorListener());