import { Request, Response, NextFunction } from "express"
import Logger from "@lib/Logger/Logger";
import { RequestData } from "@lib/API/API";
const logger = new Logger('API', 'Middleware', 'MountIP');

// 挂载IP，将请求的ip地址挂载到data._ip属性下
let MountIP = (req: Request, res: Response, next: NextFunction) => {
    let ip = req.headers['x-forwarded-for'] || req.connection.remoteAddress || req.ip;
    if (ip == undefined || ip.length <= 0) {
        logger.warn(`[${req.method}][${req.url.split('?')[0]}] ip resolution was fail`);
    } else {
        if (typeof ip === 'object')
            req.body._ip = ip.join(' ').replace('::ffff:', '');
        else
            req.body._ip = ip.replace('::ffff:', '');
        logger.info(`[${req.method}][${req.url.split('?')[0]}] ip resolution was successful: ${req.body._ip}`);
    }
    next();
}

export default MountIP;


interface MountIPRequestData extends RequestData {
    _ip: string;
} 

export type { MountIPRequestData };