Hexo 自定义插件实录：给博客加一个统计命令

function countWords(text) {
  const plain = String(text)
    .replace(/```[\s\S]*?```/g, ' ') // 代码块不计入正文
    .replace(/<[^>]+>/g, ' '); // 去 HTML 标签
  const cjk = (plain.match(/[一-龥]/g) || []).length;
  const en = (plain.replace(/[一-龥]/g, ' ').match(/[A-Za-z0-9]+/g) || []).length;
  return cjk + en;
}

hexo.extend.filter.register('before_post_render', function (data) {
  const words = countWords(data.raw || data.content || '');
  data.stats = { words, minutes: readingMinutes(words) };
  return data; // 过滤器必须 return 数据，丢了返回值文章就丢了
});

hexo.extend.generator.register('stats', function (locals) {
  const stats = aggregate(locals.posts.toArray());
  const payload = Object.assign(
    { site: hexo.config.title, url: hexo.config.url, generatedAt: new Date().toISOString() },
    stats
  );
  return { path: 'stats.json', data: JSON.stringify(payload, null, 2) };
});

hexo.extend.console.register('stats', '打印全站文章统计', {}, function () {
  const h = this; // 回调里的 this 就是 hexo 实例
  return h.load().then(() => {
    const s = aggregate(h.locals.get('posts').toArray());
    // ...打印表格...
  });
});

/* scripts/site-stats.js —— 本地插件：站点统计 */
'use strict';

function countWords(text) {
  const plain = String(text)
    .replace(/```[\s\S]*?```/g, ' ')
    .replace(/<[^>]+>/g, ' ');
  const cjk = (plain.match(/[一-龥]/g) || []).length;
  const en = (plain.replace(/[一-龥]/g, ' ').match(/[A-Za-z0-9]+/g) || []).length;
  return cjk + en;
}

function readingMinutes(words) {
  return Math.max(1, Math.round(words / 400));
}

/* ① 过滤器：文章渲染前，把字数挂到文章对象上 */
hexo.extend.filter.register('before_post_render', function (data) {
  const words = countWords(data.raw || data.content || '');
  data.stats = { words, minutes: readingMinutes(words) };
  return data;
});

/* 公共聚合逻辑：一份代码，同时喂给 ② 和 ③ */
function aggregate(posts) {
  const items = posts.map(p => ({
    title: p.title,
    year: p.date ? p.date.year() : 0,
    words: (p.stats && p.stats.words) || countWords(p.raw || ''),
    encrypted: Boolean(p.password),
    category: (p.categories.toArray()[0] || {}).name || '未分类'
  }));

  const totalWords = items.reduce((s, x) => s + x.words, 0);
  const yearMap = new Map();
  const catMap = new Map();
  for (const it of items) {
    const y = yearMap.get(it.year) || { year: it.year, posts: 0, words: 0 };
    y.posts++; y.words += it.words; yearMap.set(it.year, y);
    const c = catMap.get(it.category) || { name: it.category, posts: 0, words: 0 };
    c.posts++; c.words += it.words; catMap.set(it.category, c);
  }

  return {
    posts: items.length,
    encrypted: items.filter(x => x.encrypted).length,
    totalWords,
    readingTimeMinutes: readingMinutes(totalWords),
    byYear: [...yearMap.values()].sort((a, b) => b.year - a.year),
    byCategory: [...catMap.values()].sort((a, b) => b.posts - a.posts),
    longest: items.slice().sort((a, b) => b.words - a.words).slice(0, 3)
      .map(x => ({ title: x.title, words: x.words, minutes: readingMinutes(x.words) }))
  };
}

/* ② 生成器：产出 /stats.json */
hexo.extend.generator.register('stats', function (locals) {
  const stats = aggregate(locals.posts.toArray());
  const payload = Object.assign(
    { site: hexo.config.title, url: hexo.config.url, generatedAt: new Date().toISOString() },
    stats
  );
  return { path: 'stats.json', data: JSON.stringify(payload, null, 2) };
});

/* ③ 控制台命令：npx hexo stats */
hexo.extend.console.register('stats', '打印全站文章统计', {}, function () {
  const h = this;
  return h.load().then(() => {
    const s = aggregate(h.locals.get('posts').toArray());
    console.log('\n━━ 站点统计 · ' + h.config.title + ' ━━');
    console.log(`文章 ${s.posts} 篇（含加密 ${s.encrypted} 篇）`);
    console.log(`总字数 ${s.totalWords.toLocaleString()} 字，通读约 ${(s.readingTimeMinutes / 60).toFixed(1)} 小时`);
    console.log('\n按年：');
    const max = Math.max.apply(null, s.byYear.map(y => y.posts));
    for (const y of s.byYear) {
      const bar = '*'.repeat(Math.max(1, Math.round(y.posts / max * 24)));
      console.log(`  ${y.year}  ${bar} ${y.posts} 篇 · ${y.words.toLocaleString()} 字`);
    }
    console.log('\n最长三篇：');
    s.longest.forEach((x, i) =>
      console.log(`  ${i + 1}. 《${x.title}》${x.words.toLocaleString()} 字（约 ${x.minutes} 分钟）`));
    console.log('');
  });
});