🌈 style: 代码格式化

This commit is contained in:
zhiyu1998 2023-03-15 23:46:38 +08:00
parent cec0f2da03
commit 8426215d5f

View File

@ -1,50 +1,58 @@
import plugin from "../../../lib/plugins/plugin.js";
import axios from "axios";
import fs from 'node:fs';
import {segment} from "oicq";
import {getQrCode, getKey, getLoginStatus, getDailyRecommend, getCookies, getUserRecord, checkMusic, getSong} from '../utils/netease.js';
import {ha12store, store2ha1} from '../utils/encrypt.js';
import fs from "node:fs";
import { segment } from "oicq";
import {
getQrCode,
getKey,
getLoginStatus,
getDailyRecommend,
getCookies,
getUserRecord,
checkMusic,
getSong,
} from "../utils/netease.js";
import { ha12store, store2ha1 } from "../utils/encrypt.js";
import fetch from "node-fetch";
export class neteasepro extends plugin {
constructor() {
super({
/** 功能名称 */
name: 'R插件网易云音乐解析',
name: "R插件网易云音乐解析",
/** 功能描述 */
dsc: '网易云音乐解析Pro',
dsc: "网易云音乐解析Pro",
/** https://oicqjs.github.io/oicq/#events */
event: 'message',
event: "message",
/** 优先级,数字越小等级越高 */
priority: 1,
rule: [
{
/** 命令正则匹配 */
reg: '#网易云登陆',
reg: "#网易云登陆",
/** 执行方法 */
fnc: 'neteaseCloudLogin'
fnc: "neteaseCloudLogin",
},
{
reg: '#网易云每日推荐',
fnc: 'neteaseDailyRecommend'
reg: "#网易云每日推荐",
fnc: "neteaseDailyRecommend",
},
{
reg: '#网易云听歌排行',
fnc: 'neteaseListenRank'
reg: "#网易云听歌排行",
fnc: "neteaseListenRank",
},
{
reg: "music.163.com",
fnc: "netease",
},
]
})
],
});
}
async neteaseCloudLogin(e) {
let neteaseCookie;
// 如果不存在cookie
if (!await redis.exists(await this.getRedisKey(e.user_id))) {
if (!(await redis.exists(await this.getRedisKey(e.user_id)))) {
// 获取密钥
const key = await getKey();
// console.log(key);
@ -53,28 +61,33 @@ export class neteasepro extends plugin {
// 下载qrcode
await this.downloadQrCode(qrPic).then(path => {
// 发送二维码
e.reply(segment.image(fs.readFileSync(path)))
})
e.reply(segment.image(fs.readFileSync(path)));
});
// 定时轮询
await this.poll(key).then(async cookie => {
// 存放到redis
neteaseCookie = cookie
neteaseCookie = cookie;
});
} else {
// 已经登陆过的直接从redis取出
neteaseCookie = await store2ha1(JSON.parse(await redis.get(await this.getRedisKey(e.user_id))).cookie)
neteaseCookie = await store2ha1(
JSON.parse(await redis.get(await this.getRedisKey(e.user_id))).cookie,
);
}
// 获取用户信息
const userInfo = await getLoginStatus(neteaseCookie);
// 提取信息
const {userId, nickname, avatarUrl} = userInfo.profile;
e.reply(["欢迎使用 🎶网易云音乐 🎶," + nickname, segment.image(avatarUrl)])
const { userId, nickname, avatarUrl } = userInfo.profile;
e.reply(["欢迎使用 🎶网易云音乐 🎶," + nickname, segment.image(avatarUrl)]);
// 重组后存放到redis {uid, cookie}
await redis.set(await this.getRedisKey(e.user_id), JSON.stringify({
await redis.set(
await this.getRedisKey(e.user_id),
JSON.stringify({
uid: userId,
cookie: (await ha12store(neteaseCookie))
}));
return true
cookie: await ha12store(neteaseCookie),
}),
);
return true;
}
async neteaseDailyRecommend(e) {
@ -88,17 +101,21 @@ export class neteasepro extends plugin {
const combineMsg = await dailyRecommend.dailySongs.slice(0, 10).map(item => {
// 组合数据
return {
message: [segment.text(`${item?.id}: ${item?.name}-${item?.ar?.[0].name}-${item?.al?.name}`)
, segment.image(item?.al?.picUrl)],
message: [
segment.text(
`${item?.id}: ${item?.name}-${item?.ar?.[0].name}-${item?.al?.name}`,
),
segment.image(item?.al?.picUrl),
],
nickname: e.sender.card || e.user_id,
user_id: e.user_id,
}
})
};
});
await e.reply(await Bot.makeForwardMsg(combineMsg));
}
async neteaseListenRank(e) {
const userInfo = await this.aopBefore(e)
const userInfo = await this.aopBefore(e);
const realCookie = userInfo.cookie;
if (realCookie === "") {
return true;
@ -106,36 +123,49 @@ export class neteasepro extends plugin {
// 获取用户id
const uid = userInfo.uid;
// 获取听歌排行榜
const userRecord = await getUserRecord(uid)
const userRecord = await getUserRecord(uid);
let rankId = 0;
e.reply(" 😘亲这是你的听歌排行榜Top10")
e.reply(" 😘亲这是你的听歌排行榜Top10");
const rank = userRecord.weekData.slice(0, 10).map(item => {
// 组合数据
const song = item.song;
rankId++;
return {
message: [segment.text(`No.${rankId} ${song?.id}: ${song?.name}-${song?.ar?.[0].name}-${song?.al?.name}`)
, segment.image(song?.al?.picUrl)],
message: [
segment.text(
`No.${rankId} ${song?.id}: ${song?.name}-${song?.ar?.[0].name}-${song?.al?.name}`,
),
segment.image(song?.al?.picUrl),
],
nickname: e.sender.card || e.user_id,
user_id: e.user_id,
}
})
};
});
await e.reply(await Bot.makeForwardMsg(rank));
}
async netease(e) {
const message = e.msg === undefined ? e.message.shift().data.replaceAll("\\", "") : e.msg.trim();
const message =
e.msg === undefined ? e.message.shift().data.replaceAll("\\", "") : e.msg.trim();
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 id = musicUrlReg2.exec(message)[3] || musicUrlReg.exec(message)[2] || /id=(\d+)/.exec(message)[1];
const musicJson = JSON.parse(message)
const {musicUrl, preview, title, desc} = musicJson.meta.music || musicJson.meta.news;
console.log(musicUrl, preview, title, desc)
const id =
musicUrlReg2.exec(message)[3] ||
musicUrlReg.exec(message)[2] ||
/id=(\d+)/.exec(message)[1];
const musicJson = JSON.parse(message);
const { musicUrl, preview, title, desc } = musicJson.meta.music || musicJson.meta.news;
console.log(musicUrl, preview, title, desc);
// 如果没有登陆,就使用官方接口
e.reply([`识别:网易云音乐,${title}--${desc}`, segment.image(preview)]);
if (!await redis.exists(await this.getRedisKey(e.user_id))) {
this.downloadMp3(`music.163.com/song/media/outer/url?id=${id}`, 'follow').then(path => {
Bot.acquireGfs(e.group_id).upload(fs.readFileSync(path), '/', `${title.replace(/[\/\?<>\\:\*\|".… ]/g, '')}.mp3`)
if (!(await redis.exists(await this.getRedisKey(e.user_id)))) {
this.downloadMp3(`music.163.com/song/media/outer/url?id=${id}`, "follow")
.then(path => {
Bot.acquireGfs(e.group_id).upload(
fs.readFileSync(path),
"/",
`${title.replace(/[\/\?<>\\:\*\|".… ]/g, "")}.mp3`,
);
})
.catch(err => {
console.error(`下载音乐失败,错误信息为: ${err.message}`);
@ -144,15 +174,20 @@ export class neteasepro extends plugin {
}
// 检查当前歌曲是否可用
const checkOne = await checkMusic(id);
if (checkOne.success === 'false') {
if (checkOne.success === "false") {
e.reply(checkOne.message);
return true;
}
const userInfo = await this.aopBefore(e)
const userInfo = await this.aopBefore(e);
// 可用,开始下载
const userDownloadUrl = (await getSong(id, await userInfo.cookie))[0].url
await this.downloadMp3(userDownloadUrl).then(path => {
Bot.acquireGfs(e.group_id).upload(fs.readFileSync(path), '/', `${title.replace(/[\/\?<>\\:\*\|".… ]/g, '')}.mp3`)
const userDownloadUrl = (await getSong(id, await userInfo.cookie))[0].url;
await this.downloadMp3(userDownloadUrl)
.then(path => {
Bot.acquireGfs(e.group_id).upload(
fs.readFileSync(path),
"/",
`${title.replace(/[\/\?<>\\:\*\|".… ]/g, "")}.mp3`,
);
})
.catch(err => {
console.error(`下载音乐失败,错误信息为: ${err.message}`);
@ -171,7 +206,7 @@ export class neteasepro extends plugin {
return "";
}
// 解析cookie
userInfo.cookie = store2ha1(cookie)
userInfo.cookie = store2ha1(cookie);
return userInfo;
}
@ -185,7 +220,7 @@ export class neteasepro extends plugin {
},
responseType: "stream",
})
.then((resp) => {
.then(resp => {
const filepath = "./netease_qr.jpg";
const writer = fs.createWriteStream(filepath);
resp.data.pipe(writer);
@ -201,19 +236,21 @@ export class neteasepro extends plugin {
let timer;
return new Promise((resolve, reject) => {
timer = setInterval(async () => {
const statusRes = await getCookies(key)
console.log(statusRes)
const statusRes = await getCookies(key);
console.log(statusRes);
if (statusRes.code === 800) {
clearInterval(timer)
reject('二维码已过期,请重新获取')
clearInterval(timer);
reject("二维码已过期,请重新获取");
}
if (statusRes.code === 803) {
// 这一步会返回cookie
clearInterval(timer)
clearInterval(timer);
const cookie = statusRes.cookie;
resolve(/__csrf=[0-9a-z]+;/.exec(cookie)[0] + /MUSIC_U=[0-9a-z]+;/.exec(cookie)[0]);
resolve(
/__csrf=[0-9a-z]+;/.exec(cookie)[0] + /MUSIC_U=[0-9a-z]+;/.exec(cookie)[0],
);
}
}, 3000)
}, 3000);
});
}
@ -223,17 +260,16 @@ export class neteasepro extends plugin {
* @param redirect
* @returns {Promise<unknown>}
*/
async downloadMp3(mp3Url, redirect='manual') {
async downloadMp3(mp3Url, redirect = "manual") {
return fetch(mp3Url, {
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",
redirect: redirect
})
.then(res => {
const path = `./data/rcmp4/${this.e.group_id || this.e.user_id}/temp.mp3`
redirect: redirect,
}).then(res => {
const path = `./data/rcmp4/${this.e.group_id || this.e.user_id}/temp.mp3`;
const fileStream = fs.createWriteStream(path);
res.body.pipe(fileStream);
return new Promise((resolve, reject) => {
@ -248,7 +284,7 @@ export class neteasepro extends plugin {
});
});
});
})
});
}
// 获取redis的key