Homarr/src/pages/api/modules/nzbs.ts

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 { Config, DownloadItem } from '../../../tools/types';

dayjs.extend(duration);

async function Get(req: NextApiRequest, res: NextApiResponse) {
  try {
    const configName = getCookie('config-name', { req });
    const { config }: { config: Config } = getConfig(configName?.toString() ?? 'default').props;
    const nzbServices = config.services.filter((service) => service.type === 'Sabnzbd');

    const downloads: DownloadItem[] = [];

    await Promise.all(
      nzbServices.map(async (service) => {
        if (!service.apiKey) {
          throw new Error(`API Key for service "${service.name}" is missing`);
        }
        const queue = await new Client(service.url, service.apiKey).queue();

        queue.slots.forEach((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);
          downloads.push({
            id: slot.nzo_id,
            eta: eta.asSeconds(),
            name: slot.filename,
            progress: parseFloat(slot.percentage),
            size: parseFloat(slot.mb),
            state: slot.status.toLowerCase() as any,
          });
        });
      })
    );

    return res.status(200).json(downloads);
  } catch (err) {
    return res.status(401).json(err);
  }
}

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',
  });
};