🦄 refactor: 修改https框架为axios

This commit is contained in:
zhiyu1998 2022-11-23 19:34:40 +08:00
parent e99d6d721c
commit 0266a79862

View File

@ -1,151 +1,132 @@
// 主库
import fetch from "node-fetch";
import fs from "node:fs";
import { segment } from "oicq";
import {segment} from "oicq";
// 其他库
import md5 from "md5";
import https from "https";
import axios from "axios";
import path from 'path'
export class tools extends plugin {
constructor() {
super({
name: "工具和学习类",
dsc: "工具相关指令",
event: "message.group",
priority: 500,
rule: [
{
reg: "^#(翻译)(.*)$",
fnc: "trans",
},
{
reg: "(.*)(v.douyin.com)",
fnc: "douyin",
},
],
});
this.path = "./data/mp4/";
}
// 翻译插件
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)
.then((resp) => resp.json())
.then((text) => text.trans_result)
.then((res) => this.reply(`${res[0].dst}`, true))
.catch((err) => logger.error(err));
return true;
}
//zengjie2001
async douyinRequest() {
const urlRex = /(http:|https:)\/\/v.douyin.com\/[A-Za-z\d._?%&+\-=\/#]*/g;
const douUrl = urlRex.exec(e.msg.trim())[0];
const params = {
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,
};
return new Promise((resolve, reject) => {
axios
.request(douUrl, params)
.then((res) => {
const location = res.headers["location"];
resolve(location);
})
.catch((err) => {
reject(err);
constructor() {
super({
name: "工具和学习类",
dsc: "工具相关指令",
event: "message.group",
priority: 500,
rule: [
{
reg: "^#(翻译)(.*)$",
fnc: "trans",
},
{
reg: "(.*)(v.douyin.com)",
fnc: "douyin",
},
],
});
});
}
async douyin1(e) {
this.douyinRequest().then((res) => {
const douRex = /.*video\/(\d+)\/(.*?)/g;
const douId = douRex.exec(res)[1];
const url = `https://www.iesdouyin.com/web/api/v2/aweme/iteminfo/?item_ids=${douId}`;
e.reply("解析中...");
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)));
});
});
return true;
}
// 抖音解析
async douyin(e) {
const urlRex = /(http:|https:)\/\/v.douyin.com\/[A-Za-z\d._?%&+\-=\/#]*/g;
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("解析中...");
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}`);
this.path = "./data/rcmp4/";
}
// 翻译插件
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)
.then((resp) => resp.json())
.then((text) => text.trans_result)
.then((res) => this.reply(`${res[0].dst}`, true))
.catch((err) => logger.error(err));
return true;
}
// 抖音解析
async douyin(e) {
const urlRex = /(http:|https:)\/\/v.douyin.com\/[A-Za-z\d._?%&+\-=\/#]*/g;
const douUrl = urlRex.exec(e.msg.trim())[0];
await this.douyinRequest(douUrl).then((res) => {
const douRex = /.*video\/(\d+)\/(.*?)/g;
const douId = douRex.exec(res)[1];
const url = `https://www.iesdouyin.com/web/api/v2/aweme/iteminfo/?item_ids=${douId}`;
e.reply("解析中...");
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)));
});
});
return true;
}
// 请求参数
async douyinRequest(url) {
const params = {
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,
};
return new Promise((resolve, reject) => {
axios
.head(url, params)
.then((resp) => {
const location = resp.request.res.responseUrl
resolve(location);
})
.catch((err) => {
reject(err);
});
});
}
// 根URL据下载视频 / 音频
async downloadVideo(url) {
let target = `${this.path}${this.e.group_id || this.e.user_id}`;
if (!fs.existsSync(target)) {
this.mkdirsSync(`${this.path}${this.e.group_id || this.e.user_id}`);
}
target += '/temp.mp4'
// 待优化
if (fs.existsSync(target)) {
console.log(`视频已存在`);
fs.unlinkSync(target);
}
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;
}
// 同步递归创建文件夹
mkdirsSync(dirname) {
if (fs.existsSync(dirname)) {
return true;
} else {
if (this.mkdirsSync(path.dirname(dirname))) {
fs.mkdirSync(dirname);
return true;
}
}
}
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;
}
}