Add qBittorrent Widget

- extract cookie jar functionality into its own file
- use i18n for more strings in existing widgets

completes: #152
associated: #123

src/utils/api-helpers.js

@@ -10,6 +10,7 @@ const formats = {
   portainer: `{url}/api/endpoints/{env}/{endpoint}`,
   rutorrent: `{url}/plugins/httprpc/action.php`,
   transmission: `{url}/transmission/rpc`,
+  qbittorrent: `{url}/api/v2/{endpoint}`,
   jellyseerr: `{url}/api/v1/{endpoint}`,
   overseerr: `{url}/api/v1/{endpoint}`,
   ombi: `{url}/api/v1/{endpoint}`,

src/utils/cookie-jar.js

@@ -0,0 +1,34 @@
+/* eslint-disable no-param-reassign */
+import { Cookie, CookieJar } from 'tough-cookie';
+
+const cookieJar = new CookieJar();
+
+export function setCookieHeader(url, params) {
+  // add cookie header, if we have one in the jar
+  const existingCookie = cookieJar.getCookieStringSync(url.toString());
+  if (existingCookie) {
+    params.headers = params.headers ?? {};
+    params.headers.Cookie = existingCookie;
+  }
+}
+
+export function addCookieToJar(url, headers) {
+  let cookieHeader = headers['set-cookie'];
+  if (headers instanceof Headers) {
+    cookieHeader = headers.get('set-cookie');
+  }
+
+  if (!cookieHeader || cookieHeader.length === 0) return;
+
+  let cookies = null;
+  if (cookieHeader instanceof Array) {
+    cookies = cookieHeader.map(Cookie.parse);
+  }
+  else {
+    cookies = [Cookie.parse(cookieHeader)];
+  }
+
+  for (let i = 0; i < cookies.length; i += 1) {
+    cookieJar.setCookieSync(cookies[i], url.toString());
+  }
+}

src/utils/http.js

@@ -1,39 +1,15 @@
 /* eslint-disable prefer-promise-reject-errors */
 /* eslint-disable no-param-reassign */
 import { http, https } from "follow-redirects";
-import { Cookie, CookieJar } from 'tough-cookie';
 
-const cookieJar = new CookieJar();
-
-function setCookieHeader(url, params) {
-  // add cookie header, if we have one in the jar
-  const existingCookie = cookieJar.getCookieStringSync(url.toString());
-  if (existingCookie) {
-    params.headers = params.headers ?? {};
-    params.headers.Cookie = existingCookie;
-  }
-}
+import { addCookieToJar, setCookieHeader } from "utils/cookie-jar";
 
 function addCookieHandler(url, params) {
   setCookieHeader(url, params);
 
   // handle cookies during redirects
   params.beforeRedirect = (options, responseInfo) => {
-    const cookieHeader = responseInfo.headers['set-cookie'];
-    if (!cookieHeader || cookieHeader.length === 0) return;
-
-    let cookies = null;
-    if (cookieHeader instanceof Array) {
-      cookies = cookieHeader.map(Cookie.parse);
-    }
-    else {
-      cookies = [Cookie.parse(cookieHeader)];
-    }
-
-    for (let i = 0; i < cookies.length; i += 1) {
-      cookieJar.setCookieSync(cookies[i], options.href);
-    }
-
+    addCookieToJar(options.href, responseInfo.headers);
     setCookieHeader(options.href, options);
   };
 }
@@ -49,6 +25,7 @@ export function httpsRequest(url, params) {
       });
 
       response.on("end", () => {
+        addCookieToJar(url, response.headers);
         resolve([response.statusCode, response.headers["content-type"], Buffer.concat(data), response.headers]);
       });
     });
@@ -76,6 +53,7 @@ export function httpRequest(url, params) {
       });
 
       response.on("end", () => {
+        addCookieToJar(url, response.headers);
         resolve([response.statusCode, response.headers["content-type"], Buffer.concat(data), response.headers]);
       });
     });

src/utils/proxies/qbittorrent.js

@@ -0,0 +1,58 @@
+import { formatApiCall } from "utils/api-helpers";
+import { addCookieToJar, setCookieHeader } from "utils/cookie-jar";
+import { httpProxy } from "utils/http";
+import getServiceWidget from "utils/service-helpers";
+
+async function login(widget, params) {
+  const loginUrl = new URL(`${widget.url}/api/v2/auth/login`);
+  const loginBody = `username=${encodeURI(widget.username)}&password=${encodeURI(widget.password)}`;
+
+  // using fetch intentionally, for login only, as the httpProxy method causes qBittorrent to
+  // complain about header encoding
+  return fetch(loginUrl, {
+    method: "POST",
+    headers: { "Content-Type": "application/x-www-form-urlencoded" },
+    body: loginBody
+  })
+  .then(async response => {
+    addCookieToJar(loginUrl, response.headers);
+    setCookieHeader(loginUrl, params);
+    const data = await response.text();
+    return ([response.status, data]);
+  })
+  .catch(err => ([500, err]));
+}
+
+export default async function qbittorrentProxyHandler(req, res) {
+  const { group, service, endpoint } = req.query;
+
+  if (!group || !service) {
+    return res.status(400).json({ error: "Invalid proxy service type" });
+  }
+
+  const widget = await getServiceWidget(group, service);
+
+  if (!widget) {
+    return res.status(400).json({ error: "Invalid proxy service type" });
+  }
+
+  const url = new URL(formatApiCall(widget.type, { endpoint, ...widget }));
+  const params = { method: "GET", headers: {} };
+  setCookieHeader(url, params);
+
+  if (!params.headers.Cookie) {
+    const [status, data] = await login(widget, params);
+
+    if (status !== 200) {
+      return res.status(status).end(data);
+    }
+    if (data.toString() !== 'Ok.') {
+      return res.status(401).end(data);
+    }
+  }
+
+  const [status, contentType, data] = await httpProxy(url, params);
+
+  if (contentType) res.setHeader("Content-Type", contentType);
+  return res.status(status).send(data);
+}