307 lines
15 KiB
JavaScript
307 lines
15 KiB
JavaScript
// ==UserScript==
|
||
// @name B站 AI 收藏夹自动细化整理 (V8.2 火山方舟版)
|
||
// @namespace http://tampermonkey.net/
|
||
// @version 8.2.0
|
||
// @description 使用火山方舟大模型接口，强力约束优先匹配已有收藏夹，听从对话框指令
|
||
// @author 某不知名的根号三 & Gemini
|
||
// @match https://space.bilibili.com/*
|
||
// @grant GM_xmlhttpRequest
|
||
// ==/UserScript==
|
||
|
||
(function() {
|
||
'use strict';
|
||
|
||
// ================= 配置区 =================
|
||
// 火山方舟 Chat Completions 接口
|
||
const API_URL = 'https://ark.cn-beijing.volces.com/api/v3/chat/completions';
|
||
// 你的火山方舟 API Key（支持填写原始 Key 或 Bearer 前缀格式）
|
||
const API_KEY = 'ffcead97-0c80-4ff7-8b71-ee255a512f07';
|
||
// 火山方舟模型名称（示例：deepseek-v3-1-terminus）
|
||
const MODEL_NAME = 'deepseek-v3-1-terminus';
|
||
// ==========================================
|
||
|
||
const sleep = ms => new Promise(r => setTimeout(r, ms));
|
||
|
||
function getAuthHeader(apiKey) {
|
||
const key = (apiKey || '').trim();
|
||
if (!key) return '';
|
||
return /^Bearer\s+/i.test(key) ? key : `Bearer ${key}`;
|
||
}
|
||
|
||
// 状态日志打印小工具
|
||
function logStatus(msg) {
|
||
console.log(msg);
|
||
const logDiv = document.getElementById('ai-status-log');
|
||
if (logDiv) {
|
||
logDiv.innerHTML += `<div style="margin-top:4px;">➜ ${msg}</div>`;
|
||
logDiv.scrollTop = logDiv.scrollHeight;
|
||
}
|
||
}
|
||
|
||
function getBiliData() {
|
||
const midMatch = document.cookie.match(/DedeUserID=([^;]+)/);
|
||
const csrfMatch = document.cookie.match(/bili_jct=([^;]+)/);
|
||
return { mid: midMatch ? midMatch[1] : '', csrf: csrfMatch ? csrfMatch[1] : '' };
|
||
}
|
||
|
||
function getSourceMediaId() {
|
||
const params = new URLSearchParams(window.location.search);
|
||
return params.get('fid') || params.get('media_id') || params.get('id');
|
||
}
|
||
|
||
function buildFormData(obj) {
|
||
return new URLSearchParams(obj).toString();
|
||
}
|
||
|
||
// 获取已有收藏夹列表 (过滤掉默认收藏夹)
|
||
async function getMyFolders(biliData) {
|
||
const url = `https://api.bilibili.com/x/v3/fav/folder/created/list-all?up_mid=${biliData.mid}`;
|
||
const res = await fetch(url, { credentials: 'include' }).then(r => r.json());
|
||
if (res.code === 0 && res.data && res.data.list) {
|
||
const folderMap = {};
|
||
res.data.list.forEach(f => {
|
||
if (f.title !== '默认收藏夹') folderMap[f.title] = f.id;
|
||
});
|
||
return folderMap;
|
||
}
|
||
return {};
|
||
}
|
||
|
||
async function createFolder(title, biliData) {
|
||
logStatus(`📁 正在新建收藏夹：【${title}】`);
|
||
const url = 'https://api.bilibili.com/x/v3/fav/folder/add';
|
||
const data = buildFormData({ title: title, privacy: 1, csrf: biliData.csrf });
|
||
const res = await fetch(url, {
|
||
method: 'POST', credentials: 'include',
|
||
headers: { 'Content-Type': 'application/x-www-form-urlencoded' }, body: data
|
||
}).then(r => r.json());
|
||
if (res.code === 0) return res.data.id;
|
||
throw new Error(`新建失败: ${res.message}`);
|
||
}
|
||
|
||
async function moveVideos(sourceMediaId, tarMediaId, resourcesStr, biliData) {
|
||
const url = 'https://api.bilibili.com/x/v3/fav/resource/move';
|
||
const data = buildFormData({
|
||
src_media_id: sourceMediaId, tar_media_id: tarMediaId, mid: biliData.mid, resources: resourcesStr, csrf: biliData.csrf
|
||
});
|
||
const res = await fetch(url, {
|
||
method: 'POST', credentials: 'include',
|
||
headers: { 'Content-Type': 'application/x-www-form-urlencoded' }, body: data
|
||
}).then(r => r.json());
|
||
if (res.code !== 0) console.error("移动失败：", res.message);
|
||
}
|
||
|
||
async function startProcess() {
|
||
const biliData = getBiliData();
|
||
const btn = document.getElementById('ai-start-btn');
|
||
const customPromptInput = document.getElementById('ai-custom-prompt');
|
||
|
||
if (!biliData.mid || !biliData.csrf) return alert("请确保你在 B 站已登录！");
|
||
const sourceMediaId = getSourceMediaId();
|
||
if (!sourceMediaId) return alert("未能识别当前页面的收藏夹 ID！请确保你在某个具体的收藏夹页面内。");
|
||
|
||
const userRequirement = customPromptInput.value.trim();
|
||
|
||
btn.innerText = '🔄 整理中，请看下方日志...';
|
||
btn.disabled = true;
|
||
btn.style.background = '#ccc';
|
||
document.getElementById('ai-status-log').innerHTML = '';
|
||
|
||
try {
|
||
// 1. 获取现有的“家底”
|
||
logStatus(`正在获取现有的收藏夹列表...`);
|
||
const existingFoldersMap = await getMyFolders(biliData);
|
||
const existingFolderNames = Object.keys(existingFoldersMap);
|
||
logStatus(`📦 发现 ${existingFolderNames.length} 个已有收藏夹`);
|
||
|
||
// 2. 全量抓取视频
|
||
logStatus(`开始全量抓取当前收藏夹视频...`);
|
||
let allVideos = [];
|
||
let pn = 1;
|
||
const ps = 20;
|
||
|
||
while (true) {
|
||
logStatus(`正在读取第 ${pn} 页...`);
|
||
const listUrl = `https://api.bilibili.com/x/v3/fav/resource/list?media_id=${sourceMediaId}&pn=${pn}&ps=${ps}&platform=web`;
|
||
const listRes = await fetch(listUrl, { credentials: 'include' }).then(r => r.json());
|
||
|
||
if (listRes.code !== 0) {
|
||
logStatus(`❌ 读取出错: ${listRes.message}`);
|
||
break;
|
||
}
|
||
|
||
const videos = (listRes.data && listRes.data.medias) ? listRes.data.medias : [];
|
||
if (videos.length === 0) break;
|
||
|
||
allVideos.push(...videos);
|
||
if (videos.length < ps) break;
|
||
pn++;
|
||
await sleep(300);
|
||
}
|
||
|
||
if (allVideos.length === 0) {
|
||
logStatus("⚠️ 当前收藏夹是空的！");
|
||
resetButton(btn);
|
||
return;
|
||
}
|
||
|
||
logStatus(`✅ 读取完毕！共获取到 ${allVideos.length} 个视频。`);
|
||
logStatus(`🧠 正在呼叫 ${MODEL_NAME} 进行匹配与思考，请耐心等待...`);
|
||
|
||
const videoDataForAI = allVideos.map(v => ({
|
||
id: v.id, type: v.type, title: v.title, intro: v.intro ? v.intro.substring(0, 30) : ''
|
||
}));
|
||
|
||
// ====== 强化版：动态注入存量数据和用户需求 ======
|
||
const customRuleText = userRequirement ? `\n\n【⭐⭐⭐用户特殊需求 (最高优先级)⭐⭐⭐】\n用户的特别指示是："${userRequirement}"\n请你务必听从！\n⚠️ 致命警告：如果用户的指示中提到了要把视频放入某个分类，请你务必在上面的【已有收藏夹】列表里寻找最匹配的准确名称！如果用户打字简写了（比如用户说“音乐”，但已有的是“我的音乐”），你必须输出已有收藏夹的完整名称“我的音乐”，绝不允许凭空新建近义词分类！` : '';
|
||
|
||
const combinedPrompt = `你是一个逻辑极其严密的文件整理专家。我现在需要你帮我把一批 B 站视频分类。
|
||
非常重要：用户目前已经建好了以下这些收藏夹：
|
||
[ ${existingFolderNames.length > 0 ? existingFolderNames.join(', ') : '暂无'} ]
|
||
|
||
请你严格按照以下 3 个步骤执行：
|
||
【步骤 1：存量强制匹配】
|
||
通读所有视频。只要视频内容沾边，就必须一字不差地使用上述【已有收藏夹】的名称作为分类键名。
|
||
|
||
【步骤 2：谨慎新建】
|
||
只有当某几个视频确实与所有“已有收藏夹”都毫不相干时，你才可以创建一个新的涵盖面广的“大类”。绝不为单一视频建新分类，孤立视频请塞入最贴近的已有分类。
|
||
|
||
【步骤 3：绝无遗漏】
|
||
确保列表中的**每一个视频**都被分配到了具体的分类中，绝对不可以遗漏任何一个 ID！${customRuleText}
|
||
|
||
请严格输出合法的纯 JSON 格式数据。包含 "thoughts" 和 "categories" 两个字段。
|
||
示例：
|
||
{
|
||
"thoughts": "分析发现，视频A符合已有的'游戏实况'分类。用户要求把XX放入YY，我发现已有收藏夹中叫'YY合集'，因此我将它们放入'YY合集'中...",
|
||
"categories": {
|
||
"已有收藏夹准确名字1": [{"id": 111, "type": 2}],
|
||
"新创建的大类名字": [{"id": 222, "type": 2}]
|
||
}
|
||
}
|
||
|
||
以下是待处理的所有视频：
|
||
${JSON.stringify(videoDataForAI)}`;
|
||
|
||
GM_xmlhttpRequest({
|
||
method: "POST",
|
||
url: API_URL,
|
||
headers: {
|
||
"Content-Type": "application/json",
|
||
"Authorization": getAuthHeader(API_KEY)
|
||
},
|
||
data: JSON.stringify({
|
||
model: MODEL_NAME,
|
||
messages: [{role: "user", content: combinedPrompt}],
|
||
temperature: 0.1 // 保持极低的温度，确保严谨匹配，不乱造词
|
||
}),
|
||
onload: async function(response) {
|
||
if(response.status !== 200) {
|
||
logStatus(`❌ API报错：${response.status}`);
|
||
console.error(response.responseText);
|
||
resetButton(btn);
|
||
return;
|
||
}
|
||
|
||
let content = JSON.parse(response.responseText).choices[0].message.content;
|
||
content = content.replace(/```json/g, '').replace(/```/g, '').trim();
|
||
const aiResult = JSON.parse(content);
|
||
|
||
console.log("💡 AI 思考过程：\n", aiResult.thoughts);
|
||
logStatus(`💡 思考完毕！(按F12可看AI脑洞)`);
|
||
|
||
let processedCount = 0;
|
||
for (const [categoryName, vids] of Object.entries(aiResult.categories)) {
|
||
if(!vids || vids.length === 0) continue;
|
||
|
||
let targetFolderId = existingFoldersMap[categoryName];
|
||
if (!targetFolderId) {
|
||
targetFolderId = await createFolder(categoryName, biliData);
|
||
existingFoldersMap[categoryName] = targetFolderId;
|
||
await sleep(1000);
|
||
}
|
||
|
||
logStatus(`🚚 正将 ${vids.length} 个视频移入【${categoryName}】...`);
|
||
const resourcesStr = vids.map(v => `${v.id}:${v.type}`).join(',');
|
||
await moveVideos(sourceMediaId, targetFolderId, resourcesStr, biliData);
|
||
processedCount += vids.length;
|
||
await sleep(500);
|
||
}
|
||
|
||
logStatus(`🎉 整理完成！共处理了 ${processedCount} 个视频。请刷新页面！`);
|
||
btn.innerText = '✅ 整理完成，点我重置';
|
||
btn.style.background = '#4CAF50';
|
||
btn.disabled = false;
|
||
btn.onclick = () => window.location.reload();
|
||
},
|
||
onerror: function(err) {
|
||
logStatus(`❌ 请求大模型失败，请检查网络或 API Key`);
|
||
console.error(err);
|
||
resetButton(btn);
|
||
}
|
||
});
|
||
|
||
} catch (error) {
|
||
logStatus(`❌ 发生未知错误，请看 F12 控制台`);
|
||
console.error(error);
|
||
resetButton(btn);
|
||
}
|
||
}
|
||
|
||
function resetButton(btn) {
|
||
btn.innerText = '🚀 开始深度整理';
|
||
btn.style.background = '#fb7299';
|
||
btn.disabled = false;
|
||
btn.onclick = startProcess;
|
||
}
|
||
|
||
// ================= UI 构建区 =================
|
||
function initUI() {
|
||
if (document.getElementById('ai-sort-wrapper')) return;
|
||
|
||
const floatBtn = document.createElement('div');
|
||
floatBtn.id = 'ai-float-btn';
|
||
floatBtn.innerHTML = '🤖<br>AI整理';
|
||
floatBtn.style.cssText = 'position:fixed; bottom:30px; left:30px; z-index:9999; background:#fb7299; color:white; width:50px; height:50px; border-radius:25px; display:flex; align-items:center; justify-content:center; text-align:center; font-size:12px; font-weight:bold; cursor:pointer; box-shadow: 0 4px 10px rgba(251, 114, 153, 0.4); transition:0.3s;';
|
||
|
||
const panel = document.createElement('div');
|
||
panel.id = 'ai-sort-wrapper';
|
||
panel.style.cssText = 'position:fixed; bottom:30px; left:30px; z-index:10000; width:320px; display:none; flex-direction:column; box-shadow: 0 5px 20px rgba(0,0,0,0.2); border-radius:10px; overflow:hidden; font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif;';
|
||
|
||
panel.innerHTML = `
|
||
<div style="background:#fb7299; color:#fff; padding:12px 15px; font-weight:bold; font-size:15px; display:flex; justify-content:space-between; align-items:center;">
|
||
<span>🤖 AI 收藏夹整理助理</span>
|
||
<span id="ai-close-btn" style="cursor:pointer; font-size:18px; line-height:1;">×</span>
|
||
</div>
|
||
<div style="background:#fff; padding:15px; border:1px solid #eee; border-top:none;">
|
||
<p style="margin:0 0 8px 0; font-size:13px; color:#555;">有什么特定的整理要求吗？(选填)</p>
|
||
<textarea id="ai-custom-prompt" placeholder="例如：\n- 把所有 Vue 相关的放一个文件夹\n- 把时长超过1小时的单独拎出来\n(不填则优先放入已有收藏夹，并由 AI 自由发挥补充分类)" style="width:100%; height:80px; padding:8px; box-sizing:border-box; border:1px solid #ddd; border-radius:6px; font-size:13px; resize:none; margin-bottom:12px; outline:none;"></textarea>
|
||
<button id="ai-start-btn" style="width:100%; padding:10px; background:#fb7299; color:white; border:none; border-radius:6px; font-size:14px; font-weight:bold; cursor:pointer; transition:background 0.2s;">🚀 开始深度整理</button>
|
||
<div id="ai-status-log" style="margin-top:15px; background:#f4f4f4; padding:8px; border-radius:6px; font-size:12px; color:#333; height:120px; overflow-y:auto; word-break:break-all;">
|
||
等待指令...
|
||
</div>
|
||
</div>
|
||
`;
|
||
|
||
document.body.appendChild(floatBtn);
|
||
document.body.appendChild(panel);
|
||
|
||
floatBtn.onclick = () => {
|
||
floatBtn.style.display = 'none';
|
||
panel.style.display = 'flex';
|
||
};
|
||
|
||
document.getElementById('ai-close-btn').onclick = () => {
|
||
panel.style.display = 'none';
|
||
floatBtn.style.display = 'flex';
|
||
};
|
||
|
||
document.getElementById('ai-start-btn').onclick = startProcess;
|
||
}
|
||
|
||
if (document.readyState === 'loading') {
|
||
document.addEventListener('DOMContentLoaded', initUI);
|
||
} else {
|
||
initUI();
|
||
}
|
||
|
||
})(); |