103 lines
2.5 KiB
TypeScript
103 lines
2.5 KiB
TypeScript
import { Method } from "axios";
|
||
import axios from "axios";
|
||
import QbAddressDto from "../entity/dto/QbAddressDto";
|
||
import GlobalService from '../service/GlobalConfigService';
|
||
import { setUncaughtExceptionCaptureCallback } from "process";
|
||
|
||
//qb状态，true正常，false:无法访问
|
||
let qbStatus = true;
|
||
let qbInfo: QbAddressDto = null;
|
||
let cookie: string = null;
|
||
|
||
export function getQbStatus() {
|
||
return qbStatus;
|
||
}
|
||
|
||
export async function updateQbInfo(info: QbAddressDto, status: boolean) {
|
||
if (!info) {
|
||
let obj = await GlobalService.getMultVal(["qbAddress", "qbUsername", "qbPassword"]);
|
||
if (!obj.qbAddress) {
|
||
qbStatus = false;
|
||
return;
|
||
}
|
||
qbInfo.address = obj.qbAddress;
|
||
qbInfo.username = obj.qbUsername;
|
||
qbInfo.password = obj.qbPassword;
|
||
} else {
|
||
qbInfo = info;
|
||
}
|
||
if (status) {
|
||
qbStatus = status;
|
||
}
|
||
|
||
axios.defaults.baseURL = qbInfo.address;
|
||
}
|
||
|
||
export function get() {
|
||
|
||
}
|
||
|
||
export function post() {
|
||
|
||
}
|
||
|
||
async function request(method: Method, url: string, query: any, body: any, isForm = false) {
|
||
if (!qbStatus) {
|
||
throw new Error("qbittorrent无法连接，请检查配置");
|
||
}
|
||
let isTryLogin = false;
|
||
while (true) {
|
||
let headers = { "Cookie": cookie };
|
||
if (isForm) {
|
||
headers['content-type'] = "multipart/form-data";
|
||
} else if (method == "post") {
|
||
headers['content-type'] = "application/json";
|
||
}
|
||
let res = await axios.request({
|
||
baseURL: qbInfo.address,
|
||
url: url,
|
||
method,
|
||
params: query,
|
||
data: body,
|
||
headers,
|
||
});
|
||
if (res.status == 200) {
|
||
return res.data;
|
||
} if (res.status == 403) {
|
||
if (isTryLogin) {
|
||
throw new Error("qb用户名密码设置有误");
|
||
} else {
|
||
await tryLogin(qbInfo.address, qbInfo.username, qbInfo.password, true);
|
||
isTryLogin = true;
|
||
}
|
||
} else {
|
||
throw new Error("请求报错:" + res.data);
|
||
}
|
||
}
|
||
|
||
}
|
||
|
||
export async function tryLogin(address: string, username: string, password: string, updateStatus: boolean): Promise<void> {
|
||
let body = { username, password };
|
||
try {
|
||
let res = await axios.post(address + "/api/v2/auth/login", body, {
|
||
headers: { "Content-Type": "multipart/form-data;boundary=--------------------------125002698093981740970152" }
|
||
});
|
||
let success = res.data.toLocaleLowerCase().contains('ok');
|
||
if (updateStatus) {
|
||
qbStatus = success;
|
||
}
|
||
if (!success) {
|
||
throw new Error("登录失败");
|
||
} else {
|
||
cookie = res.headers['Cookie'];
|
||
}
|
||
} catch (error) {
|
||
console.error("登录报错：", error);
|
||
if (updateStatus) {
|
||
qbStatus = false;
|
||
}
|
||
throw new Error("登录出错");
|
||
}
|
||
}
|