webpack-hot-middleware-for-koa2/src/index.js

用于koa2的webpack-hot-middleware的中间件

"use strict";

const hotMiddleware = require("webpack-hot-middleware");

function middleware(doIt, req, res) {
  const { end: originalEnd } = res;

  return new Promise((resolve) => {
    res.end = function end() {
      originalEnd.apply(this, arguments);
      resolve(0);
    };
    doIt(req, res, () => {
      resolve(1);
    });
  });
}

module.exports = (compiler, opts) => {
  const doIt = hotMiddleware(compiler, opts);

  async function koaMiddleware(ctx, next) {
    const { req, res } = ctx;

    const runNext = await middleware(doIt, req, res);

    if (runNext) {
      await next();
    }
  }

  return koaMiddleware;
};