feat!: added PDF and screenshot archive support

lib/api/archive.ts

@@ -0,0 +1,21 @@
+import { chromium, devices } from "playwright";
+
+export default async (url: string, collectionId: number, linkId: number) => {
+  const archivePath = `data/archives/${collectionId}/${linkId}`;
+
+  const browser = await chromium.launch();
+  const context = await browser.newContext(devices["Desktop Chrome"]);
+  const page = await context.newPage();
+
+  // const contexts = browser.contexts();
+  // console.log(contexts.length);
+
+  await page.goto(url);
+
+  await page.pdf({ path: archivePath + ".pdf" });
+
+  await page.screenshot({ fullPage: true, path: archivePath + ".png" });
+
+  await context.close();
+  await browser.close();
+};

lib/api/controllers/collections/postCollection.ts

@@ -1,6 +1,7 @@
 import type { NextApiRequest, NextApiResponse } from "next";
 import { prisma } from "@/lib/api/db";
 import { Session } from "next-auth";
+import { existsSync, mkdirSync } from "fs";
 
 export default async function (
   req: NextApiRequest,
@@ -39,7 +40,7 @@ export default async function (
     return res.status(400).json({ response: "Collection already exists." });
   }
 
-  const createCollection = await prisma.collection.create({
+  const newCollection = await prisma.collection.create({
     data: {
       owner: {
         connect: {
@@ -50,7 +51,11 @@ export default async function (
     },
   });
 
+  const collectionPath = `data/archives/${newCollection.id}`;
+  if (!existsSync(collectionPath))
+    mkdirSync(collectionPath, { recursive: true });
+
   return res.status(200).json({
-    response: createCollection,
+    response: newCollection,
   });
 }

lib/api/controllers/links/postLink.ts

@@ -2,6 +2,11 @@ import type { NextApiRequest, NextApiResponse } from "next";
 import { prisma } from "@/lib/api/db";
 import { Session } from "next-auth";
 import { LinkAndTags, NewLink } from "@/types/global";
+import { existsSync, mkdirSync } from "fs";
+import getTitle from "../../getTitle";
+import archive from "../../archive";
+import { Link } from "@prisma/client";
+import AES from "crypto-js/aes";
 
 export default async function (
   req: NextApiRequest,
@@ -21,8 +26,8 @@ export default async function (
       .json({ response: "Please enter a valid name for the link." });
   }
 
-  if (link.collectionId.isNew) {
-    const collectionId = link.collectionId.id as string;
+  if (link.collection.isNew) {
+    const collectionId = link.collection.id as string;
 
     const findCollection = await prisma.user.findFirst({
       where: {
@@ -43,7 +48,7 @@ export default async function (
       return res.status(400).json({ response: "Collection already exists." });
     }
 
-    const createCollection = await prisma.collection.create({
+    const newCollection = await prisma.collection.create({
       data: {
         owner: {
           connect: {
@@ -54,12 +59,18 @@ export default async function (
       },
     });
 
-    link.collectionId.id = createCollection.id;
+    const collectionPath = `data/archives/${newCollection.id}`;
+    if (!existsSync(collectionPath))
+      mkdirSync(collectionPath, { recursive: true });
+
+    link.collection.id = newCollection.id;
   }
 
-  const collectionId = link.collectionId.id as number;
+  const collectionId = link.collection.id as number;
 
-  const createLink: LinkAndTags = await prisma.link.create({
+  const title = await getTitle(link.url);
+
+  const newLink: Link = await prisma.link.create({
     data: {
       name: link.name,
       url: link.url,
@@ -86,11 +97,34 @@ export default async function (
           },
         })),
       },
+      title,
+      isFavorites: false,
+      screenshotPath: "",
+      pdfPath: "",
     },
+  });
+
+  const AES_SECRET = process.env.AES_SECRET as string;
+
+  const screenShotHashedPath = AES.encrypt(
+    `data/archives/${newLink.collectionId}/${newLink.id}.png`,
+    AES_SECRET
+  ).toString();
+
+  const pdfHashedPath = AES.encrypt(
+    `data/archives/${newLink.collectionId}/${newLink.id}.pdf`,
+    AES_SECRET
+  ).toString();
+
+  const updatedLink: LinkAndTags = await prisma.link.update({
+    where: { id: newLink.id },
+    data: { screenshotPath: screenShotHashedPath, pdfPath: pdfHashedPath },
     include: { tags: true },
   });
 
+  archive(updatedLink.url, updatedLink.collectionId, updatedLink.id);
+
   return res.status(200).json({
-    response: createLink,
+    response: updatedLink,
   });
 }

lib/api/getTitle.ts

@@ -0,0 +1,9 @@
+export default async (url: string) => {
+  const response = await fetch(url);
+  const text = await response.text();
+
+  // regular expression to find the <title> tag
+  let match = text.match(/<title.*>([^<]*)<\/title>/);
+  if (match) return match[1];
+  else return "";
+};

lib/api/hasAccessToCollection.ts

@@ -0,0 +1,15 @@
+import { prisma } from "@/lib/api/db";
+
+export default async (userId: number, collectionId: number) => {
+  const check: any = await prisma.collection.findFirst({
+    where: {
+      AND: {
+        id: collectionId,
+        OR: [{ ownerId: userId }, { members: { some: { userId } } }],
+      },
+    },
+    include: { members: true },
+  });
+
+  return check;
+};