106 lines
3.1 KiB
TypeScript
106 lines
3.1 KiB
TypeScript
// import "./logger";
|
|
import "./err";
|
|
import "./env";
|
|
import express, { Request, Response, NextFunction } from "express";
|
|
import { Server } from "socket.io";
|
|
import http from "node:http";
|
|
import expressWs from "express-ws";
|
|
import logger from "morgan";
|
|
import cors from "cors";
|
|
import buildRoute from "@/core";
|
|
import fs from "fs";
|
|
import u from "@/utils";
|
|
import jwt from "jsonwebtoken";
|
|
import socketInit from "@/socket/index";
|
|
|
|
const app = express();
|
|
const server = http.createServer(app);
|
|
|
|
export default async function startServe(randomPort: Boolean = false) {
|
|
const io = new Server(server, { cors: { origin: "*" } });
|
|
socketInit(io);
|
|
|
|
if (process.env.NODE_ENV == "dev") await buildRoute();
|
|
|
|
expressWs(app);
|
|
|
|
app.use(logger("dev"));
|
|
app.use(cors({ origin: "*" }));
|
|
app.use(express.json({ limit: "100mb" }));
|
|
app.use(express.urlencoded({ extended: true, limit: "100mb" }));
|
|
|
|
const rootDir = u.getPath("oss");
|
|
|
|
// 确保 uploads 目录存在
|
|
if (!fs.existsSync(rootDir)) {
|
|
fs.mkdirSync(rootDir, { recursive: true });
|
|
}
|
|
console.log("文件目录:", rootDir);
|
|
|
|
app.use(express.static(rootDir));
|
|
|
|
app.use(async (req, res, next) => {
|
|
const setting = await u.db("o_setting").where("key", "tokenKey").select("value").first();
|
|
if (!setting) return res.status(444).send({ message: "服务器秘钥未配置，请联系管理员" });
|
|
const { value: tokenKey } = setting;
|
|
// 从 header 或 query 参数获取 token
|
|
const rawToken = req.headers.authorization || (req.query.token as string) || "";
|
|
const token = rawToken.replace("Bearer ", "");
|
|
// 白名单路径
|
|
if (req.path === "/api/login/login") return next();
|
|
|
|
if (!token) return res.status(401).send({ message: "未提供token" });
|
|
try {
|
|
const decoded = jwt.verify(token, tokenKey as string);
|
|
(req as any).user = decoded;
|
|
next();
|
|
} catch (err) {
|
|
return res.status(401).send({ message: "无效的token" });
|
|
}
|
|
});
|
|
|
|
const router = await import("@/router");
|
|
await router.default(app);
|
|
|
|
// 404 处理
|
|
app.use((_, res, next: NextFunction) => {
|
|
return res.status(404).send({ message: "API 404 Not Found" });
|
|
});
|
|
|
|
// 错误处理
|
|
app.use((err: any, _: Request, res: Response, __: NextFunction) => {
|
|
res.locals.message = err.message;
|
|
res.locals.error = err;
|
|
console.error(err);
|
|
res.status(err.status || 500).send(err);
|
|
});
|
|
|
|
const port = randomPort ? 0 : parseInt(process.env.PORT || "10588");
|
|
return await new Promise((resolve) => {
|
|
server.listen(port, async () => {
|
|
const address = server.address();
|
|
const realPort = typeof address === "string" ? address : address?.port;
|
|
console.log(`[服务启动成功]: http://localhost:${realPort}`);
|
|
resolve(realPort);
|
|
});
|
|
});
|
|
}
|
|
|
|
// 支持await关闭
|
|
export function closeServe(): Promise<void> {
|
|
return new Promise((resolve, reject) => {
|
|
if (server) {
|
|
server.close((err?: Error) => {
|
|
if (err) return reject(err);
|
|
console.log("[服务已关闭]");
|
|
resolve();
|
|
});
|
|
} else {
|
|
resolve();
|
|
}
|
|
});
|
|
}
|
|
|
|
const isElectron = typeof process.versions?.electron !== "undefined";
|
|
if (!isElectron) startServe();
|