import type { Logger } from '@rocket.chat/logger';
import { censorUrl } from '@rocket.chat/tools';
import type { MiddlewareHandler } from 'hono';

import { getRestPayload } from '../../../lib/logger/logPayloads';

export const loggerMiddleware =
	(logger: Logger): MiddlewareHandler =>
	async (c, next) => {
		const startTime = Date.now();

		const log = logger.logger.child(
			{
				method: c.req.method,
				url: censorUrl(c.req.url),
				userId: c.req.header('x-user-id'),
				userAgent: c.req.header('user-agent'),
				length: c.req.header('content-length'),
				host: c.req.header('host'),
				referer: c.req.header('referer'),
				remoteIP: c.get('remoteAddress'),
				...(['POST', 'PUT', 'PATCH', 'DELETE'].includes(c.req.method) && (await getRestPayload(c.req))),
			},
			{
				redact: [
					'payload.password', // Potentially logged by v1/login
				],
			},
		);

		await next();

		log.http({
			status: c.res.status,
			responseTime: Date.now() - startTime,
		});
	};