mirror of
https://github.com/jiangrui1994/CloudSaver.git
synced 2026-01-11 23:58:46 +08:00
346 lines
11 KiB
TypeScript
346 lines
11 KiB
TypeScript
import { defineStore } from "pinia";
|
|
import { cloud115Api } from "@/api/cloud115";
|
|
import { resourceApi } from "@/api/resource";
|
|
import { quarkApi } from "@/api/quark";
|
|
import type {
|
|
Resource,
|
|
ShareInfoResponse,
|
|
Save115FileParams,
|
|
SaveQuarkFileParams,
|
|
ShareInfo,
|
|
ResourceItem,
|
|
} from "@/types";
|
|
import { ElMessage } from "element-plus";
|
|
|
|
interface StorageListObject {
|
|
list: Resource[];
|
|
lastUpdateTime?: string;
|
|
}
|
|
|
|
const lastResource = (
|
|
localStorage.getItem("last_resource_list")
|
|
? JSON.parse(localStorage.getItem("last_resource_list") as string)
|
|
: { list: [] }
|
|
) as StorageListObject;
|
|
|
|
// 定义云盘驱动配置类型
|
|
interface CloudDriveConfig<
|
|
T extends Record<string, string>,
|
|
P extends Save115FileParams | SaveQuarkFileParams,
|
|
> {
|
|
name: string;
|
|
type: string;
|
|
regex: RegExp;
|
|
api: {
|
|
getShareInfo: (parsedCode: T) => Promise<ShareInfoResponse>;
|
|
saveFile: (params: P) => Promise<{ code: number; message?: string }>;
|
|
};
|
|
parseShareCode: (match: RegExpMatchArray) => T;
|
|
getSaveParams: (shareInfo: ShareInfoResponse, folderId: string) => P;
|
|
}
|
|
|
|
// 云盘类型配置
|
|
export const CLOUD_DRIVES: [
|
|
CloudDriveConfig<{ shareCode: string; receiveCode: string }, Save115FileParams>,
|
|
CloudDriveConfig<{ pwdId: string }, SaveQuarkFileParams>,
|
|
] = [
|
|
{
|
|
name: "115网盘",
|
|
type: "pan115",
|
|
regex: /(?:115|anxia|115cdn)\.com\/s\/([^?]+)(?:\?password=([^&#]+))?/,
|
|
api: {
|
|
getShareInfo: (parsedCode: { shareCode: string; receiveCode: string }) =>
|
|
cloud115Api.getShareInfo(parsedCode.shareCode, parsedCode.receiveCode),
|
|
saveFile: async (params: Save115FileParams) => {
|
|
return await cloud115Api.saveFile(params as Save115FileParams);
|
|
},
|
|
},
|
|
parseShareCode: (match: RegExpMatchArray) => ({
|
|
shareCode: match[1],
|
|
receiveCode: match[2] || "",
|
|
}),
|
|
getSaveParams: (shareInfo: ShareInfoResponse, folderId: string) => ({
|
|
shareCode: shareInfo.shareCode || "",
|
|
receiveCode: shareInfo.receiveCode || "",
|
|
fileId: shareInfo.list[0].fileId,
|
|
folderId,
|
|
}),
|
|
},
|
|
{
|
|
name: "夸克网盘",
|
|
type: "quark",
|
|
regex: /pan\.quark\.cn\/s\/([a-zA-Z0-9]+)/,
|
|
api: {
|
|
getShareInfo: (parsedCode: { pwdId: string }) => quarkApi.getShareInfo(parsedCode.pwdId),
|
|
saveFile: async (params: SaveQuarkFileParams) => {
|
|
return await quarkApi.saveFile(params as SaveQuarkFileParams);
|
|
},
|
|
},
|
|
parseShareCode: (match: RegExpMatchArray) => ({ pwdId: match[1] }),
|
|
getSaveParams: (shareInfo: ShareInfoResponse, folderId: string) => ({
|
|
fid_list: shareInfo.list.map((item: { fileId?: string }) => item.fileId || ""),
|
|
fid_token_list: shareInfo.list.map(
|
|
(item: { fileIdToken?: string }) => item.fileIdToken || ""
|
|
),
|
|
to_pdir_fid: folderId,
|
|
pwd_id: shareInfo.pwdId || "",
|
|
stoken: shareInfo.stoken || "",
|
|
pdir_fid: "0",
|
|
scene: "link",
|
|
}),
|
|
},
|
|
];
|
|
|
|
export const useResourceStore = defineStore("resource", {
|
|
state: () => ({
|
|
tagColor: {
|
|
baiduPan: "primary",
|
|
weiyun: "info",
|
|
aliyun: "warning",
|
|
pan115: "danger",
|
|
quark: "success",
|
|
},
|
|
keyword: "",
|
|
resources: lastResource.list,
|
|
lastUpdateTime: lastResource.lastUpdateTime || "",
|
|
shareInfo: {} as ShareInfoResponse,
|
|
resourceSelect: [] as ShareInfo[],
|
|
loading: false,
|
|
backupPlan: false,
|
|
loadTree: false,
|
|
}),
|
|
|
|
actions: {
|
|
setLoadTree(loadTree: boolean) {
|
|
this.loadTree = loadTree;
|
|
},
|
|
// 搜索资源
|
|
async searchResources(keyword?: string, isLoadMore = false, channelId?: string): Promise<void> {
|
|
this.loading = true;
|
|
if (!isLoadMore) this.resources = [];
|
|
try {
|
|
let lastMessageId = "";
|
|
if (isLoadMore) {
|
|
const list = this.resources.find((x) => x.id === channelId)?.list || [];
|
|
lastMessageId = list[list.length - 1].messageId || "";
|
|
if (list[list.length - 1].isLastMessage) {
|
|
ElMessage.warning("没有更多了~");
|
|
return;
|
|
}
|
|
if (!lastMessageId) {
|
|
ElMessage.error("当次搜索源不支持加载更多");
|
|
return;
|
|
}
|
|
keyword = this.keyword;
|
|
}
|
|
let { data = [] } = await resourceApi.search(keyword || "", channelId, lastMessageId);
|
|
this.keyword = keyword || "";
|
|
data = data.filter((item) => item.list.length > 0);
|
|
if (isLoadMore) {
|
|
const findedIndex = this.resources.findIndex((item) => item.id === data[0]?.id);
|
|
if (findedIndex !== -1) {
|
|
this.resources[findedIndex].list.push(...data[0].list);
|
|
}
|
|
if (data.length === 0) {
|
|
const list = this.resources.find((item) => item.id === channelId)?.list;
|
|
list && list[list.length - 1] && (list[list.length - 1]!.isLastMessage = true);
|
|
ElMessage.warning("没有更多了~");
|
|
}
|
|
} else {
|
|
this.resources = data.map((item, index) => ({ ...item, displayList: index === 0 }));
|
|
if (!keyword) {
|
|
// 获取当前时间字符串 用于存储到本地
|
|
this.lastUpdateTime = new Date().toLocaleString();
|
|
localStorage.setItem(
|
|
"last_resource_list",
|
|
JSON.stringify({ list: this.resources, lastUpdateTime: this.lastUpdateTime })
|
|
);
|
|
}
|
|
if (this.resources.length === 0) {
|
|
ElMessage.warning("未搜索到相关资源");
|
|
}
|
|
}
|
|
} catch (error) {
|
|
console.log(error);
|
|
this.handleError("搜索失败，请重试", null);
|
|
} finally {
|
|
this.loading = false;
|
|
}
|
|
},
|
|
|
|
// 设置选择资源
|
|
async setSelectedResource(resourceSelect: ShareInfo[]) {
|
|
this.resourceSelect = resourceSelect;
|
|
},
|
|
|
|
// 转存资源
|
|
async saveResource(resource: ResourceItem, folderId: string): Promise<void> {
|
|
const savePromises: Promise<void>[] = [];
|
|
CLOUD_DRIVES.forEach((drive) => {
|
|
if (resource.cloudLinks.some((link) => drive.regex.test(link))) {
|
|
savePromises.push(this.saveResourceToDrive(resource, folderId, drive));
|
|
}
|
|
});
|
|
await Promise.all(savePromises);
|
|
},
|
|
|
|
// 保存资源到网盘
|
|
async saveResourceToDrive(
|
|
resource: ResourceItem,
|
|
folderId: string,
|
|
drive:
|
|
| CloudDriveConfig<{ shareCode: string; receiveCode: string }, Save115FileParams>
|
|
| CloudDriveConfig<{ pwdId: string }, SaveQuarkFileParams>
|
|
): Promise<void> {
|
|
const link = resource.cloudLinks.find((link) => drive.regex.test(link));
|
|
if (!link) return;
|
|
|
|
const match = link.match(drive.regex);
|
|
if (!match) throw new Error("链接解析失败");
|
|
|
|
const shareInfo = {
|
|
...this.shareInfo,
|
|
list: this.resourceSelect.filter((x) => x.isChecked),
|
|
};
|
|
|
|
if (this.is115Drive(drive)) {
|
|
const params = drive.getSaveParams(shareInfo, folderId);
|
|
const result = await drive.api.saveFile(params);
|
|
|
|
if (result.code === 0) {
|
|
ElMessage.success(`${drive.name} 转存成功`);
|
|
} else {
|
|
ElMessage.error(result.message);
|
|
}
|
|
} else {
|
|
const params = drive.getSaveParams(shareInfo, folderId);
|
|
const result = await drive.api.saveFile(params);
|
|
|
|
if (result.code === 0) {
|
|
ElMessage.success(`${drive.name} 转存成功`);
|
|
} else {
|
|
ElMessage.error(result.message);
|
|
}
|
|
}
|
|
},
|
|
|
|
// 解析云盘链接
|
|
async parsingCloudLink(url: string): Promise<void> {
|
|
this.loading = true;
|
|
this.resources = [];
|
|
try {
|
|
const matchedDrive = CLOUD_DRIVES.find((drive) => drive.regex.test(url));
|
|
if (!matchedDrive) throw new Error("不支持的网盘链接");
|
|
|
|
const match = url.match(matchedDrive.regex);
|
|
if (!match) throw new Error("链接解析失败");
|
|
|
|
const parsedCode = matchedDrive.parseShareCode(match);
|
|
let shareInfo = this.is115Drive(matchedDrive)
|
|
? await matchedDrive.api.getShareInfo(
|
|
parsedCode as { shareCode: string; receiveCode: string }
|
|
)
|
|
: await matchedDrive.api.getShareInfo(parsedCode as { pwdId: string });
|
|
|
|
if (Array.isArray(shareInfo)) {
|
|
shareInfo = {
|
|
list: shareInfo,
|
|
};
|
|
}
|
|
|
|
if (shareInfo?.list?.length) {
|
|
this.resources = [
|
|
{
|
|
id: "",
|
|
channelInfo: {
|
|
name: "自定义搜索",
|
|
channelLogo: "",
|
|
channelId: "",
|
|
},
|
|
displayList: true,
|
|
list: [
|
|
{
|
|
id: "1",
|
|
title: shareInfo.list.map((item) => item.fileName).join(", "),
|
|
cloudLinks: [url],
|
|
cloudType: matchedDrive.type,
|
|
channel: matchedDrive.name,
|
|
pubDate: "",
|
|
},
|
|
],
|
|
},
|
|
];
|
|
} else {
|
|
throw new Error("解析失败，请检查链接是否正确");
|
|
}
|
|
} catch (error) {
|
|
this.handleError("解析失败，请重试", error);
|
|
} finally {
|
|
this.loading = false;
|
|
}
|
|
},
|
|
|
|
// 获取资源列表并选择
|
|
async getResourceListAndSelect(resource: ResourceItem): Promise<boolean> {
|
|
this.setSelectedResource([]);
|
|
const { cloudType } = resource;
|
|
const drive = CLOUD_DRIVES.find((x) => x.type === cloudType);
|
|
if (!drive) {
|
|
return false;
|
|
}
|
|
const link = resource.cloudLinks.find((link) => drive.regex.test(link));
|
|
if (!link) return false;
|
|
|
|
const match = link.match(drive.regex);
|
|
if (!match) throw new Error("链接解析失败");
|
|
|
|
const parsedCode = drive.parseShareCode(match);
|
|
let shareInfo = {} as ShareInfoResponse;
|
|
this.setLoadTree(true);
|
|
if (this.is115Drive(drive)) {
|
|
shareInfo = await drive.api.getShareInfo(
|
|
parsedCode as { shareCode: string; receiveCode: string }
|
|
);
|
|
} else {
|
|
shareInfo = this.is115Drive(drive)
|
|
? await drive.api.getShareInfo(parsedCode as { shareCode: string; receiveCode: string })
|
|
: await drive.api.getShareInfo(parsedCode as { pwdId: string });
|
|
}
|
|
this.setLoadTree(false);
|
|
if (shareInfo) {
|
|
if (Array.isArray(shareInfo)) {
|
|
shareInfo = {
|
|
list: shareInfo,
|
|
...parsedCode,
|
|
};
|
|
} else {
|
|
shareInfo = {
|
|
...shareInfo,
|
|
...parsedCode,
|
|
};
|
|
}
|
|
this.shareInfo = shareInfo;
|
|
this.setSelectedResource(this.shareInfo.list.map((x) => ({ ...x, isChecked: true })));
|
|
return true;
|
|
} else {
|
|
ElMessage.error("获取资源信息失败,请先检查cookie!");
|
|
return false;
|
|
}
|
|
},
|
|
|
|
// 统一错误处理
|
|
handleError(message: string, error: unknown): void {
|
|
console.error(message, error);
|
|
ElMessage.error(error instanceof Error ? error.message : message);
|
|
},
|
|
|
|
is115Drive(
|
|
drive:
|
|
| CloudDriveConfig<{ shareCode: string; receiveCode: string }, Save115FileParams>
|
|
| CloudDriveConfig<{ pwdId: string }, SaveQuarkFileParams>
|
|
): drive is CloudDriveConfig<{ shareCode: string; receiveCode: string }, Save115FileParams> {
|
|
return drive.type === "pan115";
|
|
},
|
|
},
|
|
});
|