muh

src/inc/router.mjs

@@ -1,4 +1,4 @@
-import { promises as fs } from "fs";
+import fs from "fs";
 import path from "path";
 
 export default new class Router {
@@ -19,23 +19,61 @@ export default new class Router {
   async static({ dir = path.resolve() + "/public", route = /^\/public/ }) {
     if(!this.#mimes) {
       this.#mimes = new Map();
-      (await fs.readFile("/etc/mime.types", "utf-8"))
+      (await fs.promises.readFile("/etc/mime.types", "utf-8"))
         .split("\n")
         .filter(e => !e.startsWith("#") && e)
         .map(e => e.split(/\s{2,}/))
         .filter(e => e.length > 1)
         .forEach(m => m[1].split(" ").forEach(ext => this.#mimes.set(ext, m[0])));
     }
+
     this.get(route, async (req, res) => {
       try {
+        const mime = this.#mimes.get(req.url.path.split(".").pop());
+        const file = path.join(dir, req.url.path.replace(route, ""));
+        let stat;
+        try {
+          stat = await fs.promises.stat(file);
+        } catch {
+          return res.reply({
+            code: 404,
+            body: "404 - file not found."
+          });
+        }
+
+        if(!mime.startsWith("video") && !mime.startsWith("audio")) {
+          return res.reply({
+            type: this.#mimes.get(req.url.path.split(".").pop()).toLowerCase(),
+            body: await fs.promises.readFile(path.join(dir, req.url.path.replace(route, "")))
+          });
+        }
+        
+        if(req.headers.range) {
+          const parts = req.headers.range.replace(/bytes=/, "").split("-");
+          const start = parseInt(parts[0], 10);
+          const end = parts[1] ? parseInt(parts[1], 10) : stat.size - 1;
+          res.writeHead(206, {
+            "Content-Range": `bytes ${start}-${end}/${stat.size}`,
+            "Accept-Ranges": "bytes",
+            "Content-Length": (end - start) + 1,
+            "Content-Type": mime,
+          });
+          const stream = fs.createReadStream(file, { start: start, end: end })
+            .on("open", () => stream.pipe(res))
+            .on("error", err => res.end(err));
+        }
+        else {
+          res.writeHead(200, {
+            "Content-Length": stat.size,
+            "Content-Type": mime,
+          });
+          fs.createReadStream(file).pipe(res);
+        }
+      } catch(err) {
+        console.error(err);
         return res.reply({
-          type: this.#mimes.get(req.url.path.split(".").pop()).toLowerCase(),
-          body: await fs.readFile(path.join(dir, req.url.path.replace(route, "")))
-        });
-      } catch {
-        return res.reply({
-          code: 404,
-          body: "404 - file not found"
+          code: 500,
+          body: "500 - f0ck hat keinen b0ck"
         });
       }
     });