FileLib: Use userAgent parameter

ChatTriggers · Jun 16, 2023 · 4196497 · 4196497
1 parent 318ae5d
commit 4196497
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/main/kotlin/com/chattriggers/ctjs/CTJS.kt b/src/main/kotlin/com/chattriggers/ctjs/CTJS.kt
@@ -89,8 +89,9 @@ class CTJS : ClientModInitializer {
 
         internal val gson = Gson()
 
-        internal fun makeWebRequest(url: String): URLConnection = URL(url).openConnection().apply {
-            setRequestProperty("User-Agent", "Mozilla/5.0 (ChatTriggers)")
+        @JvmOverloads
+        internal fun makeWebRequest(url: String, userAgent: String? = "Mozilla/5.0 (ChatTriggers)"): URLConnection = URL(url).openConnection().apply {
+            setRequestProperty("User-Agent", userAgent)
             connectTimeout = 3000
             readTimeout = 3000
         }

diff --git a/src/main/kotlin/com/chattriggers/ctjs/minecraft/libs/FileLib.kt b/src/main/kotlin/com/chattriggers/ctjs/minecraft/libs/FileLib.kt
@@ -165,7 +165,7 @@ object FileLib {
     @JvmStatic
     @JvmOverloads
     fun getUrlContent(theUrl: String, userAgent: String? = "Mozilla/5.0"): String {
-        val conn = CTJS.makeWebRequest(theUrl)
+        val conn = CTJS.makeWebRequest(theUrl, userAgent)
 
         return conn.getInputStream().use {
             it.readBytes()