Phishing: Improved redirect support

Kord-Extensions · Mar 7, 2024 · 5cc223a · 5cc223a
1 parent df4abf6
commit 5cc223a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/...main/kotlin/com/kotlindiscord/kord/extensions/modules/extra/phishing/PhishingExtension.kt b/...main/kotlin/com/kotlindiscord/kord/extensions/modules/extra/phishing/PhishingExtension.kt
@@ -340,7 +340,15 @@ class PhishingExtension(private val settings: ExtPhishingBuilder) : Extension()
 		}
 
 		if (response.headers.contains("Location")) {
-			val newUrl = response.headers["Location"]!!
+			var newUrl = response.headers["Location"]!!
+
+			if (newUrl.startsWith("/")) {
+				val (protocol, path) = url.split("://", limit = 2)
+
+				newUrl = "$protocol://${path.split("/", limit = 2).first()}$newUrl"
+			} else if (newUrl.startsWith("./")) {
+				newUrl = url.trimEnd('/') + newUrl.trimStart('.')
+			}
 
 			if (url.trim('/') == newUrl.trim('/')) {
 				return null  // Results in the same URL