✨ 完成 cookie_token 获取

http/request/getCookieToken.js

@@ -0,0 +1,28 @@
+/**
+ * @file http request getCookieToken.js
+ * @description 获取cookieToken
+ * @author BTMuli<bt-muli@outlook.com>
+ * @since 1.1.0
+ */
+
+// Node
+import axios from "axios";
+// TGAssistant
+import getHeader from "../tools/getHeader.js";
+
+
+/**
+ * @description 根据 stoken 获取 cookieToken
+ * @param {string} cookie cookie
+ * @param {string} SToken sToken
+ * @returns {Promise<string>} cookieToken
+ */
+export async function getCookieTokenBySToken(cookie, SToken) {
+	const urlPre = "https://passport-api.mihoyo.com/account/auth/api/getCookieAccountInfoBySToken";
+	const urlCur = `stoken=${SToken}`;
+	const url = `${urlPre}?${urlCur}`;
+	const header = getHeader(cookie, "GET", urlCur);
+	return await axios.get(url, { headers: header }).then(res => {
+		return res.data["data"]["cookie_token"];
+	});
+}

http/tools/operSQLite.js

@@ -9,6 +9,7 @@
 import sqlite3 from "sqlite3";
 // TGAssistant
 import HttpConstant from "../constant/index.js";
+import { transCookie } from "./utils.js";
 
 class LocalSqlite {
 	constructor() {
@@ -53,7 +54,7 @@ class LocalSqlite {
 	async getCookie() {
 		const sql = "select * from AppData where key='cookie'";
 		const res = await this.selectSingle(sql);
-		return JSON.parse(res.value);
+		return transCookie(JSON.parse(res.value));
 	}
 
 	/**
@@ -63,7 +64,9 @@ class LocalSqlite {
 	 * @return {Promise<string>}
 	 */
 	async getCookieItem(itemKey){
-		const cookie = await this.getCookie();
+		const sql = "select * from AppData where key='cookie'";
+		const res = await this.selectSingle(sql);
+		const cookie = JSON.parse(res.value);
 		return cookie[itemKey];
 	}
 

http/tools/utils.js

@@ -29,4 +29,21 @@ export function getRandomString(length) {
  */
 export function getRandomNumber(min, max) {
 	return Math.floor(Math.random() * (max - min + 1) + min);
+}
+
+/**
+ * @description 转换 cookie
+ * @since 1.1.0
+ * @param {object} cookie cookie
+ * @returns {string} 转换后的 cookie
+ */
+export function transCookie(cookie) {
+	let res = "";
+	const keys = Object.keys(cookie);
+	for (const key of keys){
+		if(cookie[key]!==""){
+			res += `${key}=${cookie[key]};`;
+		}
+	}
+	return res;
 }

test/http/getCookieToken.test.js

@@ -0,0 +1,23 @@
+/**
+ * @file test http getCookieToken.test.js
+ * @description 测试 CookieToken 的获取
+ * @author BTMuli<bt-muli@outlook.com>
+ * @since 1.1.0
+ */
+
+// Node
+import { describe, it } from "mocha";
+import assert from "node:assert";
+// TGAssistant
+import { getCookieTokenBySToken } from "../../http/request/getCookieToken.js";
+import LocalSqlite from "../../http/tools/operSQLite.js";
+
+describe("测试 cookieToken 获取", ()=>{
+	it("通过 stoken", async ()=>{
+		const stoken = await LocalSqlite.getCookieItem("stoken");
+		const cookie = JSON.stringify(await LocalSqlite.getCookie());
+		const res = await getCookieTokenBySToken(cookie, stoken);
+		const cookieToken = await LocalSqlite.getCookieItem("cookie_token");
+		assert.strictEqual(res, cookieToken);
+	});
+});