aboutsummaryrefslogtreecommitdiff
path: root/api/image-proxy.ts
blob: 8628b5a17f75b4142802ba2fa82a40c1d517ecdc (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
56
57
import { NowRequest, NowResponse } from "@now/node";
import fetch from "node-fetch";

export default async (request: NowRequest, response: NowResponse) => {
  res.setHeader("Access-Control-Allow-Origin", "trace.moe");
  res.setHeader("Access-Control-Allow-Methods", "GET,OPTIONS");
  const { url } = request.query;
  const imageURL = Array.isArray(url) ? url[0] : url;
  if (!imageURL) {
    return response.status(400).send("Error: No url from param");
  }

  try {
    new URL(imageURL);
  } catch (e) {
    return response.status(400).send("Error: Invalid URL string");
  }

  const res = await fetch(imageURL, {
    headers: {
      "User-Agent": request.headers["user-agent"],
      Referer: new URL(imageURL).origin,
    },
    redirect: "follow",
  });

  if (res.status >= 400) {
    return response.status(res.status).send(await res.text());
  }

  if (["image", "video"].includes(res.headers.get("Content-Type").split("/")[0].toLowerCase())) {
    response.setHeader("Content-Type", res.headers.get("Content-Type"));
    response.setHeader("Cache-Control", "s-maxage=86400");
  } else if (res.headers.get("Content-Type").toLowerCase() === "application/octet-stream") {
    const match = imageURL.match(/\.(\w{3,4})($|\?)/);
    if (!match) return response.status(400).send("Error: Cannot determine Content-Type");
    const contentType = {
      mp4: "video/mp4",
      mpeg: "video/mpeg",
      webm: "video/webm",
      mkv: "video/x-matroska",
      bmp: "image/bmp",
      gif: "image/gif",
      jpg: "image/jpeg",
      jpeg: "image/jpeg",
      png: "image/png",
      webp: "image/webp",
    }[match[1]];
    if (!contentType) return response.status(400).send("Error: Unknown Content-Type");
    response.setHeader("Content-Type", contentType);
    response.setHeader("Cache-Control", "s-maxage=86400");
  } else {
    return response.status(400).send("Error: Unsupported Content-Type");
  }

  return response.status(200).send(await res.buffer());
};