LeetCode 灵神题单状态标记

请求用户所有题目的状态,匹配HTML页面上相应的题目超链接进行展示,适用于灵神的讨论帖题单场景(也可用于其他题单)

Versión del día 8/5/2025. Echa un vistazo a la versión más reciente.

Tendrás que instalar una extensión para tu navegador como Tampermonkey, Greasemonkey o Violentmonkey si quieres utilizar este script.

Necesitarás instalar una extensión como Tampermonkey o Violentmonkey para instalar este script.

Necesitarás instalar una extensión como Tampermonkey o Violentmonkey para instalar este script.

Necesitarás instalar una extensión como Tampermonkey o Userscripts para instalar este script.

Necesitará instalar una extensión como Tampermonkey para instalar este script.

Necesitarás instalar una extensión para administrar scripts de usuario si quieres instalar este script.

(Ya tengo un administrador de scripts de usuario, déjame instalarlo)

Necesitará instalar una extensión como Stylus para instalar este estilo.

Necesitará instalar una extensión como Stylus para instalar este estilo.

Necesitará instalar una extensión como Stylus para instalar este estilo.

Necesitará instalar una extensión del gestor de estilos de usuario para instalar este estilo.

Necesitará instalar una extensión del gestor de estilos de usuario para instalar este estilo.

Necesitará instalar una extensión del gestor de estilos de usuario para instalar este estilo.

(Ya tengo un administrador de estilos de usuario, déjame instalarlo)

// ==UserScript==
// @name         LeetCode 灵神题单状态标记
// @namespace    https://github.com/qk-antares
// @version      0.3
// @description  请求用户所有题目的状态,匹配HTML页面上相应的题目超链接进行展示,适用于灵神的讨论帖题单场景(也可用于其他题单)
// @author       qk-antares
// @match        https://leetcode.cn/discuss/post/*
// @grant        GM_xmlhttpRequest
// @connect      leetcode.cn
// @icon         data:image/svg+xml;base64,PD94bWwgdmVyc2lvbj0iMS4wIiBzdGFuZGFsb25lPSJubyI/PjwhRE9DVFlQRSBzdmcgUFVCTElDICItLy9XM0MvL0RURCBTVkcgMS4xLy9FTiIgImh0dHA6Ly93d3cudzMub3JnL0dyYXBoaWNzL1NWRy8xLjEvRFREL3N2ZzExLmR0ZCI+PHN2ZyB0PSIxNzQ2NjcxNzM5NTU1IiBjbGFzcz0iaWNvbiIgdmlld0JveD0iMCAwIDEwMjQgMTAyNCIgdmVyc2lvbj0iMS4xIiB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIHAtaWQ9IjY3MDIiIHhtbG5zOnhsaW5rPSJodHRwOi8vd3d3LnczLm9yZy8xOTk5L3hsaW5rIiB3aWR0aD0iMjAwIiBoZWlnaHQ9IjIwMCI+PHBhdGggZD0iTTEwMjEuMDUgOTE0LjU5TDY1NC4yMSAxMTkuNzdBOTYuMjcgOTYuMjcgMCAwIDAgNTY3LjA1IDY0SDQ4MGEzMiAzMiAwIDAgMC0yOSAxOC41OWwtMzg0IDgzMkEzMiAzMiAwIDAgMCA5NiA5NjBoMjE1LjA1YTk2LjI3IDk2LjI3IDAgMCAwIDg3LjE2LTU1Ljc3bDQxLTg4Ljg0IDI1NS40NSAxMjcuNzJBMTYwLjg0IDE2MC44NCAwIDAgMCA3NjYuMjIgOTYwSDk5MmEzMiAzMiAwIDAgMCAyOS00NS40MXogbS02ODAuOTQtMzcuMThBMzIuMTEgMzIuMTEgMCAwIDEgMzExLjA1IDg5NkgyNzRsNjEuMjEtMTMyLjYxTDM4MiA3ODYuNzV6IG0zODMuMTcgOC40NUw0MzguNDEgNzQzLjQzbC0wLjE2LTAuMDgtMTAzLjc3LTUxLjg5LTAuODEtMC4zOWEzMS44MyAzMS44MyAwIDAgMC0yNC0xLjM3aC0wLjA1YTMxLjg0IDMxLjg0IDAgMCAwLTE4LjUzIDE2LjU5Yy0wLjA2IDAuMTItMC4xMiAwLjI1LTAuMTcgMC4zN0wyMDMuNTMgODk2SDE0NmwzNTQuNDctNzY4SDU1OEwzMzkgNjAyLjU5Yy0wLjA1IDAuMDktMC4wOCAwLjE4LTAuMTIgMC4yN2EzMiAzMiAwIDAgMCAxNC44NiA0MS43NmwyOTEuMiAxNDUuNmEzMiAzMiAwIDAgMCA0NS45MS0zMy42M2MxLTAuNDYgMC4yMi0yLjQtMi41NS04LjRMNTc5LjI0IDUxMiA2MDggNDQ5LjcgODE0IDg5NmgtNDcuNzhhOTYuNDEgOTYuNDEgMCAwIDEtNDIuOTQtMTAuMTR6TTU0NCA1ODguMzZsNDcuOTIgMTAzLjgyLTc3Ljg3LTM4Ljkzek04ODQuNDcgODk2TDYzNy4wNSAzNTkuOTJhMzIgMzIgMCAwIDAtNTgtMC4ybC0wLjEgMC4yLTEyMi4xNyAyNjQuNjktNDYuNzMtMjMuMzZMNjA4IDE3Mi4zNiA5NDIgODk2eiIgZmlsbD0iI0ZGN0E1OCIgcC1pZD0iNjcwMyI+PC9wYXRoPjwvc3ZnPg==
// @license      MIT
// ==/UserScript==

