feat:版本迭代

backend/src/controllers/quark.ts

@@ -1,40 +1,52 @@
 import { Request, Response, NextFunction } from "express";
 import { QuarkService } from "../services/QuarkService";
-import { config } from "../config";
-import { handleResponse } from "../utils/responseHandler";
-import handleError from "../utils/handleError";
+import { sendSuccess, sendError } from "../utils/response";
+import UserSetting from "../models/UserSetting";
 
-const { cookie } = config.quark;
+const quark = new QuarkService();
 
-const quark = new QuarkService(cookie);
+const setCookie = async (req: Request) => {
+  const userId = req.user?.userId;
+  const userSetting = await UserSetting.findOne({
+    where: { userId },
+  });
+  if (userSetting && userSetting.dataValues.quarkCookie) {
+    quark.setCookie(userSetting.dataValues.quarkCookie);
+  } else {
+    throw new Error("请先设置夸克网盘cookie");
+  }
+};
 
 export const quarkController = {
   async getShareInfo(req: Request, res: Response, next: NextFunction) {
     try {
       const { pwdId, passcode } = req.query;
+      await setCookie(req);
       const result = await quark.getShareInfo(pwdId as string, passcode as string);
-      handleResponse(res, result, true);
+      sendSuccess(res, result);
     } catch (error) {
-      handleError(res, error, "获取分享信息失败", next);
+      sendError(res, { message: "获取分享信息失败" });
     }
   },
 
   async getFolderList(req: Request, res: Response, next: NextFunction) {
     try {
       const { parentCid } = req.query;
+      await setCookie(req);
       const result = await quark.getFolderList(parentCid as string);
-      handleResponse(res, result, true);
+      sendSuccess(res, result);
     } catch (error) {
-      handleError(res, error, "获取目录列表失败", next);
+      sendError(res, { message: (error as Error).message || "获取目录列表失败" });
     }
   },
 
   async saveFile(req: Request, res: Response, next: NextFunction) {
     try {
+      await setCookie(req);
       const result = await quark.saveSharedFile(req.body);
-      handleResponse(res, result, true);
+      sendSuccess(res, result);
     } catch (error) {
-      handleError(res, error, "保存文件失败", next);
+      sendError(res, { message: (error as Error).message || "保存文件失败" });
     }
   },
 };