From 9f3304831b376763c05e4ec048dd761ee0c36dca Mon Sep 17 00:00:00 2001
From: Tobias Lindberg <tobias.ehlert@gmail.com>
Date: Tue, 4 Jan 2022 23:18:00 +0100
Subject: [PATCH] adding TIBIADATA_PROXY env

---
 src/webserver.go | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/webserver.go b/src/webserver.go
index bd46c2c9..01b46ac3 100644
--- a/src/webserver.go
+++ b/src/webserver.go
@@ -30,8 +30,9 @@ var TibiadataDefaultVoc string = "all"
 var TibiadataAPIversion int = 3
 var TibiadataDebug bool
 
-// Tibiadata app user-agent
+// Tibiadata app resty vars
 var TibiadataUserAgent string
+var TibiadataProxyDomain string
 
 // Tibiadata app details set to release/build on GitHub
 var TibiadataBuildRelease = "unknown"     // will be set by GitHub Actions (to release number)
@@ -164,6 +165,12 @@ func TibiaDataInitializer() {
 
 	// generating TibiadataUserAgent with TibiadataUserAgentGenerator function
 	TibiadataUserAgent = TibiadataUserAgentGenerator(TibiadataAPIversion)
+
+	// setting TibiadataProxyDomain
+	if isEnvExist("TIBIADATA_PROXY") {
+		TibiadataProxyDomain = "https://" + getEnv("TIBIADATA_PROXY", "www.tibia.com") + "/"
+	}
+
 }
 
 /*
@@ -257,6 +264,11 @@ func TibiadataHTMLDataCollectorV3(TibiaURL string) string {
 	// Disable redirection of client (so we skip parsing maintenance page)
 	client.SetRedirectPolicy(resty.NoRedirectPolicy())
 
+	// Replace domain with proxy if env TIBIADATA_PROXY set
+	if TibiadataProxyDomain != "" {
+		TibiaURL = strings.ReplaceAll(TibiaURL, "https://www.tibia.com/", TibiadataProxyDomain)
+	}
+
 	res, err := client.R().Get(TibiaURL)
 	if err != nil {
 		log.Printf("[error] TibiadataHTMLDataCollectorV3 (URL: %s) in resp1: %s", TibiaURL, err)