85 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			85 lines
		
	
	
		
			3.0 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
"use strict";
 | 
						|
Object.defineProperty(exports, "__esModule", { value: true });
 | 
						|
exports.getHandlers = exports.init = void 0;
 | 
						|
const logger_1 = require("./logger");
 | 
						|
const logger = (0, logger_1.getInstance)();
 | 
						|
function init(proxy, option) {
 | 
						|
    const handlers = getHandlers(option);
 | 
						|
    for (const eventName of Object.keys(handlers)) {
 | 
						|
        proxy.on(eventName, handlers[eventName]);
 | 
						|
    }
 | 
						|
    // https://github.com/webpack/webpack-dev-server/issues/1642
 | 
						|
    proxy.on('econnreset', (error, req, res, target) => {
 | 
						|
        logger.error(`[HPM] ECONNRESET: %O`, error);
 | 
						|
    });
 | 
						|
    // https://github.com/webpack/webpack-dev-server/issues/1642#issuecomment-1104325120
 | 
						|
    proxy.on('proxyReqWs', (proxyReq, req, socket, options, head) => {
 | 
						|
        socket.on('error', (error) => {
 | 
						|
            logger.error(`[HPM] WebSocket error: %O`, error);
 | 
						|
        });
 | 
						|
    });
 | 
						|
    logger.debug('[HPM] Subscribed to http-proxy events:', Object.keys(handlers));
 | 
						|
}
 | 
						|
exports.init = init;
 | 
						|
function getHandlers(options) {
 | 
						|
    // https://github.com/nodejitsu/node-http-proxy#listening-for-proxy-events
 | 
						|
    const proxyEventsMap = {
 | 
						|
        error: 'onError',
 | 
						|
        proxyReq: 'onProxyReq',
 | 
						|
        proxyReqWs: 'onProxyReqWs',
 | 
						|
        proxyRes: 'onProxyRes',
 | 
						|
        open: 'onOpen',
 | 
						|
        close: 'onClose',
 | 
						|
    };
 | 
						|
    const handlers = {};
 | 
						|
    for (const [eventName, onEventName] of Object.entries(proxyEventsMap)) {
 | 
						|
        // all handlers for the http-proxy events are prefixed with 'on'.
 | 
						|
        // loop through options and try to find these handlers
 | 
						|
        // and add them to the handlers object for subscription in init().
 | 
						|
        const fnHandler = options ? options[onEventName] : null;
 | 
						|
        if (typeof fnHandler === 'function') {
 | 
						|
            handlers[eventName] = fnHandler;
 | 
						|
        }
 | 
						|
    }
 | 
						|
    // add default error handler in absence of error handler
 | 
						|
    if (typeof handlers.error !== 'function') {
 | 
						|
        handlers.error = defaultErrorHandler;
 | 
						|
    }
 | 
						|
    // add default close handler in absence of close handler
 | 
						|
    if (typeof handlers.close !== 'function') {
 | 
						|
        handlers.close = logClose;
 | 
						|
    }
 | 
						|
    return handlers;
 | 
						|
}
 | 
						|
exports.getHandlers = getHandlers;
 | 
						|
function defaultErrorHandler(err, req, res) {
 | 
						|
    // Re-throw error. Not recoverable since req & res are empty.
 | 
						|
    if (!req && !res) {
 | 
						|
        throw err; // "Error: Must provide a proper URL as target"
 | 
						|
    }
 | 
						|
    const host = req.headers && req.headers.host;
 | 
						|
    const code = err.code;
 | 
						|
    if (res.writeHead && !res.headersSent) {
 | 
						|
        if (/HPE_INVALID/.test(code)) {
 | 
						|
            res.writeHead(502);
 | 
						|
        }
 | 
						|
        else {
 | 
						|
            switch (code) {
 | 
						|
                case 'ECONNRESET':
 | 
						|
                case 'ENOTFOUND':
 | 
						|
                case 'ECONNREFUSED':
 | 
						|
                case 'ETIMEDOUT':
 | 
						|
                    res.writeHead(504);
 | 
						|
                    break;
 | 
						|
                default:
 | 
						|
                    res.writeHead(500);
 | 
						|
            }
 | 
						|
        }
 | 
						|
    }
 | 
						|
    res.end(`Error occurred while trying to proxy: ${host}${req.url}`);
 | 
						|
}
 | 
						|
function logClose(req, socket, head) {
 | 
						|
    // view disconnected websocket connections
 | 
						|
    logger.info('[HPM] Client disconnected');
 | 
						|
}
 |