implement queue info and pause/resume

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

@@ -3,29 +3,28 @@ import dayjs from 'dayjs';
 import duration from 'dayjs/plugin/duration';
 import { NextApiRequest, NextApiResponse } from 'next';
 import { Client } from 'sabnzbd-api';
-import { UsenetQueueItem } from '../../../../modules';
 import { getConfig } from '../../../../tools/getConfig';
 import { getServiceById } from '../../../../tools/hooks/useGetServiceByType';
 import { Config } from '../../../../tools/types';
 
 dayjs.extend(duration);
 
-export interface UsenetQueueRequestParams {
+export interface UsenetInfoRequestParams {
   serviceId: string;
-  offset: number;
-  limit: number;
 }
 
-export interface UsenetQueueResponse {
-  items: UsenetQueueItem[];
-  total: number;
+export interface UsenetInfoResponse {
+  paused: boolean;
+  sizeLeft: number;
+  speed: number;
+  eta: number;
 }
 
 async function Get(req: NextApiRequest, res: NextApiResponse) {
   try {
     const configName = getCookie('config-name', { req });
     const { config }: { config: Config } = getConfig(configName?.toString() ?? 'default').props;
-    const { limit, offset, serviceId } = req.query as any as UsenetQueueRequestParams;
+    const { serviceId } = req.query as any as UsenetInfoRequestParams;
 
     const service = getServiceById(config, serviceId);
 
@@ -36,29 +35,21 @@ async function Get(req: NextApiRequest, res: NextApiResponse) {
     if (!service.apiKey) {
       throw new Error(`API Key for service "${service.name}" is missing`);
     }
-    const queue = await new Client(service.url, 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);
+    const queue = await new Client(service.url, service.apiKey).queue(0, -1);
 
-      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 [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: UsenetQueueResponse = {
-      items,
-      total: queue.noofslots_total,
+    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);

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

@@ -0,0 +1,49 @@
+import { getCookie } from 'cookies-next';
+import dayjs from 'dayjs';
+import duration from 'dayjs/plugin/duration';
+import { NextApiRequest, NextApiResponse } from 'next';
+import { Client } from 'sabnzbd-api';
+import { getConfig } from '../../../../tools/getConfig';
+import { getServiceById } from '../../../../tools/hooks/useGetServiceByType';
+import { Config } from '../../../../tools/types';
+
+dayjs.extend(duration);
+
+export interface UsenetPauseRequestParams {
+  serviceId: string;
+}
+
+async function Post(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const configName = getCookie('config-name', { req });
+    const { config }: { config: Config } = getConfig(configName?.toString() ?? 'default').props;
+    const { serviceId } = req.query as any as UsenetPauseRequestParams;
+
+    const service = getServiceById(config, serviceId);
+
+    if (!service) {
+      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`);
+    }
+
+    const result = await new Client(service.url, service.apiKey).queuePause();
+
+    return res.status(200).json(result);
+  } catch (err) {
+    return res.status(500).send((err as any).message);
+  }
+}
+
+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',
+  });
+};

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

@@ -0,0 +1,79 @@
+import { getCookie } from 'cookies-next';
+import dayjs from 'dayjs';
+import duration from 'dayjs/plugin/duration';
+import { NextApiRequest, NextApiResponse } from 'next';
+import { Client } from 'sabnzbd-api';
+import { UsenetQueueItem } from '../../../../modules';
+import { getConfig } from '../../../../tools/getConfig';
+import { getServiceById } from '../../../../tools/hooks/useGetServiceByType';
+import { Config } from '../../../../tools/types';
+
+dayjs.extend(duration);
+
+export interface UsenetQueueRequestParams {
+  serviceId: string;
+  offset: number;
+  limit: number;
+}
+
+export interface UsenetQueueResponse {
+  items: UsenetQueueItem[];
+  total: number;
+}
+
+async function Get(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const configName = getCookie('config-name', { req });
+    const { config }: { config: Config } = getConfig(configName?.toString() ?? 'default').props;
+    const { limit, offset, serviceId } = req.query as any as UsenetQueueRequestParams;
+
+    const service = getServiceById(config, serviceId);
+
+    if (!service) {
+      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`);
+    }
+    const queue = await new Client(service.url, 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);
+  }
+}
+
+export default async (req: NextApiRequest, res: NextApiResponse) => {
+  // Filter out if the reuqest is a POST or a GET
+  if (req.method === 'GET') {
+    return Get(req, res);
+  }
+  return res.status(405).json({
+    statusCode: 405,
+    message: 'Method not allowed',
+  });
+};

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

@@ -0,0 +1,50 @@
+import { getCookie } from 'cookies-next';
+import dayjs from 'dayjs';
+import duration from 'dayjs/plugin/duration';
+import { NextApiRequest, NextApiResponse } from 'next';
+import { Client } from 'sabnzbd-api';
+import { getConfig } from '../../../../tools/getConfig';
+import { getServiceById } from '../../../../tools/hooks/useGetServiceByType';
+import { Config } from '../../../../tools/types';
+
+dayjs.extend(duration);
+
+export interface UsenetResumeRequestParams {
+  serviceId: string;
+  nzbId?: string;
+}
+
+async function Post(req: NextApiRequest, res: NextApiResponse) {
+  try {
+    const configName = getCookie('config-name', { req });
+    const { config }: { config: Config } = getConfig(configName?.toString() ?? 'default').props;
+    const { serviceId } = req.query as any as UsenetResumeRequestParams;
+
+    const service = getServiceById(config, serviceId);
+
+    if (!service) {
+      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`);
+    }
+
+    const result = await new Client(service.url, service.apiKey).queueResume();
+
+    return res.status(200).json(result);
+  } catch (err) {
+    return res.status(500).send((err as any).message);
+  }
+}
+
+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',
+  });
+};