koishi-plugin-bilitester 1.1.0 → 1.3.0

package/README.md

@@ -8,6 +8,7 @@
 - Cookie 管理：自动存储和管理登录 Cookie
 - 账号信息查询：查看当前登录账号的详细信息
 - 视频信息查询：获取哔哩哔哩视频的详细信息
+- 视频搜索：搜索哔哩哔哩视频
 - 动态查询：获取用户的最新动态
 - 直播间信息：获取直播间状态和信息
 - 账号信息刷新：更新缓存的账号信息
@@ -70,10 +71,18 @@ bilitester logout
 ### 查询视频信息
 
 ```
-bilibili video <BV号>
+bilitester video <BV号>
 ```
 
-例如：`bilibili video BV1xx411c7mD`
+例如：`bilitester video BV1xx411c7mD`
+
+### 搜索视频
+
+```
+bilitester search <关键词>
+```
+
+例如：`bilitester search 动画`
 
 ### 查询动态
 

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;
@@ -52,6 +110,105 @@ function apply(ctx, config) {
     const getCookieString = (sessdata, biliJct, dedeUserId, dedeUserIdCkMd5) => {
         return `SESSDATA=${sessdata}; bili_jct=${biliJct}; DedeUserID=${dedeUserId}; DedeUserID__ckMd5=${dedeUserIdCkMd5}`;
     };
+    const formatNumber = (num) => {
+        if (num >= 100000000) {
+            return (num / 100000000).toFixed(1) + '亿';
+        }
+        else if (num >= 10000) {
+            return (num / 10000).toFixed(1) + '万';
+        }
+        return num.toString();
+    };
+    const formatDuration = (seconds) => {
+        const hours = Math.floor(seconds / 3600);
+        const minutes = Math.floor((seconds % 3600) / 60);
+        const secs = seconds % 60;
+        if (hours > 0) {
+            return `${hours}:${minutes.toString().padStart(2, '0')}:${secs.toString().padStart(2, '0')}`;
+        }
+        else {
+            return `${minutes}:${secs.toString().padStart(2, '0')}`;
+        }
+    };
+    const parseVideoId = (input) => {
+        const avMatch = input.match(/(?:^|[\s/])(av\d+)(?:[\s/]|$)/i);
+        if (avMatch) {
+            return { type: 'av', id: avMatch[1] };
+        }
+        const bvMatch = input.match(/(?:^|[\s/])(BV[\w]+)(?:[\s/]|$)/i);
+        if (bvMatch) {
+            return { type: 'bv', id: bvMatch[1] };
+        }
+        return { type: null, id: null };
+    };
+    const parseVideoLinks = (content) => {
+        const links = [];
+        const fullLinkPattern = /bilibili\.com\/video\/([ab]v[\w]+)/gi;
+        let match;
+        while ((match = fullLinkPattern.exec(content)) !== null) {
+            links.push({ type: 'Video', id: match[1] });
+        }
+        const shortLinkPatterns = [
+            /b23\.tv\/([\w]+)/gi,
+            /bili22\.cn\/([\w]+)/gi,
+            /bili23\.cn\/([\w]+)/gi,
+            /bili33\.cn\/([\w]+)/gi,
+            /bili2233\.cn\/([\w]+)/gi,
+        ];
+        for (const pattern of shortLinkPatterns) {
+            while ((match = pattern.exec(content)) !== null) {
+                links.push({ type: 'Short', id: match[1] });
+            }
+        }
+        return links;
+    };
+    const resolveShortLink = async (shortId) => {
+        try {
+            const response = await axios_1.default.get(`https://b23.tv/${shortId}`, {
+                maxRedirects: 0,
+                validateStatus: (status) => status >= 300 && status < 400,
+            });
+            const location = response.headers['location'];
+            if (location) {
+                return location;
+            }
+            return null;
+        }
+        catch (error) {
+            if (error.response && error.response.headers && error.response.headers.location) {
+                return error.response.headers.location;
+            }
+            return null;
+        }
+    };
+    const fetchVideoInfo = async (videoId) => {
+        const parsed = parseVideoId(videoId);
+        if (!parsed.type || !parsed.id) {
+            return null;
+        }
+        try {
+            let url;
+            if (parsed.type === 'av') {
+                url = `https://api.bilibili.com/x/web-interface/view?aid=${parsed.id.replace('av', '')}`;
+            }
+            else {
+                url = `https://api.bilibili.com/x/web-interface/view?bvid=${parsed.id}`;
+            }
+            const response = await axios_1.default.get(url, {
+                headers: {
+                    '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',
+                },
+            });
+            if (response.data.code === 0) {
+                return response.data;
+            }
+            return null;
+        }
+        catch (error) {
+            console.error('获取视频信息失败:', error);
+            return null;
+        }
+    };
     const pollLoginStatus = async (qrcodeKey, userId) => {
         const session = loginSessions.get(qrcodeKey);
         if (!session)
@@ -260,7 +417,7 @@ function apply(ctx, config) {
             return '刷新账号信息失败，请稍后重试';
         }
     });
