server.js 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. const express = require('express');
  2. const mongoose = require('mongoose');
  3. const dotenv = require('dotenv');
  4. const Logs = require('./libs/logs');
  5. const userRoutes = require('./routes/user');
  6. const systemRoutes = require('./routes/system');
  7. const psteryRoutes = require('./routes/pstery');
  8. const cookieParser = require('cookie-parser');
  9. const app = express();
  10. dotenv.config();
  11. // 添加 CORS 支持
  12. app.use((req, res, next) => {
  13. res.header('Access-Control-Allow-Origin', '*');
  14. res.header('Access-Control-Allow-Headers', 'Origin, X-Requested-With, Content-Type, Accept, Authorization');
  15. res.header('Access-Control-Allow-Methods', 'GET, POST, PUT, DELETE, OPTIONS');
  16. if (req.method === 'OPTIONS') {
  17. return res.sendStatus(200);
  18. }
  19. next();
  20. });
  21. // 中间件
  22. app.use(express.json({ limit: '10mb' }));
  23. app.use(cookieParser());
  24. app.use((req, res, next) => {
  25. res.badRequest = (msg) => {
  26. res.status(400).json({ statusCode: 400, code: -1, message: msg ?? 'Bad Request' });
  27. }
  28. res.unauthorized = (msg) => {
  29. res.status(401).json({ statusCode: 401, code: -1, message: msg ?? 'Unauthorized' });
  30. }
  31. res.notFound = (msg) => {
  32. res.status(404).json({ statusCode: 404, code: -1, message: msg ?? 'Not Found' });
  33. }
  34. res.serverError = (msg) => {
  35. res.status(500).json({ statusCode: 500, code: -1, message: msg ?? 'Internal Server Error' });
  36. }
  37. res.sendSuccess = (data, msg) => {
  38. const response = { statusCode: 200, code: 0, message: msg ?? 'OK' }
  39. if (data) {
  40. response.data = data;
  41. }
  42. res.status(200).json(response);
  43. }
  44. next();
  45. });
  46. app.use(['/api/user', '/api/auth'], userRoutes);
  47. app.use('/api/system', systemRoutes);
  48. app.use('/api/pstery', psteryRoutes);
  49. // 启动服务
  50. const PORT = process.env.PORT || 9055;
  51. mongoose.connect(process.env.MONGO_URI)
  52. .then(() => {
  53. Logs.out('MongoDB connected');
  54. app.listen(PORT, () => console.log(`Server running on port ${PORT}`));
  55. })
  56. .catch(Logs.err);