import axios from "axios"; import { exec, execSync } from "child_process"; import { HttpsProxyAgent } from 'https-proxy-agent'; import _ from "lodash"; import fetch from "node-fetch"; import { Buffer } from 'node:buffer'; import fs from "node:fs"; import PQueue from 'p-queue'; import path from "path"; import qrcode from "qrcode"; import querystring from "querystring"; import puppeteer from "../../../lib/puppeteer/puppeteer.js"; import { BILI_CDN_SELECT_LIST, BILI_DEFAULT_INTRO_LEN_LIMIT, BILI_RESOLUTION_LIST, COMMON_USER_AGENT, DIVIDING_LINE, douyinTypeMap, DOWNLOAD_WAIT_DETECT_FILE_TIME, HELP_DOC, MESSAGE_RECALL_TIME, REDIS_YUNZAI_ISOVERSEA, REDIS_YUNZAI_WHITELIST, SUMMARY_PROMPT, transMap, TWITTER_BEARER_TOKEN, XHS_NO_WATERMARK_HEADER } from "../constants/constant.js"; import { ANIME_SERIES_SEARCH_LINK, ANIME_SERIES_SEARCH_LINK2, BILI_ARTICLE_INFO, BILI_EP_INFO, BILI_NAV, BILI_NAV_STAT, BILI_ONLINE, BILI_SSID_INFO, BILI_STREAM_FLV, BILI_STREAM_INFO, BILI_SUMMARY, DY_COMMENT, DY_INFO, DY_LIVE_INFO, DY_LIVE_INFO_2, DY_TOUTIAO_INFO, GENERAL_REQ_LINK, HIBI_API_SERVICE, MIYOUSHE_ARTICLE, NETEASE_API_CN, NETEASE_SONG_DOWNLOAD, NETEASE_TEMP_API, QISHUI_MUSIC_TEMP_API, QQ_MUSIC_TEMP_API, TWITTER_TWEET_INFO, WEIBO_SINGLE_INFO, WEISHI_VIDEO_INFO, XHS_REQ_LINK } from "../constants/tools.js"; import BiliInfoModel from "../model/bili-info.js"; import config from "../model/config.js"; import NeteaseModel from "../model/netease.js"; import NeteaseMusicInfo from '../model/neteaseMusicInfo.js'; import * as aBogus from "../utils/a-bogus.cjs"; import { downloadM3u8Videos, mergeAcFileToMp4, parseM3u8, parseUrl } from "../utils/acfun.js"; import { startBBDown } from "../utils/bbdown-util.js"; import { av2BV } from "../utils/bilibili-bv-av-convert.js"; import { BILI_HEADER, downloadBFile, filterBiliDescLink, getBiliAudio, getDownloadUrl, getDynamic, getScanCodeData, getVideoInfo, m4sToMp3, mergeFileToMp4 } from "../utils/bilibili.js"; import { getWbi } from "../utils/biliWbi.js"; import { getBodianAudio, getBodianMusicInfo, getBodianMv } from "../utils/bodian.js"; import { checkToolInCurEnv, cleanFilename, downloadAudio, downloadImg, estimateReadingTime, formatBiliInfo, retryAxiosReq, secondsToTime, testProxy, truncateString, urlTransformShortLink } from "../utils/common.js"; import { convertFlvToMp4 } from "../utils/ffmpeg-util.js"; import { checkAndRemoveFile, deleteFolderRecursive, getMediaFilesAndOthers, mkdirIfNotExists } from "../utils/file.js"; import GeneralLinkAdapter from "../utils/general-link-adapter.js"; import { contentEstimator } from "../utils/link-share-summary-util.js"; import { deepSeekChat, llmRead } from "../utils/llm-util.js"; import { getDS } from "../utils/mihoyo.js"; import { OpenaiBuilder } from "../utils/openai-builder.js"; import { redisExistKey, redisGetKey, redisSetKey } from "../utils/redis-util.js"; import { saveTDL, startTDL } from "../utils/tdl-util.js"; import { genVerifyFp } from "../utils/tiktok.js"; import Translate from "../utils/trans-strategy.js"; import { mid2id } from "../utils/weibo.js"; import { convertToSeconds, removeParams, ytbFormatTime } from "../utils/youtube.js"; import { ytDlpGetDuration, ytDlpGetThumbnail, ytDlpGetTilt, ytDlpHelper } from "../utils/yt-dlp-util.js"; import { textArrayToMakeForward } from "../utils/yunzai-util.js"; export class tools extends plugin { /** * 构造安全的命令 * @type {{existsPromptKey: string, existsTransKey: string}} */ static Constants = { existsTransKey: Object.keys(transMap).join("|"), }; constructor() { super({ name: "R插件工具和学习类", dsc: "R插件工具相关指令", event: "message.group", priority: 300, rule: [ { reg: `^(翻|trans)[${ tools.Constants.existsTransKey }]`, fnc: "trans", }, { reg: "(v.douyin.com|live.douyin.com)", fnc: "douyin", }, { reg: "(www.tiktok.com)|(vt.tiktok.com)|(vm.tiktok.com)", fnc: "tiktok", }, { reg: "^#(RBQ|rbq)$", fnc: "biliScan", permission: 'master', }, { reg: "^#(RBS|rbs)$", fnc: "biliState", permission: 'master', }, { reg: "(bilibili.com|b23.tv|t.bilibili.com|^BV[1-9a-zA-Z]{10}$)", fnc: "bili", }, { reg: "https?:\\/\\/x.com\\/[0-9-a-zA-Z_]{1,20}\\/status\\/([0-9]*)", fnc: "twitter_x", }, { reg: "(acfun.cn|^ac[0-9]{8}$)", fnc: "acfun", }, { reg: "(xhslink.com|xiaohongshu.com)", fnc: "xhs", }, { reg: "(h5app.kuwo.cn)", fnc: "bodianMusic", }, { reg: "(kuaishou.com|ixigua.com|h5.pipix.com|h5.pipigx.com|s.xsj.qq.com|m.okjike.com)", fnc: "general", }, { reg: "(youtube.com|youtu.be|music.youtube.com)", fnc: "sy2b" }, { reg: "(miyoushe.com)", fnc: "miyoushe" }, { reg: "(music.163.com|163cn.tv)", fnc: "netease", }, { reg: "(weibo.com|m.weibo.cn)", fnc: "weibo", }, { reg: "(weishi.qq.com)", fnc: "weishi" }, { reg: "share.xiaochuankeji.cn", fnc: "zuiyou" }, { reg: "(music.apple.com|open.spotify.com)", fnc: "freyr" }, { reg: "(^#总结一下(http|https):\/\/.*|mp.weixin|arxiv.org|sspai.com|chinadaily.com.cn|zhihu.com)", fnc: "linkShareSummary" }, { reg: "(y.qq.com)", fnc: "qqMusic" }, { reg: "(qishui.douyin.com)", fnc: "qishuiMusic" }, { reg: "https:\\/\\/t\\.me\\/(?:c\\/\\d+\\/\\d+\\/\\d+|c\\/\\d+\\/\\d+|\\w+\\/\\d+\\/\\d+|\\w+\\/\\d+\\?\\w+=\\d+|\\w+\\/\\d+)", fnc: "aircraft" }, { reg: "tieba.baidu.com", fnc: "tieba" }, { reg: "^#(网易状态|rns|RNS)$", fnc: "neteaseStatus", permission: 'master', }, { reg: "^#(rnq|RNQ)$", fnc: 'netease_scan', permission: 'master', }, ], }); // 配置文件 this.toolsConfig = config.getConfig("tools"); // 视频保存路径 this.defaultPath = this.toolsConfig.defaultPath; // 视频限制大小 this.videoSizeLimit = this.toolsConfig.videoSizeLimit; // 魔法接口 this.proxyAddr = this.toolsConfig.proxyAddr; this.proxyPort = this.toolsConfig.proxyPort; this.myProxy = `http://${ this.proxyAddr }:${ this.proxyPort }`; // 加载识别前缀 this.identifyPrefix = this.toolsConfig.identifyPrefix; // 加载直播录制时长 this.streamDuration = this.toolsConfig.streamDuration; // 加载直播是否开启兼容模式 this.streamCompatibility = this.toolsConfig.streamCompatibility; // 加载哔哩哔哩配置 this.biliSessData = this.toolsConfig.biliSessData; // 加载哔哩哔哩的限制时长 this.biliDuration = this.toolsConfig.biliDuration; // 加载是否显示哔哩哔哩的封面 this.biliDisplayCover = this.toolsConfig.biliDisplayCover; // 加载是否显示哔哩哔哩的视频信息 this.biliDisplayInfo = this.toolsConfig.biliDisplayInfo; // 加载是否显示哔哩哔哩的简介 this.biliDisplayIntro = this.toolsConfig.biliDisplayIntro; // 加载是否显示哔哩哔哩的在线人数 this.biliDisplayOnline = this.toolsConfig.biliDisplayOnline; // 加载是否显示哔哩哔哩的总结 this.biliDisplaySummary = this.toolsConfig.biliDisplaySummary; // 加载哔哩哔哩是否使用BBDown this.biliUseBBDown = this.toolsConfig.biliUseBBDown; // 加载 BBDown 的CDN配置 this.biliCDN = this.toolsConfig.biliCDN; // 加载网易云Cookie this.neteaseCookie = this.toolsConfig.neteaseCookie; // 加载是否转化群语音 this.isSendVocal = this.toolsConfig.isSendVocal; // 加载是否自建服务器 this.useLocalNeteaseAPI = this.toolsConfig.useLocalNeteaseAPI; // 加载自建服务器API this.neteaseCloudAPIServer = this.toolsConfig.neteaseCloudAPIServer; // 加载网易云解析最高音质 this.neteaseCloudAudioQuality = this.toolsConfig.neteaseCloudAudioQuality; // 加载哔哩哔哩是否使用Aria2 this.biliDownloadMethod = this.toolsConfig.biliDownloadMethod; // 加载哔哩哔哩最高分辨率 this.biliResolution = this.toolsConfig.biliResolution; // 加载youtube的截取时长 this.youtubeClipTime = this.toolsConfig.youtubeClipTime; // 加载youtube的解析时长 this.youtubeDuration = this.toolsConfig.youtubeDuration; // 加载油管下载画质选项 this.youtubeGraphicsOptions = this.toolsConfig.youtubeGraphicsOptions; // 加载youtube的Cookie this.youtubeCookiePath = this.toolsConfig.youtubeCookiePath; // 加载抖音Cookie this.douyinCookie = this.toolsConfig.douyinCookie; // 加载抖音是否压缩 this.douyinCompression = this.toolsConfig.douyinCompression; // 加载抖音是否开启评论 this.douyinComments = this.toolsConfig.douyinComments; // 加载小红书Cookie this.xiaohongshuCookie = this.toolsConfig.xiaohongshuCookie; // 翻译引擎 this.translateEngine = new Translate({ deeplApiUrls: this.toolsConfig.deeplApiUrls, proxy: this.myProxy, }); // 并发队列 this.queue = new PQueue({ concurrency: Number(this.toolsConfig.queueConcurrency) }); // 视频下载的并发数量 this.videoDownloadConcurrency = this.toolsConfig.videoDownloadConcurrency; // ai接口 this.aiBaseURL = this.toolsConfig.aiBaseURL; // ai api key this.aiApiKey = this.toolsConfig.aiApiKey; // ai模型 this.aiModel = this.toolsConfig.aiModel; } // 翻译插件 async trans(e) { const languageReg = /翻(.)/s; let msg = e.msg.trim(); const language = languageReg.exec(msg); if (!(language[1] in transMap)) { e.reply( "输入格式有误或暂不支持该语言！\n例子：翻中 China's policy has been consistent, but Japan chooses a path of mistrust, decoupling and military expansion", ); return; } let place = msg.slice(1 + language[1].length); if (_.isEmpty(place)) { const reply = await e?.getReply(); if (reply !== undefined) { place = reply.message.find(item => item.text !== undefined).text; } else { return; } } // 如果没有百度那就Google const translateResult = await this.translateEngine.translate(place, language[1]); e.reply(translateResult.trim(), true); return true; } // 抖音解析 async douyin(e) { const urlRex = /(http:\/\/|https:\/\/)(v|live).douyin.com\/[A-Za-z\d._?%&+\-=\/#]*/; // 检测无效链接，例如：v.douyin.com if (!urlRex.test(e.msg)) { e.reply(`检测到这是一个无效链接，无法解析抖音${ HELP_DOC }`); return; } // 获取链接 let douUrl = urlRex.exec(e.msg.trim())[0]; let ttwid = ''; if (douUrl.includes("v.douyin.com")) { const { location, ttwidValue } = await this.douyinRequest(douUrl); ttwid = ttwidValue; douUrl = location; } // TODO 如果有新的好解决方案可以删除，如果遇到https://www.iesdouyin.com/share/slides，这类动图暂时交付给其他API解析 if (douUrl.includes("share/slides")) { this.general(e); return; } // 获取 ID const douId = /note\/(\d+)/g.exec(douUrl)?.[1] || /video\/(\d+)/g.exec(douUrl)?.[1] || /live.douyin.com\/(\d+)/.exec(douUrl)?.[1] || /live\/(\d+)/.exec(douUrl)?.[1] || /webcast.amemv.com\/douyin\/webcast\/reflow\/(\d+)/.exec(douUrl)?.[1]; // 当前版本需要填入cookie if (_.isEmpty(this.douyinCookie) || _.isEmpty(douId)) { e.reply(`检测到没有Cookie 或者 这是一个无效链接，无法解析抖音${ HELP_DOC }`); return; } // 以下是更新了很多次的抖音API历史，且用且珍惜 // const url = `https://www.iesdouyin.com/web/api/v2/aweme/iteminfo/?item_ids=${ douId }`; // const url = `https://www.iesdouyin.com/aweme/v1/web/aweme/detail/?aweme_id=${ douId }&aid=1128&version_name=23.5.0&device_platform=android&os_version=2333`; // 感谢 Evil0ctal（https://github.com/Evil0ctal）提供的header 和 B1gM8c（https://github.com/B1gM8c）的逆向算法X-Bogus const headers = { "Accept-Language": "zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2", "User-Agent": COMMON_USER_AGENT, Referer: "https://www.douyin.com/", cookie: this.douyinCookie, }; let dyApi; if (douUrl.includes("live.douyin.com")) { // 第一类直播类型 dyApi = DY_LIVE_INFO.replaceAll("{}", douId); } else if (douUrl.includes("webcast.amemv.com")) { // 第二类直播类型，这里必须使用客户端的 fetch 请求 dyApi = DY_LIVE_INFO_2.replace("{}", douId) + `&verifyFp=${ genVerifyFp() }` + `&msToken=${ ttwid }`; const webcastResp = await fetch(dyApi); const webcastData = await webcastResp.json(); const item = webcastData.data.room; const { title, cover, user_count, stream_url } = item; const dySendContent = `${ this.identifyPrefix }识别：抖音直播，${ title }`; e.reply([segment.image(cover?.url_list?.[0]), dySendContent, `\n🏄‍♂️在线人数：${ user_count }人正在观看`]); // 下载10s的直播流 await this.sendStreamSegment(e, stream_url?.flv_pull_url?.HD1 || stream_url?.flv_pull_url?.FULL_HD1 || stream_url?.flv_pull_url?.SD1 || stream_url?.flv_pull_url?.SD2); return; } else { // 普通类型 dyApi = DY_INFO.replace("{}", douId); } // a-bogus参数 const abParam = aBogus.generate_a_bogus( new URLSearchParams(new URL(dyApi).search).toString(), headers["User-Agent"], ); // const param = resp.data.result[0].paramsencode; const resDyApi = `${ dyApi }&a_bogus=${ abParam }`; headers['Referer'] = `https://www.douyin.com/`; // 定义一个dy请求 const dyResponse = () => axios.get(resDyApi, { headers, }); // 如果失败进行3次重试 try { const data = await retryAxiosReq(dyResponse); // saveJsonToFile(data); // 直播数据逻辑 if (douUrl.includes("live")) { const item = await data.data.data?.[0]; const { title, cover, user_count_str, stream_url } = item; const dySendContent = `${ this.identifyPrefix }识别：抖音直播，${ title }`; e.reply([segment.image(cover?.url_list?.[0]), dySendContent, `\n🏄‍♂️在线人数：${ user_count_str }人正在观看`]); // 下载10s的直播流 await this.sendStreamSegment(e, stream_url?.flv_pull_url?.HD1 || stream_url?.flv_pull_url?.FULL_HD1 || stream_url?.flv_pull_url?.SD1 || stream_url?.flv_pull_url?.SD2); return; } const item = await data.aweme_detail; // await saveJsonToFile(item); // 如果为null则退出 if (item == null) { e.reply("R插件无法识别到当前抖音内容，请换一个试试！"); return; } const urlTypeCode = item.aweme_type; const urlType = douyinTypeMap[urlTypeCode]; // 核心内容 if (urlType === "video") { // logger.info(item.video); // 多位面选择：play_addr、play_addr_265、play_addr_h264 const { play_addr: { uri: videoAddrURI }, duration, cover } = item.video; // 进行时间判断，如果超过时间阈值就不发送 const dyDuration = Math.trunc(duration / 1000); const durationThreshold = this.biliDuration; // 一些共同发送内容 let dySendContent = `${ this.identifyPrefix }识别：抖音，${ item.author.nickname }\n📝 简介：${ item.desc }`; if (dyDuration >= durationThreshold) { // 超过阈值，不发送的情况 // 封面 const dyCover = cover.url_list?.pop(); // logger.info(cover.url_list); dySendContent += `\n ${ DIVIDING_LINE.replace('{}', '限制说明') }\n当前视频时长约：${ (dyDuration / 60).toFixed(2).replace(/\.00$/, '') } 分钟，\n大于管理员设置的最大时长 ${ (durationThreshold / 60).toFixed(2).replace(/\.00$/, '') } 分钟！`; e.reply([segment.image(dyCover), dySendContent]); // 如果开启评论的就调用 await this.douyinComment(e, douId, headers); return; } e.reply(`${ dySendContent }`); // 分辨率判断是否压缩 const resolution = this.douyinCompression ? "720p" : "1080p"; // 使用今日头条 CDN 进一步加快解析速度 const resUrl = DY_TOUTIAO_INFO.replace("1080p", resolution).replace("{}", videoAddrURI); // ⚠️ 暂时废弃代码 /*if (this.douyinCompression) { // H.265压缩率更高、流量省一半. 相对于H.264 // 265 和 264 随机均衡负载 const videoAddrList = Math.random() > 0.5 ? play_addr_265.url_list : play_addr_h264.url_list; resUrl = videoAddrList[videoAddrList.length - 1] || videoAddrList[0]; } else { // 原始格式，ps. videoAddrList这里[0]、[1]是 http，[最后一个]是 https const videoAddrList = play_addr.url_list; resUrl = videoAddrList[videoAddrList.length - 1] || videoAddrList[0]; }*/ // logger.info(resUrl); const path = `${ this.getCurDownloadPath(e) }/temp.mp4`; // 加入队列 await this.downloadVideo(resUrl).then(() => { this.sendVideoToUpload(e, path); }); } else if (urlType === "image") { // 发送描述 e.reply(`${ this.identifyPrefix }识别：抖音, ${ item.desc }`); // 无水印图片列表 let no_watermark_image_list = []; // 有水印图片列表 // let watermark_image_list = []; for (let i of item.images) { // 无水印图片列表 no_watermark_image_list.push({ message: segment.image(i.url_list[0]), nickname: this.e.sender.card || this.e.user_id, user_id: this.e.user_id, }); // 有水印图片列表 // watermark_image_list.push(i.download_url_list[0]); // e.reply(segment.image(i.url_list[0])); } // console.log(no_watermark_image_list) await e.reply(await Bot.makeForwardMsg(no_watermark_image_list)); } // 如果开启评论的就调用 await this.douyinComment(e, douId, headers); } catch (err) { logger.error(err); logger.mark(`Cookie 过期或者 Cookie 没有填写，请参考\n${ HELP_DOC }\n尝试无效后可以到官方QQ群[575663150]提出 bug 等待解决`); } return true; } /** * 下载直播片段 * @param e * @param stream_url * @param second */ async sendStreamSegment(e, stream_url, second = this.streamDuration) { let outputFilePath = `${ this.getCurDownloadPath(e) }/stream_${ second }s.flv`; // 删除临时文件 if (this.streamCompatibility) { await checkAndRemoveFile(outputFilePath.replace("flv", "mp4")); } else { await checkAndRemoveFile(outputFilePath); } // 创建一个取消令牌 const CancelToken = axios.CancelToken; const source = CancelToken.source(); try { const response = await axios.get(stream_url, { responseType: 'stream', cancelToken: source.token, }); logger.info("[R插件][发送直播流] 正在下载直播流..."); const file = fs.createWriteStream(outputFilePath); response.data.pipe(file); // 设置 streamDuration 秒后停止下载 setTimeout(async () => { logger.info(`[R插件][发送直播流] 直播下载 ${ second } 秒钟到，停止下载！`); // 取消请求 source.cancel('[R插件][发送直播流] 下载时间到，停止请求'); response.data.unpipe(file); // 取消管道连接 file.end(); // 结束写入 // 这里判断是否开启兼容模式 if (this.streamCompatibility) { logger.info(`[R插件][发送直播流] 开启兼容模式，开始转换mp4格式...`); const resolvedOutputPath = await convertFlvToMp4(outputFilePath, outputFilePath.replace(".flv", ".mp4")); fs.unlinkSync(outputFilePath); outputFilePath = resolvedOutputPath; logger.info(`[R插件][发送直播流] 转换完成，开始发送视频...`); } await this.sendVideoToUpload(e, outputFilePath); }, second * 1000); // 监听请求被取消的情况 response.data.on('error', (err) => { if (axios.isCancel(err)) { logger.info('请求已取消:', err.message); } else { logger.error('下载过程中发生错误:', err.message); } }); } catch (error) { if (axios.isCancel(error)) { logger.info('请求已取消:', error.message); } else { logger.error(`下载失败: ${ error.message }`); } await fs.promises.unlink(outputFilePath); // 下载失败时删除文件 } } /** * 获取 DY 评论 * @param e * @param douId * @param headers */ async douyinComment(e, douId, headers) { if (!this.douyinComments) { return; } const dyCommentUrl = DY_COMMENT.replace("{}", douId); const abParam = aBogus.generate_a_bogus( new URLSearchParams(new URL(dyCommentUrl).search).toString(), headers["User-Agent"], ); const commentsResp = await axios.get(`${ dyCommentUrl }&a_bogus=${ abParam }`, { headers }); // logger.info(headers) // saveJsonToFile(commentsResp.data, "data.json", _); const comments = commentsResp.data.comments; const replyComments = comments.map(item => { return { message: item.text, nickname: this.e.sender.card || this.e.user_id, user_id: this.e.user_id, }; }); e.reply(await Bot.makeForwardMsg(replyComments)); } // tiktok解析 async tiktok(e) { // 判断海外 const isOversea = await this.isOverseasServer(); // 如果不是海外用户且没有梯子直接返回 if (!isOversea && !(await testProxy(this.proxyAddr, this.proxyPort))) { e.reply("检测到没有梯子，无法解析TikTok"); return false; } // 处理链接 let url = e.msg.trim(); // 去除多余参数 const parsedUrl = new URL(url); parsedUrl.search = ''; const cleanedTiktokUrl = parsedUrl.toString(); // 下载逻辑 const path = this.getCurDownloadPath(e); await checkAndRemoveFile(path + "/temp.mp4"); const title = ytDlpGetTilt(url, isOversea, this.myProxy); e.reply(`${ this.identifyPrefix }识别：TikTok，视频下载中请耐心等待 \n${ title }`); await ytDlpHelper(path, cleanedTiktokUrl, isOversea, this.myProxy, this.videoDownloadConcurrency); await this.sendVideoToUpload(e, `${ path }/temp.mp4`); return true; } // 哔哩哔哩扫码登录 async biliScan(e) { e.reply('R插件开源免责声明:\n您将通过扫码完成获取哔哩哔哩refresh_token以及ck。\n本Bot将不会保存您的登录状态。\n我方仅提供视频解析及相关B站内容服务,若您的账号封禁、被盗等处罚与我方无关。\n害怕风险请勿扫码 ~', { recallMsg: 180 }); // 图片发送钩子 const imgSendHook = function (e, path) { e.reply([segment.image(path), segment.at(e.user_id), '请扫码以完成获取'], { recallMsg: 180 }); }; // 检查路径是否存在文件夹 await mkdirIfNotExists(this.defaultPath); // 发送请求 const saveCodePath = `${ this.defaultPath }qrcode.png`; const { SESSDATA, refresh_token } = await getScanCodeData(saveCodePath, 8, () => imgSendHook(e, saveCodePath)); // 更新到配置文件 config.updateField("tools", "biliSessData", SESSDATA); e.reply('登录成功！相关信息已保存至配置文件', true); return true; } // B站状态 async biliState(e) { if (!this.biliSessData) { e.reply("未检测到 B 站登录信息，请填写 SessData"); return; } // 封装 fetch 请求为函数 const fetchData = async (url) => { try { const res = await fetch(url, { headers: { Cookie: `SESSDATA=${ this.biliSessData }` } }); const data = await res.json(); return data.data; } catch (error) { e.reply("请求失败，请稍后重试"); throw error; // 确保错误传播 } }; // 并行请求用户基本信息和状态信息 const [biliData, biliStat] = await Promise.all([fetchData(BILI_NAV), fetchData(BILI_NAV_STAT)]); // 解构所需的字段 const { face, uname, level_info, money, wallet, vipStatus } = biliData; const { following = 0, // 默认值为 0 follower = 0, // 默认值为 0 dynamic_count = 0 // 默认值为 0 } = biliStat || {}; // 如果 biliStat 为 undefined，使用空对象解构 // 获取屏幕截图所需的数据 const screenData = await new BiliInfoModel(e).getData({ face, uname, level_info, money, wallet, vipStatus, following, follower, dynamic_count }); // 使用 puppeteer 生成截图 try { let img = await puppeteer.screenshot("bili-info", screenData); e.reply(img, true); } catch (error) { e.reply("截图生成失败，请稍后重试"); } } // B 站解析 async bili(e) { const urlRex = /(?:https?:\/\/)?www\.bilibili\.com\/[A-Za-z\d._?%&+\-=\/#]*/g; const bShortRex = /(http:|https:)\/\/b23.tv\/[A-Za-z\d._?%&+\-=\/#]*/g; let url = e.msg === undefined ? e.message.shift().data.replaceAll("\\", "") : e.msg.trim().replaceAll("\\", ""); // 直接发送BV号的处理 if (/^BV[1-9a-zA-Z]{10}$/.exec(url)?.[0]) { url = `https://www.bilibili.com/video/${ url }`; logger.info(url); } // 短号处理 if (url.includes("b23.tv")) { const bShortUrl = bShortRex.exec(url)?.[0]; await fetch(bShortUrl, { method: "HEAD" }).then(resp => { url = resp.url; }); } else if (url.includes("www.bilibili.com")) { url = urlRex.exec(url)[0]; } // 补充https url = url.startsWith("https://") ? url : "https://" + url; // av处理 const matched = url.match(/\/(AV|av)(\w+)/); if (matched) { url = url.replace(matched[0].replace("\/", ""), av2BV(Number(matched[2]))); } // 直播间分享 // logger.info(url) if (url.includes("live")) { // 提取直播间id const idPattern = /\/(\d+)$/; const parsedUrl = new URL(url); const streamId = parsedUrl.pathname.match(idPattern)?.[1]; // logger.info(streamId) // 提取相关信息 const liveData = await this.getBiliStreamInfo(streamId); // saveJsonToFile(liveData.data); const { title, user_cover, keyframe, description, tags, live_time, parent_area_name, area_name } = liveData.data.data; e.reply([ segment.image(user_cover), segment.image(keyframe), [`${ this.identifyPrefix }识别：哔哩哔哩直播，${ title }`, `${ description ? `📝 简述：${ description.replace(`<p>`, '').replace(`</p>`, '') }` : '' }`, `${ tags ? `🔖 标签：${ tags }` : '' }`, `📍 分区：${ parent_area_name ? `${ parent_area_name }` : '' }${ area_name ? `-${ area_name }` : '' }`, `${ live_time ? `⏰ 直播时间：${ live_time }` : '' }`, ].filter(item => item.trim() !== "").join("\n") ]); const streamData = await this.getBiliStream(streamId); const { url: streamUrl } = streamData.data.data.durl[0]; await this.sendStreamSegment(e, streamUrl); return true; } // 处理专栏 if (e.msg !== undefined && url.includes("read\/cv") || url.includes("read\/mobile")) { await this.biliArticle(e, url); return true; } // 动态处理 if (url.includes("t.bilibili.com") || url.includes("bilibili.com\/opus")) { if (_.isEmpty(this.biliSessData)) { e.reply("检测到没有填写biliSessData，无法解析动态"); return true; } url = this.biliDynamic(e, url, this.biliSessData); return true; } // 创建文件，如果不存在， const path = `${ this.getCurDownloadPath(e) }/`; await mkdirIfNotExists(path); // 处理番剧 if (url.includes("play\/ep") || url.includes("play\/ss")) { const ep = await this.biliEpInfo(url, e); // 如果使用了BBDown && 没有填写session 就放开下载 if (this.biliUseBBDown) { // 下载文件 await this.biliDownloadStrategy(e, `https://www.bilibili.com/bangumi/play/ep${ ep }`, path); } return true; } // 视频信息获取例子：http://api.bilibili.com/x/web-interface/view?bvid=BV1hY411m7cB // 请求视频信息 const videoInfo = await getVideoInfo(url); const { duration, bvid, cid, owner, pages } = videoInfo; // 限制时长 & 考虑分页视频情况 const query = querystring.parse(url); const curPage = query?.p || 0; const curDuration = pages?.[curPage]?.duration || duration; const isLimitDuration = curDuration > this.biliDuration; // 动态构造哔哩哔哩信息 let biliInfo = await this.constructBiliInfo(videoInfo); // 总结 if (this.biliDisplaySummary) { const summary = await this.getBiliSummary(bvid, cid, owner.mid); // 封装总结 summary && e.reply(await Bot.makeForwardMsg(textArrayToMakeForward(e, [`「R插件 x bilibili」联合为您总结内容：`, summary]))); } // 限制视频解析 if (isLimitDuration) { const durationInMinutes = (curDuration / 60).toFixed(0); biliInfo.push(`${ DIVIDING_LINE.replace('{}', '限制说明') }\n当前视频时长约：${ durationInMinutes }分钟，\n大于管理员设置的最大时长 ${ (this.biliDuration / 60).toFixed(2).replace(/\.00$/, '') } 分钟！`); e.reply(biliInfo); return true; } else { e.reply(biliInfo); } // 只提取音乐处理 if (e.msg !== undefined && e.msg.startsWith("音乐")) { return await this.biliMusic(e, url); } // 下载文件 await this.biliDownloadStrategy(e, url, path); return true; } /** * 提取哔哩哔哩专栏 * @param e * @param url * @returns {Promise} */ async biliArticle(e, url) { const cvid = url.match(/read\/cv(\d+)/)?.[1] || url.match(/read\/mobile\?id=(\d+)/)?.[1]; const articleResp = await fetch(BILI_ARTICLE_INFO.replace("{}", cvid), { headers: { ...BILI_HEADER } }); const articleData = (await articleResp.json()).data; const { title, author_name, origin_image_urls } = articleData; if (origin_image_urls) { const titleMsg = { message: { type: "text", text: `标题：${ title }\n作者：${ author_name }` }, nickname: e.sender.card || e.user_id, user_id: e.user_id, }; await e.reply(Bot.makeForwardMsg(origin_image_urls.map(item => { return { message: segment.image(item), nickname: e.sender.card || e.user_id, user_id: e.user_id, }; }).concat(titleMsg))); } } /** * 构造哔哩哔哩信息 * @param videoInfo * @returns {Promise<(string|string)[]>} */ async constructBiliInfo(videoInfo) { const { title, desc, bvid, cid, pic } = videoInfo; // 视频信息 const { view, danmaku, reply, favorite, coin, share, like } = videoInfo.stat; // 格式化数据 let combineContent = ""; // 是否显示信息 if (this.biliDisplayInfo) { // 构造一个可扩展的Map const dataProcessMap = { "点赞": like, "硬币": coin, "收藏": favorite, "分享": share, "总播放量": view, "弹幕数量": danmaku, "评论": reply }; combineContent += `\n${ formatBiliInfo(dataProcessMap) }`; } // 是否显示简介 if (this.biliDisplayIntro) { // 过滤简介中的一些链接 const filteredDesc = await filterBiliDescLink(desc); combineContent += `\n📝 简介：${ truncateString(filteredDesc, this.toolsConfig.biliIntroLenLimit || BILI_DEFAULT_INTRO_LEN_LIMIT) }`; } // 是否显示在线人数 if (this.biliDisplayOnline) { // 拼接在线人数 const onlineTotal = await this.biliOnlineTotal(bvid, cid); combineContent += `\n🏄‍♂️️ 当前视频有 ${ onlineTotal.total } 人在观看，其中 ${ onlineTotal.count } 人在网页端观看`; } let biliInfo = [`${ this.identifyPrefix }识别：哔哩哔哩，${ title }`, combineContent]; // 是否显示封面 if (this.biliDisplayCover) { // 加入图片 biliInfo.unshift(segment.image(pic)); } return biliInfo; } /** * 获取哔哩哔哩番剧信息 * @param url * @param e * @returns {Promise} */ async biliEpInfo(url, e) { let ep; // 处理ssid if (url.includes("play\/ss")) { const ssid = url.match(/\/ss(\d+)/)?.[1]; let resp = await (await fetch(BILI_SSID_INFO.replace("{}", ssid), { headers: BILI_HEADER })).json(); ep = (resp.result.main_section.episodes[0].share_url).replace("https://www.bilibili.com/bangumi/play/ep", ""); } // 处理普通情况，上述情况无法处理的 if (_.isEmpty(ep)) { ep = url.match(/\/ep(\d+)/)?.[1]; } const resp = await (await fetch(BILI_EP_INFO.replace("{}", ep), { headers: BILI_HEADER })).json(); const result = resp.result; const { views, danmakus, likes, coins, favorites, favorite } = result.stat; // 封装成可以format的数据 const dataProcessMap = { "播放": views, "弹幕": danmakus, "点赞": likes, "分享": coins, "追番": favorites, "收藏": favorite, }; // 截断标题，查看Redis中是否存在，避免频繁走网络连接 const title = result.title; e.reply([ segment.image(resp.result.cover), `${ this.identifyPrefix }识别：哔哩哔哩番剧，${ title }\n🎯 评分: ${ result?.rating?.score ?? '-' } / ${ result?.rating?.count ?? '-' }\n📺 ${ result.new_ep.desc }, ${ result.seasons[0].new_ep.index_show }\n`, `${ formatBiliInfo(dataProcessMap) }`, `\n\n🪶 在线观看： ${ await urlTransformShortLink(ANIME_SERIES_SEARCH_LINK + title) }`, `\n🌸 在线观看： ${ await urlTransformShortLink(ANIME_SERIES_SEARCH_LINK2 + title) }` ], true); return ep; } /** * 哔哩哔哩下载策略 * @param e 事件 * @param url 链接 * @param path 保存路径 * @returns {Promise} */ async biliDownloadStrategy(e, url, path) { return this.queue.add(async () => { // =================以下是调用BBDown的逻辑===================== // 下载视频和音频 const tempPath = `${ path }temp`; // 检测是否开启BBDown if (this.biliUseBBDown) { // 检测环境的 BBDown const isExistBBDown = await checkToolInCurEnv("BBDown"); // 存在 BBDown if (isExistBBDown) { // 删除之前的文件 await checkAndRemoveFile(`${ tempPath }.mp4`); // 下载视频 await startBBDown(url, path, { biliSessData: this.biliSessData, biliUseAria2: this.biliDownloadMethod === 1, biliCDN: BILI_CDN_SELECT_LIST.find(item => item.value === this.biliCDN)?.sign, biliResolution: this.biliResolution, }); // 发送视频 return this.sendVideoToUpload(e, `${ tempPath }.mp4`); } e.reply("🚧 R插件提醒你：开启但未检测到当前环境有【BBDown】，即将使用默认下载方式 ( ◡̀_◡́)ᕤ"); } // =================默认下载方式===================== try { // 获取分辨率参数 QN，如果没有默认使用 480p --> 32 const qn = BILI_RESOLUTION_LIST.find(item => item.value === this.biliResolution).qn || 32; // 获取下载链接 const data = await getDownloadUrl(url, this.biliSessData, qn); if (data.audioUrl != null) { await this.downBili(tempPath, data.videoUrl, data.audioUrl); } else { // 处理无音频的情况 await downloadBFile(data.videoUrl, `${ tempPath }.mp4`, _.throttle( value => logger.mark("视频下载进度", { data: value, }), 1000, )); } // 上传视频 return this.sendVideoToUpload(e, `${ tempPath }.mp4`); } catch (err) { // 错误处理 logger.error('[R插件][哔哩哔哩视频发送]下载错误，具体原因为:', err); e.reply("解析失败，请重试一下"); } }); } /** * 获取在线人数 * @param bvid * @param cid * @returns {Promise<{total: *, count: *}>} */ async biliOnlineTotal(bvid, cid) { const onlineResp = await axios.get(BILI_ONLINE.replace("{0}", bvid).replace("{1}", cid)); const online = onlineResp.data.data; return { total: online.total, count: online.count }; } // 下载哔哩哔哩音乐 async biliMusic(e, url) { const videoId = /video\/[^\?\/ ]+/.exec(url)[0].split("/")[1]; this.queue.add(() => { getBiliAudio(videoId, "").then(async audioUrl => { const path = this.getCurDownloadPath(e); const biliMusicPath = await m4sToMp3(audioUrl, path); // 发送语音 e.reply(segment.record(biliMusicPath)); // 上传群文件 await this.uploadGroupFile(e, biliMusicPath); }); }); return true; } // 发送哔哩哔哩动态的算法 biliDynamic(e, url, session) { // 去除多余参数 if (url.includes("?")) { url = url.substring(0, url.indexOf("?")); } const dynamicId = /[^/]+(?!.*\/)/.exec(url)[0]; getDynamic(dynamicId, session).then(async resp => { if (resp.dynamicSrc.length > 0 || resp.dynamicDesc) { e.reply(`${ this.identifyPrefix }识别：哔哩哔哩动态\n${ resp.dynamicDesc }`); let dynamicSrcMsg = []; resp.dynamicSrc.forEach(item => { dynamicSrcMsg.push({ message: segment.image(item), nickname: e.sender.card || e.user_id, user_id: e.user_id, }); }); await e.reply(await Bot.makeForwardMsg(dynamicSrcMsg)); } else { e.reply(`识别：哔哩哔哩动态, 但是失败！`); } }); return url; } /** * 哔哩哔哩总结 * @author zhiyu1998 * @param bvid 稿件 * @param cid 视频 cid * @param up_mid UP主 mid * @return {Promise} */ async getBiliSummary(bvid, cid, up_mid) { // 这个有点用，但不多 let wbi = "wts=1701546363&w_rid=1073871926b3ccd99bd790f0162af634"; if (!_.isEmpty(this.biliSessData)) { wbi = await getWbi({ bvid, cid, up_mid }, this.biliSessData); } // 构造API const summaryUrl = `${ BILI_SUMMARY }?${ wbi }`; logger.info(summaryUrl); // 构造结果：https://api.bilibili.com/x/web-interface/view/conclusion/get?bvid=BV1L94y1H7CV&cid=1335073288&up_mid=297242063&wts=1701546363&w_rid=1073871926b3ccd99bd790f0162af634 return axios.get(summaryUrl) .then(resp => { const data = resp.data.data?.model_result; // logger.info(data) const summary = data?.summary; const outline = data?.outline; let resReply = ""; // 总体总结 if (summary) { resReply = `\n摘要：${ summary }\n`; } // 分段总结 if (outline) { const specificTimeSummary = outline.map(item => { const smallTitle = item.title; const keyPoint = item?.part_outline; // 时间点的总结 const specificContent = keyPoint.map(point => { const { timestamp, content } = point; const specificTime = secondsToTime(timestamp); return `${ specificTime } ${ content }\n`; }).join(""); return `- ${ smallTitle }\n${ specificContent }\n`; }); resReply += specificTimeSummary.join(""); } return resReply; }); } /** * 获取直播间信息 * @param liveId * @returns {Promise<*>} */ async getBiliStreamInfo(liveId) { return axios.get(`${ BILI_STREAM_INFO }?room_id=${ liveId }`, { headers: { 'User-Agent': COMMON_USER_AGENT, } }); } /** * 获取直播流 * @param liveId * @returns {Promise<*>} */ async getBiliStream(liveId) { return axios.get(`${ BILI_STREAM_FLV }?cid=${ liveId }`, { headers: { 'User-Agent': COMMON_USER_AGENT, } }); } /** * @deprecated Use newFunction instead. */ async twitter(e) { console.warn('警告: 函数已弃用，将在未来版本中移除'); // 配置参数及解析 const reg = /https?:\/\/twitter.com\/[0-9-a-zA-Z_]{1,20}\/status\/([0-9]*)/; const twitterUrl = reg.exec(e.msg); const id = twitterUrl[1]; // 判断是否是海外服务器，默认为false const isOversea = !(await this.isOverseasServer()); // 请求 const params = { "ids": id, "media.fields": "duration_ms,height,media_key,preview_image_url,public_metrics,type,url,width,alt_text,variants", "expansions": ["entities.mentions.username", "attachments.media_keys"], }; await fetch(TWITTER_TWEET_INFO.replace("{}", id), { headers: { "User-Agent": "v2TweetLookupJS", "authorization": `Bearer ${ Buffer.from(TWITTER_BEARER_TOKEN, "base64").toString() }` }, ...params, agent: !isOversea ? '' : new HttpsProxyAgent(this.myProxy), }).then(async resp => { logger.info(resp); e.reply(`${ this.identifyPrefix }识别：小蓝鸟学习版，${ resp.data.text }`); const downloadPath = `${ this.getCurDownloadPath(e) }`; // 创建文件夹（如果没有过这个群） if (!fs.existsSync(downloadPath)) { mkdirsSync(downloadPath); } // 逐个遍历判断 let task = []; for (let item of resp.includes.media) { if (item.type === "photo") { // 图片 task.push(downloadImg(item.url, downloadPath, "", true)); } else if (item.type === "video") { // 视频 await this.downloadVideo(resp.includes.media[0].variants[0].url, true).then( _ => { e.reply(segment.video(`${ downloadPath }/temp.mp4`)); }, ); } } // 如果没有图片直接返回走 if (task.length === 0) { return true; } // 下面是有图片的情况 let images = []; let path = []; // 获取所有图片的promise await Promise.all(task).then(resp => { // console.log(resp) resp.forEach(item => { path.push(item); images.push({ message: segment.image(fs.readFileSync(item)), nickname: this.e.sender.card || this.e.user_id, user_id: this.e.user_id, }); }); }); await e.reply(await Bot.makeForwardMsg(images)); // 清理文件 path.forEach(item => { fs.unlinkSync(item); }); }); return true; } // 使用现有api解析小蓝鸟 async twitter_x(e) { if (!(await this.isTrustUser(e.user_id))) { e.reply("你没有权限使用此命令"); return; } // 配置参数及解析 const reg = /https:\/\/x\.com\/[\w]+\/status\/\d+(\/photo\/\d+)?/; const twitterUrl = reg.exec(e.msg)[0]; // 检测 const isOversea = await this.isOverseasServer(); if (!isOversea && !(await testProxy(this.proxyAddr, this.proxyPort))) { e.reply("检测到没有梯子，无法解析小蓝鸟"); return false; } // 提取视频 let videoUrl = GENERAL_REQ_LINK.link.replace("{}", twitterUrl); e.reply(`${ this.identifyPrefix }识别：小蓝鸟学习版`); const config = { headers: { 'Accept': 'ext/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7', 'Accept-Encoding': 'gzip, deflate', 'Accept-Language': 'zh-CN,zh;q=0.9', 'Host': '47.99.158.118', 'Proxy-Connection': 'keep-alive', 'Upgrade-Insecure-Requests': '1', 'User-Agent': COMMON_USER_AGENT, }, timeout: 10000 // 设置超时时间 }; let resp = await axios.get(videoUrl, config); if (resp.data.data == null) { videoUrl += '/photo/1'; logger.info(videoUrl); resp = await axios.get(videoUrl, config); } const url = resp.data.data?.url; if (url && (url.endsWith(".jpg") || url.endsWith(".png"))) { if (isOversea) { // 海外直接下载 e.reply(segment.image(url)); } else { // 非海外使用🪜下载 const localPath = this.getCurDownloadPath(e); const xImgPath = await downloadImg({ img: url, dir: localPath, isProxy: !isOversea, proxyInfo: { proxyAddr: this.proxyAddr, proxyPort: this.proxyPort }, downloadMethod: this.biliDownloadMethod, }); e.reply(segment.image(xImgPath)); } } else { this.downloadVideo(url, !isOversea).then(path => { e.reply(segment.video(path + "/temp.mp4")); }); } return true; } // acfun解析 async acfun(e) { const path = `${ this.getCurDownloadPath(e) }/temp/`; await mkdirIfNotExists(path); let inputMsg = e.msg; // 适配手机分享：https://m.acfun.cn/v/?ac=32838812&sid=d2b0991bd6ad9c09 if (inputMsg.includes("m.acfun.cn")) { inputMsg = `https://www.acfun.cn/v/ac${ /ac=([^&?]*)/.exec(inputMsg)[1] }`; } else if (inputMsg.includes("ac")) { // 如果是 inputMsg = "https://www.acfun.cn/v/" + /ac\d+/.exec(inputMsg)[0]; } parseUrl(inputMsg).then(res => { e.reply(`${ this.identifyPrefix }识别：猴山，${ res.videoName }`); parseM3u8(res.urlM3u8s[res.urlM3u8s.length - 1]).then(res2 => { downloadM3u8Videos(res2.m3u8FullUrls, path).then(_ => { mergeAcFileToMp4(res2.tsNames, path, `${ path }out.mp4`).then(_ => { this.sendVideoToUpload(e, `${ path }out.mp4`); }); }); }); }); return true; } // 小红书解析 async xhs(e) { // 正则说明：匹配手机链接、匹配小程序、匹配PC链接 let msgUrl = /(http:|https:)\/\/(xhslink|xiaohongshu).com\/[A-Za-z\d._?%&+\-=\/#@]*/.exec( e.msg, )?.[0] || /(http:|https:)\/\/www\.xiaohongshu\.com\/discovery\/item\/(\w+)/.exec( e.message[0].data, )?.[0] || /(http:|https:)\/\/www\.xiaohongshu\.com\/explore\/(\w+)/.exec( e.msg, )?.[0] || /(http:|https:)\/\/www\.xiaohongshu\.com\/discovery\/item\/(\w+)/.exec( e.msg, )?.[0]; // 注入ck XHS_NO_WATERMARK_HEADER.cookie = this.xiaohongshuCookie; // 解析短号 let id, xsecToken, xsecSource; if (msgUrl.includes("xhslink")) { await fetch(msgUrl, { headers: XHS_NO_WATERMARK_HEADER, redirect: "follow", }).then(resp => { const uri = decodeURIComponent(resp.url); const parsedUrl = new URL(resp.url); // 如果出现了网页验证uri:https://www.xiaohongshu.com/website-login/captcha?redirectPath=https://www.xiaohongshu.com/discovery/item/63c93ac3000000002203b28a?app_platform=android&app_version=8.23.1&author_share=1&ignoreEngage=true&share_from_user_hidden=true&type=normal&xhsshare=CopyLink&appuid=62c58b90000000000303dc54&apptime=1706149572&exSource=&verifyUuid=a5f32b62-453e-426b-98fe-2cfe0c16776d&verifyType=102&verifyBiz=461 const verify = uri.match(/\/item\/([0-9a-fA-F]+)/); // 一般情况下不会出现问题就使用这个正则 id = /noteId=(\w+)/.exec(uri)?.[1] ?? verify?.[1]; // 提取 xsec_source 和 xsec_token 参数 xsecSource = parsedUrl.searchParams.get("xsec_source") ?? "pc_feed"; xsecToken = parsedUrl.searchParams.get("xsec_token"); }); } else { // 新版 xhs 这里必须是e.msg.trim()，因为要匹配参数：xsec_source 和 xsec_token const xhsUrlMatch = e.msg.trim().replace("amp;", "").match(/(http|https)?:\/\/(www\.)?xiaohongshu\.com[^\s]+/); if (!xhsUrlMatch) { logger.info("[R插件][xhs] 无法匹配到链接"); return; } const parsedUrl = new URL(xhsUrlMatch[0]); id = /explore\/(\w+)/.exec(msgUrl)?.[1] || /discovery\/item\/(\w+)/.exec(msgUrl)?.[1]; // 提取 xsec_source 和 xsec_token 参数 xsecSource = parsedUrl.searchParams.get("xsec_source") ?? "pc_feed"; xsecToken = parsedUrl.searchParams.get("xsec_token"); } const downloadPath = `${ this.getCurDownloadPath(e) }`; // 检测没有 cookie 则退出 if (_.isEmpty(this.xiaohongshuCookie) || _.isEmpty(id) || _.isEmpty(xsecToken) || _.isEmpty(xsecSource)) { e.reply(`请检查以下问题：\n1. 是否填写 Cookie\n2. 链接是否有id\n3. 链接是否有 xsec_token 和 xsec_source\n${ HELP_DOC }`); return; } // 获取信息 const resp = await fetch(`${ XHS_REQ_LINK }${ id }?xsec_token=${ xsecToken }&xsec_source=${ xsecSource }`, { headers: XHS_NO_WATERMARK_HEADER, }); // 从网页获取数据 const xhsHtml = await resp.text(); const reg = /window\.__INITIAL_STATE__=(.*?)<\/script>/; const res = xhsHtml.match(reg)[1].replace(/undefined/g, "null"); const resJson = JSON.parse(res); // saveJsonToFile(resJson); // 检测无效 Cookie if (resJson?.note === undefined || resJson?.note?.noteDetailMap?.[id]?.note === undefined) { e.reply(`检测到无效的小红书 Cookie，可以尝试清除缓存和cookie 或者 换一个浏览器进行获取\n${ HELP_DOC }`); return; } // 提取出数据 const noteData = resJson?.note?.noteDetailMap?.[id]?.note; const { title, desc, type } = noteData; if (type === "video") { // 封面 const cover = noteData.imageList?.[0].urlDefault; e.reply([segment.image(cover), `${ this.identifyPrefix }识别：小红书, ${ title }\n${ desc }`]); // ⚠️ （暂时废弃）构造xhs视频链接（有水印） const xhsVideoUrl = noteData.video.media.stream.h264?.[0]?.masterUrl; // 构造无水印 // const xhsVideoUrl = `http://sns-video-bd.xhscdn.com/${ noteData.video.consumer.originVideoKey }` // 下载视频 this.downloadVideo(xhsVideoUrl).then(path => { if (path === undefined) { // 创建文件，如果不存在 path = `${ this.getCurDownloadPath(e) }/`; } this.sendVideoToUpload(e, `${ path }/temp.mp4`); }); return true; } else if (type === "normal") { e.reply(`${ this.identifyPrefix }识别：小红书, ${ title }\n${ desc }`); const imagePromises = []; // 使用 for..of 循环处理异步下载操作 for (let [index, item] of noteData.imageList.entries()) { imagePromises.push(downloadImg({ img: item.urlDefault, dir: downloadPath, fileName: `${ index }.png`, downloadMethod: this.biliDownloadMethod, })); } // 等待所有图片下载完成 const paths = await Promise.all(imagePromises); // 直接构造 imagesData 数组 const imagesData = await Promise.all(paths.map(async (item) => { return { message: segment.image(await fs.promises.readFile(item)), nickname: e.sender.card || e.user_id, user_id: e.user_id, }; })); // 回复带有转发消息的图片数据 e.reply(await Bot.makeForwardMsg(imagesData)); // 批量删除下载的文件 await Promise.all(paths.map(item => fs.promises.rm(item, { force: true }))); } return true; } // 波点音乐解析 async bodianMusic(e) { // 音频例子：https://h5app.kuwo.cn/m/bodian/playMusic.html?uid=3216773&musicId=192015898&opusId=&extendType=together // 视频例子：https://h5app.kuwo.cn/m/bodian/play.html?uid=3216773&mvId=118987&opusId=770096&extendType=together const id = /(?=musicId).*?(?=&)/.exec(e.msg.trim())?.[0].replace("musicId=", "") || /(?=mvId).*?(?=&)/.exec(e.msg.trim())?.[0].replace("mvId=", ""); const { name, album, artist, albumPic120, categorys } = await getBodianMusicInfo(id); e.reply([ `${ this.identifyPrefix }识别：波点音乐，${ name }-${ album }-${ artist }\n标签：${ categorys .map(item => item.name) .join(" | ") }`, segment.image(albumPic120), ]); if (e.msg.includes("musicId")) { const path = `${ this.getCurDownloadPath(e) }`; await getBodianAudio(id, path, `${ name }-${ artist }`).then(sendPath => { // 发送语音 e.reply(segment.record(sendPath)); // 上传群文件 this.uploadGroupFile(e, sendPath); // 删除文件 checkAndRemoveFile(sendPath); }); } else if (e.msg.includes("mvId")) { await getBodianMv(id).then(res => { // 下载 && 发送 const { coverUrl, highUrl, lowUrl, shortLowUrl } = res; this.downloadVideo(lowUrl).then(path => { e.reply(segment.video(path + "/temp.mp4")); }); }); } return true; } // 网易云登录状态 async neteaseStatus(e, reck) { // 优先判断是否使用自建 API let autoSelectNeteaseApi = this.useLocalNeteaseAPI ? this.neteaseCloudAPIServer : (await this.isOverseasServer() ? NETEASE_SONG_DOWNLOAD : NETEASE_API_CN); const statusUrl = `${ autoSelectNeteaseApi }/login/status`; try { const statusResponse = await axios.get(statusUrl, { headers: { "User-Agent": COMMON_USER_AGENT, "Cookie": reck ? reck : this.neteaseCookie, }, }); const userInfo = statusResponse.data?.data?.profile; if (!userInfo) { e.reply('暂未登录，请发 #RNQ 或者 #rnq 进行登陆绑定ck'); return; } const vipResponse = await axios.get(`${ autoSelectNeteaseApi }/vip/info?uid=${ userInfo.userId }`, { headers: { "User-Agent": COMMON_USER_AGENT, "Cookie": reck ? reck : this.neteaseCookie, }, }); const vipInfo = vipResponse.data?.data; const checkVipStatus = async (vipLevel, expireTime, nickname, avatarUrl) => { const expireDate = new Date(expireTime); if (expireDate > Date.now()) { const vipLevelData = vipLevel.split("\n"); const neteaseData = await new NeteaseModel(e).getData({ avatarUrl: `${ avatarUrl }?param=170y170`, nickname, vipLevel: vipLevelData[0], musicQuality: vipLevelData[2], expireDate: expireDate.toLocaleString(), }); let img = await puppeteer.screenshot("netease", neteaseData); e.reply(img, true); return true; } return false; }; if (vipInfo.redplus.vipCode !== 0 && await checkVipStatus(`SVIP${ vipInfo.redplus.vipLevel }\n最高解析音质:\n jymaster(超清母带)`, vipInfo.redplus.expireTime, userInfo.nickname, userInfo.avatarUrl)) { return; } if (vipInfo.associator.vipCode !== 0 && await checkVipStatus(`VIP${ vipInfo.associator.vipLevel }\n最高解析音质:\n jyeffect(高清环绕音)`, vipInfo.associator.expireTime, userInfo.nickname, userInfo.avatarUrl)) { return; } // 如果都已过期，发送 VIP 已过期信息 const neteaseData = await new NeteaseModel(e).getData({ avatarUrl: `${ userInfo.avatarUrl }?param=170y170`, nickname: userInfo.nickname, vipLevel: vipInfo.redplus.vipCode !== 0 ? `SVIP${ vipInfo.redplus.vipLevel }(已过期)` : vipInfo.associator.vipCode !== 0 ? `VIP${ vipInfo.associator.vipLevel }(已过期)` : '未开通', musicQuality: 'standard(标准)', expireDate: '未开通', }); let img = await puppeteer.screenshot("netease", neteaseData); e.reply(img, true); } catch (error) { logger.error('获取网易云状态时出错:', error); e.reply('获取网易云状态时出错，请稍后再试'); } } // 轮询网易云状态 async pollLoginStatus(autoSelectNeteaseApi, unikey, e) { let pollCount = 0; const maxPolls = 8; const intervalTime = 5000; const pollRequest = async () => { try { const pollUrl = `${ autoSelectNeteaseApi }/login/qr/check?key=${ unikey }×tamp=${ Date.now() }`; const res = await axios.get(pollUrl, { headers: { "User-Agent": COMMON_USER_AGENT } }); if (res.data.code == '800') { e.reply("二维码过期，请重新获取"); clearInterval(intervalId); return; } if (res.data.code == '803') { const regex = /music_u=([^;]+)/i; const match = res.data.cookie.match(regex); if (match) { try { const ck = `${ match[0] }; os=pc`; await config.updateField("tools", "neteaseCookie", ck); this.neteaseStatus(e, ck); e.reply(`扫码登录成功，ck已自动保存`); } catch (error) { logger.error('更新ck时出错:', error); e.reply('更新ck时出错，请稍后重试'); } } clearInterval(intervalId); return; } pollCount++; if (pollCount > maxPolls) { clearInterval(intervalId); logger.info('超时轮询已停止'); e.reply('扫码超时，请重新获取'); } } catch (error) { logger.error('轮询过程中出错:', error); clearInterval(intervalId); e.reply('轮询过程中发生错误，请稍后再试'); } }; const intervalId = setInterval(pollRequest, intervalTime); } // 网易扫码登录 async netease_scan(e) { try { // 优先判断是否使用自建 API const isOversea = await this.isOverseasServer(); let autoSelectNeteaseApi; if (this.useLocalNeteaseAPI) { autoSelectNeteaseApi = this.neteaseCloudAPIServer; } else { autoSelectNeteaseApi = (isOversea ? NETEASE_SONG_DOWNLOAD : NETEASE_API_CN); await e.reply('未使用自建服务器，高概率#rnq失败'); } // 获取登录key const keyUrl = `${ autoSelectNeteaseApi }/login/qr/key`; const keyResponse = await axios.get(keyUrl, { headers: { "User-Agent": COMMON_USER_AGENT } }); const unikey = keyResponse.data.data.unikey; // 获取登录二维码 const qrUrl = `${ autoSelectNeteaseApi }/login/qr/create?key=${ unikey }&qrimg=true`; const qrResponse = await axios.get(qrUrl, { headers: { "User-Agent": COMMON_USER_AGENT } }); await mkdirIfNotExists(this.defaultPath); const saveCodePath = `${ this.defaultPath }NeteaseQrcode.png`; await qrcode.toFile(saveCodePath, qrResponse.data.data.qrurl); e.reply([segment.image(saveCodePath), '请在40秒内使用网易云APP进行扫码']); // 轮询检查登录状态 await this.pollLoginStatus(autoSelectNeteaseApi, unikey, e); } catch (error) { if (error.code == 'ERR_INVALID_URL') { logger.error('执行网易云扫码登录时出错:非法地址，请检查API服务地址', error); e.reply(`执行网易云扫码登录时出错${ error.code }请检查API服务器地址`); } else if (error.code == 'ECONNRESET') { logger.error('执行网易云扫码登录时出错:API请求错误，请检查API服务状态', error); e.reply(`执行扫码登录时发生错误${ error.code }请检查API服务状态`); } else { logger.error('执行网易云扫码登录时出错:', error); e.reply('执行扫码登录时发生错误，请稍后再试'); } } } // 网易云解析 async netease(e) { let message = e.msg === undefined ? e.message.shift().data.replaceAll("\\", "") : e.msg.trim(); // 处理短号，此时会变成y.music.163.com if (message.includes("163cn.tv")) { message = /(http:|https:)\/\/163cn\.tv\/([a-zA-Z0-9]+)/.exec(message)?.[0]; // logger.info(message) message = await axios.head(message).then((resp) => { return resp.request.res.responseUrl; }); } // 处理网页链接 const musicUrlReg = /(http:|https:)\/\/music.163.com\/song\/media\/outer\/url\?id=(\d+)/; const musicUrlReg2 = /(http:|https:)\/\/y.music.163.com\/m\/song\?(.*)&id=(\d+)/; const musicUrlReg3 = /(http:|https:)\/\/music.163.com\/m\/song\/(\d+)/; const id = musicUrlReg2.exec(message)?.[3] || musicUrlReg.exec(message)?.[2] || musicUrlReg3.exec(message)?.[2] || /(? { const userInfo = res.data.data.profile; if (userInfo) { logger.info('[R插件][netease]ck活着，使用ck进行高音质下载'); return true; } else { logger.info('[R插件][netease]ck失效，将启用临时接口下载'); return false; } }); // mv截断 if (/mv\?/.test(message)) { const AUTO_NETEASE_MV_DETAIL = autoSelectNeteaseApi + "/mv/detail?mvid={}"; const AUTO_NETEASE_MV_URL = autoSelectNeteaseApi + "/mv/url?id={}"; // logger.info(AUTO_NETEASE_MV_DETAIL.replace("{}", id)); // logger.info(AUTO_NETEASE_MV_URL.replace("{}", id)); const [mvDetailData, mvUrlData] = await Promise.all([ axios.get(AUTO_NETEASE_MV_DETAIL.replace("{}", id), { headers: { "User-Agent": COMMON_USER_AGENT, "Cookie": this.neteaseCookie } }), axios.get(AUTO_NETEASE_MV_URL.replace("{}", id), { headers: { "User-Agent": COMMON_USER_AGENT, "Cookie": this.neteaseCookie } }) ]); const { name: mvName, artistName: mvArtist, cover: mvCover } = mvDetailData.data?.data; e.reply([segment.image(mvCover), `${ this.identifyPrefix }识别：网易云MV，${ mvName } - ${ mvArtist }`]); // logger.info(mvUrlData.data) const { url: mvUrl } = mvUrlData.data?.data; this.downloadVideo(mvUrl).then(path => { this.sendVideoToUpload(e, `${ path }/temp.mp4`); }); return; } const songWikiUrl = autoSelectNeteaseApi + '/song/wiki/summary?id=' + id; // 国内解决方案，替换为国内API (其中，NETEASE_API_CN是国内基址) const AUTO_NETEASE_SONG_DOWNLOAD = autoSelectNeteaseApi + "/song/url/v1?id={}&level=" + this.neteaseCloudAudioQuality; const AUTO_NETEASE_SONG_DETAIL = autoSelectNeteaseApi + "/song/detail?ids={}"; // logger.info(AUTO_NETEASE_SONG_DOWNLOAD.replace("{}", id)); const downloadUrl = AUTO_NETEASE_SONG_DOWNLOAD.replace("{}", id); const detailUrl = AUTO_NETEASE_SONG_DETAIL.replace("{}", id); // 请求netease数据 axios.get(downloadUrl, { headers: { "User-Agent": COMMON_USER_AGENT, "Cookie": this.neteaseCookie }, }).then(async resp => { // 国内解决方案，替换API后这里也需要修改 // 英转中字典匹配 const translationDict = { 'standard': '标准', 'higher': '较高', 'exhigh': '极高', 'lossless': '无损', 'hires': 'Hi-Res', 'jyeffect': '高清环绕声', 'sky': '沉浸环绕声', 'dolby': '杜比全景声', 'jymaster': '超清母带' }; // 英转中 function translateToChinese(word) { return translationDict[word] || word; // 如果找不到对应翻译，返回原词 } // 字节转MB function bytesToMB(sizeInBytes) { const sizeInMB = sizeInBytes / (1024 * 1024); // 1 MB = 1024 * 1024 bytes return sizeInMB.toFixed(2); // 保留两位小数 } let url = await resp.data.data?.[0]?.url || null; const AudioLevel = translateToChinese(resp.data.data?.[0]?.level); const AudioSize = bytesToMB(resp.data.data?.[0]?.size); // 获取歌曲信息 let { songName, artistName } = await axios.get(detailUrl).then(res => { const song = res.data.songs[0]; return { songName: cleanFilename(song?.name), artistName: cleanFilename(song?.ar?.[0].name) }; }); let title = artistName + '-' + songName; // 获取歌曲封面 let coverUrl = await axios.get(detailUrl).then(res => { const song = res.data.songs[0]; return song?.al?.picUrl; }); let typelist = []; // 歌曲百科API await axios.get(songWikiUrl, { headers: { "User-Agent": COMMON_USER_AGENT, // "Cookie": this.neteaseCookie }, }).then(res => { const wikiData = res.data.data.blocks[1].creatives; try { typelist.push(wikiData[0].resources[0]?.uiElement?.mainTitle?.title || ""); // 防止数据过深出错 const recTags = wikiData[1]; if (recTags?.resources[0]) { for (let i = 0; i < Math.min(3, recTags.resources.length); i++) { if (recTags.resources[i] && recTags.resources[i].uiElement && recTags.resources[i].uiElement.mainTitle.title) { typelist.push(recTags.resources[i].uiElement.mainTitle.title); } } } else { if (recTags.uiElement.textLinks[0].text) typelist.push(recTags.uiElement.textLinks[0].text); } if (wikiData[2].uiElement.mainTitle.title == 'BPM') { typelist.push('BPM ' + wikiData[2].uiElement.textLinks[0].text); } else { typelist.push(wikiData[2].uiElement.textLinks[0].text || ''); } } catch (error) { logger.error('获取标签报错：', error); } typelist.push(AudioLevel); }); let musicInfo = { 'cover': coverUrl, 'songName': songName, 'singerName': artistName, 'size': AudioSize + " MB", 'musicType': typelist }; // 一般这个情况是VIP歌曲 (如果没有url或者是国内,公用接口暂时不可用，必须自建并且ck可用状态才能进行高质量解析) if (!isCkExpired || url == null) { url = await this.musicTempApi(e, title, "网易云音乐", musicInfo); } else { // 拥有ck，并且有效，直接进行解析 let audioInfo = AudioLevel; if (AudioLevel == '杜比全景声') { audioInfo += '\n(杜比下载文件为MP4，编码格式为AC-4，需要设备支持才可播放)'; } const data = await new NeteaseMusicInfo(e).getData(musicInfo); let img = await puppeteer.screenshot("neteaseMusicInfo", data); await e.reply(img); // e.reply([segment.image(coverUrl), `${this.identifyPrefix}识别：网易云音乐，${title}\n当前下载音质: ${audioInfo}\n预估大小: ${AudioSize}MB`]); } // 动态判断后缀名 let musicExt = resp.data.data?.[0]?.type; // 下载音乐 downloadAudio(url, this.getCurDownloadPath(e), title, 'follow', musicExt).then(async path => { // 发送群文件 await this.uploadGroupFile(e, path); // 发送语音 if (musicExt != 'mp4' && this.isSendVocal) { await e.reply(segment.record(path)); } // 删除文件 await checkAndRemoveFile(path); }).catch(err => { logger.error(`下载音乐失败，错误信息为: ${ err }`); }); }); return true; } // 临时接口 async musicTempApi(e, title, musicType, musicInfo = {}) { let musicReqApi = NETEASE_TEMP_API; if (musicType === "QQ音乐") { musicReqApi = QQ_MUSIC_TEMP_API; } else if (musicType === "汽水音乐") { musicReqApi = QISHUI_MUSIC_TEMP_API; } // 临时接口，title经过变换后搜索到的音乐质量提升 const vipMusicData = await axios.get(musicReqApi.replace("{}", title.replace("-", " ")), { headers: { "User-Agent": COMMON_USER_AGENT, }, }); // const messageTitle = title + "\nR插件检测到当前为VIP音乐，正在转换..."; // ??后的内容是适配`QQ_MUSIC_TEMP_API`、最后是汽水 const url = vipMusicData.data?.music_url ?? vipMusicData.data?.data?.music_url ?? vipMusicData.data?.music; const cover = vipMusicData.data?.cover ?? vipMusicData.data?.data?.cover ?? vipMusicData.data?.cover; const name = vipMusicData.data?.title ?? vipMusicData.data?.data?.song_name ?? vipMusicData.data?.title; const singer = vipMusicData.data?.singer ?? vipMusicData.data?.data?.song_singer ?? vipMusicData.data?.singer; const id = vipMusicData.data?.id ?? vipMusicData.data?.data?.quality ?? vipMusicData.data?.pay; if (musicType === "网易云音乐") { musicInfo.size = id; musicInfo.musicType = musicInfo.musicType.slice(0, -1); const data = await new NeteaseMusicInfo(e).getData(musicInfo); let img = await puppeteer.screenshot("neteaseMusicInfo", data); await e.reply(img); } else { musicInfo = { 'cover': cover, 'songName': name, 'singerName': singer, 'size': id, 'musicType': "" }; const data = await new NeteaseMusicInfo(e).getData(musicInfo); let img = await puppeteer.screenshot("neteaseMusicInfo", data); await e.reply(img); } // await e.reply([segment.image(cover), `${this.identifyPrefix}识别：${musicType}，${messageTitle}`]); return url; } // 微博解析 async weibo(e) { let weiboId; const weiboUrl = e.msg === undefined ? e.message.shift().data.replaceAll("\\", "") : e.msg.trim().replaceAll("\\", ""); // 对已知情况进行判断 if (weiboUrl.includes("m.weibo.cn")) { // https://m.weibo.cn/detail/4976424138313924 weiboId = /(?<=detail\/)[A-Za-z\d]+/.exec(weiboUrl)?.[0] || /(?<=m.weibo.cn\/)[A-Za-z\d]+\/[A-Za-z\d]+/.exec(weiboUrl)?.[0]; } else if (weiboUrl.includes("weibo.com\/tv\/show") && weiboUrl.includes("mid=")) { // https://weibo.com/tv/show/1034:5007449447661594?mid=5007452630158934 weiboId = /(?<=mid=)[A-Za-z\d]+/.exec(weiboUrl)?.[0]; weiboId = mid2id(weiboId); } else if (weiboUrl.includes("weibo.com")) { // https://weibo.com/1707895270/5006106478773472 weiboId = /(?<=weibo.com\/)[A-Za-z\d]+\/[A-Za-z\d]+/.exec(weiboUrl)?.[0]; } // 无法获取id就结束 if (!weiboId) { e.reply("解析失败：无法获取到wb的id"); return; } const id = weiboId.split("/")[1] || weiboId; axios.get(WEIBO_SINGLE_INFO.replace("{}", id), { headers: { "User-Agent": COMMON_USER_AGENT, "accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9", "cookie": "_T_WM=40835919903; WEIBOCN_FROM=1110006030; MLOGIN=0; XSRF-TOKEN=4399c8", "Referer": `https://m.weibo.cn/detail/${ id }`, } }) .then(async resp => { const wbData = resp.data.data; const { text, status_title, source, region_name, pics, page_info } = wbData; e.reply(`${ this.identifyPrefix }识别：微博，${ text.replace(/<[^>]+>/g, '') }\n${ status_title }\n${ source }\t${ region_name ?? '' }`); if (pics) { // 下载图片并格式化消息 const imagesPromise = pics.map(item => { return downloadImg({ img: item?.large.url || item.url, dir: this.getCurDownloadPath(e), headersExt: { "Referer": "http://blog.sina.com.cn/", }, downloadMethod: this.biliDownloadMethod, }).then(async (filePath) => { // 格式化为消息对象 return { message: segment.image(await fs.promises.readFile(filePath)), nickname: e.sender.card || e.user_id, user_id: e.user_id, // 返回路径以便后续删除 filePath }; }); }); // 等待所有图片处理完 const images = await Promise.all(imagesPromise); // 回复合并的消息 await e.reply(await Bot.makeForwardMsg(images)); // 并行删除文件 await Promise.all(images.map(({ filePath }) => checkAndRemoveFile(filePath))); } if (page_info) { // 视频 const videoUrl = page_info.urls?.mp4_720p_mp4 || page_info.urls?.mp4_hd_mp4; // 文章 if (!videoUrl) return true; try { // wb 视频只能强制使用 1，由群友@非酋提出 this.downloadVideo(videoUrl, false, { "User-Agent": COMMON_USER_AGENT, "accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9", "referer": "https://weibo.com/", }, 1).then(path => { this.sendVideoToUpload(e, `${ path }/temp.mp4`); }); } catch (err) { e.reply("视频资源获取失败"); logger.error("403错误：", err); } } }); return true; } /** * 通用解析 * @param e * @return {Promise} */ async general(e) { try { const adapter = await GeneralLinkAdapter.create(e.msg); e.reply(`${ this.identifyPrefix }识别：${ adapter.name }${ adapter.desc ? `, ${ adapter.desc }` : '' }`); logger.mark(adapter); if (adapter.images && adapter.images.length > 0) { const images = adapter.images.map(item => { return { message: segment.image(item), nickname: this.e.sender.card || this.e.user_id, user_id: this.e.user_id, }; }); e.reply(Bot.makeForwardMsg(images)); } else if (adapter.video && adapter.video !== '') { // 视频：https://www.kuaishou.com/short-video/3xhjgcmir24m4nm const url = adapter.video; this.downloadVideo(url).then(path => { logger.info(path); this.sendVideoToUpload(e, `${ path }/temp.mp4`); }); } else { e.reply("解析失败：无法获取到资源"); } } catch (err) { logger.error("解析失败 ", err); return true; } return true; } // 油管解析 async sy2b(e) { const timeRange = ytbFormatTime(this.youtubeClipTime); const isOversea = await this.isOverseasServer(); if (!isOversea && !(await testProxy(this.proxyAddr, this.proxyPort))) { e.reply("检测到没有梯子，无法解析油管"); return false; } try { const urlRex = /(?:https?:\/\/)?(www\.|music\.)?youtube\.com\/[A-Za-z\d._?%&+\-=\/#]*/g; const url2Rex = /(?:https?:\/\/)?youtu\.be\/[A-Za-z\d._?%&+\-=\/#]*/g; // 检测操作系统平台 const isWindows = process.platform === 'win32'; // 匹配并转义 URL 中的 & 符号（仅对 Windows 进行转义） let url = removeParams(urlRex.exec(e.msg)?.[0] || url2Rex.exec(e.msg)?.[0]).replace(/&/g, isWindows ? '^&' : '&'); //非最高画质，就按照设定的来 let graphics = ""; if (this.youtubeGraphicsOptions != 0) { graphics = `[height<=${ this.youtubeGraphicsOptions }]`; } const path = this.getCurDownloadPath(e); await checkAndRemoveFile(path + "/temp.mp4"); await checkAndRemoveFile(path + "/temp.flac"); await checkAndRemoveFile(path + "/thumbnail.png"); await ytDlpGetThumbnail(path, url, isOversea, this.myProxy, this.youtubeCookiePath); const title = ytDlpGetTilt(url, isOversea, this.myProxy, this.youtubeCookiePath).toString().replace(/\n/g, ''); // 音频逻辑 if (url.includes("music")) { e.reply([ segment.image(`${ path }/thumbnail.png`), `${ this.identifyPrefix }识别：油管音乐\n视频标题：${ title }` ]); await ytDlpHelper(path, url, isOversea, this.myProxy, this.videoDownloadConcurrency, true, graphics, timeRange, this.youtubeCookiePath); if (this.isSendVocal) { e.reply(segment.record(`${ path }/temp.flac`)); } this.uploadGroupFile(e, `${ path }/temp.flac`); // 发送完就截断 return; } // 下面为视频逻辑 const Duration = convertToSeconds(ytDlpGetDuration(url, isOversea, this.myProxy, this.youtubeCookiePath).toString().replace(/\n/g, '')); // logger.info('时长------',Duration) if (Duration > this.youtubeDuration) { e.reply([ segment.image(`${ path }/thumbnail.png`), `${ this.identifyPrefix }识别：油管，视频时长超限 \n视频标题：${ title }\n⌚${ DIVIDING_LINE.replace('{}', '限制说明').replace(/\n/g, '') }⌚\n视频时长：${ (Duration / 60).toFixed(2).replace(/\.00$/, '') } 分钟\n大于管理员限定解析时长：${ (this.youtubeDuration / 60).toFixed(2).replace(/\.00$/, '') } 分钟` ]); } else if (Duration > this.youtubeClipTime && timeRange != '00:00:00-00:00:00') { e.reply([ segment.image(`${ path }/thumbnail.png`), `${ this.identifyPrefix }识别：油管，视频截取中请耐心等待 \n视频标题：${ title }\n✂️${ DIVIDING_LINE.replace('{}', '截取说明').replace(/\n/g, '') }✂️\n视频时长：${ (Duration / 60).toFixed(2).replace(/\.00$/, '') } 分钟\n大于管理员限定截取时长：${ (this.youtubeClipTime / 60).toFixed(2).replace(/\.00$/, '') } 分钟\n将截取视频片段` ]); await ytDlpHelper(path, url, isOversea, this.myProxy, this.videoDownloadConcurrency, true, graphics, timeRange, this.youtubeCookiePath); this.sendVideoToUpload(e, `${ path }/temp.mp4`); } else { e.reply([segment.image(`${ path }/thumbnail.png`), `${ this.identifyPrefix }识别：油管，视频下载中请耐心等待 \n视频标题：${ title }\n视频时长：${ (Duration / 60).toFixed(2).replace(/\.00$/, '') } 分钟`]); await ytDlpHelper(path, url, isOversea, this.myProxy, this.videoDownloadConcurrency, true, graphics, timeRange, this.youtubeCookiePath); this.sendVideoToUpload(e, `${ path }/temp.mp4`); } } catch (error) { logger.error(error); throw error; // Rethrow the error so it can be handled by the caller } return true; } // 米游社 async miyoushe(e) { let url = e.msg === undefined ? e.message.shift().data.replaceAll("\\", "") : e.msg.trim(); let msg = /(?:https?:\/\/)?(m|www)\.miyoushe\.com\/[A-Za-z\d._?%&+\-=\/#]*/.exec(url)?.[0]; const id = /\/(\d+)$/.exec(msg)?.[0].replace("\/", ""); fetch(MIYOUSHE_ARTICLE.replace("{}", id), { headers: { "Accept-Encoding": "gzip, deflate, br", "Accept-Language": "zh-cn", "Connection": "keep-alive", "Host": "api-takumi.mihoyo.com", "x-rpc-app_version": "2.11.0", "x-rpc-client_type": "4", "Referer": "https://bbs.mihoyo.com/", "DS": getDS(), } }).then(async resp => { const respJson = await resp.json(); // debug专用 // fs.writeFile('data.json', JSON.stringify(respJson), (err) => { // if (err) { // logger.error('Error writing file:', err); // } else { // console.log('JSON saved to file successfully.'); // } // }); // return; const data = respJson.data.post.post; // 分别获取：封面、主题、内容、图片 const { cover, subject, content, images } = data; let realContent = ""; // safe JSON.parse try { realContent = JSON.parse(content); } catch (e) { realContent = content; } const normalMsg = `${ this.identifyPrefix }识别：米游社，${ subject }\n${ realContent?.describe || "" }`; const replyMsg = cover ? [segment.image(cover), normalMsg] : normalMsg; e.reply(replyMsg); // 图片 if (images && images.length > 1) { const replyImages = images.map(item => { return { message: segment.image(item), nickname: this.e.sender.card || this.e.user_id, user_id: this.e.user_id, }; }); e.reply(Bot.makeForwardMsg(replyImages)); } // 视频 let vod_list = respJson.data.post?.vod_list; if (vod_list.length > 0) { const resolutions = vod_list?.[0]?.resolutions; // 逐个遍历是否包含url for (let i = 0; i < resolutions.length; i++) { if (resolutions) { // 暂时选取分辨率较低的video进行解析 const videoUrl = resolutions[i].url; this.downloadVideo(videoUrl).then(path => { this.sendVideoToUpload(e, `${ path }/temp.mp4`); }); break; } } } }); } // 微视 async weishi(e) { let url = e.msg; const urlRegex = /https?:\/\/video\.weishi\.qq\.com\/\S+/g; // 执行匹配 url = url.match(urlRegex)[0]; // 消除短链接 await fetch(url, { method: "HEAD" }).then(resp => { url = resp.url; }); try { const idMatch = url.match(/id=(.*)&spid/); if (!idMatch || idMatch.length !== 2) { e.reply("识别：微视，但无法完整检测到视频ID"); // 打个日志 方便后面出bug知道位置 logger.error("[R插件][微视] 无法检测到ID，逻辑大概问题在正则表达式"); return true; } const feedId = idMatch[1]; const response = await axios.get(WEISHI_VIDEO_INFO.replace("{}", feedId), { headers: { "User-Agent": COMMON_USER_AGENT, } }); const weishiResponse = response.data; const firstFeed = weishiResponse.data.feeds[0]; // 标题、封面、视频链接 const title = firstFeed.feed_desc; const cover = firstFeed.images[0].url; const noWatermarkDownloadUrl = firstFeed.video_url; e.reply([segment.image(cover), `${ this.identifyPrefix }识别：微视，${ title }`]); this.downloadVideo(noWatermarkDownloadUrl).then(path => { this.sendVideoToUpload(e, `${ path }/temp.mp4`); }); } catch (err) { logger.error(err); return true; } return true; } async zuiyou(e) { // #最右#分享一条有趣的内容给你，不好看算我输。请戳链接>>https://share.xiaochuankeji.cn/hybrid/share/post?pid=365367131&zy_to=applink&share_count=1&m=dc114ccc8e55492642f6a702b510c1f6&d=9e18ca2dace030af656baea96321e0ea353fe5c46097a7f3962b93f995641e962796dd5faa231feea5531ac65547045f&app=zuiyou&recommend=r0&name=n0&title_type=t0 let msg = e.msg === undefined ? e.message.shift().data.replaceAll("\\", "") : e.msg.trim(); const url = /(?:https?:\/\/)?(share|share.xiaochuankeji)\.cn\/[A-Za-z\d._?%&+\-=\/#]*/.exec(msg)[0]; try { const response = await axios.get(url, { headers: { "User-Agent": COMMON_USER_AGENT, } }); const html = response.data; const videoUrlRegex = /fullscreen="false" src="(.*?)"/; const videoTitleRegex = /:<\/span>

(.*?)<\/h1><\/div>
(.*?)<\/span>/; const videoUrlMatch = html.match(videoUrlRegex); const videoTitleMatch = html.match(videoTitleRegex); const videoCoverMatch = html.match(videoCoverRegex); const videoAuthorMatch = html.match(videoAuthorRegex); const imgSrcRegex = /]*src="([^"]*)"[^>]*\/>/gi; let match; const imgSrcs = []; while ((match = imgSrcRegex.exec(html)) !== null) { imgSrcs.push(match[1]); // Adds the content of the src attribute to the array } const images = imgSrcs.filter(item => item.includes("\/img\/view\/id")); // Construct the response object const shortVideoInfo = { authorName: videoAuthorMatch ? videoAuthorMatch[1] : '', title: videoTitleMatch ? videoTitleMatch[1] : '', cover: videoCoverMatch ? videoCoverMatch[1] : '' || images[0], noWatermarkDownloadUrl: videoUrlMatch ? videoUrlMatch[1] : '', images, }; e.reply(`${ this.identifyPrefix }识别：最右，${ shortVideoInfo.authorName }\n${ shortVideoInfo.title }`); if (shortVideoInfo.images.length > 0) { const replyImages = shortVideoInfo.images.map(item => { return { message: segment.image(item), nickname: this.e.sender.card || this.e.user_id, user_id: this.e.user_id, }; }); e.reply(Bot.makeForwardMsg(replyImages)); } if (shortVideoInfo.noWatermarkDownloadUrl) { this.downloadVideo(shortVideoInfo.noWatermarkDownloadUrl).then(path => { this.sendVideoToUpload(e, `${ path }/temp.mp4`); }); } } catch (error) { logger.error(error); throw error; // Rethrow the error so it can be handled by the caller } } async freyr(e) { // https://music.apple.com/cn/album/hectopascal-from-yagate-kimi-ni-naru-piano-arrangement/1468323115?i=1468323724 // 过滤参数 const message = e.msg.replace("&ls", ""); // 匹配名字 const freyrName = message.includes("spotify") ? "Spotify" : "Apple Music"; // 找到R插件保存目录 const currentWorkingDirectory = path.resolve(this.getCurDownloadPath(e)); // 如果没有文件夹就创建一个 await mkdirIfNotExists(currentWorkingDirectory + "/am"); // 检测是否存在框架 const isExistFreyr = await checkToolInCurEnv("freyr"); if (!isExistFreyr) { e.reply(`检测到没有${ freyrName }需要的环境，无法解析！${ HELP_DOC }`); return; } // 执行命令 const result = await execSync(`freyr -d ${ currentWorkingDirectory + "/am/" } get ${ message }`); logger.info(result.toString()); // 获取信息 let { title, album, artist } = await this.parseFreyrLog(result.toString()); // 兜底策略 if (freyrName === "Apple Music" && (title === "N/A" || album === "N/A" || artist === "N/A")) { const data = await axios.get(`https://api.fabdl.com/apple-music/get?url=${ message }`, { headers: { "User-Agent": COMMON_USER_AGENT, "Referer": "https://apple-music-downloader.com/", "Origin": "https://apple-music-downloader.com", "Accept": "application/json, text/plain, */*", }, }); const { name, artists } = data.data.result; title = name; artist = artists; } // 判断是否是海外服务器 const isOversea = await this.isOverseasServer(); // 国内服务器解决方案 if (!isOversea) { // 临时接口 const url = await this.musicTempApi(e, `${ title } ${ artist }`, freyrName); // 下载音乐 downloadAudio(url, this.getCurDownloadPath(e), title, 'follow').then(async path => { // 发送语音 if (this.isSendVocal) { await e.reply(segment.record(path)); } // 判断是不是icqq await this.uploadGroupFile(e, path); await checkAndRemoveFile(path); }).catch(err => { logger.error(`下载音乐失败，错误信息为: ${ err.message }`); }); } else { // freyr 逻辑 e.reply(`${ this.identifyPrefix }识别：${ freyrName }，${ title }--${ artist }`); // 检查目录是否存在 const musicPath = currentWorkingDirectory + "/am/" + artist + "/" + album; // 找到音频文件 const mediaFiles = await getMediaFilesAndOthers(musicPath); for (let other of mediaFiles.others) { await this.uploadGroupFile(e, `${ musicPath }/${ other }`); } } return true; } /** * 用于Apple Music抓取部分信息的函数 * @link {applemusic} * @param log * @returns {Promise<{artist: (*|string), album: (*|string), title: (*|string)}>} */ async parseFreyrLog(log) { const titleMatch = log.match(/Title: (.*)/); const albumMatch = log.match(/Album: (.*)/); const artistMatch = log.match(/Artist: (.*)/); const title = titleMatch ? titleMatch[1] : 'N/A'; const album = albumMatch ? albumMatch[1] : 'N/A'; const artist = artistMatch ? artistMatch[1] : 'N/A'; return { title, album, artist }; } // 链接总结 async linkShareSummary(e) { let name, summaryLink; if (e.msg.startsWith("#总结一下")) { name = "网页总结"; summaryLink = e.msg.replace("#总结一下", ""); // 如果需要进一步处理 summaryLink，可以在这里添加相关逻辑 } else { ({ name: name, summaryLink: summaryLink } = contentEstimator(e.msg)); } // 判断是否有总结的条件 if (_.isEmpty(this.aiApiKey) || _.isEmpty(this.aiApiKey)) { // e.reply(`没有配置 Kimi，无法为您总结！${ HELP_DOC }`) await this.tempSummary(name, summaryLink, e); return true; } const builder = await new OpenaiBuilder() .setBaseURL(this.aiBaseURL) .setApiKey(this.aiApiKey) .setModel(this.aiModel) .setPrompt(SUMMARY_PROMPT) .build(); e.reply(`${ this.identifyPrefix }识别：${ name }，正在为您总结，请稍等...`, true, { recallMsg: MESSAGE_RECALL_TIME }); const { ans: kimiAns, model } = await builder.kimi(summaryLink); // 计算阅读时间 const stats = estimateReadingTime(kimiAns); const titleMatch = kimiAns.match(/(Title|标题)([:：])\s*(.*?)\n/)?.[3]; e.reply(`《${ titleMatch }》 预计阅读时间: ${ stats.minutes } 分钟，总字数: ${ stats.words }`); const Msg = await Bot.makeForwardMsg(textArrayToMakeForward(e, [`「R插件 x ${ model }」联合为您总结内容：`, kimiAns])); await e.reply(Msg); return true; } /** * 临时AI接口 * @param name * @param summaryLink * @param e * @returns {Promise} */ async tempSummary(name, summaryLink, e) { const content = await llmRead(summaryLink); const titleMatch = content.match(/Title:\s*(.*?)\n/)?.[1]; e.reply(`${ this.identifyPrefix }识别：${ name } - ${ titleMatch }，正在为您总结，请稍等...`, true); const summary = await deepSeekChat(content, SUMMARY_PROMPT); const Msg = await Bot.makeForwardMsg(textArrayToMakeForward(e, [`「R插件 x DeepSeek」联合为您总结内容：`, summary])); await e.reply(Msg); } // q q m u s i c 解析 async qqMusic(e) { // case1:　Taylor Swift/Bleachers《Anti-Hero (Feat. Bleachers) (Explicit)》 https://c6.y.qq.com/base/fcgi-bin/u?__=lg19lFgQerbo @QQ音乐 /** case 2: * {"app":"com.tencent.structmsg","config":{"ctime":1722497864,"forward":1,"token":"987908ab4a1c566d3645ef0ca52a162a","type":"normal"},"extra":{"app_type":1,"appid":100497308,"uin":542716863},"meta":{"news":{"action":"","android_pkg_name":"","app_type":1,"appid":100497308,"ctime":1722497864,"desc":"Taylor Swift/Bleachers","jumpUrl":"https://i.y.qq.com/v8/playsong.html?hosteuin=7KvA7i6sNeCi&sharefrom=gedan&from_id=1674373010&from_idtype=10014&from_name=(7rpl)&songid=382775503&songmid=&type=0&platform=1&appsongtype=1&_wv=1&source=qq&appshare=iphone&media_mid=000dKYJS3KCzpu&ADTAG=qfshare","preview":"https://pic.ugcimg.cn/1070bf5a6962b75263eee1404953c9b2/jpg1","source_icon":"https://p.qpic.cn/qqconnect/0/app_100497308_1626060999/100?max-age=2592000&t=0","source_url":"","tag":"QQ音乐","title":"Anti-Hero (Feat. Bleachers) (E…","uin":542716863}},"prompt":"[分享]Anti-Hero (Feat. Bleachers) (E…","ver":"0.0.0.1","view":"news"} */ let musicInfo; // applet判定 if (e.msg.includes(`"app":"com.tencent.structmsg"`)) { logger.info("[R插件][qqMusic] 识别为小程序分享"); const musicInfoJson = JSON.parse(e.msg); // 歌手和歌名 const prompt = musicInfoJson.meta?.news?.title ?? musicInfoJson.meta?.music?.title; const desc = musicInfoJson.meta?.news?.desc ?? musicInfoJson.meta?.music?.desc; // 必要性拼接 musicInfo = prompt + "-" + desc; // 空判定 if (musicInfo.trim() === "-" || prompt === undefined || desc === undefined) { logger.info(`没有识别到QQ音乐小程序，帮助文档如下：${ HELP_DOC }`); return true; } } else { // 连接判定 const normalRegex = /^(.*?)\s*https?:\/\//; musicInfo = normalRegex.exec(e.msg)?.[1].trim(); } // 删除特殊字符 musicInfo = cleanFilename(musicInfo); logger.info(`[R插件][qqMusic] 识别音乐为：${ musicInfo }`); // 使用临时接口下载 const url = await this.musicTempApi(e, musicInfo, "QQ音乐"); // 下载音乐 await downloadAudio(url, this.getCurDownloadPath(e), musicInfo, 'follow').then(async path => { // 发送语音 if (this.isSendVocal) { await e.reply(segment.record(path)); } // 判断是不是icqq await this.uploadGroupFile(e, path); await checkAndRemoveFile(path); }).catch(err => { logger.error(`下载音乐失败，错误信息为: ${ err.message }`); }); return true; } // 汽水音乐 async qishuiMusic(e) { const normalRegex = /^(.*?)\s*https?:\/\//; const musicInfo = normalRegex.exec(e.msg)?.[1].trim().replace("@汽水音乐", ""); logger.info(`[R插件][qishuiMusic] 识别音乐为：${ musicInfo }`); // 使用临时接口下载 const url = await this.musicTempApi(e, musicInfo, "汽水音乐"); // 下载音乐 await downloadAudio(url, this.getCurDownloadPath(e), musicInfo, 'follow').then(async path => { // 发送语音 if (this.isSendVocal) { await e.reply(segment.record(path)); } // 判断是不是icqq await this.uploadGroupFile(e, path); await checkAndRemoveFile(path); }).catch(err => { logger.error(`下载音乐失败，错误信息为: ${ err.message }`); }); return true; } // 小飞机下载 async aircraft(e) { if (!(await this.isTrustUser(e.user_id))) { e.reply("你没有权限使用此命令"); return; } const isOversea = await this.isOverseasServer(); if (!isOversea && !(await testProxy(this.proxyAddr, this.proxyPort))) { e.reply("检测到没有梯子，无法解析小飞机"); return false; } const urlRex = /(?:https?:\/\/)?t\.me\/[A-Za-z\d._?%&+\-=\/#]*/g; // 检查当前环境 const isExistTdl = await checkToolInCurEnv("tdl"); if (!isExistTdl) { e.reply(`未检测到必要的环境，无法解析小飞机${ HELP_DOC }`); return; } const url = urlRex.exec(e.msg)[0]; if (e.msg.startsWith("保存")) { // 发送文件到 SaveMessages await saveTDL(url, isOversea, this.myProxy); e.reply("文件已保存到 Save Messages！"); return true; } e.reply(`${ this.identifyPrefix }识别：小飞机（学习版）`); const tgSavePath = `${ this.getCurDownloadPath(e) }/tg`; // 如果没有文件夹则创建 await mkdirIfNotExists(tgSavePath); // 删除之前的文件 await deleteFolderRecursive(tgSavePath); await startTDL(url, tgSavePath, isOversea, this.myProxy, this.videoDownloadConcurrency); // 过滤当前文件 const mediaFiles = await getMediaFilesAndOthers(tgSavePath); if (mediaFiles.images.length > 0) { const imagesData = mediaFiles.images.map(item => { const fileContent = fs.readFileSync(`${ tgSavePath }/${ item }`); return { message: segment.image(fileContent), nickname: e.sender.card || e.user_id, user_id: e.user_id, }; }); e.reply(await Bot.makeForwardMsg(imagesData), true, { recallMsg: MESSAGE_RECALL_TIME }); } else if (mediaFiles.videos.length > 0) { for (const item of mediaFiles.videos) { await this.sendVideoToUpload(e, `${ tgSavePath }/${ item }`); } } else { for (let other of mediaFiles.others) { await this.uploadGroupFile(e, `${ tgSavePath }/${ other }`); } } return true; } // 贴吧 async tieba(e) { // 提取链接和ID const msg = /https:\/\/tieba\.baidu\.com\/p\/[A-Za-z0-9]+/.exec(e.msg)?.[0]; const id = /\/p\/([A-Za-z0-9]+)/.exec(msg)?.[1]; // 获取帖子详情 const hibi = HIBI_API_SERVICE + `/tieba/post_detail?tid=${ id }`; const hibiResp = await fetch(hibi, { headers: { "User-Agent": COMMON_USER_AGENT, } }); const postList = (await hibiResp.json()).post_list; // 获取楼主的消息 const top = postList[0]; // 提取标题和内容 const { title, content } = top; let sendContent = `${ this.identifyPrefix }识别：贴吧，${ title }`; let extractImages = []; // 如果内容中有图片、文本或视频，它会将它们添加到 sendContent 消息中 if (content && content.length > 0) { sendContent = [sendContent]; for (const { cdn_src, text, link } of content) { logger.info({ cdn_src, text, link }); // 可以一次性输出多个属性 // 处理图片 if (cdn_src) extractImages.push(segment.image(cdn_src)); // 处理文本 if (text) sendContent.push(`\n\n📝 简介：${ text }`); // 处理视频 if (link) { const filePath = await this.downloadVideo(link); this.sendVideoToUpload(e, `${ filePath }/temp.mp4`); } } } e.reply(sendContent, true); extractImages && e.reply(Bot.makeForwardMsg(extractImages.map(item => ({ message: item, nickname: e.sender.card || e.user_id, user_id: e.user_id, })))); // 切除楼主的消息 const others = postList.slice(1); // 贴吧楼层的消息处理：如果响应中有其他帖子，代码创建一条转发消息，包含其他帖子的内容，并回复原始消息 const reply = others.flatMap(item => { if (!item.content || item.content.length === 0) return []; return item.content.map(floor => { const commonData = { nickname: e.sender.card || e.user_id, user_id: e.user_id, }; if (floor?.cdn_src) { return { ...commonData, message: segment.image(floor.cdn_src) }; } else if (floor?.text) { return { ...commonData, message: { type: 'text', text: floor.text || '-' } }; } return null; }).filter(Boolean); // 过滤掉 null 的值 }); e.reply(await Bot.makeForwardMsg(reply)); return true; } /** * 哔哩哔哩下载 * @param title * @param videoUrl * @param audioUrl * @returns {Promise} */ async downBili(title, videoUrl, audioUrl) { return Promise.all([ downloadBFile( videoUrl, title + "-video.m4s", _.throttle( value => logger.mark("视频下载进度", { data: value, }), 1000, ), this.biliDownloadMethod, this.videoDownloadConcurrency ), downloadBFile( audioUrl, title + "-audio.m4s", _.throttle( value => logger.mark("音频下载进度", { data: value, }), 1000, ), this.biliDownloadMethod, this.videoDownloadConcurrency ), ]).then(data => { return mergeFileToMp4(data[0].fullFileName, data[1].fullFileName, `${ title }.mp4`); }); } /** * douyin 请求参数 * @param url * @returns {Promise} */ async douyinRequest(url) { const params = { headers: { "User-Agent": COMMON_USER_AGENT, }, timeout: 10000, }; try { const resp = await axios.get(url, params); const location = resp.request.res.responseUrl; const setCookieHeaders = resp.headers['set-cookie']; let ttwidValue; if (setCookieHeaders) { setCookieHeaders.forEach(cookie => { // 使用正则表达式提取 ttwid 的值 const ttwidMatch = cookie.match(/ttwid=([^;]+)/); if (ttwidMatch) { ttwidValue = ttwidMatch[1]; } }); } return new Promise((resolve, reject) => { if (location != null) { return resolve({ location: location, ttwidValue: ttwidValue }); } else { return reject("获取失败"); } }); } catch (error) { logger.error(error); throw error; } } /** * 获取当前发送人/群的下载路径 * @param e Yunzai 机器人事件 * @returns {string} */ getCurDownloadPath(e) { return `${ this.defaultPath }${ e.group_id || e.user_id }`; } /** * 提取视频下载位置 * @returns {{groupPath: string, target: string}} */ getGroupPathAndTarget() { const groupPath = `${ this.defaultPath }${ this.e.group_id || this.e.user_id }`; const target = `${ groupPath }/temp.mp4`; return { groupPath, target }; } /** * 工具：根据URL多线程下载视频 / 音频 * @param url * @param isProxy * @param headers * @param numThreads * @returns {Promise} */ async downloadVideo(url, isProxy = false, headers = null, numThreads = this.videoDownloadConcurrency) { // 构造群信息参数 const { groupPath, target } = this.getGroupPathAndTarget.call(this); await mkdirIfNotExists(groupPath); // 构造header部分内容 const userAgent = "Mozilla/5.0 (Linux; Android 5.0; SM-G900P Build/LRX21T) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.25 Mobile Safari/537.36"; // 构造代理参数 const proxyOption = { ...(isProxy && { httpAgent: new HttpsProxyAgent(`http://${ this.proxyAddr }:${ this.proxyPort }`), }), }; /** * 构造下载视频参数 * 构造信息：链接、头信息、userAgent、代理信息、下载位置、返回的路径 * @type {{headers: null, userAgent: string, groupPath: string, url, proxyOption: {}, target: string}} */ const downloadVideoParams = { url, headers, userAgent, proxyOption, target, groupPath, }; logger.info(`[R插件][视频下载]：当前队列长度为 ${ this.queue.size + 1 }`); return await this.queue.add(async () => { // 如果是用户设置了单线程，则不分片下载 if (numThreads === 1) { return this.downloadVideoWithSingleThread(downloadVideoParams); } else if (numThreads !== 1 && this.biliDownloadMethod === 1) { return this.downloadVideoWithAria2(downloadVideoParams, numThreads); } else if (numThreads !== 1 && this.biliDownloadMethod === 2) { return this.downloadVideoUseAxel(downloadVideoParams, numThreads); } else { return this.downloadVideoWithMultiThread(downloadVideoParams, numThreads); } }); } /** * 多线程下载视频 * @link {downloadVideo} * @param downloadVideoParams * @param numThreads * @returns {Promise<*>} */ async downloadVideoWithMultiThread(downloadVideoParams, numThreads) { const { url, headers, userAgent, proxyOption, target, groupPath } = downloadVideoParams; try { // Step 1: 请求视频资源获取 Content-Length const headRes = await axios.head(url, { headers: headers || { "User-Agent": userAgent }, ...proxyOption }); const contentLength = headRes.headers['content-length']; if (!contentLength) { throw new Error("无法获取视频大小"); } // Step 2: 计算每个线程应该下载的文件部分 const partSize = Math.ceil(contentLength / numThreads); let promises = []; for (let i = 0; i < numThreads; i++) { const start = i * partSize; let end = start + partSize - 1; if (i === numThreads - 1) { end = contentLength - 1; // 确保最后一部分可以下载完整 } // Step 3: 并发下载文件的不同部分 const partAxiosConfig = { headers: { "User-Agent": userAgent, "Range": `bytes=${ start }-${ end }` }, responseType: "stream", ...proxyOption }; promises.push(axios.get(url, partAxiosConfig).then(res => { return new Promise((resolve, reject) => { const partPath = `${ target }.part${ i }`; logger.mark(`[R插件][视频下载引擎] 正在下载 part${ i }`); const writer = fs.createWriteStream(partPath); res.data.pipe(writer); writer.on("finish", () => { logger.mark(`[R插件][视频下载引擎] part${ i + 1 } 下载完成`); // 记录线程下载完成 resolve(partPath); }); writer.on("error", reject); }); })); } // 等待所有部分都下载完毕 const parts = await Promise.all(promises); // Step 4: 合并下载的文件部分 await checkAndRemoveFile(target); // 确保目标文件不存在 const writer = fs.createWriteStream(target, { flags: 'a' }); for (const partPath of parts) { await new Promise((resolve, reject) => { const reader = fs.createReadStream(partPath); reader.pipe(writer, { end: false }); reader.on('end', () => { fs.unlinkSync(partPath); // 删除部分文件 resolve(); }); reader.on('error', reject); }); } writer.close(); return groupPath; } catch (err) { logger.error(`下载视频发生错误！\ninfo:${ err }`); } } /** * 使用Aria2进行多线程下载 * @param downloadVideoParams * @param numThreads * @returns {Promise} */ async downloadVideoWithAria2(downloadVideoParams, numThreads) { const { url, headers, userAgent, proxyOption, target, groupPath } = downloadVideoParams; // 构造aria2c命令参数 const aria2cArgs = [ `"${ url }"`, `--out="temp.mp4"`, `--dir="${ groupPath }"`, `--user-agent="${ userAgent }"`, `--max-connection-per-server=${ numThreads }`, // 每个服务器的最大连接数 `--split=${ numThreads }`, // 分成 6 个部分进行下载 ]; // 如果有自定义头信息 if (headers) { for (const [key, value] of Object.entries(headers)) { aria2cArgs.push(`--header="${ key }: ${ value }"`); } } // 如果使用代理 if (proxyOption && proxyOption.httpAgent) { const proxyUrl = proxyOption.httpAgent.proxy.href; aria2cArgs.push(`--all-proxy="${ proxyUrl }"`); } try { await checkAndRemoveFile(target); logger.mark(`开始下载: ${ url }`); // 执行aria2c命令 const command = `aria2c ${ aria2cArgs.join(' ') }`; exec(command, (error, stdout, stderr) => { if (error) { logger.error(`下载视频发生错误！\ninfo:${ stderr }`); throw error; } else { logger.mark(`下载完成: ${ url }`); } }); // 监听文件生成完成 let count = 0; return new Promise((resolve, reject) => { const checkInterval = setInterval(() => { logger.info(logger.red(`[R插件][Aria2] 没有检测到文件！重试第${ count + 1 }次`)); count += 1; if (fs.existsSync(target)) { logger.info("[R插件][Aria2] 检测到文件！"); clearInterval(checkInterval); resolve(groupPath); } if (count === 6) { logger.error(`[R插件][Aria2] 下载视频发生错误！`); clearInterval(checkInterval); reject(); } }, DOWNLOAD_WAIT_DETECT_FILE_TIME); }); } catch (err) { logger.error(`下载视频发生错误！\ninfo:${ err }`); throw err; } } /** * 使用Axel进行多线程下载 * @param downloadVideoParams * @param numThreads * @returns {Promise} */ async downloadVideoUseAxel(downloadVideoParams, numThreads) { const { url, headers, userAgent, proxyOption, target, groupPath } = downloadVideoParams; // 构造axel命令参数 const axelArgs = [ `-n ${ numThreads }`, `-o "${ target }"`, `-U "${ userAgent }"`, url ]; // 如果有自定义头信息 if (headers) { for (const [key, value] of Object.entries(headers)) { axelArgs.push(`-H "${ key }: ${ value }"`); } } // 如果使用代理 if (proxyOption && proxyOption.httpAgent) { const proxyUrl = proxyOption.httpAgent.proxy.href; axelArgs.push(`--proxy="${ proxyUrl }"`); } try { await checkAndRemoveFile(target); logger.mark(`开始下载: ${ url }`); // 执行axel命令 const command = `axel ${ axelArgs.join(' ') }`; exec(command, (error, stdout, stderr) => { if (error) { logger.error(`下载视频发生错误！\ninfo:${ stderr }`); throw error; } else { logger.mark(`下载完成: ${ url }`); } }); let count = 0; // 监听文件生成完成 return new Promise((resolve, reject) => { const checkInterval = setInterval(() => { logger.info(logger.red(`[R插件][Aria2] 没有检测到文件！重试第${ count + 1 }次`)); count += 1; if (fs.existsSync(target)) { logger.info("[R插件][Axel] 检测到文件！"); clearInterval(checkInterval); logger.info(`[R插件][Axel] 下载到${ groupPath }`); resolve(groupPath); } if (count === 6) { logger.error(`[R插件][Axel] 下载视频发生错误！`); clearInterval(checkInterval); reject(); } }, DOWNLOAD_WAIT_DETECT_FILE_TIME); }); } catch (err) { logger.error(`下载视频发生错误！\ninfo:${ err }`); throw err; } } /** * 单线程下载视频 * @link {downloadVideo} * @returns {Promise} * @param downloadVideoParams */ async downloadVideoWithSingleThread(downloadVideoParams) { const { url, headers, userAgent, proxyOption, target, groupPath } = downloadVideoParams; const axiosConfig = { headers: headers || { "User-Agent": userAgent }, responseType: "stream", ...proxyOption }; try { await checkAndRemoveFile(target); const res = await axios.get(url, axiosConfig); logger.mark(`开始下载: ${ url }`); const writer = fs.createWriteStream(target); res.data.pipe(writer); return new Promise((resolve, reject) => { writer.on("finish", () => resolve(groupPath)); writer.on("error", reject); }); } catch (err) { logger.error(`下载视频发生错误！\ninfo:${ err }`); } } /** * 判断是否是海外服务器 * @return {Promise} */ async isOverseasServer() { // 如果第一次使用没有值就设置 if (!(await redisExistKey(REDIS_YUNZAI_ISOVERSEA))) { await redisSetKey(REDIS_YUNZAI_ISOVERSEA, { os: false, }); return true; } // 如果有就取出来 return (await redisGetKey(REDIS_YUNZAI_ISOVERSEA)).os; } /** * 判断当前用户是否是信任用户 * @param userId * @returns {Promise} */ async isTrustUser(userId) { // 如果当前用户是机器人主人 if (this.e.isMaster) { return true; } // 如果不存在则返回 if (!(await redisExistKey(REDIS_YUNZAI_WHITELIST))) { return false; } const whiteList = await redisGetKey(REDIS_YUNZAI_WHITELIST); return whiteList.includes(userId.toString()) || whiteList.includes(userId); } /** * 发送转上传视频 * @param e 交互事件 * @param path 视频所在路径 * @param videoSizeLimit 发送转上传视频的大小限制，默认70MB */ async sendVideoToUpload(e, path, videoSizeLimit = this.videoSizeLimit) { try { // 判断文件是否存在 if (!fs.existsSync(path)) { return e.reply('视频不存在'); } const stats = fs.statSync(path); const videoSize = Math.floor(stats.size / (1024 * 1024)); // 正常发送视频 if (videoSize > videoSizeLimit) { e.reply(`当前视频大小：${ videoSize }MB，\n大于设置的最大限制：${ videoSizeLimit }MB，\n改为上传群文件`); await this.uploadGroupFile(e, path); } else { e.reply(segment.video(path)); } } catch (err) { logger.error(`[R插件][发送视频判断是否需要上传] 发生错误:\n ${ err }`); // logger.info(logger.yellow(`上传发生错误，R插件正在为你采用备用策略，请稍等，如果发不出来请再次尝试！`)); // e.reply(segment.video(path)); } } /** * 上传到群文件 * @param e 交互事件 * @param path 上传的文件所在路径 * @return {Promise} */ async uploadGroupFile(e, path) { // 判断是否是ICQQ if (e.bot?.sendUni) { await e.group.fs.upload(path); } else { await e.group.sendFile(path); } } }