You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.6 KiB
TypeScript
75 lines
2.6 KiB
TypeScript
import fs from 'fs';
|
|
import path from 'path';
|
|
import serialize from 'serialize-javascript';
|
|
import nunjucks from 'nunjucks';
|
|
import * as markdown from './markdown';
|
|
import { STATUS, PERM, PRIV } from '../model/builtin';
|
|
import * as misc from './misc';
|
|
|
|
class Loader extends nunjucks.Loader {
|
|
// eslint-disable-next-line class-methods-use-this
|
|
getSource(name: string) {
|
|
if (!process.env.debug) {
|
|
if (!global.Hydro.template[name]) throw new Error(`Cannot get template ${name}`);
|
|
return {
|
|
src: global.Hydro.template[name],
|
|
path: name,
|
|
noCache: false,
|
|
};
|
|
}
|
|
let fullpath = null;
|
|
const base = path.join(process.cwd(), 'templates');
|
|
const p = path.resolve(base, name);
|
|
if (fs.existsSync(p)) fullpath = p;
|
|
if (!fullpath) throw new Error(`Cannot get template ${name}`);
|
|
return {
|
|
src: fs.readFileSync(fullpath, 'utf-8'),
|
|
path: fullpath,
|
|
noCache: true,
|
|
};
|
|
}
|
|
}
|
|
|
|
class Nunjucks extends nunjucks.Environment {
|
|
constructor() {
|
|
super(new Loader(), { autoescape: true, trimBlocks: true });
|
|
this.addFilter('json', (self) => JSON.stringify(self), false);
|
|
this.addFilter('serialize', (self, ignoreFunction = true) => serialize(self, { ignoreFunction }));
|
|
this.addFilter('assign', (self, data) => Object.assign(self, data));
|
|
this.addFilter('markdown', (self, safe = true) => {
|
|
if (safe) return markdown.safe.render(self);
|
|
return markdown.unsafe.render(self);
|
|
});
|
|
this.addFilter('ansi', (self) => misc.ansiToHtml(self));
|
|
this.addFilter('base64_encode', (s) => Buffer.from(s).toString('base64'));
|
|
this.addFilter('bitand', (self, val) => self & val);
|
|
}
|
|
}
|
|
const env = new Nunjucks();
|
|
|
|
export function render(name: string, state: any): Promise<string> {
|
|
return new Promise((resolve, reject) => {
|
|
env.render(name, {
|
|
page_name: name.split('.')[0],
|
|
...state,
|
|
typeof: (o) => typeof o,
|
|
static_url: (str) => `/${str}`,
|
|
datetimeSpan: misc.datetimeSpan,
|
|
paginate: misc.paginate,
|
|
perm: PERM,
|
|
PRIV,
|
|
STATUS,
|
|
size: misc.size,
|
|
gravatar: misc.gravatar,
|
|
model: global.Hydro.model,
|
|
Context: global.Hydro.ui,
|
|
isIE: (str: string) => str.includes('MSIE') || str.includes('rv:11.0'),
|
|
}, (err, res) => {
|
|
if (err) reject(err);
|
|
else resolve(res);
|
|
});
|
|
});
|
|
}
|
|
|
|
global.Hydro.lib.template = { render };
|