Allow Epub generation from multiple bookmarks hoarder-app#295

Changed dependency
added proper authentication check
added a flag to ignore errors while downloading

apps/web/app/api/epub/route.ts

@@ -1,10 +1,18 @@
-import { inArray } from "drizzle-orm";
-import epub, { Chapter } from "epub-gen-memory";
+import { getServerAuthSession } from "@/server/auth";
+import epub, { Chapter } from "@epubkit/epub-gen-memory";
+import { and, eq, inArray } from "drizzle-orm";
 
 import { db } from "@hoarder/db";
-import { bookmarkLinks } from "@hoarder/db/schema";
+import { bookmarkLinks, bookmarks } from "@hoarder/db/schema";
 
 export async function GET(request: Request) {
+  const session = await getServerAuthSession();
+  if (!session || !session.user) {
+    return new Response("", {
+      status: 401,
+    });
+  }
+
   const { searchParams } = new URL(request.url);
   const assetIds = searchParams.getAll("assetId");
 
@@ -20,15 +28,21 @@ export async function GET(request: Request) {
       htmlContent: bookmarkLinks.htmlContent,
     })
     .from(bookmarkLinks)
-    .where(inArray(bookmarkLinks.id, assetIds));
+    .leftJoin(bookmarks, eq(bookmarks.id, bookmarkLinks.id))
+    .where(
+      and(
+        eq(bookmarks.userId, session.user.id),
+        inArray(bookmarkLinks.id, assetIds),
+      ),
+    );
 
   if (!bookmarkInformation || bookmarkInformation.length === 0) {
     return new Response("", {
       status: 404,
     });
   }
 
-  const chapters = bookmarkInformation.map((information) => {
+  const chapters: Chapter[] = bookmarkInformation.map((information) => {
     return {
       content: information.htmlContent ?? "",
       title: information.title ?? "",
@@ -37,7 +51,10 @@ export async function GET(request: Request) {
 
   const title = getTitle(chapters);
 
-  const generatedEpub = await epub(title, chapters);
+  const generatedEpub = await epub(
+    { title, ignoreFailedDownloads: true },
+    chapters,
+  );
 
   return new Response(generatedEpub, {
     status: 200,

apps/web/package.json

@@ -18,6 +18,7 @@
     "@auth/drizzle-adapter": "^1.4.2",
     "@emoji-mart/data": "^1.1.2",
     "@emoji-mart/react": "^1.1.1",
+    "@epubkit/epub-gen-memory": "1.0.10-aplha.8",
     "@hoarder/db": "workspace:^0.1.0",
     "@hoarder/shared": "workspace:^0.1.0",
     "@hoarder/shared-react": "workspace:^0.1.0",