import { IncomingMessage } from '../../types';
import { RootHandler } from './RootHandler';

export const augmentRequest = (req: IncomingMessage, rootHandler: RootHandler) => {
    req.context = rootHandler.context;
    req.host = getHost(req);
    req.hostname = getHostname(req);
};

const getHost = (req: IncomingMessage): string => {
    return req.headers['host'];
};

const getHostname = (req: IncomingMessage) => {
    return getHost(req).split(':')[0];
};