Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Deluge widget #551

Merged
merged 6 commits into from Nov 26, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
6 changes: 6 additions & 0 deletions public/locales/en/common.json
Expand Up @@ -112,6 +112,12 @@
"leech": "Leech",
"seed": "Seed"
},
"deluge": {
"download": "Download",
"upload": "Upload",
"leech": "Leech",
"seed": "Seed"
},
"sonarr": {
"wanted": "Wanted",
"queued": "Queued",
Expand Down
82 changes: 82 additions & 0 deletions src/utils/proxy/handlers/jsonrpc.js
@@ -0,0 +1,82 @@
import { JSONRPCClient, JSONRPCErrorException } from "json-rpc-2.0";

import { formatApiCall } from "utils/proxy/api-helpers";
import { httpProxy } from "utils/proxy/http";
import getServiceWidget from "utils/config/service-helpers";
import createLogger from "utils/logger";
import widgets from "widgets/widgets";

const logger = createLogger("jsonrpcProxyHandler");

export async function sendJsonRpcRequest(url, method, params, username, password) {
const headers = {
"content-type": "application/json",
"accept": "application/json"
}

if (username && password) {
headers.authorization = `Basic ${Buffer.from(`${username}:${password}`).toString("base64")}`;
}

const client = new JSONRPCClient(async (rpcRequest) => {
const body = JSON.stringify(rpcRequest);
const httpRequestParams = {
method: "POST",
headers,
body
};

// eslint-disable-next-line no-unused-vars
const [status, contentType, data] = await httpProxy(url, httpRequestParams);
if (status === 200) {
const json = JSON.parse(data.toString());

// in order to get access to the underlying error object in the JSON response
// you must set `result` equal to undefined
if (json.error && (json.result === null)) {
json.result = undefined;
}
return client.receive(json);
}

return Promise.reject(data?.error ? data : new Error(data.toString()));
});

try {
const response = await client.request(method, params);
return [200, "application/json", JSON.stringify(response)];
}
catch (e) {
if (e instanceof JSONRPCErrorException) {
logger.debug("Error calling JSONPRC endpoint: %s. %s", url, e.message);
return [200, "application/json", JSON.stringify({result: null, error: {code: e.code, message: e.message}})];
}

logger.warn("Error calling JSONPRC endpoint: %s. %s", url, e);
return [500, "application/json", JSON.stringify({result: null, error: {code: 2, message: e.toString()}})];
}
}

export default async function jsonrpcProxyHandler(req, res) {
const { group, service, endpoint: method } = req.query;

if (group && service) {
const widget = await getServiceWidget(group, service);
const api = widgets?.[widget.type]?.api;

if (!api) {
return res.status(403).json({ error: "Service does not support API calls" });
}

if (widget) {
const url = formatApiCall(api, { ...widget });

// eslint-disable-next-line no-unused-vars
const [status, contentType, data] = await sendJsonRpcRequest(url, method, null, widget.username, widget.password);
return res.status(status).end(data);
}
}

logger.debug("Invalid or missing proxy service type '%s' in group '%s'", service, group);
return res.status(400).json({ error: "Invalid proxy service type" });
}
41 changes: 13 additions & 28 deletions src/utils/proxy/http.js
Expand Up @@ -18,10 +18,15 @@ function addCookieHandler(url, params) {
};
}

