♻️ Refactor torrent network traffic widget #616
This commit is contained in:
Manuel
191
src/pages/api/modules/downloads/index.ts
Normal file
191
src/pages/api/modules/downloads/index.ts
Normal file
@@ -0,0 +1,191 @@
|
||||
import { Deluge } from '@ctrl/deluge';
|
||||
import { AllClientData } from '@ctrl/shared-torrent';
|
||||
import { getCookie } from 'cookies-next';
|
||||
import dayjs from 'dayjs';
|
||||
import { NextApiRequest, NextApiResponse } from 'next';
|
||||
import { Client } from 'sabnzbd-api';
|
||||
import { getConfig } from '../../../../tools/config/getConfig';
|
||||
import {
|
||||
NormalizedDownloadAppStat,
|
||||
NormalizedDownloadQueueResponse,
|
||||
} from '../../../../types/api/downloads/queue/NormalizedDownloadQueueResponse';
|
||||
import { ConfigAppType, IntegrationField } from '../../../../types/app';
|
||||
import { UsenetQueueItem } from '../../../../widgets/useNet/types';
|
||||
import { NzbgetClient } from '../usenet/nzbget/nzbget-client';
|
||||
import { NzbgetQueueItem, NzbgetStatus } from '../usenet/nzbget/types';
|
||||
|
||||
const Get = async (request: NextApiRequest, response: NextApiResponse) => {
|
||||
const configName = getCookie('config-name', { req: request });
|
||||
const config = getConfig(configName?.toString() ?? 'default');
|
||||
|
||||
const clientData: Promise<NormalizedDownloadAppStat | undefined>[] = config.apps.map((app) =>
|
||||
GetDataFromClient(app)
|
||||
);
|
||||
|
||||
const settledPromises = await Promise.allSettled(clientData);
|
||||
|
||||
const data: NormalizedDownloadAppStat[] = settledPromises
|
||||
.filter((x) => x.status === 'fulfilled')
|
||||
.map((promise) => (promise as PromiseFulfilledResult<NormalizedDownloadAppStat>).value)
|
||||
.filter((x) => x !== undefined);
|
||||
|
||||
const responseBody = { apps: data } as NormalizedDownloadQueueResponse;
|
||||
|
||||
return response.status(200).json(responseBody);
|
||||
};
|
||||
|
||||
const GetDataFromClient = async (
|
||||
app: ConfigAppType
|
||||
): Promise<NormalizedDownloadAppStat | undefined> => {
|
||||
const reduceTorrent = (data: AllClientData): NormalizedDownloadAppStat => ({
|
||||
type: 'torrent',
|
||||
appId: app.id,
|
||||
success: true,
|
||||
torrents: data.torrents,
|
||||
totalDownload: data.torrents
|
||||
.map((torrent) => torrent.downloadSpeed)
|
||||
.reduce((acc, torrent) => acc + torrent),
|
||||
totalUpload: data.torrents
|
||||
.map((torrent) => torrent.uploadSpeed)
|
||||
.reduce((acc, torrent) => acc + torrent),
|
||||
});
|
||||
|
||||
const findField = (app: ConfigAppType, field: IntegrationField) =>
|
||||
app.integration?.properties.find((x) => x.field === field)?.value ?? undefined;
|
||||
|
||||
switch (app.integration?.type) {
|
||||
case 'deluge': {
|
||||
return reduceTorrent(
|
||||
await new Deluge({
|
||||
baseUrl: app.url,
|
||||
password: findField(app, 'password'),
|
||||
}).getAllData()
|
||||
);
|
||||
}
|
||||
case 'transmission': {
|
||||
return reduceTorrent(
|
||||
await new Deluge({
|
||||
baseUrl: app.url,
|
||||
username: findField(app, 'username'),
|
||||
password: findField(app, 'password'),
|
||||
}).getAllData()
|
||||
);
|
||||
}
|
||||
case 'qBittorrent': {
|
||||
return reduceTorrent(
|
||||
await new Deluge({
|
||||
baseUrl: app.url,
|
||||
username: findField(app, 'username'),
|
||||
password: findField(app, 'password'),
|
||||
}).getAllData()
|
||||
);
|
||||
}
|
||||
case 'sabnzbd': {
|
||||
const { origin } = new URL(app.url);
|
||||
const client = new Client(origin, findField(app, 'apiKey') ?? '');
|
||||
const queue = await client.queue();
|
||||
const items: UsenetQueueItem[] = queue.slots.map((slot) => {
|
||||
const [hours, minutes, seconds] = slot.timeleft.split(':');
|
||||
const eta = dayjs.duration({
|
||||
hour: parseInt(hours, 10),
|
||||
minutes: parseInt(minutes, 10),
|
||||
seconds: parseInt(seconds, 10),
|
||||
} as any);
|
||||
|
||||
return {
|
||||
id: slot.nzo_id,
|
||||
eta: eta.asSeconds(),
|
||||
name: slot.filename,
|
||||
progress: parseFloat(slot.percentage),
|
||||
size: parseFloat(slot.mb) * 1000 * 1000,
|
||||
state: slot.status.toLowerCase() as any,
|
||||
};
|
||||
});
|
||||
const killobitsPerSecond = Number(queue.kbpersec);
|
||||
const bytesPerSecond = killobitsPerSecond * 1024; // convert killobytes to bytes
|
||||
return {
|
||||
type: 'usenet',
|
||||
appId: app.id,
|
||||
totalDownload: bytesPerSecond,
|
||||
nzbs: items,
|
||||
success: true,
|
||||
};
|
||||
}
|
||||
case 'nzbGet': {
|
||||
const url = new URL(app.url);
|
||||
const options = {
|
||||
host: url.hostname,
|
||||
port: url.port,
|
||||
login: app.integration.properties.find((x) => x.field === 'username')?.value ?? undefined,
|
||||
hash: app.integration.properties.find((x) => x.field === 'password')?.value ?? undefined,
|
||||
};
|
||||
|
||||
const nzbGet = NzbgetClient(options);
|
||||
const nzbgetQueue: NzbgetQueueItem[] = await new Promise((resolve, reject) => {
|
||||
nzbGet.listGroups((err: any, result: NzbgetQueueItem[]) => {
|
||||
if (!err) {
|
||||
resolve(result);
|
||||
} else {
|
||||
reject(err);
|
||||
}
|
||||
});
|
||||
});
|
||||
if (!nzbgetQueue) {
|
||||
throw new Error('Error while getting NZBGet queue');
|
||||
}
|
||||
|
||||
const nzbgetStatus: NzbgetStatus = await new Promise((resolve, reject) => {
|
||||
nzbGet.status((err: any, result: NzbgetStatus) => {
|
||||
if (!err) {
|
||||
resolve(result);
|
||||
} else {
|
||||
reject(err);
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
if (!nzbgetStatus) {
|
||||
throw new Error('Error while getting NZBGet status');
|
||||
}
|
||||
|
||||
const nzbgetItems: UsenetQueueItem[] = nzbgetQueue.map((item: NzbgetQueueItem) => ({
|
||||
id: item.NZBID.toString(),
|
||||
name: item.NZBName,
|
||||
progress: (item.DownloadedSizeMB / item.FileSizeMB) * 100,
|
||||
eta: (item.RemainingSizeMB * 1000000) / nzbgetStatus.DownloadRate,
|
||||
// Multiple MB to get bytes
|
||||
size: item.FileSizeMB * 1000 * 1000,
|
||||
state: getNzbgetState(item.Status),
|
||||
}));
|
||||
|
||||
return {
|
||||
type: 'usenet',
|
||||
appId: app.id,
|
||||
nzbs: nzbgetItems,
|
||||
success: true,
|
||||
totalDownload: 0,
|
||||
};
|
||||
}
|
||||
default:
|
||||
return undefined;
|
||||
}
|
||||
};
|
||||
|
||||
export default async (request: NextApiRequest, response: NextApiResponse) => {
|
||||
if (request.method === 'GET') {
|
||||
return Get(request, response);
|
||||
}
|
||||
|
||||
return response.status(405);
|
||||
};
|
||||
|
||||
function getNzbgetState(status: string) {
|
||||
switch (status) {
|
||||
case 'QUEUED':
|
||||
return 'queued';
|
||||
case 'PAUSED ':
|
||||
return 'paused';
|
||||
default:
|
||||
return 'downloading';
|
||||
}
|
||||
}
|
||||
@@ -1,131 +0,0 @@
|
||||
import { Deluge } from '@ctrl/deluge';
|
||||
import { QBittorrent } from '@ctrl/qbittorrent';
|
||||
import { AllClientData } from '@ctrl/shared-torrent';
|
||||
import { Transmission } from '@ctrl/transmission';
|
||||
import Consola from 'consola';
|
||||
import { getCookie } from 'cookies-next';
|
||||
import { NextApiRequest, NextApiResponse } from 'next';
|
||||
import { getConfig } from '../../../tools/config/getConfig';
|
||||
import { NormalizedTorrentListResponse } from '../../../types/api/NormalizedTorrentListResponse';
|
||||
import { ConfigAppType, IntegrationType } from '../../../types/app';
|
||||
|
||||
const supportedTypes: IntegrationType[] = ['deluge', 'qBittorrent', 'transmission'];
|
||||
|
||||
async function Post(req: NextApiRequest, res: NextApiResponse<NormalizedTorrentListResponse>) {
|
||||
// Get the type of app from the request url
|
||||
const configName = getCookie('config-name', { req });
|
||||
const config = getConfig(configName?.toString() ?? 'default');
|
||||
|
||||
const clientApps = config.apps.filter(
|
||||
(app) =>
|
||||
app.integration && app.integration.type && supportedTypes.includes(app.integration.type)
|
||||
);
|
||||
|
||||
if (clientApps.length < 1) {
|
||||
return res.status(500).json({
|
||||
allSuccess: false,
|
||||
missingDownloadClients: true,
|
||||
labels: [],
|
||||
torrents: [],
|
||||
});
|
||||
}
|
||||
|
||||
const promiseList: Promise<ConcatenatedClientData>[] = [];
|
||||
|
||||
for (let i = 0; i < clientApps.length; i += 1) {
|
||||
const app = clientApps[i];
|
||||
const getAllData = getAllDataForClient(app);
|
||||
if (!getAllData) {
|
||||
continue;
|
||||
}
|
||||
const concatenatedPromise = async (): Promise<ConcatenatedClientData> => ({
|
||||
clientData: await getAllData,
|
||||
appId: app.id,
|
||||
});
|
||||
promiseList.push(concatenatedPromise());
|
||||
}
|
||||
|
||||
const settledPromises = await Promise.allSettled(promiseList);
|
||||
const fulfilledPromises = settledPromises.filter(
|
||||
(settledPromise) => settledPromise.status === 'fulfilled'
|
||||
);
|
||||
|
||||
const fulfilledClientData: ConcatenatedClientData[] = fulfilledPromises.map(
|
||||
(fulfilledPromise) => (fulfilledPromise as PromiseFulfilledResult<ConcatenatedClientData>).value
|
||||
);
|
||||
|
||||
const notFulfilledClientData = settledPromises
|
||||
.filter((x) => x.status === 'rejected')
|
||||
.map(
|
||||
(fulfilledPromise) =>
|
||||
(fulfilledPromise as PromiseRejectedResult)
|
||||
);
|
||||
|
||||
notFulfilledClientData.forEach((result) => {
|
||||
Consola.error(`Error while communicating with torrent download client: ${result.reason}`);
|
||||
});
|
||||
|
||||
Consola.info(
|
||||
`Successfully fetched data from ${fulfilledPromises.length} torrent download clients`
|
||||
);
|
||||
|
||||
return res.status(200).json({
|
||||
labels: fulfilledClientData.flatMap((clientData) => clientData.clientData.labels),
|
||||
torrents: fulfilledClientData.map((clientData) => ({
|
||||
appId: clientData.appId,
|
||||
torrents: clientData.clientData.torrents,
|
||||
})),
|
||||
allSuccess: settledPromises.length === fulfilledPromises.length,
|
||||
missingDownloadClients: false,
|
||||
});
|
||||
}
|
||||
|
||||
const getAllDataForClient = (app: ConfigAppType) => {
|
||||
switch (app.integration?.type) {
|
||||
case 'deluge': {
|
||||
const password =
|
||||
app.integration?.properties.find((x) => x.field === 'password')?.value ?? undefined;
|
||||
return new Deluge({
|
||||
baseUrl: app.url,
|
||||
password,
|
||||
}).getAllData();
|
||||
}
|
||||
case 'transmission': {
|
||||
return new Transmission({
|
||||
baseUrl: app.url,
|
||||
username:
|
||||
app.integration!.properties.find((x) => x.field === 'username')?.value ?? undefined,
|
||||
password:
|
||||
app.integration!.properties.find((x) => x.field === 'password')?.value ?? undefined,
|
||||
}).getAllData();
|
||||
}
|
||||
case 'qBittorrent': {
|
||||
return new QBittorrent({
|
||||
baseUrl: app.url,
|
||||
username:
|
||||
app.integration!.properties.find((x) => x.field === 'username')?.value ?? undefined,
|
||||
password:
|
||||
app.integration!.properties.find((x) => x.field === 'password')?.value ?? undefined,
|
||||
}).getAllData();
|
||||
}
|
||||
default:
|
||||
Consola.error(`unable to find torrent client of type '${app.integration?.type}'`);
|
||||
return undefined;
|
||||
}
|
||||
};
|
||||
|
||||
type ConcatenatedClientData = {
|
||||
appId: string;
|
||||
clientData: AllClientData;
|
||||
};
|
||||
|
||||
export default async (req: NextApiRequest, res: NextApiResponse) => {
|
||||
// Filter out if the reuqest is a POST or a GET
|
||||
if (req.method === 'POST') {
|
||||
return Post(req, res);
|
||||
}
|
||||
return res.status(405).json({
|
||||
statusCode: 405,
|
||||
message: 'Method not allowed',
|
||||
});
|
||||
};
|
||||
Reference in New Issue
Block a user