From 43e36ee6fc64812fd0b9a6705305b4c384683ede Mon Sep 17 00:00:00 2001
From: Harvey Tindall <hrfee@protonmail.ch>
Date: Thu, 19 Oct 2023 17:19:52 +0100
Subject: [PATCH] setup: Include proxy, test JF with it

Found on the 2nd page.
---
 html/setup.html       | 26 ++++++++++++++++++++++++++
 lang.go               |  1 +
 lang/setup/en-us.json | 11 +++++++++--
 package-lock.json     |  1 -
 setup.go              | 36 ++++++++++++++++++++++++++++++++----
 ts/setup.ts           | 22 +++++++++++++++++++---
 6 files changed, 87 insertions(+), 10 deletions(-)

diff --git a/html/setup.html b/html/setup.html
index c69b4b7..f72628a 100644
--- a/html/setup.html
+++ b/html/setup.html
@@ -122,6 +122,32 @@
                                 </select>
                             </div>
                         </label>
+                        <span class="heading">{{ .lang.Proxy.title }}</span>
+                        <p class="content my-2" id="proxy-description">{{ .lang.Proxy.description }}</p>
+                        <label class="row switch pb-4">
+                            <input type="checkbox" class="mr-2" id="advanced-proxy"><span>{{ .lang.Strings.enabled }}</span>
+                        </label>
+                        <label class="label">
+                            <span>{{ .lang.Proxy.protocol }}</span>
+                            <div class="select ~neutral @low mt-4 mb-2">
+                                <select id="advanced-proxy_protocol">
+                                    <option value="http">HTTP</option>
+                                    <option value="socks">SOCKS5</option>
+                                </select>
+                            </div>
+                        </label>
+                        <label class="label">
+                            <span class="mt-4">{{ .lang.Proxy.address }}</span>
+                            <input type="text" class="input ~neutral @low mt-4 mb-2" id="advanced-proxy_address">
+                        </label>
+                        <label class="label">
+                            <span class="mt-4">{{ .lang.Strings.username }}</span>
+                            <input type="text" class="input ~neutral @low mt-4 mb-2" id="advanced-proxy_user">
+                        </label>
+                        <label class="label">
+                            <span class="mt-4">{{ .lang.Strings.password }}</span>
+                            <input type="text" class="input ~neutral @low mt-4 mb-2" id="advanced-proxy_password">
+                        </label>
                     </div>
                 </div>
                 <section class="section ~neutral banner footer flex-expand middle">
diff --git a/lang.go b/lang.go
index c6026b7..8dbdc46 100644
--- a/lang.go
+++ b/lang.go
@@ -116,6 +116,7 @@ type setupLang struct {
 	EndPage            langSection `json:"endPage"`
 	General            langSection `json:"general"`
 	Updates            langSection `json:"updates"`
+	Proxy              langSection `json:"proxy"`
 	Language           langSection `json:"language"`
 	Login              langSection `json:"login"`
 	JellyfinEmby       langSection `json:"jellyfinEmby"`
diff --git a/lang/setup/en-us.json b/lang/setup/en-us.json
index a60853a..d3b2a76 100644
--- a/lang/setup/en-us.json
+++ b/lang/setup/en-us.json
@@ -18,11 +18,12 @@
         "apiKey": "API Key",
         "error": "Error",
         "errorInvalidUserPass": "Invalid username/password.",
-        "errorNotAdmin": "User is not allowed to manage server.",
+        "errorNotAdmin": "User is not aEnabledllowed to manage server.",
         "errorUserDisabled": "User may be disabled.",
         "error404": "404, check the internal URL.",
         "errorConnectionRefused": "Connection refused.",
-        "errorUnknown": "Unknown error, check app logs."
+        "errorUnknown": "Unknown error, check app logs.",
+        "errorProxy": "Proxy configuration invalid."
     },
     "startPage": {
         "welcome": "Welcome!",
@@ -62,6 +63,12 @@
         "stable": "Stable",
         "unstable": "Unstable"
     },
