diff options
| author | MohamedBassem <me@mbassem.com> | 2024-02-07 18:29:52 +0000 |
|---|---|---|
| committer | MohamedBassem <me@mbassem.com> | 2024-02-07 18:37:20 +0000 |
| commit | 3ec45e8bbb8285b17c703907d4c161b633663096 (patch) | |
| tree | ee52a753740a4a45e0ffe34840fc878ac383a5e2 /workers/crawler.ts | |
| parent | b12b964e0617f410b5c7b0989754cf94d01177cf (diff) | |
| download | karakeep-3ec45e8bbb8285b17c703907d4c161b633663096.tar.zst | |
[refactor] Rename the crawlers package to workers
Diffstat (limited to 'workers/crawler.ts')
| -rw-r--r-- | workers/crawler.ts | 72 |
1 files changed, 72 insertions, 0 deletions
diff --git a/workers/crawler.ts b/workers/crawler.ts new file mode 100644 index 00000000..c0f433af --- /dev/null +++ b/workers/crawler.ts @@ -0,0 +1,72 @@ +import logger from "@remember/shared/logger"; +import { + ZCrawlLinkRequest, + zCrawlLinkRequestSchema, +} from "@remember/shared/queues"; +import { Job } from "bullmq"; + +import prisma from "@remember/db"; + +import metascraper from "metascraper"; + +const metascraperParser = metascraper([ + require("metascraper-description")(), + require("metascraper-image")(), + require("metascraper-logo-favicon")(), + require("metascraper-title")(), + require("metascraper-url")(), +]); + +export default async function runCrawler(job: Job<ZCrawlLinkRequest, void>) { + const jobId = job.id || "unknown"; + + const request = zCrawlLinkRequestSchema.safeParse(job.data); + if (!request.success) { + logger.error( + `[Crawler][${jobId}] Got malformed job request: ${request.error.toString()}`, + ); + return; + } + + const { url, linkId } = request.data; + + logger.info( + `[Crawler][${jobId}] Will crawl "${url}" for link with id "${linkId}"`, + ); + // TODO(IMPORTANT): Run security validations on the input URL (e.g. deny localhost, etc) + + const resp = await fetch(url); + const respBody = await resp.text(); + + const meta = await metascraperParser({ + url, + html: respBody, + }); + + await prisma.bookmarkedLink.update({ + where: { + id: linkId, + }, + data: { + details: { + upsert: { + create: { + title: meta.title, + description: meta.description, + imageUrl: meta.image, + favicon: meta.logo, + }, + update: { + title: meta.title, + description: meta.description, + imageUrl: meta.image, + favicon: meta.logo, + }, + }, + }, + }, + include: { + details: true, + }, + }); +} |
