source: Dev/trunk/src/server/util/simple-http-proxy.js @ 531

/**
 * Module dependencies
 */
var url = require('url'),
    join = require('path').join,
    protocols = {
        http: require('http'),
        https: require('https')
    };

module.exports = function(endpoint, opts) {
    opts = opts || {};

    var parsedUrl = url.parse(endpoint);

    return function simpleHttpProxy(req, res, next) {
        // Get our forwarding info
        var hostInfo = req.headers.host.split(":");

        // Remove the host header
        delete req.headers.host;

        // Optionally delete cookie
        if (opts.cookies === false) {
            delete req.headers.cookie;
        }

        // Should we keep the trailing slash?
        var trailingSlash = req.originalUrl[req.originalUrl.length-1] === "/";

        // Setup the options
        var options = {
            hostname: parsedUrl.hostname,
            auth: parsedUrl.auth,
            port: parsedUrl.port,
            headers: req.headers,
            path: join(parsedUrl.pathname, trailingSlash ? req.url : req.url.substring(1)),
            method: req.method
        };

        // Enable forwarding headers
        if(opts.xforward) {
            // Get the path at which the middleware is mounted
            var resPath = req.originalUrl.replace(req.url, "");

            // We'll need to add a / if it's not on there
            if (resPath.indexOf("/") !== 0) {
                resPath = join("/", resPath);
            }

            // Pass along our headers
            options.headers[opts.xforward.proto || "x-forwarded-proto"] = req.connection.encrypted ? "https" : "http";
            options.headers[opts.xforward.host || "x-forwarded-host"] = hostInfo[0];
            options.headers[opts.xforward.path || "x-forwarded-path"] = resPath;

            if (hostInfo[1]) {
                options.headers[opts.xforward.port || "x-forwarded-port"] = hostInfo[1];
            }
        }

        /**
         * Hack for nginx backends where node, by default, sends no 'content-length'
         * header if no data is sent with the request and sets the 'transfer-encoding'
         * to 'chunked'. Nginx will then send a 411 because the body contains a '0'
         * which is the length of the chunk
         *
         *     GET /proxy
         *     transfer-encoding: chunked
         *     
         *     0
         *
         */
        if( ["POST", "DELETE"].indexOf(req.method) !== -1 && options.headers['transfer-encoding'] !== 'chunked') {
            options.headers['content-length'] = options.headers['content-length'] || '0';
        }

        // Make the request with the correct protocol
        var request = protocols[(parsedUrl.protocol || 'http').replace(":", "")].request(options, function(response) {

            // Send down the statusCode and headers
            res.writeHead(response.statusCode, response.headers);

            // Pipe the response
            response.pipe(res);
        });

        // Handle any timeouts that occur
        request.setTimeout(opts.timeout || 10000, function() {
            // Clean up the socket
            // TODO is there a better way to do this? There's a 'socket hang up' error being emitted...
            request.setSocketKeepAlive(false);
            request.socket.destroy();

            // Pass down the error
            var err = new Error("Proxy to '"+endpoint+"' timed out");
            request.emit("error", err);
        });

        // Pipe the client request upstream
        req.pipe(request);

        // Pass on our errors
        request.on('error', next);
    };
};