const express = require('express');
const mongoose = require('mongoose');
const dotenv = require('dotenv');
const userRoutes = require('./routes/user');
const triangleRoutes = require('./routes/triangle');
const app = express();

dotenv.config();

// 中间件
app.use(express.json({ limit: '10mb' }));

app.use((req, res, next) => {
  res.badRequest = (msg) => {
    res.status(400).json({ statusCode: 400, code: -1, message: msg ?? 'Bad Request' });
  }
  res.unauthorized = (msg) => {
    res.status(401).json({ statusCode: 401, code: -1,  message: msg ?? 'Unauthorized' });
  }
  res.serverError = (msg) => {
    res.status(500).json({ statusCode: 500, code: -1,  message: msg ?? 'Internal Server Error' });
  }
  res.sendSuccess = (data, msg) => {
    const response = { statusCode: 200, code: 0,  message: msg ?? 'OK' }
    if (data) {
      response.data = data;
    }
    res.status(200).json(response);
  }
  next();
});

// 路由
app.use('/api/user', userRoutes);
app.use('/api/triangle', triangleRoutes);

// 启动服务
const PORT = process.env.PORT || 9055;
mongoose.connect(process.env.MONGO_URI)
.then(() => {
  console.log('MongoDB connected');
  app.listen(PORT, () => console.log(`Server running on port ${PORT}`));
})
.catch(err => console.error(err));