mirror of
https://github.com/BobbyWibowo/lolisafe.git
synced 2024-12-13 16:06:21 +00:00
ea30e5dee5
also slight improvement to NunjucksRenderer class
68 lines
1.6 KiB
JavaScript
68 lines
1.6 KiB
JavaScript
const nunjucks = require('nunjucks')
|
|
|
|
class NunjucksRenderer {
|
|
directory
|
|
environment
|
|
|
|
#persistentCaches = new Map()
|
|
|
|
constructor (directory = '', options = {}) {
|
|
if (typeof directory !== 'string') {
|
|
throw new TypeError('Root directory must be a string value')
|
|
}
|
|
|
|
this.directory = directory
|
|
|
|
this.environment = nunjucks.configure(
|
|
this.directory,
|
|
Object.assign(options, {
|
|
autoescape: true
|
|
})
|
|
)
|
|
}
|
|
|
|
#middleware (req, res, next) {
|
|
// Inject render() method into Response on each requests
|
|
res.render = (...args) => this.#render(res, ...args)
|
|
return next()
|
|
}
|
|
|
|
#render (res, path, context, usePersistentCache = false) {
|
|
// If usePersistentCache is true, the rendered templates will be cached forever,
|
|
// thus only use for absolutely static pages
|
|
if (!path) {
|
|
throw new Error('Missing Nunjucks template name.')
|
|
}
|
|
|
|
return new Promise((resolve, reject) => {
|
|
const template = `${path}.njk`
|
|
|
|
if (usePersistentCache) {
|
|
const cached = this.#persistentCaches.get(template)
|
|
if (cached) {
|
|
return resolve(cached)
|
|
}
|
|
}
|
|
|
|
this.environment.render(template, context, (err, html) => {
|
|
if (err) {
|
|
return reject(err)
|
|
}
|
|
if (usePersistentCache) {
|
|
this.#persistentCaches.set(template, html)
|
|
}
|
|
resolve(html)
|
|
})
|
|
}).then(html => {
|
|
res.type('html').send(html)
|
|
return html
|
|
})
|
|
}
|
|
|
|
get middleware () {
|
|
return this.#middleware.bind(this)
|
|
}
|
|
}
|
|
|
|
module.exports = NunjucksRenderer
|