fix(cool-links): restore video summarization functionality (#141)

Author: potb

package.json

@@ -7,7 +7,7 @@
   "license": "MIT",
   "scripts": {
     "build": "tsup-node src/main.ts",
-    "dev": "tsup-node src/main.ts --watch --on-success \"clear && node --enable-source-maps -r dotenv/config dist/main.js\"",
+    "dev": "tsup-node src/main.ts --watch src --watch .env --watch pnpm-lock.yaml --on-success \"clear && node --enable-source-maps -r dotenv/config dist/main.js\"",
     "test": "vitest",
     "check:lint": "eslint src --report-unused-disable-directives",
     "check:format": "prettier -c src",
@@ -21,6 +21,7 @@
     "cron": "3.1.7",
     "discord.js": "14.14.1",
     "keyv": "4.5.4",
+    "nanoid": "5.0.7",
     "open-graph-scraper": "6.5.1",
     "zod": "3.23.5"
   },

pnpm-lock.yaml

@@ -1,41 +1,39 @@
+import { nanoid } from 'nanoid';
+import { z } from 'zod';
+
 const baseUrl = 'https://www.summarize.tech/api/summary';
 
 // Example usage:
 // const videoUrl = 'https://www.youtube.com/watch?v=ruUlK6zRwS8';
 // const summary = await getVideoSummary(videoUrl);
 // console.log({ summary });
 
-type SummaryResponse = {
-  rollups: Record<number, SummaryChunk>;
-  title: string;
-};
-
-type SummaryChunk = {
-  children: Record<number, string>;
-  summary: string;
-};
-
-const makeId = (length: number) => {
-  let result = '';
-  const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789_';
-  const charactersLength = characters.length;
-
-  for (let i = 0; i < length; i++)
-    result += characters.charAt(Math.floor(Math.random() * charactersLength));
-
-  return result;
-};
+const summarySchema = z.object({
+  title: z.string(),
+  rollups: z.record(
+    z.coerce.number().int().gte(0),
+    z.object({
+      children: z.record(z.coerce.number().int().gte(0), z.string()),
+      summary: z.object({
+        before: z.string(),
+        keyword: z.string(),
+        after: z.string(),
+      }),
+    }),
+  ),
+});
 
-export const getVideoSummary = async (videoUrl: string) => {
-  return await fetch(baseUrl, {
+export const getVideoSummary = async (videoUrl: string) =>
+  fetch(baseUrl, {
     method: 'POST',
-    body: JSON.stringify({ url: videoUrl, deviceId: makeId(21) }),
+    body: JSON.stringify({ url: videoUrl, deviceId: nanoid(21) }),
     headers: { 'content-type': 'application/json' },
   })
     .then((res) => res.json())
-    .then((res) =>
-      (Object.values((res as SummaryResponse).rollups) ?? [])
-        .map((chunk) => chunk.summary)
-        .join(' '),
+    .then((json) => summarySchema.parse(json))
+    .then((result) =>
+      Object.values(result.rollups)
+        .map(({ summary }) => summary.before + summary.keyword + summary.after)
+        .join('\n')
+        .replace(/\. /g, '.\n'),
     );
-};