webpack-dev-middleware-for-koa/dist/index.js

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

"use strict";

require("core-js/modules/es.array.for-each");

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

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

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

require("core-js/modules/web.dom-collections.for-each");

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 devMiddleware = require("webpack-dev-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, option) {
  var doIt = devMiddleware(compiler, option);

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

  function _koaMiddleware() {
    _koaMiddleware = _asyncToGenerator( /*#__PURE__*/regeneratorRuntime.mark(function _callee(ctx, next) {
      var req, locals, runNext;
      return regeneratorRuntime.wrap(function _callee$(_context) {
        while (1) {
          switch (_context.prev = _context.next) {
            case 0:
              req = ctx.req;
              locals = ctx.locals || ctx.state;
              ctx.webpack = doIt;
              _context.next = 5;
              return middleware(doIt, req, {
                end: function end(content) {
                  ctx.body = content;
                },
                locals: locals,
                setHeader: function setHeader() {
                  ctx.set.apply(ctx, arguments);
                }
              });

            case 5:
              runNext = _context.sent;

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

              _context.next = 9;
              return next();

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

  Object.keys(doIt).forEach(function (p) {
    koaMiddleware[p] = doIt[p];
  });
  return koaMiddleware;
};