-
Notifications
You must be signed in to change notification settings - Fork 2
/
index.js
50 lines (44 loc) · 1.58 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
// Dependencies
const path = require("path");
const ejs = require("ejs");
// Export
module.exports = function polkaEjs({
// Global options that will be used in all calls
// unless the render call overrides an option then that one is used
ext: globalExt = "ejs",
cache: globalCache = process.env.NODE_ENV === "production",
views: globalViews = path.join(process.cwd(), "views"),
contentType: globalContentType = "text/html",
status: globalStatus = 200,
options: globalOptions = {},
} = {}) {
// The actual middleware that will be passed to app.use()
return function middleware(req, res, next) {
// Define the res.render function on response object
res.render = function render(file, locals = {}, {
// Per call options that overrides the global call
cache = globalCache,
status = globalStatus,
contentType = globalContentType,
ext = globalExt,
views = globalViews,
options = globalOptions
} = {}) {
// Add cache option
options.cache = cache;
// Validate and process input
if (isNaN(status)) throw new Error("Status must be a number.");
if (typeof file !== "string") throw new Error("File must be a string.");
if (!file.endsWith(`.${ext}`)) file = file + `.${ext}`;
// Render!
return ejs.renderFile(path.join(views, file), locals, options, (err, html) => {
if (err) throw err; // Handle error
// Send the response
res.writeHead(status, { "Content-Type": contentType });
return res.end(html);
});
};
// We are done.
return next();
};
};