server/server.ts

import {
    clientResponse,
    errorResponse,
    jsonResponse,
    response,
} from "@response";
import type { Config } from "config-manager";
import { matches } from "ip-matching";
import type { LogManager, MultiLogManager } from "log-manager";
import { LogLevel } from "log-manager";
import { RequestParser } from "request-parser";
import { getFromRequest } from "~database/entities/User";
import { matchRoute } from "~routes";

export const createServer = (
    config: Config,
    logger: LogManager | MultiLogManager,
    isProd: boolean,
) =>
    Bun.serve({
        port: config.http.bind_port,
        hostname: config.http.bind || "0.0.0.0", // defaults to "0.0.0.0"
        async fetch(req) {
            // Check for banned IPs
            const request_ip = this.requestIP(req)?.address ?? "";

            for (const ip of config.http.banned_ips) {
                try {
                    if (matches(ip, request_ip)) {
                        return errorResponse("Forbidden", 403);
                    }
                } catch (e) {
                    console.error(`[-] Error while parsing banned IP "${ip}" `);
                    throw e;
                }
            }

            // Check for banned user agents (regex)
            const ua = req.headers.get("User-Agent") ?? "";

            for (const agent of config.http.banned_user_agents) {
                if (new RegExp(agent).test(ua)) {
                    return errorResponse("Forbidden", 403);
                }
            }

            if (config.http.bait.enabled) {
                // Check for bait IPs
                for (const ip of config.http.bait.bait_ips) {
                    try {
                        if (matches(ip, request_ip)) {
                            const file = Bun.file(
                                config.http.bait.send_file ||
                                    "./pages/beemovie.txt",
                            );

                            if (await file.exists()) {
                                return response(file);
                            }
                            await logger.log(
                                LogLevel.ERROR,
                                "Server.Bait",
                                `Bait file not found: ${config.http.bait.send_file}`,
                            );
                        }
                    } catch (e) {
                        console.error(
                            `[-] Error while parsing bait IP "${ip}" `,
                        );
                        throw e;
                    }
                }

                // Check for bait user agents (regex)
                for (const agent of config.http.bait.bait_user_agents) {
                    if (new RegExp(agent).test(ua)) {
                        const file = Bun.file(
                            config.http.bait.send_file ||
                                "./pages/beemovie.txt",
                        );

                        if (await file.exists()) {
                            return response(file);
                        }
                        await logger.log(
                            LogLevel.ERROR,
                            "Server.Bait",
                            `Bait file not found: ${config.http.bait.send_file}`,
                        );
                    }
                }
            }

            if (config.logging.log_requests) {
                await logger.logRequest(
                    req.clone(),
                    config.logging.log_ip ? request_ip : undefined,
                    config.logging.log_requests_verbose,
                );
            }

            if (req.method === "OPTIONS") {
                return jsonResponse({});
            }

            const { file: filePromise, matchedRoute } = await matchRoute(
                req.url,
            );

            const file = filePromise;

            if (matchedRoute && file === undefined) {
                await logger.log(
                    LogLevel.ERROR,
                    "Server",
                    `Route file ${matchedRoute.filePath} not found or not registered in the routes file`,
                );

                return errorResponse("Route not found", 500);
            }

            if (matchedRoute && matchedRoute.name !== "/[...404]" && file) {
                const meta = file.meta;

                // Check for allowed requests
                // @ts-expect-error Stupid error
                if (!meta.allowedMethods.includes(req.method)) {
                    return errorResponse(
                        `Method not allowed: allowed methods are: ${meta.allowedMethods.join(
                            ", ",
                        )}`,
                        405,
                    );
                }

                // TODO: Check for ratelimits
                const auth = await getFromRequest(req);

                // Check for authentication if required
                if (
                    (meta.auth.required ||
                        (meta.auth.requiredOnMethods ?? []).includes(
                            // @ts-expect-error Stupid error
                            req.method,
                        )) &&
                    !auth.user
                ) {
                    return errorResponse("Unauthorized", 401);
                }

                let parsedRequest = {};

                try {
                    parsedRequest = await new RequestParser(req).toObject();
                } catch (e) {
                    await logger.logError(
                        LogLevel.ERROR,
                        "Server.RouteRequestParser",
                        e as Error,
                    );
                    return errorResponse("Bad request", 400);
                }

                return await file.default(req.clone(), matchedRoute, {
                    auth,
                    parsedRequest,
                    // To avoid having to rewrite each route
                    configManager: {
                        getConfig: () => Promise.resolve(config),
                    },
                });
            }

            if (new URL(req.url).pathname.startsWith("/api")) {
                return errorResponse("Route not found", 404);
            }

            // Proxy response from Vite at localhost:5173 if in development mode
            if (isProd) {
                let file = Bun.file("./pages/dist/index.html");
                if (new URL(req.url).pathname.startsWith("/assets")) {
                    file = Bun.file(`./pages/dist${new URL(req.url).pathname}`);
                }

                // Serve from pages/dist/assets
                if (await file.exists()) {
                    return clientResponse(file, 200, {
                        "Content-Type": file.type,
                    });
                }
                return errorResponse("Asset not found", 404);
            }

            const proxy = await fetch(
                req.url.replace(config.http.base_url, "http://localhost:5173"),
            ).catch(async (e) => {
                await logger.logError(
                    LogLevel.ERROR,
                    "Server.Proxy",
                    e as Error,
                );
                await logger.log(
                    LogLevel.ERROR,
                    "Server.Proxy",
                    `The development Vite server is not running or the route is not found: ${req.url.replace(
                        config.http.base_url,
                        "http://localhost:5173",
                    )}`,
                );
                return errorResponse("Route not found", 404);
            });

            if (
                proxy.status !== 404 &&
                !(await proxy.clone().text()).includes("404 Not Found")
            ) {
                return proxy;
            }

            return errorResponse("Route not found", 404);
        },
    });