在 qBittorrent WebUI 中根据辅种数为种子添加标签，支持all和list两种模式，并输出调试信息
当前为
// ==UserScript==
// @name qB-WebUI 根据辅种数添加标签
// @name:en qB-WebUI Add Tags Based on Reseed Count
// @namespace localhost
// @version 0.5.0
// @author Schalkiii
// @description 在 qBittorrent WebUI 中根据辅种数为种子添加标签，支持all和list两种模式，并输出调试信息
// @description:en Add tags to torrents in qBittorrent WebUI based on reseed count, supporting all and list modes with debug info
// @license MIT
// @run-at document-end
// @match http://192.168.10.72:9091/*
// @match http://127.0.0.1:9091/*
// @grant GM_xmlhttpRequest
// ==/UserScript==
/* globals torrentsTable */
const baseURL = window.location.origin + '/api/v2/torrents/';
const reseedAPI = 'http://api.iyuu.cn/index.php?s=App.Api.GetSubject&info_hash='; // 替换为实际的辅种查询API
const tagPrefix = 'Reseed-'; // 标签前缀，例如 Reseed-0, Reseed-1, ..., Reseed-6, Reseed-7+
// 获取种子列表（支持all和list模式）
function getTorrentList(scope) {
if (scope === 'all') {
return getFetch('info'); // 获取所有种子
} else if (scope === 'list') {
return torrentsTable.getFilteredAndSortedRows().map(row => row.full_data); // 获取当前显示的种子
} else {
return null;
}
}
// 发起 fetch 请求
async function getFetch(route) {
try {
const response = await fetch(baseURL + route);
if (!response.ok) {
throw new Error('Error fetching data!');
}
return await response.json();
} catch (error) {
console.error('Error fetching data:', error);
return null;
}
}
// 检查种子是否已有Reseed-开头的标签
function hasReseedTag(tags) {
if (!tags) return false;
const tagArray = tags.split(', ');
return tagArray.some(tag => tag.startsWith(tagPrefix));
}
// 根据哈希值查询辅种数
function queryReseedCount(hash) {
return new Promise((resolve) => {
console.log(`查询辅种数：正在查询哈希值 ${hash}...`);
GM_xmlhttpRequest({
url: `${reseedAPI}${hash}`,
method: 'GET',
responseType: 'json',
onload: function (response) {
console.log(`查询辅种数：请求成功，状态码：${response.status}`);
if (response.status === 200) {
try {
const json = JSON.parse(response.responseText);
if (json.ret === 200) {
const reseedCount = Math.max(json.data.pid_total, json.data.tid_total);
console.log(`查询辅种数：哈希值 ${hash} 的辅种数为 ${reseedCount}`);
resolve({ hash, reseedCount });
} else {
console.warn(`查询辅种数：API返回的ret不是200，实际返回: ${json.ret}，默认辅种数为 0`);
resolve({ hash, reseedCount: -1 }); // 返回辅种数为 0
}
} catch (e) {
console.error(`查询辅种数：JSON解析失败: ${e}，默认辅种数为 0`);
resolve({ hash, reseedCount: -1 }); // 返回辅种数为 0
}
} else {
console.warn(`查询辅种数：请求失败，状态码: ${response.status}，默认辅种数为 0`);
resolve({ hash, reseedCount: -1 }); // 返回辅种数为 0
}
},
onerror: function (error) {
console.error(`查询辅种数：请求发生错误: ${error}，默认辅种数为 0`);
resolve({ hash, reseedCount: 0 }); // 返回辅种数为 0
},
});
});
}
// 为种子添加标签
async function addTagToTorrent(hash, tag) {
const url = `${baseURL}addTags`;
const data = new URLSearchParams();
data.append('hashes', hash);
data.append('tags', tag);
try {
const response = await fetch(url, {
method: 'POST',
headers: {
'Content-Type': 'application/x-www-form-urlencoded',
},
body: data,
});
if (!response.ok) {
throw new Error('Error adding tag!');
}
console.log(`标签添加成功：哈希值 ${hash} 已添加标签 "${tag}"`);
} catch (error) {
console.error(`标签添加失败：哈希值 ${hash} 添加标签 "${tag}" 时出错:`, error);
}
}
// 根据辅种数生成标签
function getReseedTag(reseedCount) {
if (reseedCount >= 7) {
return `${tagPrefix}7+`; // 大于等于7的合并为一个标签
} else {
return `${tagPrefix}${reseedCount}`; // 0, 1, 2, ..., 6
}
}
// 主函数：处理种子并添加标签
async function processTorrents(scope) {
const torrentList = await getTorrentList(scope);
if (!torrentList || torrentList.length === 0) {
console.log('未找到种子。');
return;
}
console.log(`在模式 "${scope}" 下找到 ${torrentList.length} 个种子。`);
console.log('获取到的种子哈希值列表：', torrentList.map(torrent => torrent.hash));
let successCount = 0;
let errorCount = 0;
let skippedCount = 0;
for (const torrent of torrentList) {
try {
// 检查是否已有Reseed-开头的标签
if (hasReseedTag(torrent.tags)) {
console.log(`跳过种子：名称 "${torrent.name}" 已有Reseed标签 "${torrent.tags}"`);
skippedCount++;
continue;
}
const { hash, reseedCount } = await queryReseedCount(torrent.hash);
console.log(`处理种子：名称 "${torrent.name}", 哈希值 ${hash}, 辅种数 ${reseedCount}`);
// 根据辅种数生成标签并添加
const tag = getReseedTag(reseedCount);
await addTagToTorrent(hash, tag);
successCount++;
} catch (error) {
console.error(`处理种子失败：名称 "${torrent.name}", 哈希值 ${torrent.hash}:`, error);
errorCount++;
}
}
alert(`处理完成！\n成功：${successCount} 个，失败：${errorCount} 个，跳过已有标签：${skippedCount} 个`);
}
// 添加按钮到页面
function addButton() {
const newBtn = document.createElement("li");
newBtn.innerHTML = "<a class='js-modal'><b> 打可辅种数标签 </b></a>";
document.querySelector("#desktopNavbar > ul").append(newBtn);
newBtn.addEventListener("click", async function() {
const scope = window.prompt("!!!请先手动删除[tagNames]中包含的标签!!!\n检查全部种子请输入 [all]\n仅检查当前表格中显示的种子请输入 [list]", "all")
if (!scope) return;
await processTorrents(scope);
});
}
// 初始化
(function () {
addButton();
})();