feat: 增加 mellow 规则处理方法

surgioproject · Nov 13, 2019 · b646199 · b646199
1 parent bef00c7
commit b646199
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 0 deletions.
diff --git a/lib/template.ts b/lib/template.ts
@@ -42,6 +42,27 @@ export default function getEngine(templateDir: string): nunjucks.Environment {
       .join('\n');
   });
 
+  engine.addFilter('mellow', (str: string) => {
+    const array = str.split('\n');
+
+    return array
+      .filter(item => {
+        return item && item.trim() !== '' &&
+          item.toUpperCase().indexOf('URL-REGEX') === -1 &&
+          item.toUpperCase().indexOf('USER-AGENT') === -1;
+      })
+      .map((item: string) => {
+        if (item.startsWith('#')) {
+          return item;
+        }
+        return item
+          .replace(/,no-resolve/, '')
+          .replace(/\/\/.*$/, '')
+          .trim();
+      })
+      .join('\n');
+  });
+
   // istanbul ignore next
   engine.addFilter('yaml', (obj: JsonObject) => YAML.stringify(obj));
 

diff --git a/test/template.test.ts b/test/template.test.ts
@@ -84,3 +84,23 @@ test('quantumultx filter', t => {
 
   t.is(result, '');
 });
+
+test('mellow filter 1', t => {
+  const body = `{{ str | mellow }}`;
+  const str = `IP-CIDR,67.198.55.0/24,Proxy,no-resolve // test rule`;
+  const result = templateEngine.renderString(body, {
+    str,
+  });
+
+  t.is(result, 'IP-CIDR,67.198.55.0/24,Proxy');
+});
+
+test('mellow filter 2', t => {
+  const body = `{{ str | mellow }}`;
+  const str = `URL-REGEX,xxxxxxxxxxxx`;
+  const result = templateEngine.renderString(body, {
+    str,
+  });
+
+  t.is(result, '');
+});