const util = require('util') const format = util.format /* * Clean version of https://github.com/jaredhanson/connect-flash * Included here to avoid includng ridiculously small modules */ module.exports = function (options) { options = options || {} let safe = (options.unsafe === undefined) ? true : !options.unsafe return function (req, res, next) { if (req.flash && safe) { return next() } req.flash = _flash next() } } function _flash (type, msg) { if (this.session === undefined) throw Error('req.flash() requires sessions') let msgs = this.session.flash = this.session.flash || {} if (type && msg) { if (arguments.length > 2 && format) { let args = Array.prototype.slice.call(arguments, 1) msg = format.apply(undefined, args) } else if (Array.isArray(msg)) { msg.forEach((val) => { (msgs[type] = msgs[type] || []).push(val) }) return msgs[type].length } return (msgs[type] = msgs[type] || []).push(msg) } else if (type) { let arr = msgs[type] delete msgs[type] return arr || [] } else { this.session.flash = {} return msgs } }