✨ Add NZBGet download client

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

@@ -7,6 +7,8 @@ import { UsenetQueueItem } from '../../../../modules';
 import { getConfig } from '../../../../tools/getConfig';
 import { getServiceById } from '../../../../tools/hooks/useGetServiceByType';
 import { Config } from '../../../../tools/types';
+import { NzbgetClient } from './nzbget/nzbget-client';
+import { NzbgetQueueItem, NzbgetStatus } from './nzbget/types';
 
 dayjs.extend(duration);
 
@@ -33,42 +35,116 @@ 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: UsenetQueueResponse;
+    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 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),
+        }));
+
+        response = {
+          items: nzbgetItems,
+          total: nzbgetItems.length,
+        };
+        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(offset, limit);
+
+        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,
+          };
+        });
+
+        response = {
+          items,
+          total: queue.noofslots,
+        };
+        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(offset, limit);
-
-    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 response: UsenetQueueResponse = {
-      items,
-      total: queue.noofslots,
-    };
-
     return res.status(200).json(response);
   } catch (err) {
     return res.status(500).send((err as any).message);
   }
 }
 
+function getNzbgetState(status: string) {
+  switch (status) {
+    case 'QUEUED':
+      return 'queued';
+    case 'PAUSED ':
+        return 'paused';
+    default:
+      return 'downloading';
+  }
+}
+
 export default async (req: NextApiRequest, res: NextApiResponse) => {
   // Filter out if the reuqest is a POST or a GET
   if (req.method === 'GET') {