Skip to content

Commit 28c4a94

Browse files
authored
Adding TIBIADATA_PROXY env support (#25)
1 parent 2834daf commit 28c4a94

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

src/webserver.go

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,9 @@ var TibiadataDefaultVoc string = "all"
3030
var TibiadataAPIversion int = 3
3131
var TibiadataDebug bool
3232

33-
// Tibiadata app user-agent
33+
// Tibiadata app resty vars
3434
var TibiadataUserAgent string
35+
var TibiadataProxyDomain string
3536

3637
// Tibiadata app details set to release/build on GitHub
3738
var TibiadataBuildRelease = "unknown" // will be set by GitHub Actions (to release number)
@@ -164,6 +165,12 @@ func TibiaDataInitializer() {
164165

165166
// generating TibiadataUserAgent with TibiadataUserAgentGenerator function
166167
TibiadataUserAgent = TibiadataUserAgentGenerator(TibiadataAPIversion)
168+
169+
// setting TibiadataProxyDomain
170+
if isEnvExist("TIBIADATA_PROXY") {
171+
TibiadataProxyDomain = "https://" + getEnv("TIBIADATA_PROXY", "www.tibia.com") + "/"
172+
}
173+
167174
}
168175

169176
/*
@@ -257,6 +264,11 @@ func TibiadataHTMLDataCollectorV3(TibiaURL string) string {
257264
// Disable redirection of client (so we skip parsing maintenance page)
258265
client.SetRedirectPolicy(resty.NoRedirectPolicy())
259266

267+
// Replace domain with proxy if env TIBIADATA_PROXY set
268+
if TibiadataProxyDomain != "" {
269+
TibiaURL = strings.ReplaceAll(TibiaURL, "https://www.tibia.com/", TibiadataProxyDomain)
270+
}
271+
260272
res, err := client.R().Get(TibiaURL)
261273
if err != nil {
262274
log.Printf("[error] TibiadataHTMLDataCollectorV3 (URL: %s) in resp1: %s", TibiaURL, err)

0 commit comments

Comments
 (0)