115 lines
3.1 KiB
JavaScript
115 lines
3.1 KiB
JavaScript
const { Middleware } = require('libflitter')
|
|
const moment = require('moment')
|
|
const uuid = require('uuid/v4')
|
|
|
|
class TrustManager {
|
|
assume_trust = false
|
|
|
|
constructor(request, response) {
|
|
this.request = request
|
|
this.response = response
|
|
this.init_store()
|
|
}
|
|
|
|
init_store() {
|
|
if ( !Array.isArray(this.request.session.trust_tokens) ) {
|
|
this.request.session.trust_tokens = []
|
|
}
|
|
|
|
const now = moment()
|
|
this.request.session.trust_tokens = this.request.session.trust_tokens.filter(x => {
|
|
return moment(new Date(x.expires)) > now
|
|
})
|
|
|
|
this.assume_trust = !!this.request.session.trust_assume_trust
|
|
}
|
|
|
|
assume() {
|
|
this.request.session.trust_assume_trust = true
|
|
this.assume_trust = true
|
|
}
|
|
|
|
unassume() {
|
|
this.request.session.trust_assume_trust = false
|
|
this.assume_trust = false
|
|
this.purge()
|
|
}
|
|
|
|
init_flow(scope, next) {
|
|
this.request.session.trust_flow = { scope, next, in_progress: false, requested: `${new Date}` }
|
|
}
|
|
|
|
has_flow() {
|
|
const flow = this.request.session?.trust_flow
|
|
return flow && flow.scope && flow.next && moment(new Date(flow.requested)) < moment(new Date(flow.requested)).add('20', 'minutes')
|
|
}
|
|
|
|
flow_scope() {
|
|
if ( this.has_flow() ) {
|
|
return this.request.session.trust_flow.scope
|
|
}
|
|
}
|
|
|
|
flow() {
|
|
if ( this.has_flow() ) {
|
|
return this.request.session.trust_flow.next
|
|
}
|
|
}
|
|
|
|
start() {
|
|
delete this.request.session.user_id
|
|
const grant_token = uuid()
|
|
this.request.session.trust_flow.grant_token = grant_token
|
|
this.request.session.trust_flow.in_progress = true
|
|
this.request.session.trust_flow.started = `${new Date}`
|
|
return grant_token
|
|
}
|
|
|
|
check_grant(grant_token) {
|
|
return grant_token === this.request.session?.trust_flow?.grant_token
|
|
}
|
|
|
|
end() {
|
|
const next = this.request.session?.trust_flow?.next
|
|
delete this.request.session.trust_flow
|
|
return next
|
|
}
|
|
|
|
in_progress() {
|
|
const flow = this.request.session.trust_flow
|
|
return flow && flow.in_progress && flow.started && moment(new Date(flow.started)) < moment(new Date(flow.started)).add('10', 'minutes')
|
|
}
|
|
|
|
has(scope) {
|
|
return this.assume_trust || this.request.session.trust_tokens.some(x => x.scope === scope)
|
|
}
|
|
|
|
grant(scope) {
|
|
this.request.session.trust_tokens.push({
|
|
scope,
|
|
expires: moment().add('1', 'hour').toDate().toString(),
|
|
})
|
|
}
|
|
|
|
deplete(scope) {
|
|
this.request.session.trust_tokens = this.request.session.trust_tokens.filter(x => x.scope !== scope)
|
|
}
|
|
|
|
purge() {
|
|
this.end()
|
|
this.request.session.trust_tokens = []
|
|
}
|
|
}
|
|
|
|
class TrustTokenUtilityMiddleware extends Middleware {
|
|
async test(req, res, next, args = {}) {
|
|
if ( req.session ) {
|
|
req.trust = new TrustManager(req, res)
|
|
}
|
|
|
|
return next()
|
|
}
|
|
}
|
|
|
|
module.exports = exports = TrustTokenUtilityMiddleware
|