feat: add caching to thumbnail creator

server/api/thumbnail.ts

@@ -1,24 +1,44 @@
 import { Request, Response } from "express";
 import { screenshot } from "../lib/screenshot";
 
-const cache = new Map<string, Buffer>();
+const cache = new Map<string, { data: Buffer; timestamp: number }>();
 
-export const thumbnail = async (req: Request, res: Response) => {
+const regenerateThumbnail = async (slug: string) => {
-  const { slug } = req.params;
+  const curCache = cache.get(slug);
-  const cacheData = cache.get(slug);
 
-  if (cacheData) {
+  if (curCache?.data) {
-    res.contentType("image/jpeg");
+    cache.set(slug, {
-    return res.send(cacheData);
+      data: curCache.data,
+      timestamp: Date.now(),
+    });
   }
 
   const result = await screenshot(slug);
   if (!result) {
-    return res.status(400).send("Cannot generate thumbnail!");
+    return curCache;
   }
 
-  cache.set(slug, result);
+  const data = {
+    data: result,
+    timestamp: Date.now(),
+  };
+
+  cache.set(slug, data);
+  return data;
+};
+
+export const thumbnail = async (req: Request, res: Response) => {
+  const { slug } = req.params;
+  let cacheData = cache.get(slug);
+
+  if (!cacheData) {
+    cacheData = await regenerateThumbnail(slug);
+  }
+
+  if (cacheData && Date.now() - cacheData.timestamp > 10000) {
+    regenerateThumbnail(slug);
+  }
 
   res.contentType("image/jpeg");
-  res.send(result);
+  res.send(cacheData?.data);
 };

server/lib/screenshot.ts

@@ -28,6 +28,7 @@ export const screenshot = async (slug: string) => {
 
     closeHandler = setTimeout(() => {
       browser?.close();
+      browser = null;
       closeHandler = null;
     }, 60000);