Misc: add meta generator injection filter

hexojs · Apr 20, 2018 · 017d355 · 017d355
1 parent 373b9c7
commit 017d355
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/lib/plugins/filter/index.js b/lib/plugins/filter/index.js
@@ -11,4 +11,5 @@ module.exports = ctx => {
 
   filter.register('new_post_path', require('./new_post_path'));
   filter.register('post_permalink', require('./post_permalink'));
+  filter.register('after_render:html', require('./meta_generator'));
 };
diff --git a/lib/plugins/filter/meta_generator.js b/lib/plugins/filter/meta_generator.js
@@ -0,0 +1,18 @@
+'use strict';
+
+const rMetaGenerator = /<meta\s+name=['|"]?generator['|"]?/i;
+const hexoGeneratorTag = '<meta name="generator" content="Hexo %s" />';
+let cheerio;
+
+function hexoMetaGeneratorInject(data) {
+  if (!rMetaGenerator.test(data)) {
+    if (!cheerio) cheerio = require('cheerio');
+    const $ = cheerio.load(data, {decodeEntities: false});
+
+    $('head').prepend(hexoGeneratorTag.replace('%s', this.version));
+
+    return $.html();
+  }
+}
+
+module.exports = hexoMetaGeneratorInject;