+    "proxy": {
+        "title": "Proxy",
+        "description": "Have jfa-go make all connections through a HTTP/SOCKS5 proxy. Connection to Jellyfin will be tested through this.",
+        "protocol": "Protocol",
+        "address": "Address (Including Port)"
+    },
     "login": {
         "title": "Login",
         "description": "To access the admin page, you need to login with a method below:",
diff --git a/package-lock.json b/package-lock.json
index 235e343..c7afbe7 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -15,7 +15,6 @@
         "any-date-parser": "^1.5.4",
         "browserslist": "^4.21.7",
         "cheerio": "^1.0.0-rc.12",
-        "esbuild": "^0.18.20",
         "fs-cheerio": "^3.0.0",
         "inline-source": "^8.0.2",
         "jsdom": "^22.1.0",
diff --git a/setup.go b/setup.go
index 38393f8..2e23dcf 100644
--- a/setup.go
+++ b/setup.go
@@ -7,6 +7,7 @@ import (
 	"strings"
 
 	"github.com/gin-gonic/gin"
+	"github.com/hrfee/jfa-go/easyproxy"
 	"github.com/hrfee/mediabrowser"
 )
 
@@ -47,10 +48,15 @@ func (app *appContext) ServeSetup(gc *gin.Context) {
 }
 
 type testReq struct {
-	ServerType string `json:"type"`
-	Server     string `json:"server"`
-	Username   string `json:"username"`
-	Password   string `json:"password"`
+	ServerType    string `json:"type"`
+	Server        string `json:"server"`
+	Username      string `json:"username"`
+	Password      string `json:"password"`
+	Proxy         bool   `json:"proxy"`
+	ProxyProtocol string `json:"proxy_protocol,omitempty"`
+	ProxyAddress  string `json:"proxy_address,omitempty"`
+	ProxyUsername string `json:"proxy_user,omitempty"`
+	ProxyPassword string `json:"proxy_password,omitempty"`
 }
 
 func (app *appContext) TestJF(gc *gin.Context) {
@@ -64,6 +70,26 @@ func (app *appContext) TestJF(gc *gin.Context) {
 		serverType = mediabrowser.EmbyServer
 	}
 	tempjf, _ := mediabrowser.NewServer(serverType, req.Server, "jfa-go-setup", app.version, "auth", "auth", mediabrowser.NewNamedTimeoutHandler("authJF", req.Server, true), 30)
+
+	if req.Proxy {
+		conf := easyproxy.ProxyConfig{
+			Protocol: easyproxy.HTTP,
+			Addr:     req.ProxyAddress,
+			User:     req.ProxyUsername,
+			Password: req.ProxyPassword,
+		}
+		if strings.Contains(req.ProxyProtocol, "socks") {
+			conf.Protocol = easyproxy.SOCKS5
+		}
+
+		transport, err := easyproxy.NewTransport(conf)
+		if err != nil {
+			respond(400, "errorProxy", gc)
+			return
+		}
+		tempjf.SetTransport(transport)
+	}
+
 	user, status, err := tempjf.Authenticate(req.Username, req.Password)
 	if !(status == 200 || status == 204) || err != nil {
 		msg := ""
@@ -126,6 +152,7 @@ func (st *Storage) loadLangSetup(filesystems ...fs.FS) error {
 					patchLang(&lang.Strings, &fallback.Strings, &english.Strings)
 					patchLang(&lang.StartPage, &fallback.StartPage, &english.StartPage)
 					patchLang(&lang.Updates, &fallback.Updates, &english.Updates)
+					patchLang(&lang.Proxy, &fallback.Proxy, &english.Proxy)
 					patchLang(&lang.EndPage, &fallback.EndPage, &english.EndPage)
 					patchLang(&lang.Language, &fallback.Language, &english.Language)
 					patchLang(&lang.Login, &fallback.Login, &english.Login)
@@ -144,6 +171,7 @@ func (st *Storage) loadLangSetup(filesystems ...fs.FS) error {
 				patchLang(&lang.Strings, &english.Strings)
 				patchLang(&lang.StartPage, &english.StartPage)
 				patchLang(&lang.Updates, &english.Updates)
+				patchLang(&lang.Proxy, &english.Proxy)
 				patchLang(&lang.EndPage, &english.EndPage)
 				patchLang(&lang.Language, &english.Language)
 				patchLang(&lang.Login, &english.Login)
diff --git a/ts/setup.ts b/ts/setup.ts
index e445fa1..3e9c4cb 100644
--- a/ts/setup.ts
+++ b/ts/setup.ts
@@ -85,6 +85,13 @@ class Checkbox {
                 }
             });
         }
+
+        if (this._el.hasAttribute("checked")) {
+            this._el.checked = true;
+        } else {
+            this._el.checked = false;
+        }
+        this.broadcast();
     }
 }
 
@@ -315,10 +322,14 @@ const settings = {
         "tls": new Checkbox(get("advanced-tls"), "", false, "advanced", "tls"),
         "tls_port": new Input(get("advanced-tls_port"), "", "", "tls", true, "advanced"),
         "tls_cert": new Input(get("advanced-tls_cert"), "", "", "tls", true, "advanced"),
-        "tls_key": new Input(get("advanced-tls_key"), "", "", "tls", true, "advanced")
+        "tls_key": new Input(get("advanced-tls_key"), "", "", "tls", true, "advanced"),
+        "proxy": new Checkbox(get("advanced-proxy"), "", false, "advanced", "proxy"),
+        "proxy_protocol": new Select(get("advanced-proxy_protocol"), "proxy", true, "advanced"),
+        "proxy_address": new Input(get("advanced-proxy_address"), "", "", "proxy", true, "advanced"),
+        "proxy_user": new Input(get("advanced-proxy_user"), "", "", "proxy", true, "advanced"),
+        "proxy_password": new Input(get("advanced-proxy_password"), "", "", "proxy", true, "advanced")
     }
 };
-
 const checkTheme = () => {
     if (settings["ui"]["theme"].value.includes("Dark")) {
         document.documentElement.classList.add("dark-theme");
@@ -553,7 +564,12 @@ window.onpopstate = (event: PopStateEvent) => {
             "type": settings["jellyfin"]["type"].value,
             "server": settings["jellyfin"]["server"].value,
             "username": settings["jellyfin"]["username"].value,
-            "password": settings["jellyfin"]["password"].value
+            "password": settings["jellyfin"]["password"].value,
+            "proxy": settings["advanced"]["proxy"].value == "true",
+            "proxy_protocol": settings["advanced"]["proxy_protocol"].value,
+            "proxy_address": settings["advanced"]["proxy_address"].value,
+            "proxy_user": settings["advanced"]["proxy_user"].value,
+            "proxy_password": settings["advanced"]["proxy_password"].value
         };
         _post("/jellyfin/test", send, (req: XMLHttpRequest) => {
             if (req.readyState == 4) {