feat: admin image

2023-06-10 00:35:26 +08:00
parent 2378615887
commit e19ac56fe5
21 changed files with 6045 additions and 416 deletions
--- a/admin/service/route/user.js
+++ b/admin/service/route/user.js
@@ -0,0 +1,138 @@
+import { User, Pay } from '../schema.js';
+import dayjs from 'dayjs';
+import { auth } from './system.js';
+
+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;
+
+      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 });
+    }
+  });
+};