lib/upload.js


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

import { fileTypeFromStream } from "file-type";
import pump from "pump";
import Clone from "readable-stream-clone";
import fs from "fs";
import path from "path";
import sharp from "sharp";
import { pipeline } from "stream";
import { g, s, db } from "$lib/db";

export default async (req, res) => {
  let {
    params: { type },
    user
  } = req;
  
  if (!["banner", "profile"].includes(type)) throw new Error("invalid upload");

  let data = await req.file();
  let s1 = new Clone(data.file);
  let s2 = new Clone(data.file);

  let [format, ext] = (await fileTypeFromStream(s1)).mime.split("/");

  if (format !== "image" && !["jpg", "jpeg", "png"].includes(ext))
    throw new Error("unsupported file type");

  let name = `${user.id}-${type}.webp`;
  let path = `/app/data/uploads/${name}`;
  let size = type === "banner" ? 1920 : 240;

  let t = sharp()
    .rotate()
    .resize(size)
    .webp();

  await new Promise(r => pipeline(s2, t, fs.createWriteStream(path), r));

  user[type] = true;
  await s(`user:${user.id}`, user);

  res.send(name);
};