refactor: #42 Optimize project file structure

Author: CrazyMrYan

index.js

@@ -1,62 +1,32 @@
 // app.js
 const Koa = require("koa");
-const { koaBody } = require("koa-body");
-const path = require("path");
-const fs = require("fs");
 const sequelize = require("./utils/dbInstance");
 const filesRouter = require("./routers/files");
 const usersRouter = require("./routers/users");
 const redisClient = require("./redis");
 const authenticateToken = require("./middleware/authenticateToken");
-const cors = require("@koa/cors");
+const koaBody = require("./middleware/koaBody");
+const cors = require("./middleware/cors");
+const createInitDir = require("./utils/createInitDir");
+
 require("./models");
 require("./schedules/fileRecover");
 require("dotenv").config({ path: ".env.local" });
 
 const app = new Koa();
 
-app.use(
-  cors({
-    origin: "http://localhost:5173/", // 允许的来源
-    allowMethods: ["GET", "POST"], // 允许的方法
-  })
-);
-
-app.use(require("koa-static")(path.join(__dirname, "public")));
-
-const createDirectories = () => {
-  const dirs = [
-    path.join(__dirname, "provisional"),
-    path.join(__dirname, "resource"),
-  ];
-  dirs.forEach((dir) => {
-    if (!fs.existsSync(dir)) {
-      fs.mkdirSync(dir, { recursive: true });
-    }
-  });
-};
-
-createDirectories();
+app.use(cors());
 
 app.use(authenticateToken);
 
-app.use(
-  koaBody({
-    multipart: true,
-    // 解决 DELETE 没法获取ids的问题
-    parsedMethods: ["POST", "PUT", "PATCH", "DELETE"],
-    formidable: {
-      uploadDir: path.join(__dirname, "provisional"), // 临时上传目录
-      keepExtensions: true, // 保留文件扩展名
-    },
-  })
-);
+app.use(koaBody());
 
 // 挂载文件路由
 app.use(usersRouter.routes()).use(usersRouter.allowedMethods());
 app.use(filesRouter.routes()).use(filesRouter.allowedMethods());
 
 app.listen(process.env.SERVER_PORT, async () => {
+  createInitDir();
   await redisClient.connect();
   await sequelize.sync();
   console.log(`Server is running on ${process.env.INTERNAL_NETWORK_DOMAIN}`);

middleware/cors.js

@@ -0,0 +1,7 @@
+const cors = require("@koa/cors");
+
+module.exports = () =>
+  cors({
+    origin: "http://localhost:5173/", // 允许的来源
+    allowMethods: ["GET", "POST"], // 允许的方法
+  });

middleware/koaBody.js

@@ -0,0 +1,13 @@
+const { koaBody } = require("koa-body");
+const path = require("path");
+
+module.exports = () =>
+  koaBody({
+    multipart: true,
+    // 解决 DELETE 没法获取ids的问题
+    parsedMethods: ["POST", "PUT", "PATCH", "DELETE"],
+    formidable: {
+      uploadDir: path.join(__dirname, "..", "provisional"), // 临时上传目录
+      keepExtensions: true, // 保留文件扩展名
+    },
+  });

public/index.html

@@ -0,0 +1,14 @@
+const path = require("path");
+const fs = require("fs");
+
+module.exports = () => {
+  const dirs = [
+    path.join(__dirname, "..", "provisional"),
+    path.join(__dirname, "..", "resource"),
+  ];
+  dirs.forEach((dir) => {
+    if (!fs.existsSync(dir)) {
+      fs.mkdirSync(dir, { recursive: true });
+    }
+  });
+};