44 lines
1.1 KiB
JavaScript
44 lines
1.1 KiB
JavaScript
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
|
|
*/
|
|
|
|
export default function (options) {
|
|
options = options || {}
|
|
const 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')
|
|
|
|
const msgs = this.session.flash = this.session.flash || {}
|
|
if (type && msg) {
|
|
if (arguments.length > 2 && format) {
|
|
const 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) {
|
|
const arr = msgs[type]
|
|
delete msgs[type]
|
|
return arr || []
|
|
} else {
|
|
this.session.flash = {}
|
|
return msgs
|
|
}
|
|
}
|