export function httpsRequest(url, params) {
function handleRequest(requestor, url, params) {
return new Promise((resolve, reject) => {
addCookieHandler(url, params);
const request = https.request(url, params, (response) => {
if (params?.body) {
params.headers = params.headers ?? {};
params.headers['content-length'] = Buffer.byteLength(params.body);
}

const request = requestor.request(url, params, (response) => {
const data = [];

response.on("data", (chunk) => {
Expand All @@ -38,40 +43,20 @@ export function httpsRequest(url, params) {
reject([500, error]);
});

if (params.body) {
if (params?.body) {
request.write(params.body);
}

request.end();
});
}

export function httpRequest(url, params) {
return new Promise((resolve, reject) => {
addCookieHandler(url, params);
const request = http.request(url, params, (response) => {
const data = [];

response.on("data", (chunk) => {
data.push(chunk);
});

response.on("end", () => {
addCookieToJar(url, response.headers);
resolve([response.statusCode, response.headers["content-type"], Buffer.concat(data), response.headers]);
});
});

request.on("error", (error) => {
reject([500, error]);
});

if (params.body) {
request.write(params.body);
}
export function httpsRequest(url, params) {
return handleRequest(https, url, params);
}

request.end();
});
export function httpRequest(url, params) {
return handleRequest(http, url, params);
}

export async function httpProxy(url, params = {}) {
Expand Down
1 change: 1 addition & 0 deletions src/widgets/components.js
Expand Up @@ -7,6 +7,7 @@ const components = {
bazarr: dynamic(() => import("./bazarr/component")),
changedetectionio: dynamic(() => import("./changedetectionio/component")),
coinmarketcap: dynamic(() => import("./coinmarketcap/component")),
deluge: dynamic(() => import("./deluge/component")),
docker: dynamic(() => import("./docker/component")),
emby: dynamic(() => import("./emby/component")),
gluetun: dynamic(() => import("./gluetun/component")),
Expand Down
52 changes: 52 additions & 0 deletions src/widgets/deluge/component.jsx
@@ -0,0 +1,52 @@
import { useTranslation } from "next-i18next";

import Container from "components/services/widget/container";
import Block from "components/services/widget/block";
import useWidgetAPI from "utils/proxy/use-widget-api";

export default function Component({ service }) {
const { t } = useTranslation();

const { widget } = service;

const { data: torrentData, error: torrentError } = useWidgetAPI(widget);

if (torrentError) {
return <Container error={torrentError} />;
}

if (!torrentData) {
return (
<Container service={service}>
<Block label="deluge.leech" />
<Block label="deluge.download" />
<Block label="deluge.seed" />
<Block label="deluge.upload" />
</Container>
);
}

const { torrents } = torrentData;
const keys = torrents ? Object.keys(torrents) : [];

let rateDl = 0;
let rateUl = 0;
let completed = 0;
for (let i = 0; i < keys.length; i += 1) {
const torrent = torrents[keys[i]];
rateDl += torrent.download_payload_rate;
rateUl += torrent.upload_payload_rate;
completed += torrent.total_remaining === 0 ? 1 : 0;
}

const leech = keys.length - completed || 0;

return (
<Container service={service}>
<Block label="deluge.leech" value={t("common.number", { value: leech })} />
<Block label="deluge.download" value={t("common.bitrate", { value: rateDl })} />
<Block label="deluge.seed" value={t("common.number", { value: completed })} />
<Block label="deluge.upload" value={t("common.bitrate", { value: rateUl })} />
</Container>
);
}
63 changes: 63 additions & 0 deletions src/widgets/deluge/proxy.js
@@ -0,0 +1,63 @@
import { formatApiCall } from "utils/proxy/api-helpers";
import { sendJsonRpcRequest } from "utils/proxy/handlers/jsonrpc";
import getServiceWidget from "utils/config/service-helpers";
import createLogger from "utils/logger";
import widgets from "widgets/widgets";

const logger = createLogger("delugeProxyHandler");

const dataMethod = "web.update_ui";
const dataParams = [
["queue", "name", "total_wanted", "state", "progress", "download_payload_rate", "upload_payload_rate", "total_remaining"],
{}
];
const loginMethod = "auth.login";

async function sendRpc(url, method, params, username, password) {
const [status, contentType, data] = await sendJsonRpcRequest(url, method, params, username, password);
const json = JSON.parse(data.toString());
if (json?.error) {
if (json.error.code === 1) {
return [403, contentType, data];
}
return [500, contentType, data];
}

return [status, contentType, data];
}

function login(url, username, password) {
return sendRpc(url, loginMethod, [password], username, password);
}

export default async function delugeProxyHandler(req, res) {
const { group, service } = 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 api = widgets?.[widget.type]?.api
const url = new URL(formatApiCall(api, { ...widget }));

let [status, contentType, data] = await sendRpc(url, dataMethod, dataParams, widget.username, widget.password);
if (status === 403) {
[status, contentType, data] = await login(url, widget.username, widget.password);
if (status !== 200) {
return res.status(status).end(data);
}

// eslint-disable-next-line no-unused-vars
[status, contentType, data] = await sendRpc(url, dataMethod, dataParams, widget.username, widget.password);
}

return res.status(status).end(data);
}
8 changes: 8 additions & 0 deletions src/widgets/deluge/widget.js
@@ -0,0 +1,8 @@
import delugeProxyHandler from "./proxy";

const widget = {
api: "{url}/json",
proxyHandler: delugeProxyHandler,
};

export default widget;
40 changes: 0 additions & 40 deletions src/widgets/nzbget/proxy.js

This file was deleted.

5 changes: 3 additions & 2 deletions src/widgets/nzbget/widget.js
@@ -1,7 +1,8 @@
import nzbgetProxyHandler from "./proxy";
import jsonrpcProxyHandler from "utils/proxy/handlers/jsonrpc";

const widget = {
proxyHandler: nzbgetProxyHandler,
api: "{url}/jsonrpc",
proxyHandler: jsonrpcProxyHandler,
};

export default widget;