import type { MiddlewareHandler } from 'hono';

import { getPath, time } from '@/utils/helpers';
import logger from '@/utils/logger';
import { requestMetric } from '@/utils/otel';

enum LogPrefix {
    Outgoing = '-->',
    Incoming = '<--',
    Error = 'xxx',
}

const colorStatus = (status: number) => {
    const out: { [key: string]: string } = {
        7: `\u{1B}[35m${status}\u{1B}[0m`,
        5: `\u{1B}[31m${status}\u{1B}[0m`,
        4: `\u{1B}[33m${status}\u{1B}[0m`,
        3: `\u{1B}[36m${status}\u{1B}[0m`,
        2: `\u{1B}[32m${status}\u{1B}[0m`,
        1: `\u{1B}[32m${status}\u{1B}[0m`,
        0: `\u{1B}[33m${status}\u{1B}[0m`,
    };

    const calculateStatus = Math.trunc(status / 100);

    return out[calculateStatus];
};

const middleware: MiddlewareHandler = async (ctx, next) => {
    const { method, raw, routePath } = ctx.req;
    const path = getPath(raw);

    logger.info(`${LogPrefix.Incoming} ${method} ${path}`);

    const start = Date.now();

    await next();

    const status = ctx.res.status;

    logger.info(`${LogPrefix.Outgoing} ${method} ${path} ${colorStatus(status)} ${time(start)}`);
    requestMetric.success(Date.now() - start, { path: routePath, method, status });
};

export default middleware;