Refactoring the backend

backend/src/services/Cloud115Service.ts

@@ -1,7 +1,11 @@
 import { AxiosHeaders, AxiosInstance } from "axios"; // 导入 AxiosHeaders
 import { createAxiosInstance } from "../utils/axiosInstance";
-import { Logger } from "../utils/logger";
 import { ShareInfoResponse } from "../types/cloud115";
+import { injectable } from "inversify";
+import { Request } from "express";
+import UserSetting from "../models/UserSetting";
+import { ICloudService } from "../types/services";
+import { logger } from "@/utils/logger";
 
 interface Cloud115ListItem {
   cid: string;
@@ -20,11 +24,12 @@ interface Cloud115PathItem {
   name: string;
 }
 
-export class Cloud115Service {
+@injectable()
+export class Cloud115Service implements ICloudService {
   private api: AxiosInstance;
   private cookie: string = "";
 
-  constructor(cookie?: string) {
+  constructor() {
     this.api = createAxiosInstance(
       "https://webapi.115.com",
       AxiosHeaders.from({
@@ -44,19 +49,23 @@ export class Cloud115Service {
         "Accept-Language": "zh-CN,zh;q=0.9",
       })
     );
-    if (cookie) {
-      this.setCookie(cookie);
-    } else {
-      console.log("请注意:115网盘需要提供cookie进行身份验证");
-    }
+
     this.api.interceptors.request.use((config) => {
-      config.headers.cookie = cookie || this.cookie;
+      config.headers.cookie = this.cookie;
       return config;
     });
   }
 
-  public setCookie(cookie: string): void {
-    this.cookie = cookie;
+  async setCookie(req: Request): Promise<void> {
+    const userId = req.user?.userId;
+    const userSetting = await UserSetting.findOne({
+      where: { userId },
+    });
+    if (userSetting && userSetting.dataValues.cloud115Cookie) {
+      this.cookie = userSetting.dataValues.cloud115Cookie;
+    } else {
+      throw new Error("请先设置115网盘cookie");
+    }
   }
 
   async getShareInfo(shareCode: string, receiveCode = ""): Promise<ShareInfoResponse> {
@@ -114,7 +123,7 @@ export class Cloud115Service {
           })),
       };
     } else {
-      Logger.error("获取目录列表失败:", response.data.error);
+      logger.error("获取目录列表失败:", response.data.error);
       throw new Error("获取115pan目录列表失败:" + response.data.error);
     }
   }
@@ -132,14 +141,14 @@ export class Cloud115Service {
       file_id: params.fileId,
     });
     const response = await this.api.post("/share/receive", param.toString());
-    Logger.info("保存文件:", response.data);
+    logger.info("保存文件:", response.data);
     if (response.data.state) {
       return {
         message: response.data.error,
         data: response.data.data,
       };
     } else {
-      Logger.error("保存文件失败:", response.data.error);
+      logger.error("保存文件失败:", response.data.error);
       throw new Error("保存115pan文件失败:" + response.data.error);
     }
   }