ysjj-system/node_modules/egg-static/app/middleware/static.js

'use strict';

const range = require('koa-range');
const compose = require('koa-compose');
const staticCache = require('koa-static-cache');
const assert = require('assert');
const mkdirp = require('mkdirp');
const LRU = require('ylru');
const is = require('is-type-of');

module.exports = (options, app) => {
  let dirs = options.dir;
  if (!is.array(dirs)) dirs = [ dirs ];

  const prefixs = [];

  function rangeMiddleware(ctx, next) {
    // if match static file, and use range middleware.
    const isMatch = prefixs.some(p => ctx.path.startsWith(p));
    if (isMatch) {
      return range(ctx, next);
    }
    return next();
  }

  const middlewares = [ rangeMiddleware ];

  for (const dirObj of dirs) {
    assert(is.object(dirObj) || is.string(dirObj), '`config.static.dir` must be `string | Array<string|object>`.');

    let newOptions;

    if (is.string(dirObj)) {
      // copy origin options to new options ensure the safety of objects
      newOptions = Object.assign({}, options, { dir: dirObj });
    } else {
      assert(is.string(dirObj.dir), '`config.static.dir` should contains `[].dir` property when object style.');
      newOptions = Object.assign({}, options, dirObj);
    }

    if (newOptions.dynamic && !newOptions.files) {
      newOptions.files = new LRU(newOptions.maxFiles);
    }

    if (newOptions.prefix) {
      prefixs.push(newOptions.prefix);
    }

    // ensure directory exists
    mkdirp.sync(newOptions.dir);

    app.loggers.coreLogger.info('[egg-static] starting static serve %s -> %s', newOptions.prefix, newOptions.dir);

    middlewares.push(staticCache(newOptions));
  }

  return compose(middlewares);
};