请求用户所有题目的状态，匹配HTML页面上相应的题目超链接进行展示，适用于灵神的讨论帖题单场景（也可用于其他题单）
// ==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);
})();