feat: reschedule checkChapter jobs on startup. closes #14

Author: oae

src/server/index.ts

@@ -4,7 +4,7 @@ import { ExpressAdapter } from '@bull-board/express';
 import express, { Request, Response } from 'express';
 import next from 'next';
 import { logger } from '../utils/logging';
-import { checkChaptersQueue } from './queue/checkChapters';
+import { checkChaptersQueue, scheduleAll } from './queue/checkChapters';
 import { downloadQueue } from './queue/download';
 import { notificationQueue } from './queue/notify';
 
@@ -24,6 +24,7 @@ createBullBoard({
 (async () => {
   try {
     await app.prepare();
+    await scheduleAll();
     const server = express();
     server.use('/bull/queues', serverAdapter.getRouter()).all('*', (req: Request, res: Response) => {
       return handle(req, res);

src/server/queue/checkChapters.ts

@@ -119,12 +119,16 @@ export const removeJob = async (title: string) => {
   );
 };
 
-export const schedule = async (manga: MangaWithLibraryAndMetadata) => {
+export const schedule = async (manga: MangaWithLibraryAndMetadata, runImmediately: boolean) => {
+  await removeJob(manga.title);
+
+  if (runImmediately === true) {
+    await checkChapters(manga);
+  }
+
   if (manga.interval === 'never') {
     return;
   }
-
-  await removeJob(manga.title);
   const jobId = getJobIdFromTitle(manga.title);
 
   await checkChaptersQueue.add(
@@ -140,6 +144,13 @@ export const schedule = async (manga: MangaWithLibraryAndMetadata) => {
       },
     },
   );
+};
 
-  await checkChapters(manga);
+export const scheduleAll = async () => {
+  const mangaList = await prisma.manga.findMany({ include: { library: true, metadata: true } });
+  await Promise.all(
+    mangaList.map(async (manga) => {
+      await schedule(manga, false);
+    }),
+  );
 };

src/server/trpc/router/manga.ts

@@ -177,7 +177,7 @@ export const mangaRouter = t.router({
         },
       });
 
-      schedule(manga);
+      schedule(manga, true);
 
       return manga;
     }),

src/utils/index.ts

@@ -0,0 +1,38 @@
+import cronParser from 'cron-parser';
+import cronstrue from 'cronstrue';
+import { logger } from './logging';
+
+export const sanitizer = (value: string): string => {
+  return value
+    .replaceAll(/[\\/<>:;"'|?!*{}#%&^+,~\s]/g, '_')
+    .replaceAll(/__+/g, '_')
+    .replaceAll(/^[_\-.]+|[_\-.]+$/g, '_');
+};
+
+export const getCronLabel = (cron: string): string | undefined => {
+  if (cron === 'never') {
+    return 'never';
+  }
+
+  try {
+    return cronstrue.toString(cron, { use24HourTimeFormat: true });
+  } catch (err) {
+    logger.error('failed to parse cron');
+  }
+
+  return undefined;
+};
+
+export const isCronValid = (cron: string): boolean => {
+  if (cron === 'never') {
+    return true;
+  }
+
+  try {
+    cronParser.parseExpression(cron);
+    cronstrue.toString(cron);
+    return true;
+  } catch (err) {
+    return false;
+  }
+};