(function () {
    'use strict';

    const THRESHOLD = 10; // 阈值
    const csrfToken = document.cookie.match(/csrftoken=([^;]+)/)?.[1];
    if (!csrfToken) {
        console.warn("⚠️ 未获取到 CSRF Token,用户可能未登录。");
    }
    console.log("🚀 [题单状态标记] 脚本启动");

    function getStatusIcon(status) {
        switch (status) {
            case 'ac': return '✅';
            case 'notac': return '❌';
            case null: return '🕓';
            default: return '❓';
        }
    }

    function extractLinks() {
        return Array.from(document.querySelectorAll("ul li > a[href*='/problems/']"))
            .filter(link => /\/problems\/[^/]+\/$/.test(link.href));
    }

    function insertStatusIcon(link, status) {
        const icon = getStatusIcon(status);
        const span = document.createElement('span');
        span.textContent = icon + ' ';
        link.parentNode.insertBefore(span, link);
    }

    function fetchAllProblemStatuses() {
        return new Promise((resolve) => {
            GM_xmlhttpRequest({
                method: "GET",
                url: "https://leetcode.cn/api/problems/all/",
                onload: function (res) {
                    try {
                        const json = JSON.parse(res.responseText);
                        const map = new Map();
                        json.stat_status_pairs.forEach(item => {
                            const slug = item.stat.question__title_slug;
                            const status = item.status;
                            map.set(slug, status);
                        });
                        console.log(`📦 批量模式:获取状态成功,共 ${map.size} 题`);
                        resolve(map);
                    } catch (e) {
                        console.error("❌ 批量解析失败", e);
                        resolve(new Map());
                    }
                },
                onerror: function (err) {
                    console.error("❌ 批量请求失败", err);
                    resolve(new Map());
                }
            });
        });
    }

    function fetchSingleStatus(slug) {
        return new Promise((resolve) => {
            GM_xmlhttpRequest({
                method: "POST",
                url: "https://leetcode.cn/graphql/",
                headers: {
                    'Content-Type': 'application/json',
                    'x-csrftoken': csrfToken || '',
                },
                data: JSON.stringify({
                    operationName: "getQuestionProgress",
                    variables: { titleSlug: slug },
                    query: `
                        query getQuestionProgress($titleSlug: String!) {
                          question(titleSlug: $titleSlug) {
                            status
                          }
                        }
                    `
                }),
                onload: function (res) {
                    try {
                        const json = JSON.parse(res.responseText);
                        const status = json.data?.question?.status ?? null;
                        resolve(status);
                    } catch (e) {
                        console.error(`⚠️ 单题解析失败:${slug}`, e);
                        resolve(null);
                    }
                },
                onerror: function (err) {
                    console.error(`❌ 单题请求失败:${slug}`, err);
                    resolve(null);
                }
            });
        });
    }

    async function run() {
        const links = extractLinks();
        console.log(`🔍 检测到 ${links.length} 个题目链接`);

        if (links.length === 0) return;

        if (links.length <= THRESHOLD) {
            console.log(`🧪 小于等于 ${THRESHOLD},逐题请求模式`);
            for (const link of links) {
                const slug = link.href.match(/problems\/([^/]+)\//)?.[1];
                if (!slug) continue;
                const status = await fetchSingleStatus(slug);
                insertStatusIcon(link, status);
            }
        } else {
            console.log(`📦 超过 ${THRESHOLD},进入批量模式`);
            const statusMap = await fetchAllProblemStatuses();
            for (const link of links) {
                const slug = link.href.match(/problems\/([^/]+)\//)?.[1];
                const status = statusMap.get(slug) ?? null;
                insertStatusIcon(link, status);
            }
        }
    }

    // 等待页面加载完成
    setTimeout(run, 1000);
})();