feat: 网易云功能重构

This commit is contained in:
zhiyu1998 2023-03-14 21:15:57 +08:00
parent ce9449b781
commit bddfa2f148
3 changed files with 282 additions and 0 deletions

138
apps/neteasepro.js Normal file
View File

@ -0,0 +1,138 @@
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} from '../utils/netease.js';
import {ha12store, store2ha1} from '../utils/encrypt.js';
export class neteasepro extends plugin {
constructor() {
super({
/** 功能名称 */
name: 'R插件网易云音乐解析',
/** 功能描述 */
dsc: '网易云音乐解析Pro',
/** https://oicqjs.github.io/oicq/#events */
event: 'message',
/** 优先级,数字越小等级越高 */
priority: 1,
rule: [
{
/** 命令正则匹配 */
reg: '#网易云登陆',
/** 执行方法 */
fnc: 'neteaseCloudLogin'
},
{
reg: '#网易云每日推荐',
fnc: 'neteaseDailyRecommend'
}
]
})
}
async neteaseCloudLogin(e) {
let userInfo;
// 如果不存在cookie
if (!await redis.exists(await this.getRedisKey(e.user_id))) {
// 获取密钥
const key = await getKey();
// console.log(key);
// 获取二维码
const qrPic = await getQrCode(key);
// 下载qrcode
await this.downloadQrCode(qrPic).then(path => {
// 发送二维码
e.reply(segment.image(fs.readFileSync(path)))
})
// 定时轮询
await this.poll(key).then(async cookie => {
// 存放到redis
await redis.set(await this.getRedisKey(e.user_id), ha12store(cookie))
});
}
// 从redis中获取
const realCookie = await store2ha1(await redis.get(await this.getRedisKey(e.user_id)));
// 获取用户信息
userInfo = await getLoginStatus(realCookie);
// 提取信息
const {nickname, avatarUrl} = userInfo.profile;
e.reply(["欢迎使用 🎶网易云音乐 🎶," + nickname, segment.image(avatarUrl)])
}
async neteaseDailyRecommend(e) {
const realCookie = await this.aopBefore(e);
if (realCookie === "") {
return true;
}
// 获取每日推荐所有数据
const dailyRecommend = await getDailyRecommend(realCookie);
// 由于数据过大取前10
let combineMsg = []
dailyRecommend.dailySongs.slice(0, 10).forEach(item => {
combineMsg.push([`${item?.id}: ${item?.name}-${item?.ar?.[0].name}-${item?.al?.name}`, segment.image(item?.al?.picUrl)])
})
await e.reply(await Bot.makeForwardMsg(combineMsg));
}
// 切面方法检测cookie
async aopBefore(e) {
// 取出cookie
const cookie = await redis.get(await this.getRedisKey(e.user_id));
// 如果不存在cookie
if (!cookie) {
e.reply("请先#网易云登录");
return "";
}
// 解析cookie
return store2ha1(cookie);
}
// 下载二维码
async downloadQrCode(qrPic) {
return axios
.get(qrPic, {
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",
})
.then((resp) => {
const filepath = "./netease_qr.jpg";
const writer = fs.createWriteStream(filepath);
resp.data.pipe(writer);
return new Promise((resolve, reject) => {
writer.on("finish", () => resolve(filepath));
writer.on("error", reject);
});
});
}
// 定时轮询
async poll(key) {
let timer;
return new Promise((resolve, reject) => {
timer = setInterval(async () => {
const statusRes = await getCookies(key)
console.log(statusRes)
if (statusRes.code === 800) {
clearInterval(timer)
reject('二维码已过期,请重新获取')
}
if (statusRes.code === 803) {
// 这一步会返回cookie
clearInterval(timer)
const cookie = statusRes.cookie;
resolve(/__csrf=[0-9a-z]+;/.exec(cookie)[0] + /MUSIC_U=[0-9a-z]+;/.exec(cookie)[0]);
}
}, 3000)
});
}
// 获取redis的key
async getRedisKey(user_id) {
return `Yz:rconsole:netease:${user_id}`;
}
}

57
utils/encrypt.js Normal file
View File

@ -0,0 +1,57 @@
// AES加密
import crypto from "crypto";
const key = crypto.createHash("sha256").update("rconsole").digest();
/**
* AES解密
* @param ha1
* @returns {Promise<string>}
*/
async function ha12store(ha1) {
// IV.E
const iv = crypto.randomBytes(16);
const c = crypto.createCipheriv("aes-256-cbc", key, iv);
let e = c.update(ha1, "binary", "hex");
e += c.final("hex");
return iv.toString("hex") + "." + e;
}
/**
* AES解密
* @param passstore
* @returns {Promise<string>}
*/
async function store2ha1(passstore) {
try {
const parts = passstore.split(".");
if (parts.length === 2) {
// 新的加密方式 with IV: IV.E
const c = crypto.createDecipheriv(
"aes-256-cbc",
key,
Buffer.from(parts[0], "hex")
);
let d = c.update(parts[1], "hex", "binary");
d += c.final("binary");
return d;
} else {
// 旧加密方式 without IV: E
const c = crypto.createDecipher("aes192", key);
let d = c.update(passstore, "hex", "binary");
d += c.final("binary");
return d;
}
} catch (e) {
console.log(
"在[default]部分设置的passwordSecret无法解密信息。请确保所有节点的passwordSecret相同。如果您更改了密码保密信息可能需要重新添加用户。",
e
);
process.exit(1);
}
}
export {
ha12store,
store2ha1
}

87
utils/netease.js Normal file
View File

@ -0,0 +1,87 @@
// 获取cookie
import fetch from "node-fetch";
import axios from "axios";
const BASE_URL = "http://127.0.0.1:3000"
/**
* 获取cookie
* @param key
* @returns {Promise<unknown>}
*/
async function getCookies(key) {
const cookieUrl = `${BASE_URL}/login/qr/check?key=${key}&timestamp=${Date.now()}`;
return fetch(cookieUrl).then(async (resp) => {
return await resp.json();
});
}
/**
* 获取登陆状态
* @param cookie
* @returns {Promise<AxiosResponse<any>>}
*/
async function getLoginStatus(cookie) {
return axios({
url: `${BASE_URL}/login/status?timestamp=${Date.now()}`,
method: 'post',
data: {
cookie,
},
})
.then(resp => {
console.log(resp.data.data)
return resp.data.data
});
}
/**
* 获取每日推荐
* @param cookie
* @returns {Promise<AxiosResponse<any>>}
*/
async function getDailyRecommend(cookie) {
return axios({
url: `${BASE_URL}/recommend/songs?timestamp=${Date.now()}`,
method: 'get',
data: {
cookie,
},
})
.then(resp => {
return resp.data.data
});
}
/**
* 获取密匙
* @returns {Promise<*>}
*/
async function getKey() {
const keyUrl = `${BASE_URL}/login/qr/key?timestamp=${Date.now()}`;
return await fetch(keyUrl).then(async (resp) => {
const respJson = await resp.json();
return respJson.data.unikey;
});
}
/**
* 获取二维码
* @param key
* @returns {Promise<*>}
*/
async function getQrCode(key) {
const qrPicUrl = `${BASE_URL}/login/qr/create?key=${key}&qrimg=true&timestamp=${Date.now()}`;
return await fetch(qrPicUrl).then(async (resp) => {
const respJson = await resp.json();
return respJson.data.qrimg;
});
}
export {
getCookies,
getLoginStatus,
getDailyRecommend,
getKey,
getQrCode,
}