dominik/homepage-plus
1
0
Fork 0
mirror of https://github.com/DI0IK/homepage-plus.git synced 2025-07-09 14:58:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Better login strategy for qbittorrent proxy

Browse source 
- Additional logging in httpProxy
This commit is contained in:
Jason Fischer 2022-09-28 09:55:18 -07:00
parent 1ea8e38372
commit 65755a08aa
No known key found for this signature in database
2 changed files with 42 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

34
src/widgets/qbittorrent/proxy.js
View file

@ -2,8 +2,12 @@ import { formatApiCall } from "utils/proxy/api-helpers";
import { addCookieToJar, setCookieHeader } from "utils/proxy/cookie-jar";
import { httpProxy } from "utils/proxy/http";
import getServiceWidget from "utils/config/service-helpers";
import createLogger from "utils/logger";
const logger = createLogger("qbittorrentProxyHandler");
async function login(widget, params) {
logger.debug("qBittorrent is rejecting the request, logging in.");
const loginUrl = new URL(`${widget.url}/api/v2/auth/login`).toString();
const loginBody = `username=${encodeURI(widget.username)}&password=${encodeURI(widget.password)}`;
@ -14,46 +18,54 @@ async function login(widget, params) {
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]);
.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) {
logger.debug("Invalid or missing service '%s' or group '%s'", service, group);
return res.status(400).json({ error: "Invalid proxy service type" });
}
const widget = await getServiceWidget(group, service);
if (!widget) {
logger.debug("Invalid or missing widget for service '%s' in group '%s'", service, group);
return res.status(400).json({ error: "Invalid proxy service type" });
}
const url = new URL(formatApiCall("{url}/api/v2/{endpoint}", { endpoint, ...widget }));
const params = { method: "GET", headers: {} };
setCookieHeader(url, params);
if (!params.headers.Cookie) {
const [status, data] = await login(widget, params);
let [status, contentType, data] = await httpProxy(url, params);
if (status === 403) {
[status, data] = await login(widget, params);
if (status !== 200) {
logger.error("HTTP %d logging in to qBittorrent. Data: %s", status, data);
return res.status(status).end(data);
}
if (data.toString() !== "Ok.") {
logger.error("Error logging in to qBittorrent: Data: %s", data);
return res.status(401).end(data);
}
}
const [status, contentType, data] = await httpProxy(url, params);
[status, contentType, data] = await httpProxy(url, params);
if (status !== 200) {
logger.error("HTTP %d getting data from qBittorrent. Data: %s", status, data);
}
if (contentType) res.setHeader("Content-Type", contentType);
return res.status(status).send(data);