server/middlewares/ip-bans.ts

import { logger } from "@/loggers";
import { errorResponse } from "@/response";
import type { SocketAddress } from "bun";
import { createMiddleware } from "hono/factory";
import { matches } from "ip-matching";
import { config } from "~/packages/config-manager";
import { LogLevel } from "~/packages/log-manager";

export const ipBans = createMiddleware(async (context, next) => {
    // Check for banned IPs

    const requestIp = context.env?.ip as SocketAddress | undefined | null;

    if (!requestIp?.address) {
        await next();
        return;
    }

    for (const ip of config.http.banned_ips) {
        try {
            if (matches(ip, requestIp?.address)) {
                return errorResponse("Forbidden", 403);
            }
        } catch (e) {
            logger.log(
                LogLevel.Error,
                "Server.IPCheck",
                `Error while parsing banned IP "${ip}" `,
            );
            logger.logError(LogLevel.Error, "Server.IPCheck", e as Error);

            return errorResponse(
                `A server error occured: ${(e as Error).message}`,
                500,
            );
        }
    }

    await next();
});