summary refs log tree commit diff
path: root/src/routes/attachments.ts
blob: 730d2bb1d378a17cbc6545e2feb9e5ba28f86a74 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
import { Router } from "express";
import multer from "multer";
import { Config, Snowflake } from "@fosscord/server-util";
import { storage } from "../util/Storage";

const multer_ = multer({
	storage: multer.memoryStorage(),
	limits: {
		fields: 0,
		files: 1,
		fileSize: 1024 * 1024 * 100, // 100 mb
	},
});
const router = Router();

router.post("/:channel_id", multer_.single("attachment"), async (req, res) => {
	const { buffer, mimetype, stream, size, originalname, fieldname } = req.file;
	const { channel_id } = req.params;
	const filename = originalname.replaceAll(" ", "_").replace(/\W+/g, "");
	
	const endpoint = Config.get().cdn.endpoint || "http://localhost:3003";

	await storage.set(originalname, buffer);

	const id = Snowflake.generate();

	const file = {
		id,
		type: mimetype,
		content_type: mimetype,
		filename: originalname,
		size,
		url: `${endpoint}/attachments/${channel_id}/${id}/`,
	};

	return res.json(file);
});

router.get("/:hash/:filename", async (req, res) => {
	const { hash, filename } = req.params;

	const File = await db.data.attachments({ id: hash, filename: filename }).get();

	res.set("Content-Type", File.type);
	return res.send(Buffer.from(File.file, "base64"));
});

router.delete("/:hash/:filename", async (req, res) => {
	const { hash, filename } = req.params;

	await db.data.attachments({ id: hash, filename: filename }).delete();
	return res.send({ success: true, message: "attachment deleted" });
});

export default router;