-    cmd.subcommand('video <bvid>', '获取哔哩哔哩视频信息')
+    cmd.subcommand('video <bvid>', '获取B站视频信息')
         .action(async ({ session }, bvid) => {
         if (!session) {
             return '无法获取会话信息';
@@ -269,39 +426,99 @@ function apply(ctx, config) {
         try {
             const accounts = await ctx.database.get('bilibili', { userId });
             if (!accounts || accounts.length === 0) {
-                return '您还未登录哔哩哔哩账号，请使用 "bilitester login" 进行登录';
+                return '您还未登录B站账号，请使用 "bilitester login" 进行登录';
             }
             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/view', {
-                params: { bvid }
-            });
-            if (response.data.code === 0) {
-                const video = response.data.data;
-                const duration = Math.floor(video.duration / 60) + ':' + (video.duration % 60).toString().padStart(2, '0');
-                const pubdate = new Date(video.pubdate * 1000).toLocaleString('zh-CN');
-                return `视频信息：\n` +
-                    `标题: ${video.title}\n` +
-                    `BV号: ${video.bvid}\n` +
-                    `AV号: ${video.aid}\n` +
-                    `UP主: ${video.owner.name}\n` +
-                    `时长: ${duration}\n` +
-                    `发布时间: ${pubdate}\n` +
-                    `播放: ${video.stat.view} | 弹幕: ${video.stat.danmaku} | 点赞: ${video.stat.like}\n` +
-                    `投币: ${video.stat.coin} | 收藏: ${video.stat.favorite} | 分享: ${video.stat.share}\n` +
-                    `简介: ${video.desc.substring(0, 100)}${video.desc.length > 100 ? '...' : ''}\n` +
-                    `${koishi_1.h.image(video.pic)}`;
-            }
-            else {
-                return '获取视频信息失败，请检查BV号是否正确';
+            const videoInfo = await fetchVideoInfo(bvid);
+            if (!videoInfo) {
+                return '获取视频信息失败，请检查BV/AV号是否正确';
             }
+            const video = videoInfo.data;
+            const duration = formatDuration(video.duration);
+            const pubdate = new Date(video.pubdate * 1000).toLocaleString('zh-CN');
+            return `视频信息：\n` +
+                `标题: ${video.title}\n` +
+                `BV号: ${video.bvid}\n` +
+                `AV号: ${video.aid}\n` +
+                `UP主: ${video.owner.name}\n` +
+                `时长: ${duration}\n` +
+                `发布时间: ${pubdate}\n` +
+                `播放: ${formatNumber(video.stat.view)} | 弹幕: ${formatNumber(video.stat.danmaku)} | 点赞: ${formatNumber(video.stat.like)}\n` +
+                `投币: ${formatNumber(video.stat.coin)} | 收藏: ${formatNumber(video.stat.favorite)} | 分享: ${formatNumber(video.stat.share)}\n` +
+                `简介: ${video.desc.substring(0, 100)}${video.desc.length > 100 ? '...' : ''}\n` +
+                `${koishi_1.h.image(video.pic)}`;
         }
         catch (error) {
             console.error('获取视频信息失败:', error);
             return '获取视频信息失败，请稍后重试';
         }
     });
+    cmd.subcommand('parse <link>', '解析B站视频链接')
+        .action(async ({ session }, link) => {
+        if (!session) {
+            return '无法获取会话信息';
+        }
+        const userId = session.userId || session.guildId || 'unknown';
+        try {
+            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);
+            let videoId = null;
+            const parsed = parseVideoId(link);
+            if (parsed.type && parsed.id) {
+                videoId = parsed.id;
+            }
+            else {
+                const links = parseVideoLinks(link);
+                if (links.length > 0) {
+                    const videoLink = links[0];
+                    if (videoLink.type === 'Short') {
+                        const resolvedLink = await resolveShortLink(videoLink.id);
+                        if (resolvedLink) {
+                            const resolvedParsed = parseVideoId(resolvedLink);
+                            if (resolvedParsed.type && resolvedParsed.id) {
+                                videoId = resolvedParsed.id;
+                            }
+                        }
+                    }
+                    else {
+                        videoId = videoLink.id;
+                    }
+                }
+            }
+            if (!videoId) {
+                return '无法解析视频链接，请检查链接格式是否正确';
+            }
+            const videoInfo = await fetchVideoInfo(videoId);
+            if (!videoInfo) {
+                return '获取视频信息失败，请检查链接是否正确';
+            }
+            const video = videoInfo.data;
+            const duration = formatDuration(video.duration);
+            const pubdate = new Date(video.pubdate * 1000).toLocaleString('zh-CN');
+            return `视频信息：\n` +
+                `标题: ${video.title}\n` +
+                `BV号: ${video.bvid}\n` +
+                `AV号: ${video.aid}\n` +
+                `UP主: ${video.owner.name}\n` +
+                `时长: ${duration}\n` +
+                `发布时间: ${pubdate}\n` +
+                `播放: ${formatNumber(video.stat.view)} | 弹幕: ${formatNumber(video.stat.danmaku)} | 点赞: ${formatNumber(video.stat.like)}\n` +
+                `投币: ${formatNumber(video.stat.coin)} | 收藏: ${formatNumber(video.stat.favorite)} | 分享: ${formatNumber(video.stat.share)}\n` +
+                `简介: ${video.desc.substring(0, 100)}${video.desc.length > 100 ? '...' : ''}\n` +
+                `${koishi_1.h.image(video.pic)}`;
+        }
+        catch (error) {
+            console.error('解析视频链接失败:', error);
+            return '解析视频链接失败，请稍后重试';
+        }
+    });
     cmd.subcommand('dynamic', '获取哔哩哔哩动态')
         .action(async ({ session }) => {
         if (!session) {
@@ -404,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;
@@ -418,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';
@@ -429,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) {
@@ -450,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.1.0",
+  "version": "1.3.0",
   "description": "哔哩哔哩登录和API调用插件，支持二维码登录和Cookie管理",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",