Data Generator for Forge

1 files changed, 13 insertions, 0 deletions

diff --git a/src/main/java/net/fabricmc/loom/configuration/ide/RunConfig.java b/src/main/java/net/fabricmc/loom/configuration/ide/RunConfig.java
index fff6e9f3..1e75b56b 100644
--- a/src/main/java/net/fabricmc/loom/configuration/ide/RunConfig.java
+++ b/src/main/java/net/fabricmc/loom/configuration/ide/RunConfig.java
@@ -213,6 +213,19 @@ public class RunConfig {
 		return ideaServer;
 	}
 
+	public static RunConfig dataRunConfig(Project project) {
+		LoomGradleExtension extension = project.getExtensions().getByType(LoomGradleExtension.class);
+
+		RunConfig ideaServer = new RunConfig();
+		ideaServer.configName = "Generate Data";
+		ideaServer.programArgs = "";
+		populate(project, extension, ideaServer, "data");
+		ideaServer.vmArgs += " -Dfabric.dli.main=" + getMainClass("data", extension);
+		ideaServer.vscodeBeforeRun = new ArrayList<>(extension.getTasksBeforeRun());
+
+		return ideaServer;
+	}
+
 	// This can be removed at somepoint, its not ideal but its the best solution I could thing of
 	public static boolean needsUpgrade(File file) throws IOException {
 		String contents = FileUtils.readFileToString(file, StandardCharsets.UTF_8);