ufe/lib/http-server/index.js

var fs = require('fs'),
  path = require('path'),
  union = require('union'),
  ecstatic = require('ecstatic'),
  httpProxy = require('http-proxy'),
  corser = require('corser'),
  colors = require('colors'),
  logger = require('../util/logger'),
  util = require('../util/util'),
  php_fpm = require('./lib/php-fpm'),
  contentTypesMap = require('./lib/contentType'),
  findSimilarFile = require('./lib/findSimilarFile');

exports.HttpServer = exports.HTTPServer = HttpServer;

exports.createServer = function (options) {
  return new HttpServer(options);
}

function HttpServer(options) {
  options = options || {};
  this.root = options['root']
  this.headers = options.headers || {}
  this.contentType = options.contentType || 'application/octet-stream'
  // 允许跨域
  this.headers['Access-Control-Allow-Origin'] = '*';
  this.headers['Access-Control-Allow-Credentials'] = true;
  this.headers['Access-Control-Allow-Headers'] = 'Origin, X-Requested-With, Content-Type, Accept, Range';

  var before = options.before ? options.before.slice() : []

  //记录访问日志
  before.push(function (req, res, next) {
    var err
    try {
      logger.http(req, res);
    } catch(ex) {
      ex.status = 503
      err = ex
    }
    next(err)
  })

  //调用本地 php-fpm 解析 php 文件
  if(options['php']){
    var php = new php_fpm({
      documentRoot: path.resolve(process.cwd(), this.root)
    });
    before.push(function (req, res, next) {
      var err;
      if(util.getFileExtension(req.url) == 'php') {
        try {
          php.run(req, function(phpParseErr, output){
            if(output && output.indexOf('<b>Parse error</b>') > -1){
              if(!phpParseErr){
                phpParseErr = {}
              }
              if(!phpParseErr['stack']){
                phpParseErr['stack'] = output
              }
            }
      			if(!phpParseErr){
              res.setHeader('Content-Type', 'text/html');
              res.write(output, "utf-8");
              return res.end()
            } else {
              phpParseErr.status = 503
              next(phpParseErr)
            }
      		})
        } catch(ex) {
          ex.status = 503
          err = ex
          next(err)
        }
      } else {
        next(err)
      }
    })
  }

  // favicon 图标处理.
  before.push(function (req, res) {
    if (req.url === '/favicon.ico') {
      var icoPath = path.resolve( __dirname, './assets/images/favicon.ico' )
      res.setHeader('Content-Type', 'image/x-icon');
      res.write(fs.readFileSync(icoPath, "binary"), "binary");
      return res.end()
    }
    res.emit('next');
  });

  //处理静态资源.
  before.push(ecstatic({
    root: this.root,
    cache: 1,
    showDir: true,
    autoIndex: true,
    gzip: false,
    contentType: this.contentType,
    handleError: typeof options.proxy !== 'string'
  }));

  //使用代理服务器
  if (options.proxy && typeof options.proxy === 'string') {
    var proxy = httpProxy.createProxyServer({});
    proxy.on('error', function (err, req, res){})
    proxy.on('proxyRes', function (proxyRes, req, res) {
      console.log('proxy ', res['statusCode'], req.url)
    })
    before.push(function (req, res) {
      proxy.web(req, res, {
        target: options.proxy,
        changeOrigin: true
      })
    })
  }

  //处理一部分特殊扩展名，以指定的内容类型响应。
  // before.push(function (req, res) {
  //   let fileExt = util.getFileExtension(req.url)
  //   if(contentTypesMap[fileExt]){
  //     console.log(contentTypesMap[fileExt])
  //     res.setHeader('Content-Type', contentTypesMap[fileExt]);
  //   }
  //   res.emit('next');
  // })

  //创建内部的 union server
  var unionOptions = {
    before: before,
    headers: this.headers,
    onError: function (err, req, res) {
      if(err.status == 404){
        var cwd = process.cwd()
        var filePath = cwd + req.url;
        var similarFile = findSimilarFile(filePath)
        if(similarFile){
          res.writeHead(200, {})  //重置响应状态码
          let fileContent = fs.readFileSync(similarFile['pathName'], 'utf-8');
          res.write(fileContent, "utf-8");
          err['transfer-url'] = similarFile['pathName'].replace(cwd, '')
          logger.http(req, res, err)
          return res.end()
        } else {
          logger.http(req, res, err)
          return require('./lib/page/index.js')(404, req, res, err);
        }
      }
      logger.http(req, res, err)
      return require('./lib/page/index.js')(503, req, res, err);
    }
  }
  if (options.ssl) {
    unionOptions.https = {
      cert: path.resolve(__dirname, './assets/pem/cert.pem'),
      key: path.resolve(__dirname, './assets/pem/key.pem')
    }
  }
  this.server = union.createServer(unionOptions)
}

HttpServer.prototype.listen = function () {
  this.server.listen.apply(this.server, arguments)
}

HttpServer.prototype.close = function () {
  return this.server.close()
}