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

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

"use strict";

require("core-js/modules/es.object.to-string");

require("core-js/modules/es.promise");

require("regenerator-runtime/runtime");

function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }

function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }

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

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

    doIt(req, res, function () {
      resolve(1);
    });
  });
}

module.exports = function (compiler, opts) {
  var doIt = hotMiddleware(compiler, opts);

  function koaMiddleware(_x, _x2) {
    return _koaMiddleware.apply(this, arguments);
  }

  function _koaMiddleware() {
    _koaMiddleware = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee(ctx, next) {
      var req, res, runNext;
      return regeneratorRuntime.wrap(function _callee$(_context) {
        while (1) {
          switch (_context.prev = _context.next) {
            case 0:
              req = ctx.req, res = ctx.res;
              _context.next = 3;
              return middleware(doIt, req, res);

            case 3:
              runNext = _context.sent;

              if (!runNext) {
                _context.next = 7;
                break;
              }

              _context.next = 7;
              return next();

            case 7:
            case "end":
              return _context.stop();
          }
        }
      }, _callee);
    }));
    return _koaMiddleware.apply(this, arguments);
  }

  return koaMiddleware;
};