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