From ca99e5de86c8c2937747eea3d94d208194ba72e9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B2=88=E5=AE=87?= <448766534@qq.com>
Date: Tue, 7 Jan 2020 14:05:02 +0800
Subject: [PATCH] feat: WordCount

---
 _config.yml                          |  8 ++++++
 layout/_partial/article.ejs          | 11 +++++---
 layout/_partial/post/word.ejs        | 22 ++++++++++++++++
 scripts/default_config.js            |  8 ++++++
 scripts/filters/index.js             | 13 ++++++++++
 scripts/filters/meta_generator.js    | 17 ++++++++++++
 scripts/helpers/ayer-plus-vendors.js |  9 +++++++
 scripts/helpers/wordcount.js         | 39 ++++++++++++++++++++++++++++
 scripts/lib/core.js                  | 25 ++++++++++++++++++
 9 files changed, 148 insertions(+), 4 deletions(-)
 create mode 100644 layout/_partial/post/word.ejs
 create mode 100644 scripts/default_config.js
 create mode 100644 scripts/filters/index.js
 create mode 100644 scripts/filters/meta_generator.js
 create mode 100644 scripts/helpers/ayer-plus-vendors.js
 create mode 100644 scripts/helpers/wordcount.js
 create mode 100644 scripts/lib/core.js

diff --git a/_config.yml b/_config.yml
index df276bc..2737c7a 100644
--- a/_config.yml
+++ b/_config.yml
@@ -36,6 +36,14 @@ nav_text:
 # 文章页是否显示目录
 toc: true
 
+# https://github.com/willin/hexo-wordcount
+# 是否开启字数统计(关闭请设置enable为false)
+# 也可以单独在md文件里Front-matter设置`no_word_count: true`属性，来自定义关闭字数统计
+word_count:
+  enable: true
+  # 只在文章详情显示(不在首页显示)
+  only_article_visit: true
+
 # 打赏
 # 打赏type设定：0-关闭打赏； 1-文章对应的md文件里有reward:true属性，才有打赏； 2-所有文章均有打赏
 reward_type: 2
diff --git a/layout/_partial/article.ejs b/layout/_partial/article.ejs
index cb0d557..b565dc4 100644
--- a/layout/_partial/article.ejs
+++ b/layout/_partial/article.ejs
@@ -12,6 +12,9 @@
     <div class="article-meta">
       <%- partial('post/date', {class_name: 'article-date', date_format: null}) %>
       <%- partial('post/category') %>
+      <% if(theme.word_count && theme.word_count.enable && !post.no_word_count){%>
+      <%- partial('post/word') %>
+      <% } %>
     </div>
     <% } %>
 
@@ -32,10 +35,10 @@
       <% } %>
       <!-- 打赏 -->
       <% if ((theme.reward_type === 2 || (theme.reward_type === 1 && post.reward)) && !index){ %>
-        <div id="reward-btn">
-          打赏
-        </div>
-        <% } %>
+      <div id="reward-btn">
+        打赏
+      </div>
+      <% } %>
     </div>
     <footer class="article-footer">
       <a data-url="<%- post.permalink %>" data-id="<%= post._id %>"
