Scribble Hub: Display Current Chapter Word Count

Adds the word count of the current chapter between the chapter navigation buttons at the top of the chapter
当前为 2024-01-24 提交的版本，查看最新版本。
换行
// ==UserScript==
// @name           Scribble Hub: Display Current Chapter Word Count
// @namespace      https://github.com/w4tchdoge
// @version        1.0.0-20240124_033315
// @description    Adds the word count of the current chapter between the chapter navigation buttons at the top of the chapter
// @author         w4tchdoge
// @homepage       https://github.com/w4tchdoge/MISC-UserScripts
// @match          *://*.scribblehub.com/read/*/chapter/*
// @license        AGPL-3.0-or-later
// @history        1.0.0 — Initial release
// ==/UserScript==

(function () {
	`use strict`;

	// function taken from https://stackoverflow.com/a/2901298/11750206
	function numberWithCommas(x) {

		var parts = x.toString().split(`.`);
		parts[0] = parts[0].replace(/\B(?=(\d{3})+(?!\d))/g, `,`);
		return parts.join(`.`);
	}

	// Attempted conversion of the Ruby regex code AO3 uses to JavaScript by looking at:
	// https://github.com/otwcode/otwarchive/blob/943f585818005be8df269d84ca454af478150e75/config/config.yml#L453
	// https://github.com/otwcode/otwarchive/blob/943f585818005be8df269d84ca454af478150e75/lib/word_counter.rb#L26
	// https://github.com/otwcode/otwarchive/blob/943f585818005be8df269d84ca454af478150e75/lib/word_counter.rb#L30C9-L31C95
	// Has not been tested on non-English works, feedback would be appreciated
	const word_count_regex = /\p{Script=Han}|\p{Script=Hiragana}|\p{Script=Katakana}|\p{Script=Thai}|((?!\p{Script=Han}|\p{Script=Hiragana}|\p{Script=Katakana}|\p{Script=Thai})\w)+/gu;

	// Get element that contains the top navigation buttons
	const top_chp_nav_btns = document.querySelector(`.nav_chp_fi`);

	// Def vars used in word counting
	var chp_raw_node, auth_note_node, chp_raw_text, auth_note_text, chp_raw_word_count, auth_note_word_count, word_count;

	// Get elems to be word counted
	chp_raw_node = document.querySelector(`#chp_raw`).cloneNode(true);
	auth_note_node = document.querySelector(`#chp_raw div .wi_authornotes_body`).cloneNode(true);

	// Remove auth note from chp_raw_node
	chp_raw_node.removeChild(chp_raw_node.querySelector(`div.wi_authornotes`));

	// Get the text context of the elements to be counted
	chp_raw_text = chp_raw_node.textContent;
	auth_note_text = auth_note_node.textContent;

	// Count the number of words
	// Counting method from:
	// https://stackoverflow.com/a/76673564/11750206
	// Regex substitutions from:
	// https://github.com/otwcode/otwarchive/blob/943f585818005be8df269d84ca454af478150e75/lib/word_counter.rb#L30C33-L30C68
	chp_raw_word_count = [...chp_raw_text.replaceAll(/--/g, `—`).replaceAll(/['’‘-]/g, ``).matchAll(word_count_regex)].length;
	auth_note_word_count = [...auth_note_text.replaceAll(/--/g, `—`).replaceAll(/['’‘-]/g, ``).matchAll(word_count_regex)].length;

	// Check if chapter word count is very small
	// If true use auth_note_word_count as word count
	const chp_raw_frac_total_count = chp_raw_word_count / (chp_raw_word_count + auth_note_word_count);
	if (chp_raw_frac_total_count < 0.05) {
		word_count = auth_note_word_count;
	} else {
		word_count = chp_raw_word_count;
	}

	// console.log(word_count);

	// Add thousands seperators to the word count
	word_count = numberWithCommas(word_count);

	// Create element that will display the current chapter word count
	var WordCount_elm = Object.assign(document.createElement(`span`), {
		id: `userscript-word-count-span`,
		style: `display: inline-block;`,
		innerHTML: `${word_count} words`
	});

	// Add additional style to the parent div so that the word count is center-aligned
	top_chp_nav_btns.querySelector(`.prenext`).style.textAlign = `center`;

	// Add the word count element
	top_chp_nav_btns.querySelector(`.btn-prev`).after(WordCount_elm);

})();