feat: 增加抖音解析

This commit is contained in:
zhiyu1998 2022-11-23 16:16:23 +08:00
parent 851e486b36
commit 20016c1011
2 changed files with 65 additions and 22 deletions

View File

@ -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/

View File

@ -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 page = await browser.newPage();
await page.goto(urlRex.exec(e.msg.trim())[0]);
const headers = page.headers()
console.log(headers)
await browser.close();
});
const douyinRex = /.*video\/(\d+)\/(.*?)/g
const resolver = douyinRex.exec(location)
const url = `https://www.iesdouyin.com/web/api/v2/aweme/iteminfo/?item_ids=${resolver}`
const douUrl = urlRex.exec(e.msg.trim())[0]
await https.request(douUrl, {
method: 'HEAD', 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'
}, timeout: 10000
}, (res) => {
const location = res.headers['location']
const douRex = /.*video\/(\d+)\/(.*?)/g
const douId = douRex.exec(location)[1]
const url = `https://www.iesdouyin.com/web/api/v2/aweme/iteminfo/?item_ids=${douId}`
e.reply('解析中...')
const res = fetch(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"
},
timeout: 10000
}).then(resp => {
return resp.body
return fetch(url)
.then(resp => resp.json())
.then(json => json.item_list[0])
.then(item => item.video.play_addr.url_list[0])
.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
};
}