NotionNext/components/WordCount.js

import { useEffect } from 'react'

/**
 * 字数统计
 * @returns
 */
export default function WordCount () {
  useEffect(() => {
    countWords()
  })

  return <div id='wordCountWrapper' className='hidden'>
    本文共<strong id='wordCount'>0</strong>字，阅读需要约<strong id='readTime'>0</strong>分钟
    </div>
}

/**
 * 更新字数统计和阅读时间
 */
function countWords () {
  if (window) {
    const articleElement = document.getElementById('notion-article')
    if (articleElement) {
      const articleText = deleteHtmlTag(articleElement.innerHTML)
      const wordCount = fnGetCpmisWords(articleText)
      // 阅读速度 300-500每分钟
      document.getElementById('wordCount').innerHTML = wordCount
      document.getElementById('readTime').innerHTML = Math.floor(wordCount / 400) + 1
      const wordCountWrapper = document.getElementById('wordCountWrapper')
      wordCountWrapper.classList.remove('hidden')
    }
  }
}

// 去除html标签
function deleteHtmlTag (str) {
  str = str.replace(/<[^>]+>|&[^>]+;/g, '').trim()// 去掉所有的html标签和&nbsp;之类的特殊符合
  return str
}

// 用word方式计算正文字数
function fnGetCpmisWords (str) {
  let sLen = 0
  try {
    // eslint-disable-next-line no-irregular-whitespace
    str = str.replace(/(\r\n+|\s+|　+)/g, '龘')
    // eslint-disable-next-line no-control-regex
    str = str.replace(/[\x00-\xff]/g, 'm')
    str = str.replace(/m+/g, '*')
    str = str.replace(/龘+/g, '')
    sLen = str.length
  } catch (e) {

  }
  return sLen
}