myblog/node_modules/hexo-server/lib/middlewares/route.js

'use strict';

var pathFn = require('path');
var mime = require('mime');

module.exports = function(app) {
  var config = this.config;
  var args = this.env.args || {};
  var root = config.root;
  var route = this.route;

  if (args.s || args.static) return;

  app.use(root, function(req, res, next) {
    var method = req.method;
    if (method !== 'GET' && method !== 'HEAD') return next();

    var url = route.format(decodeURIComponent(req.url));
    var data = route.get(url);
    var extname = pathFn.extname(url);

    // When the URL is `foo/index.html` but users access `foo`, redirect to `foo/`.
    if (!data) {
      if (extname) return next();

      url = encodeURI(url);
      res.statusCode = 302;
      res.setHeader('Location', root + url + '/');
      res.end('Redirecting');
      return;
    }

    res.setHeader('Content-Type', extname ? mime.lookup(extname) : 'application/octet-stream');

    if (method === 'GET') {
      data.pipe(res).on('error', next);
    } else {
      res.end();
    }
  });
};