FastGPT/admin/service/route/user.js

import { User, Pay } from '../schema.js';
import dayjs from 'dayjs';
import { auth } from './system.js';
import crypto from 'crypto';

// 加密
const hashPassword = (psw) => {
  return crypto.createHash('sha256').update(psw).digest('hex');
};

export const useUserRoute = (app) => {
  // 获取用户列表
  app.get('/users', auth(), async (req, res) => {
    try {
      const start = parseInt(req.query._start) || 0;
      const end = parseInt(req.query._end) || 20;
      const order = req.query._order === 'DESC' ? -1 : 1;
      const sort = req.query._sort || 'createTime';
      const username = req.query.username || '';
      const where = {
        username: { $regex: username, $options: 'i' }
      };

      const usersRaw = await User.find(where)
        .skip(start)
        .limit(end - start)
        .sort({ [sort]: order });

      const users = usersRaw.map((user) => {
        const obj = user.toObject();
        return {
          ...obj,
          id: obj._id,
          createTime: dayjs(obj.createTime).format('YYYY/MM/DD HH:mm')
        };
      });

      const totalCount = await User.countDocuments(where);

      res.header('Access-Control-Expose-Headers', 'X-Total-Count');
      res.header('X-Total-Count', totalCount);
      res.json(users);
    } catch (err) {
      console.log(`Error fetching users: ${err}`);
      res.status(500).json({ error: 'Error fetching users' });
    }
  });

  // 创建用户
  app.post('/users', auth(), async (req, res) => {
    try {
      const {
        username,
        password,
        balance,
        promotion,
        openaiKey = '',
        avatar = '/icon/human.png'
      } = req.body;
      if (!username || !password || !balance) {
        return res.status(400).json({ error: 'Invalid user information' });
      }
      const existingUser = await User.findOne({ username });
      if (existingUser) {
        return res.status(400).json({ error: 'Username already exists' });
      }
      const user = new User({
        _id: new mongoose.Types.ObjectId(),
        username,
        password,
        balance,
        promotion: {
          rate: promotion?.rate || 0
        },
        openaiKey,
        avatar,
        createTime: new Date()
      });
      const result = await user.save();
      res.json(result);
    } catch (err) {
      console.log(`Error creating user: ${err}`);
      res.status(500).json({ error: 'Error creating user' });
    }
  });

  // 修改用户信息
  app.put('/users/:id', auth(), async (req, res) => {
    try {
      const _id = req.params.id;
  
      // Check if a new password is provided in the request body
      if (req.body.password) {
        // Hash the new password
        const hashedPassword = hashPassword(req.body.password);
        req.body.password = hashedPassword;
      }
  
      const result = await User.updateOne({ _id: _id }, { $set: req.body });
      res.json(result);
    } catch (err) {
      console.log(`Error updating user: ${err}`);
      res.status(500).json({ error: 'Error updating user' });
    }
  });

  // 新增: 获取 pays 列表
  app.get('/pays', auth(), async (req, res) => {
    try {
      const start = parseInt(req.query._start) || 0;
      const end = parseInt(req.query._end) || 20;
      const order = req.query._order === 'DESC' ? -1 : 1;
      const sort = req.query._sort || '_id';
      const userId = req.query.userId || '';
      const where = userId ? { userId: userId } : {};

      const paysRaw = await Pay.find({
        ...where
      })
        .skip(start)
        .limit(end - start)
        .sort({ [sort]: order });

      const pays = [];

      for (const payRaw of paysRaw) {
        const pay = payRaw.toObject();

        const orderedPay = {
          id: pay._id.toString(),
          userId: pay.userId,
          price: pay.price,
          orderId: pay.orderId,
          status: pay.status,
          createTime: dayjs(pay.createTime).format('YYYY/MM/DD HH:mm')
        };

        pays.push(orderedPay);
      }
      const totalCount = await Pay.countDocuments({
        ...where
      });
      res.header('Access-Control-Expose-Headers', 'X-Total-Count');
      res.header('X-Total-Count', totalCount);
      res.json(pays);
    } catch (err) {
      console.log(`Error fetching pays: ${err}`);
      res.status(500).json({ error: 'Error fetching pays', details: err.message });
    }
  });
};