✨ Add NZBGet download client

src/pages/api/modules/usenet/index.ts

@@ -6,6 +6,8 @@ import { Client } from 'sabnzbd-api';
 import { getConfig } from '../../../../tools/getConfig';
 import { getServiceById } from '../../../../tools/hooks/useGetServiceByType';
 import { Config } from '../../../../tools/types';
+import { NzbgetStatus } from './nzbget/types';
+import { NzbgetClient } from './nzbget/nzbget-client';
 
 dayjs.extend(duration);
 
@@ -32,28 +34,71 @@ async function Get(req: NextApiRequest, res: NextApiResponse) {
       throw new Error(`Service with ID "${req.query.serviceId}" could not be found.`);
     }
 
-    if (!service.apiKey) {
-      throw new Error(`API Key for service "${service.name}" is missing`);
+    let response: UsenetInfoResponse;
+    switch (service.type) {
+      case 'NZBGet': {
+        const url = new URL(service.url);
+        const options = {
+          host: url.hostname,
+          port: url.port,
+          login: service.username,
+          hash: service.password,
+        };
+
+        const nzbGet = NzbgetClient(options);
+
+        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 bytesRemaining = nzbgetStatus.RemainingSizeMB * 1000000;
+        const eta = bytesRemaining / nzbgetStatus.DownloadRate;
+        response = {
+          paused: nzbgetStatus.DownloadPaused,
+          sizeLeft: bytesRemaining,
+          speed: nzbgetStatus.DownloadRate,
+          eta,
+        };
+        break;
+      }
+      case 'Sabnzbd': {
+        if (!service.apiKey) {
+          throw new Error(`API Key for service "${service.name}" is missing`);
+        }
+
+        const { origin } = new URL(service.url);
+
+        const queue = await new Client(origin, service.apiKey).queue(0, -1);
+
+        const [hours, minutes, seconds] = queue.timeleft.split(':');
+        const eta = dayjs.duration({
+          hour: parseInt(hours, 10),
+          minutes: parseInt(minutes, 10),
+          seconds: parseInt(seconds, 10),
+        } as any);
+
+        response = {
+          paused: queue.paused,
+          sizeLeft: parseFloat(queue.mbleft) * 1024 * 1024,
+          speed: parseFloat(queue.kbpersec) * 1000,
+          eta: eta.asSeconds(),
+        };
+        break;
+      }
+      default:
+        throw new Error(`Service type "${service.type}" unrecognized.`);
     }
 
-    const { origin } = new URL(service.url);
-
-    const queue = await new Client(origin, service.apiKey).queue(0, -1);
-
-    const [hours, minutes, seconds] = queue.timeleft.split(':');
-    const eta = dayjs.duration({
-      hour: parseInt(hours, 10),
-      minutes: parseInt(minutes, 10),
-      seconds: parseInt(seconds, 10),
-    } as any);
-
-    const response: UsenetInfoResponse = {
-      paused: queue.paused,
-      sizeLeft: parseFloat(queue.mbleft) * 1024 * 1024,
-      speed: parseFloat(queue.kbpersec) * 1000,
-      eta: eta.asSeconds(),
-    };
-
     return res.status(200).json(response);
   } catch (err) {
     return res.status(500).send((err as any).message);