From 064d4051d597598f703453606915ea35b3c7ee3c Mon Sep 17 00:00:00 2001
From: Juuxel <6596629+Juuxel@users.noreply.github.com>
Date: Thu, 3 Dec 2020 11:19:38 +0200
Subject: [PATCH] Add environment variable support for idea run configs

---
 .../net/fabricmc/loom/util/RunConfig.java     | 29 +++++++++++++++++++
 .../resources/idea_run_config_template.xml    |  3 +-
 2 files changed, 31 insertions(+), 1 deletion(-)

diff --git a/src/main/java/net/fabricmc/loom/util/RunConfig.java b/src/main/java/net/fabricmc/loom/util/RunConfig.java
index 6f4dea88..0a45c884 100644
--- a/src/main/java/net/fabricmc/loom/util/RunConfig.java
+++ b/src/main/java/net/fabricmc/loom/util/RunConfig.java
@@ -30,6 +30,7 @@ import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
 import java.nio.charset.StandardCharsets;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Locale;
 import java.util.Map;
@@ -57,6 +58,7 @@ public class RunConfig {
 	public String runDir;
 	public String vmArgs;
 	public String programArgs;
+	public final Map<String, String> envVariables = new HashMap<>();
 
 	public Element genRuns(Element doc) {
 		Element root = this.addXml(doc, "component", ImmutableMap.of("name", "ProjectRunConfigurationManager"));
@@ -74,6 +76,14 @@ public class RunConfig {
 			this.addXml(root, "option", ImmutableMap.of("name", "PROGRAM_PARAMETERS", "value", programArgs));
 		}
 
+		if (!envVariables.isEmpty()) {
+			Element envs = this.addXml(root, "envs", ImmutableMap.of());
+
+			for (Map.Entry<String, String> envEntry : envVariables.entrySet()) {
+				this.addXml(envs, "env", ImmutableMap.of("name", envEntry.getKey(), "value", envEntry.getValue()));
+			}
+		}
+
 		return root;
 	}
 
@@ -193,6 +203,25 @@ public class RunConfig {
 		dummyConfig = dummyConfig.replace("%PROGRAM_ARGS%", programArgs.replaceAll("\"", "&quot;"));
 		dummyConfig = dummyConfig.replace("%VM_ARGS%", vmArgs.replaceAll("\"", "&quot;"));
 
+		String envs = "";
+
+		if (!envVariables.isEmpty()) {
+			StringBuilder builder = new StringBuilder("<envs>");
+
+			for (Map.Entry<String, String> env : envVariables.entrySet()) {
+				builder.append("<env name=\"");
+				builder.append(env.getKey().replaceAll("\"", "&quot;"));
+				builder.append("\" value=\"");
+				builder.append(env.getValue().replaceAll("\"", "&quot;"));
+				builder.append("\"/>");
+			}
+
+			builder.append("</envs>");
+			envs = builder.toString();
+		}
+
+		dummyConfig = dummyConfig.replace("%ENVS%", envs);
+
 		return dummyConfig;
 	}
 
diff --git a/src/main/resources/idea_run_config_template.xml b/src/main/resources/idea_run_config_template.xml
index 1e85bbbe..f462051d 100644
--- a/src/main/resources/idea_run_config_template.xml
+++ b/src/main/resources/idea_run_config_template.xml
@@ -1,5 +1,6 @@
 <component name="ProjectRunConfigurationManager">
   <configuration default="false" name="%NAME%" type="Application" factoryName="Application">
+    %ENVS%
     <option name="MAIN_CLASS_NAME" value="%MAIN_CLASS%" />
     <module name="%IDEA_MODULE%" />
     <option name="PROGRAM_PARAMETERS" value="%PROGRAM_ARGS%" />
@@ -9,4 +10,4 @@
       <option name="Make" enabled="true" />
     </method>
   </configuration>
-</component>
\ No newline at end of file
+</component>