From b71963a4911c2c1433a4530032476d12678edac9 Mon Sep 17 00:00:00 2001
From: "Jack W." <git@jack.cab>
Date: Tue, 24 Oct 2023 16:50:14 -0400
Subject: [PATCH] refactor: :truck: Move preview to new file

---
 src/server/index.ts   | 110 ++----------------------------------------
 src/server/preview.ts | 109 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 112 insertions(+), 107 deletions(-)
 create mode 100644 src/server/preview.ts

diff --git a/src/server/index.ts b/src/server/index.ts
index f023eaa..730145d 100644
--- a/src/server/index.ts
+++ b/src/server/index.ts
@@ -6,15 +6,15 @@ import bytes from "bytes"
 
 import ServeError from "./lib/errors"
 import Files from "./lib/files"
-import * as auth from "./lib/auth"
 import * as Accounts from "./lib/accounts"
 import { getAccount } from "./lib/middleware"
 
 import APIRouter from "./routes/api"
+import preview from "./preview"
 
 require("dotenv").config()
 
-let pkg = require(`${process.cwd()}/package.json`)
+const pkg = require(`${process.cwd()}/package.json`)
 let app = express()
 let config = require(`${process.cwd()}/config.json`)
 
@@ -79,111 +79,7 @@ app.get("/", function (req, res) {
 
 // serve download page
 
-app.get("/download/:fileId", getAccount, (req, res) => {
-    let acc = res.locals.acc as Accounts.Account
-
-    if (files.getFilePointer(req.params.fileId)) {
-        let file = files.getFilePointer(req.params.fileId)
-
-        if (file.visibility == "private" && acc?.id != file.owner) {
-            ServeError(res, 403, "you do not own this file")
-            return
-        }
-
-        fs.readFile(process.cwd() + "/pages/download.html", (err, buf) => {
-            let fileOwner = file.owner
-                ? Accounts.getFromId(file.owner)
-                : undefined
-            if (err) {
-                res.sendStatus(500)
-                console.log(err)
-                return
-            }
-            res.send(
-                buf
-                    .toString()
-                    .replaceAll("$FileId", req.params.fileId)
-                    .replaceAll("$Version", pkg.version)
-                    .replaceAll(
-                        "$FileSize",
-                        file.sizeInBytes
-                            ? bytes(file.sizeInBytes)
-                            : "[File size unknown]"
-                    )
-                    .replaceAll(
-                        "$FileName",
-                        file.filename
-                            .replaceAll("&", "&amp;")
-                            .replaceAll("<", "&lt;")
-                            .replaceAll(">", "&gt;")
-                    )
-                    .replace(
-                        "<!--metaTags-->",
-                        (file.mime.startsWith("image/")
-                            ? `<meta name="og:image" content="https://${req.headers.host}/file/${req.params.fileId}" />`
-                            : file.mime.startsWith("video/")
-                            ? `<meta property="og:video:url" content="https://${
-                                  req.headers.host
-                              }/cpt/${req.params.fileId}/video.${
-                                  file.mime.split("/")[1] == "quicktime"
-                                      ? "mov"
-                                      : file.mime.split("/")[1]
-                              }" />
-                                <meta property="og:video:secure_url" content="https://${
-                                    req.headers.host
-                                }/cpt/${req.params.fileId}/video.${
-                                  file.mime.split("/")[1] == "quicktime"
-                                      ? "mov"
-                                      : file.mime.split("/")[1]
-                              }" />
-                                <meta property="og:type" content="video.other">
-                                <!-- honestly probably good enough for now -->
-                                <meta property="twitter:image" content="0">` +
-                              // quick lazy fix as a fallback
-                              // maybe i'll improve this later, but probably not.
-                              ((file.sizeInBytes || 0) >= 26214400
-                                  ? `
-                                <meta property="og:video:width" content="1280">
-                                <meta property="og:video:height" content="720">`
-                                  : "")
-                            : "") +
-                            (fileOwner?.embed?.largeImage &&
-                            file.visibility != "anonymous" &&
-                            file.mime.startsWith("image/")
-                                ? `<meta name="twitter:card" content="summary_large_image">`
-                                : "") +
-                            `\n<meta name="theme-color" content="${
-                                fileOwner?.embed?.color &&
-                                file.visibility != "anonymous" &&
-                                (req.headers["user-agent"] || "").includes(
-                                    "Discordbot"
-                                )
-                                    ? `#${fileOwner.embed.color}`
-                                    : "rgb(30, 33, 36)"
-                            }">`
-                    )
-                    .replace(
-                        "<!--preview-->",
-                        file.mime.startsWith("image/")
-                            ? `<div style="min-height:10px"></div><img src="/file/${req.params.fileId}" />`
-                            : file.mime.startsWith("video/")
-                            ? `<div style="min-height:10px"></div><video src="/file/${req.params.fileId}" controls></video>`
-                            : file.mime.startsWith("audio/")
-                            ? `<div style="min-height:10px"></div><audio src="/file/${req.params.fileId}" controls></audio>`
-                            : ""
-                    )
-                    .replaceAll(
-                        "$Uploader",
-                        !file.owner || file.visibility == "anonymous"
-                            ? "Anonymous"
-                            : `@${fileOwner?.username || "Deleted User"}`
-                    )
-            )
-        })
-    } else {
-        ServeError(res, 404, "file not found")
-    }
-})
+app.get("/download/:fileId", getAccount, preview(files))
 
 /*
     routes should be in this order:
diff --git a/src/server/preview.ts b/src/server/preview.ts
new file mode 100644
index 0000000..af52cb2
--- /dev/null
+++ b/src/server/preview.ts
@@ -0,0 +1,109 @@
+import fs from "fs/promises"
+import bytes from "bytes"
+import ServeError from "./lib/errors"
+import * as Accounts from "./lib/accounts"
+import type { Handler } from "express"
+import type Files from "./lib/files"
+const pkg = require(`${process.cwd()}/package.json`)
+export = (files: Files): Handler =>
+    async (req, res) => {
+        let acc = res.locals.acc as Accounts.Account
+        const file = files.getFilePointer(req.params.fileId)
+        if (file) {
+            if (file.visibility == "private" && acc?.id != file.owner) {
+                ServeError(res, 403, "you do not own this file")
+                return
+            }
+
+            const template = await fs
+                .readFile(process.cwd() + "/pages/download.html", "utf8")
+                .catch(() => {
+                    throw res.sendStatus(500)
+                })
+            let fileOwner = file.owner
+                ? Accounts.getFromId(file.owner)
+                : undefined
+
+            res.send(
+                template
+                    .replaceAll("$FileId", req.params.fileId)
+                    .replaceAll("$Version", pkg.version)
+                    .replaceAll(
+                        "$FileSize",
+                        file.sizeInBytes
+                            ? bytes(file.sizeInBytes)
+                            : "[File size unknown]"
+                    )
+                    .replaceAll(
+                        "$FileName",
+                        file.filename
+                            .replaceAll("&", "&amp;")
+                            .replaceAll("<", "&lt;")
+                            .replaceAll(">", "&gt;")
+                    )
+                    .replace(
+                        "<!--metaTags-->",
+                        (file.mime.startsWith("image/")
+                            ? `<meta name="og:image" content="https://${req.headers.host}/file/${req.params.fileId}" />`
+                            : file.mime.startsWith("video/")
+                            ? `<meta property="og:video:url" content="https://${
+                                  req.headers.host
+                              }/cpt/${req.params.fileId}/video.${
+                                  file.mime.split("/")[1] == "quicktime"
+                                      ? "mov"
+                                      : file.mime.split("/")[1]
+                              }" />
+                                <meta property="og:video:secure_url" content="https://${
+                                    req.headers.host
+                                }/cpt/${req.params.fileId}/video.${
+                                  file.mime.split("/")[1] == "quicktime"
+                                      ? "mov"
+                                      : file.mime.split("/")[1]
+                              }" />
+                                <meta property="og:type" content="video.other">
+                                <!-- honestly probably good enough for now -->
+                                <meta property="twitter:image" content="0">` +
+                              // quick lazy fix as a fallback
+                              // maybe i'll improve this later, but probably not.
+                              ((file.sizeInBytes || 0) >= 26214400
+                                  ? `
+                                <meta property="og:video:width" content="1280">
+                                <meta property="og:video:height" content="720">`
+                                  : "")
+                            : "") +
+                            (fileOwner?.embed?.largeImage &&
+                            file.visibility != "anonymous" &&
+                            file.mime.startsWith("image/")
+                                ? `<meta name="twitter:card" content="summary_large_image">`
+                                : "") +
+                            `\n<meta name="theme-color" content="${
+                                fileOwner?.embed?.color &&
+                                file.visibility != "anonymous" &&
+                                (req.headers["user-agent"] || "").includes(
+                                    "Discordbot"
+                                )
+                                    ? `#${fileOwner.embed.color}`
+                                    : "rgb(30, 33, 36)"
+                            }">`
+                    )
+                    .replace(
+                        "<!--preview-->",
+                        file.mime.startsWith("image/")
+                            ? `<div style="min-height:10px"></div><img src="/file/${req.params.fileId}" />`
+                            : file.mime.startsWith("video/")
+                            ? `<div style="min-height:10px"></div><video src="/file/${req.params.fileId}" controls></video>`
+                            : file.mime.startsWith("audio/")
+                            ? `<div style="min-height:10px"></div><audio src="/file/${req.params.fileId}" controls></audio>`
+                            : ""
+                    )
+                    .replaceAll(
+                        "$Uploader",
+                        !file.owner || file.visibility == "anonymous"
+                            ? "Anonymous"
+                            : `@${fileOwner?.username || "Deleted User"}`
+                    )
+            )
+        } else {
+            ServeError(res, 404, "file not found")
+        }
+    }