✨ feat: 增加抖音解析

README.md

@@ -29,7 +29,9 @@ index -- 主入口
 2. 下载mongodb
 > linux系统下自己装一个mongodb，上一个密码(不上有风险)
 3. 在`Yunzai-Bot`安装mongodb依赖
-> pnpm i mongodb
+> pnpm add mongodb -w
+
+> pnpm add axios -w
 4. 下载插件
 > git clone https://gitee.com/kyrzy0416/rconsole-plugin.git ./plugins/rconsole-plugin/
 

apps/tools.js

@@ -1,8 +1,14 @@
+// 主库
 import fetch from 'node-fetch'
+import fs from "node:fs";
+import { segment } from 'oicq'
+// 其他库
 import md5 from 'md5'
+import https from 'https'
+import axios from 'axios'
 
 export class tools extends plugin {
-    constructor () {
+    constructor() {
         super({
             name: '工具和学习类',
             dsc: '工具相关指令',
@@ -11,13 +17,17 @@ export class tools extends plugin {
             rule: [
                 {
                     reg: '^#(翻译)(.*)$', fnc: 'trans'
+                },
+                {
+                    reg: '(.*)(v.douyin.com)', fnc: 'douyin'
                 }
             ]
         })
+        this.path = "./data/mp4/";
     }
 
     // 翻译插件
-    async trans (e) {
+    async trans(e) {
         let place = e.msg.replace(/#|翻译/g, '').trim()
         let url = /[\u4E00-\u9FFF]+/g.test(place) ? `http://api.fanyi.baidu.com/api/trans/vip/translate?from=zh&to=en&appid=20210422000794040&salt=542716863&sign=${md5('20210422000794040' + place + '542716863' + 'HooD_ndgwcGH6SAnxGrM')}&q=${place}` : `http://api.fanyi.baidu.com/api/trans/vip/translate?from=en&to=zh&appid=20210422000794040&salt=542716863&sign=${md5('20210422000794040' + place + '542716863' + 'HooD_ndgwcGH6SAnxGrM')}&q=${place}`
         await fetch(url)
@@ -31,24 +41,55 @@ export class tools extends plugin {
     // 抖音解析
     async douyin(e) {
         const urlRex = /(http:|https:)\/\/v.douyin.com\/[A-Za-z\\d._?%&+\-=\/#]*/g
-        puppeteer.launch().then(async browser => {
+        const douUrl = urlRex.exec(e.msg.trim())[0]
-            const page = await browser.newPage();
+
-            await page.goto(urlRex.exec(e.msg.trim())[0]);
+        await https.request(douUrl, {
-            const headers = page.headers()
+            method: 'HEAD', headers: {
-            console.log(headers)
+                'User-Agent':
-            await browser.close();
+                    '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'
-        });
+            }, timeout: 10000
-        const douyinRex = /.*video\/(\d+)\/(.*?)/g
+        }, (res) => {
-        const resolver = douyinRex.exec(location)
+            const location = res.headers['location']
-        const url = `https://www.iesdouyin.com/web/api/v2/aweme/iteminfo/?item_ids=${resolver}`
+            const douRex = /.*video\/(\d+)\/(.*?)/g
-        e.reply('解析中...')
+            const douId = douRex.exec(location)[1]
-        const res = fetch(url, {
+            const url = `https://www.iesdouyin.com/web/api/v2/aweme/iteminfo/?item_ids=${douId}`
-            headers: {
+            e.reply('解析中...')
-                'User-Agent': "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"
+            return fetch(url)
-            },
+                .then(resp => resp.json())
-            timeout: 10000
+                .then(json => json.item_list[0])
-        }).then(resp => {
+                .then(item => item.video.play_addr.url_list[0])
-            return resp.body
+                .then(async url => {
-        })
+                    await e.reply(await segment.video(await this.downloadVideo(url)))
+                })
+        }).on('error', (err) => {
+            console.error(err);
+        }).end();
+        return true;
     }
+
+    // 根URL据下载视频 / 音频
+    async downloadVideo (url) {
+        const target = `${this.path}${this.e.group_id || this.e.user_id}/temp.mp4`
+        if (fs.existsSync(target)) {
+            console.log(`视频已存在`);
+            fs.unlinkSync(target)
+        } else {
+            fs.mkdirSync(`${this.path}${this.e.group_id || this.e.user_id}`)
+        }
+        const res = await axios.get(url, {
+            headers: {
+                'User-Agent':
+                    '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'
+            },
+            responseType: "stream",
+        });
+        console.log(`开始下载: ${url}`);
+        const writer = fs.createWriteStream(target);
+        res.data.pipe(writer);
+        new Promise((resolve, reject) => {
+            writer.on("finish", resolve);
+            writer.on("error", reject);
+        });
+        return target
+    };
 }