feat: 模板 base64 filter

surgioproject · Aug 25, 2019 · 3d45a23 · 3d45a23
1 parent 2f2d340
commit 3d45a23
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/lib/template.spec.ts b/lib/template.spec.ts
@@ -43,3 +43,14 @@ test('renderString #4', t => {
 
   t.is(result, '# Comment');
 });
+
+test('base64', t => {
+  const body = `{{ str | base64 }}`;
+  const str = `testtesttesttest`;
+
+  const result = templateEngine.renderString(body, {
+    str,
+  });
+
+  t.is(result, 'dGVzdHRlc3R0ZXN0dGVzdA==');
+});
diff --git a/lib/template.ts b/lib/template.ts
@@ -1,6 +1,7 @@
 import nunjucks from 'nunjucks';
 import { JsonObject } from 'type-fest';
 import YAML from 'yaml';
+import { toBase64 } from './utils';
 
 export default function getEngine(templateDir: string): nunjucks.Environment {
   const engine = nunjucks.configure(templateDir, {
@@ -28,8 +29,10 @@ export default function getEngine(templateDir: string): nunjucks.Environment {
       .join('\n');
   });
 
-// istanbul ignore next
+  // istanbul ignore next
   engine.addFilter('yaml', (obj: JsonObject) => YAML.stringify(obj));
 
+  engine.addFilter('base64', (str: string) => toBase64(str));
+
   return engine;
 };