AO3 Word Count Script

Greasy Fork is available in English.
AO3 Word Count Script

Adds word counts to chapter links on AO3 Chapter Index pages.
Verze ze dne 19. 03. 2023. Zobrazit nejnovější verzi.
Zeptejte se na otázku, přidejte recenzi nebo skript nahlašte.
Wrap lines
// ==UserScript==
// @name         AO3 Word Count Script
// @namespace    ao3chapterwordcounter
// @version      3
// @description  Adds word counts to chapter links on AO3 Chapter Index pages.
// @author       Anton Dumov
// @license      MIT
// @match        https://archiveofourown.org/*/navigate
// @grant        none
// ==/UserScript==

(function() {
    'use strict';

    const wordCountRegex = /\s+/g;
    const cacheKeyPrefix = "ao3-word-count-cache-";
    const cacheDurationMs = 24 * 60 * 60 * 1000; // 24 hours in milliseconds

    const getCachedWordCount = link => {
        const cacheKey = cacheKeyPrefix + link.href;
        const cachedValue = localStorage.getItem(cacheKey);
        if (cachedValue) {
            const { timestamp, wordCount } = JSON.parse(cachedValue);
            if (Date.now() - timestamp < cacheDurationMs) {
                return wordCount;
            } else {
                localStorage.removeItem(cacheKey);
            }
        }
        return null;
    };

    const setCachedWordCount = (link, wordCount) => {
        const cacheKey = cacheKeyPrefix + link.href;
        const cacheValue = JSON.stringify({ timestamp: Date.now(), wordCount });
        localStorage.setItem(cacheKey, cacheValue);
    };

    const chapterLinks = document.querySelectorAll("ol.chapter.index.group li a");

    let maxWidth = 0;
    chapterLinks.forEach(link => {
        const width = link.getBoundingClientRect().width;
        if (width > maxWidth) {
            maxWidth = width;
        }
    });

    chapterLinks.forEach(link => {
        const cachedWordCount = getCachedWordCount(link);
        let wordCount
        if (cachedWordCount) {
            wordCount = cachedWordCount
        } else {
            fetch(link.href).then(response => response.text())
                .then(text => {
                    const parser = new DOMParser();
                    const doc = parser.parseFromString(text, "text/html");
                    const article = doc.querySelector("div[role=article]");
                    wordCount = article ? article.textContent.trim().split(wordCountRegex).length : 0;
                    setCachedWordCount(link, wordCount);
                })
                .catch(error => console.log(error));
        }
        const spanElement = link.parentElement.querySelector('span.datetime');
        const wordCountElement = document.createElement("span");
        wordCountElement.textContent = `(${wordCount} words)`;
        const margin = maxWidth - link.getBoundingClientRect().width + 7;
        wordCountElement.style.marginLeft = `${margin}px`;
        spanElement.parentNode.insertBefore(wordCountElement, spanElement.nextSibling);
    });

})();