Commit d0d44966 authored by Vadym Gidulian's avatar Vadym Gidulian

Closed #3 (Add Pug support)

parent 2ddafb83
......@@ -4,6 +4,7 @@ const fs = require('fs');
const ejs = require('ejs');
const glob = require('glob');
const handlebars = require('handlebars');
const pug = require('pug');
const TEMPLATES_DIR = '/app/templates';
......@@ -33,13 +34,16 @@ module.exports = {
compile(templateName, vars) {
const [, ext] = templateName.split('.');
const template = TEMPLATES.get(templateName) || '';
const templatePath = `${TEMPLATES_DIR}/${templateName}`;
switch (ext) {
case 'ejs':
return ejs.render(template, vars, {filename: `${TEMPLATES_DIR}/${templateName}`});
return ejs.render(template, vars, {filename: templatePath});
case 'hbs':
case 'handlebars':
return handlebars.compile(template)(vars);
case 'pug':
return pug.compile(template, {basedir: TEMPLATES_DIR, filename: templatePath})(vars);
default:
return template;
}
......
This diff is collapsed.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment