auth-center/app/middlewares/flash.js

auth center with TOTP

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = _default;

function _default(app) {
  const key = 'messages';
  app.use(async function flashHandler(ctx, next) {
    ctx.state[key] = ctx.session[key] || {};
    delete ctx.session[key];

    ctx.flash = function (type, msg) {
      ctx.session[key] = ctx.session[key] || {};
      ctx.session[key][type] = msg;
    };

    await next();

    if (ctx.status === 302 && ctx.session && !ctx.session[key]) {
      ctx.session[key] = ctx.state[key];
    }
  });
}

module.exports = exports.default;