summary refs log tree commit diff
path: root/src/cdn/routes/external.ts
diff options
context:
space:
mode:
Diffstat (limited to 'src/cdn/routes/external.ts')
-rw-r--r--src/cdn/routes/external.ts91
1 files changed, 91 insertions, 0 deletions
diff --git a/src/cdn/routes/external.ts b/src/cdn/routes/external.ts
new file mode 100644

index 00000000..cb17ff9b --- /dev/null +++ b/src/cdn/routes/external.ts
@@ -0,0 +1,91 @@ +import { Router, Response, Request } from "express"; +import fetch from "node-fetch"; +import { HTTPError } from "lambert-server"; +import { Snowflake } from "@fosscord/util"; +import { storage } from "../util/Storage"; +import FileType from "file-type"; +import { Config } from "@fosscord/util"; +import sharp from "sharp"; + +// TODO: somehow handle the deletion of images posted to the /external route + +const router = Router(); +const DEFAULT_FETCH_OPTIONS: any = { + redirect: "follow", + follow: 1, + headers: { + "user-agent": + "Mozilla/5.0 (compatible Fosscordbot/0.1; +https://fosscord.com)", + }, + size: 1024 * 1024 * 8, + compress: true, + method: "GET", +}; + +router.post("/", async (req: Request, res: Response) => { + if (req.headers.signature !== Config.get().security.requestSignature) + throw new HTTPError("Invalid request signature"); + + if (!req.body) throw new HTTPError("Invalid Body"); + + const { url } = req.body; + if (!url || typeof url !== "string") throw new HTTPError("Invalid url"); + + const id = Snowflake.generate(); + + try { + const response = await fetch(url, DEFAULT_FETCH_OPTIONS); + const buffer = await response.buffer(); + + await storage.set(`/external/${id}`, buffer); + + res.send({ id }); + } catch (error) { + throw new HTTPError("Couldn't fetch website"); + } +}); + +router.get("/:id", async (req: Request, res: Response) => { + const { id } = req.params; + + const file = await storage.get(`/external/${id}`); + if (!file) throw new HTTPError("File not found"); + const result = await FileType.fromBuffer(file); + + res.set("Content-Type", result?.mime); + + return res.send(file); +}); + +// this method is gross lol don't care +router.get("/resize/:url", async (req: Request, res: Response) => { + const url = decodeURIComponent(req.params.url); + const { width, height } = req.query; + if (!width || !height) throw new HTTPError("Must provide width and height"); + + const { resizeHeightMax, resizeWidthMax } = Config.get().cdn; + const w = Math.min(parseInt(width as string), resizeWidthMax ?? 100); + const h = Math.min(parseInt(height as string), resizeHeightMax ?? 100); + if (w < 1 || h < 1) throw new HTTPError("Width and height must be greater than 0"); + + let buffer, response; + try { + response = await fetch(url, DEFAULT_FETCH_OPTIONS); + buffer = await response.buffer(); + } + catch (e) { + throw new HTTPError("Couldn't fetch website"); + } + + const resizedBuffer = await sharp(buffer) + .resize(parseInt(width as string), parseInt(height as string), { + fit: "inside", + }) + .toBuffer(); + + res.setHeader("Content-Disposition", "attachment"); + res.setHeader("Content-Type", response.headers.get("content-type") ?? "image/png"); + return res.end(resizedBuffer); +}); + +export default router;