ApiUtil: Add developer only user agent override (#590)

Co-authored-by: NopoTheGamer <40329022+NopoTheGamer@users.noreply.github.com>

1 files changed, 7 insertions, 2 deletions

diff --git a/src/main/java/io/github/moulberry/notenoughupdates/util/ApiUtil.java b/src/main/java/io/github/moulberry/notenoughupdates/util/ApiUtil.java
index 0c590bab..023be060 100644
--- a/src/main/java/io/github/moulberry/notenoughupdates/util/ApiUtil.java
+++ b/src/main/java/io/github/moulberry/notenoughupdates/util/ApiUtil.java
@@ -61,7 +61,12 @@ import java.util.zip.GZIPInputStream;
 public class ApiUtil {
 	private static final Gson gson = new Gson();
 	private static final ExecutorService executorService = Executors.newFixedThreadPool(3);
-	private static final String USER_AGENT = "NotEnoughUpdates/" + NotEnoughUpdates.VERSION;
+	private static String getUserAgent() {
+		if (NotEnoughUpdates.INSTANCE.config.hidden.customUserAgent != null) {
+			return NotEnoughUpdates.INSTANCE.config.hidden.customUserAgent;
+		}
+		return "NotEnoughUpdates/" + NotEnoughUpdates.VERSION;
+	}
 	private static SSLContext ctx;
 	private final Map<String, CompletableFuture<Void>> updateTasks = new HashMap<>();
 
@@ -170,7 +175,7 @@ public class ApiUtil {
 						}
 						conn.setConnectTimeout(10000);
 						conn.setReadTimeout(10000);
-						conn.setRequestProperty("User-Agent", USER_AGENT);
+						conn.setRequestProperty("User-Agent", getUserAgent());
 						if (this.postContentType != null) {
 							conn.setRequestProperty("Content-Type", this.postContentType);
 						}