diff --git a/layout/_partial/post/word.ejs b/layout/_partial/post/word.ejs
new file mode 100644
index 0000000..b9da616
--- /dev/null
+++ b/layout/_partial/post/word.ejs
@@ -0,0 +1,22 @@
+<% if (!theme.word_count.only_article_visit || (!index && theme.word_count.only_article_visit)){ %>
+<div style="margin-top:10px;">
+    <span class="post-time">
+        <span class="post-meta-item-icon">
+            <!-- fonts.scss -->
+            <!-- 百度字体平台:http://fontstore.baidu.com/static/editor/index.html -->
+            <i class="icon-statistics"></i>
+            <span class="post-meta-item-text"> 字数统计:</span>
+            <span class="post-count"><%= wordcount(post.content) %>字</span>
+        </span>
+    </span>
+
+    <span class="post-time">
+        &nbsp; | &nbsp;
+        <span class="post-meta-item-icon">
+            <i class="icon-book icon"></i>
+            <span class="post-meta-item-text"> 阅读时长≈</span>
+            <span class="post-count"><%= min2read(post.content) %>分</span>
+        </span>
+    </span>
+</div>
+<% } %>
\ No newline at end of file
diff --git a/scripts/default_config.js b/scripts/default_config.js
new file mode 100644
index 0000000..924c2c7
--- /dev/null
+++ b/scripts/default_config.js
@@ -0,0 +1,8 @@
+'use strict';
+
+module.exports = {
+  
+  // 生成meta `generator`
+  meta_generator: true
+
+};
diff --git a/scripts/filters/index.js b/scripts/filters/index.js
new file mode 100644
index 0000000..a556544
--- /dev/null
+++ b/scripts/filters/index.js
@@ -0,0 +1,13 @@
+'use strict';
+
+const metaGeneratorPath = './meta_generator';
+
+module.exports = hexo => {
+  /* const {
+    filter
+  } = hexo.extend; */
+  // filter.register('after_render:html', require('./meta_generator'));
+};
+
+// 保持过滤器最先执行
+hexo.extend.filter.register('after_render:html', require(metaGeneratorPath), 1);
diff --git a/scripts/filters/meta_generator.js b/scripts/filters/meta_generator.js
new file mode 100644
index 0000000..880fa3b
--- /dev/null
+++ b/scripts/filters/meta_generator.js
@@ -0,0 +1,17 @@
+'use strict';
+
+const defaultConfig = require('../default_config');
+
+function hexoMetaGeneratorInject(data) {
+  const config = defaultConfig;
+  if (!config.meta_generator || !data ||
+    data.match(/<meta\s+name=['|"]?generator['|"]?/i)) {
+    return;
+  }
+
+  const hexoGeneratorTag = `\n  <meta name="generator" content="hexo-theme-yilia-plus">`;
+
+  return data.replace('</title>', '</title>' + hexoGeneratorTag);
+}
+
+module.exports = hexoMetaGeneratorInject;
diff --git a/scripts/helpers/ayer-plus-vendors.js b/scripts/helpers/ayer-plus-vendors.js
new file mode 100644
index 0000000..0c9b575
--- /dev/null
+++ b/scripts/helpers/ayer-plus-vendors.js
@@ -0,0 +1,9 @@
+/* global hexo */
+
+'use strict';
+
+/** 自定义链接生成 */
+hexo.extend.helper.register('ayer_plus_vendors', function (url) {
+  if (url.startsWith('//')) return url;
+  return this.url_for(`${url}`);
+});
diff --git a/scripts/helpers/wordcount.js b/scripts/helpers/wordcount.js
new file mode 100644
index 0000000..5548678
--- /dev/null
+++ b/scripts/helpers/wordcount.js
@@ -0,0 +1,39 @@
+var util = require('hexo-util');
+var stripHTML = util.stripHTML;
+
+var counter = function (content) {
+  content = stripHTML(content);
+  const cn = (content.match(/[\u4E00-\u9FA5]/g) || []).length;
+  const en = (content.replace(/[\u4E00-\u9FA5]/g, '').match(/[a-zA-Z0-9_\u0392-\u03c9\u0400-\u04FF]+|[\u4E00-\u9FFF\u3400-\u4dbf\uf900-\ufaff\u3040-\u309f\uac00-\ud7af\u0400-\u04FF]+|[\u00E4\u00C4\u00E5\u00C5\u00F6\u00D6]+|\w+/g) || []).length;
+  return [cn, en];
+};
+
+hexo.extend.helper.register('min2read', function (content, {
+  cn = 300,
+  en = 160
+} = {}) {
+  var len = counter(content);
+  var readingTime = len[0] / cn + len[1] / en;
+  return readingTime < 1 ? '1' : parseInt(readingTime, 10);
+});
+
+hexo.extend.helper.register('wordcount', function (content) {
+  var len = counter(content);
+  var count = len[0] + len[1];
+  if (count < 1000) {
+    return count;
+  }
+  return Math.round(count / 100) / 10 + 'k';
+});
+
+hexo.extend.helper.register('totalcount', function (site) {
+  var count = 0;
+  site.posts.forEach(function (post) {
+    var len = counter(post.content);
+    count += len[0] + len[1];
+  });
+  if (count < 1000) {
+    return count;
+  }
+  return Math.round(count / 100) / 10 + 'k';
+});
diff --git a/scripts/lib/core.js b/scripts/lib/core.js
new file mode 100644
index 0000000..886a24c
--- /dev/null
+++ b/scripts/lib/core.js
@@ -0,0 +1,25 @@
+/* global hexo */
+
+'use strict';
+
+// Hexo事件:https://hexo.io/zh-cn/api/events
+
+/** 在静态文件生成前发布 */
+hexo.on('generateBefore', () => {
+  // Merge config.
+  //require('./../filters/index')(hexo);
+});
+
+/** 在文章文件建立后发布。该事件返回文章参数。 */
+hexo.on('new', function (post) {
+  //console.log(post)
+});
+
+/** 在文章开始渲染前执行 */
+hexo.extend.filter.register('before_post_render', function (data) {
+  //var config = hexo.config;
+  //console.log(3);
+  //console.log(config);
+  //console.log(hexo.theme.config);
+
+});