koishi-plugin-bilitester 1.2.0 → 1.3.0

package/lib/index.js

@@ -7,6 +7,7 @@ exports.name = exports.Config = void 0;
 exports.apply = apply;
 const koishi_1 = require("koishi");
 const axios_1 = __importDefault(require("axios"));
+const crypto_1 = __importDefault(require("crypto"));
 exports.Config = koishi_1.Schema.object({
     pollInterval: koishi_1.Schema.number()
         .description('轮询扫码状态的时间间隔（毫秒）')
@@ -16,7 +17,64 @@ exports.Config = koishi_1.Schema.object({
         .default(180),
 });
 exports.name = 'bilitester';
+const searchResultsCache = new Map();
 const loginSessions = new Map();
+const MIXIN_KEY_ENC_TAB = [
+    46, 47, 18, 2, 53, 8, 23, 32, 15, 50, 10, 31, 58, 3, 45, 35, 27, 43, 5, 49,
+    33, 9, 42, 19, 29, 28, 14, 39, 12, 38, 41, 13, 37, 48, 7, 16, 24, 55, 40,
+    61, 26, 17, 0, 1, 60, 51, 30, 4, 22, 25, 54, 21, 56, 59, 6, 63, 57, 62, 11,
+    36, 20, 34, 44, 52
+];
+const getMixinKey = (orig) => {
+    return MIXIN_KEY_ENC_TAB.map(n => orig[n]).join('').slice(0, 32);
+};
+let cachedWbiKeys = null;
+let wbiKeysExpireTime = 0;
+const getWbiKeys = async (cookie) => {
+    const now = Date.now();
+    if (cachedWbiKeys && now - wbiKeysExpireTime < 3600000) {
+        return cachedWbiKeys;
+    }
+    try {
+        const response = await axios_1.default.get('https://api.bilibili.com/x/web-interface/nav', {
+            headers: {
+                'Cookie': cookie,
+                'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
+                'Referer': 'https://www.bilibili.com/',
+            },
+        });
+        const { data } = response.data;
+        if (!data || !data.wbi_img) {
+            throw new Error('Failed to get WBI keys');
+        }
+        const img_url = data.wbi_img.img_url;
+        const sub_url = data.wbi_img.sub_url;
+        const imgKey = img_url.slice(img_url.lastIndexOf('/') + 1, img_url.lastIndexOf('.'));
+        const subKey = sub_url.slice(sub_url.lastIndexOf('/') + 1, sub_url.lastIndexOf('.'));
+        cachedWbiKeys = { imgKey, subKey };
+        wbiKeysExpireTime = now;
+        return { imgKey, subKey };
+    }
+    catch (error) {
+        console.error('获取WBI keys失败:', error);
+        throw error;
+    }
+};
+const encWbi = (params, imgKey, subKey) => {
+    const mixinKey = getMixinKey(imgKey + subKey);
+    const currTime = Math.round(Date.now() / 1000);
+    const chrFilter = /[!'()*]/g;
+    Object.assign(params, { wts: currTime });
+    const sortedKeys = Object.keys(params).sort();
+    const query = sortedKeys
+        .map(key => {
+        const value = params[key].toString().replace(chrFilter, '');
+        return `${encodeURIComponent(key)}=${encodeURIComponent(value)}`;
+    })
+        .join('&');
+    const wbiSign = crypto_1.default.createHash('md5').update(query + mixinKey).digest('hex');
+    return query + `&w_rid=${wbiSign}`;
+};
 function apply(ctx, config) {
     const pollInterval = config.pollInterval || 2000;
     const qrCodeTimeout = config.qrCodeTimeout || 180;
@@ -563,13 +621,17 @@ function apply(ctx, config) {
             }
             const acc = accounts[0];
             const cookie = getCookieString(acc.sessdata, acc.biliJct, acc.dedeUserId, acc.dedeUserIdCkMd5);
-            const axiosInstance = createAxiosInstance(cookie);
-            const response = await axiosInstance.get('https://api.bilibili.com/x/web-interface/search/type', {
-                params: {
-                    search_type: 'video',
-                    keyword: keyword,
-                    page: 1,
-                }
+            const wbiKeys = await getWbiKeys(cookie);
+            const wbiSign = encWbi({
+                keyword: keyword,
+                page: 1,
+            }, wbiKeys.imgKey, wbiKeys.subKey);
+            const response = await axios_1.default.get(`https://api.bilibili.com/x/web-interface/search/type?search_type=video&keyword=${encodeURIComponent(keyword)}&page=1&${wbiSign}`, {
+                headers: {
+                    'Cookie': cookie,
+                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
+                    'Referer': 'https://www.bilibili.com/',
+                },
             });
             if (response.data.code === 0) {
                 const { data } = response.data;
@@ -577,6 +639,8 @@ function apply(ctx, config) {
                 if (videos.length === 0) {
                     return `未找到与"${keyword}"相关的视频`;
                 }
+                const cacheKey = `search_${userId}_${keyword}_1`;
+                searchResultsCache.set(cacheKey, videos);
                 let result = `搜索结果：${keyword}\n`;
                 result += `共找到 ${data.numResults} 个视频，第 ${data.page}/${data.numPages} 页\n`;
                 result += '─'.repeat(40) + '\n';
@@ -588,13 +652,14 @@ function apply(ctx, config) {
                     result += `   UP主: ${video.author}\n`;
                     result += `   BV号: ${video.bvid}\n`;
                     result += `   时长: ${duration}\n`;
-                    result += `   播放: ${video.play} | 弹幕: ${video.video_review} | 收藏: ${video.favorites}\n`;
+                    result += `   播放: ${formatNumber(video.play)} | 弹幕: ${formatNumber(video.video_review)} | 收藏: ${formatNumber(video.favorites)}\n`;
                     result += `   发布: ${pubdate}\n`;
                     result += '─'.repeat(40) + '\n';
                 }
                 if (videos.length > 5) {
                     result += `还有 ${videos.length - 5} 个结果未显示...\n`;
                 }
+                result += `\n使用 "bilitester select <序号>" 选择视频`;
                 return result;
             }
             else if (response.data.code === -412) {
@@ -609,6 +674,72 @@ function apply(ctx, config) {
             return '搜索视频失败，请稍后重试';
         }
     });
+    cmd.subcommand('select <index>', '选择搜索结果中的视频')
+        .action(async ({ session }, index) => {
+        if (!session) {
+            return '无法获取会话信息';
+        }
+        const userId = session.userId || session.guildId || 'unknown';
+        try {
+            const cachedResults = Array.from(searchResultsCache.entries()).find(([key]) => key.startsWith(`search_${userId}_`));
+            if (!cachedResults || cachedResults[1].length === 0) {
+                return '请先使用 "bilitester search <关键词>" 搜索视频';
+            }
+            const videos = cachedResults[1];
+            const idx = parseInt(index);
+            if (isNaN(idx) || idx < 1 || idx > videos.length) {
+                return `请输入有效的视频序号 (1-${videos.length})`;
+            }
+            const video = videos[idx - 1];
+            const accounts = await ctx.database.get('bilibili', { userId });
+            if (!accounts || accounts.length === 0) {
+                return '您还未登录B站账号，请使用 "bilitester login" 进行登录';
+            }
+            const acc = accounts[0];
+            const cookie = getCookieString(acc.sessdata, acc.biliJct, acc.dedeUserId, acc.dedeUserIdCkMd5);
+            const axiosInstance = createAxiosInstance(cookie);
+            const playUrlResponse = await axiosInstance.get(`https://api.bilibili.com/x/player/playurl`, {
+                params: {
+                    bvid: video.bvid,
+                    fnval: 80,
+                }
+            });
+            if (playUrlResponse.data.code === 0 && playUrlResponse.data.data && playUrlResponse.data.data.dash) {
+                const dash = playUrlResponse.data.data.dash;
+                const videoStream = dash.video?.[0];
+                const audioStream = dash.audio?.[0];
+                if (videoStream && audioStream) {
+                    const duration = formatDuration(dash.duration);
+                    const pubdate = new Date(video.pubdate * 1000).toLocaleString('zh-CN');
+                    let result = `视频信息：\n`;
+                    result += `标题: ${video.title}\n`;
+                    result += `BV号: ${video.bvid}\n`;
+                    result += `AV号: ${video.aid}\n`;
+                    result += `UP主: ${video.author}\n`;
+                    result += `时长: ${duration}\n`;
+                    result += `发布时间: ${pubdate}\n`;
+                    result += `播放: ${formatNumber(video.play)} | 弹幕: ${formatNumber(video.video_review)} | 收藏: ${formatNumber(video.favorites)}\n`;
+                    result += `简介: ${video.description.substring(0, 100)}${video.description.length > 100 ? '...' : ''}\n`;
+                    result += `${koishi_1.h.image(video.pic)}\n`;
+                    const videoUrl = videoStream.baseUrl + videoStream.segment_base_url + videoStream.segments[0].url;
+                    const audioUrl = audioStream.baseUrl + audioStream.segment_base_url + audioStream.segments[0].url;
+                    result += `\n视频链接:\n${videoUrl}\n`;
+                    result += `\n音频链接:\n${audioUrl}`;
+                    return result;
+                }
+                else {
+                    return '获取视频播放链接失败，可能是大会员专属视频';
+                }
+            }
+            else {
+                return '获取视频播放链接失败，请稍后重试';
+            }
+        }
+        catch (error) {
+            console.error('获取视频播放链接失败:', error);
+            return '获取视频播放链接失败，请稍后重试';
+        }
+    });
     ctx.on('dispose', () => {
         for (const session of loginSessions.values()) {
             if (session.timer) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koishi-plugin-bilitester",
-  "version": "1.2.0",
+  "version": "1.3.0",
   "description": "哔哩哔哩登录和API调用插件，支持二维码登录和Cookie管理",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",