lastleaf / node-fastcgi-client Goto Github PK

const path = require('path');

const fastCGIClient = require('fastcgi-client');

const defaultOptions = {
    host: '127.0.0.1',
    port: 9000,
    documentRoot: path.dirname(require.main.filename || '.'),
    skipCheckServer: true,
};

module.exports = function(options = {}, input_fastcgi_params = {}) {

    options = Object.assign({}, defaultOptions, options);
    const fpm = new Promise((resolve, reject) => {

        const loader = fastCGIClient(options);
        loader.on('ready', () => resolve(loader));
        loader.on('error', reject);
    });

    return async function(req, res, next) {

        // 以 /api/node 开始的请求全部由 express 处理
        if (req.url.startsWith('/api/node')) {
            return next();
        }
        
        let [_, ...queryString] = req.url.split('?');
        queryString = queryString ? queryString[0] : '';

        let fastcgi_params = Object.assign({}, input_fastcgi_params, {
            REQUEST_URI: `/index.php${req.url}`,
            DOCUMENT_URI: '/index.php',
            SCRIPT_NAME: '/index.php',
            QUERY_STRING: queryString,
            REQUEST_METHOD: req.method,
            CONTENT_TYPE: req.headers['content-type'],
            CONTENT_LENGTH: req.headers['content-length'],
            SERVER_PROTOCOL: 'HTTP/1.1',
            REQUEST_SCHEME: req.protocol,
            HTTPS: req.protocol === 'https' ? 'on': undefined,
            GATEWAY_INTERFACE: 'CGI/1.1',
            SERVER_SOFTWARE: 'PHP-FPM FOR Node',
            REMOTE_ADDR: req.connection.remoteAddress,
            REMOTE_PORT:  req.connection.remotePort,
            SERVER_NAME: req.connection.domain,
            REDIRECT_STATUS: 200,

            // 下面的由外部传入
            // SERVER_ADDR: 127.0.0.1,
            // SERVER_PORT: 8080,
            // DOCUMENT_ROOT: /var/www/public
            // SCRIPT_FILENAME: /var/www/public/index.php
        });

         if (! fastcgi_params.REQUEST_URI || ! fastcgi_params.REQUEST_URI.startsWith('/')) {
             throw new Error('Invalid URI');
         }

         for (const fastcgi_param in fastcgi_params) {
             if (typeof fastcgi_params[fastcgi_param] === 'undefined') {
                 delete fastcgi_params[fastcgi_param];
             }
         }

         console.log(fastcgi_params);

         const phpFPM = await fpm;

         return new Promise(function(resolve, reject) {
             
            phpFPM.request(fastcgi_params, function(err, request) {
                
                if (err) return reject(err);

                var output = '';
                var errors = '';

                // console.log(req.body);
                // console.log(req.files.file);

                req.on('end', function(){
                    console.log('没有数据了');
                });

                req.on('data', function(chunk) {
                    // console.log(chunk.toString());
                    console.log(`接收到 ${chunk.length} 个字节的数据`);
                    request.stdin.write(chunk, null, null);
                });

                // request.stdin.on('pipe', function(src) {
                //     console.error('有数据正通过管道流入写入器');
                // });

                request.stdout.on('data', function(data) {
                    output += data.toString('utf-8');
                });

                request.stderr.on('data', function(data) {
                    errors += data.toString('utf-8');
                });

                request.stdout.on('end', function() {
                    
                    if (errors) {
                        return reject(new Error(errors));
                    }

                    const head = output.match(/^[\s\S]*?\r\n\r\n/)[0];
                    const parseHead = head.split('\r\n').filter(_ => _);
                    const responseHeaders = {};
                    let statusCode = 200;
                    let statusMessage = '';
          
                    for (const item of parseHead) {
                        const pair = item.split(': ');
                        
                        if (pair.length > 1 && pair[0] && pair[1]) {
                            if (pair[0] in responseHeaders) {
                                responseHeaders[pair[0]].push(pair[1]);
                            } else {
                                responseHeaders[pair[0]] = [pair[1]];
                            }

                            if (pair[0] === 'Status') {
                                const match = pair[1].match(/(\d+) (.*)/);
                                statusCode = parseInt(match[1]);
                                statusMessage = match[2];
                            }
                        }
                    }
                    res.writeHead(statusCode, statusMessage, responseHeaders);
                    const body = output.slice(head.length);
                    res.write(body);
                    res.end();
                    resolve({ fastcgi_params, body });
                });
            });
         });
    };
};