Files
ai-xiaocai/server/routes/rooms.js

71 lines
3.1 KiB
JavaScript

// 群聊管理路由
const express = require('express');
const { v4: uuidv4 } = require('uuid');
const { authMiddleware, adminOnly } = require('../middleware/auth');
const { broadcast } = require('../ws');
const db = require('../db');
const { timestamp } = require('../utils');
const router = express.Router();
router.get('/', authMiddleware, (req, res) => {
const username = req.user.username;
const isAdmin = req.user.isAdmin;
let rooms;
if (isAdmin) {
rooms = db.prepare('SELECT * FROM rooms').all();
} else {
rooms = db.prepare("SELECT * FROM rooms WHERE white_list = '' OR (',' || white_list || ',' LIKE ?)").all('%,' + username + ',%');
}
const result = rooms.map(room => {
const lastMsg = db.prepare('SELECT text, timestamp FROM messages WHERE room_id = ? ORDER BY id DESC LIMIT 1').get(room.id);
return { ...room, last_message: lastMsg ? lastMsg.text : '', last_time: lastMsg ? lastMsg.timestamp : '' };
});
// 按最后消息时间排序,最近的在前,无消息的排最后
result.sort((a, b) => {
if (!a.last_time && !b.last_time) return 0;
if (!a.last_time) return 1;
if (!b.last_time) return -1;
return b.last_time.localeCompare(a.last_time);
});
res.json(result);
});
// 普通用户可新建,白名单自动设为用户名
router.post('/', authMiddleware, (req, res) => {
const { name, whiteList } = req.body;
const isAdmin = req.user.isAdmin;
const finalWhiteList = isAdmin ? (whiteList || '') : req.user.username;
const id = uuidv4();
const now = timestamp();
db.prepare('INSERT INTO rooms (id, name, created_by, white_list, created_at) VALUES (?,?,?,?,?)').run(id, name, req.user.username, finalWhiteList, now);
broadcast({ type: 'room_created', room: { id, name, white_list: finalWhiteList } });
res.json({ id, name });
});
// 普通用户可编辑名称,白名单不变
router.put('/:roomId', authMiddleware, (req, res) => {
const { name, whiteList } = req.body;
const roomId = req.params.roomId;
const isAdmin = req.user.isAdmin;
const room = db.prepare('SELECT * FROM rooms WHERE id = ?').get(roomId);
if (!room) return res.status(404).json({ error: '群聊不存在' });
const finalWhiteList = isAdmin ? (whiteList !== undefined ? whiteList : room.white_list) : room.white_list;
db.prepare('UPDATE rooms SET name = ?, white_list = ? WHERE id = ?').run(name, finalWhiteList, roomId);
broadcast({ type: 'room_updated', room: { id: roomId, name, white_list: finalWhiteList } });
res.json({ success: true });
});
router.delete('/:roomId', authMiddleware, adminOnly, (req, res) => {
const roomId = req.params.roomId;
db.prepare('DELETE FROM messages WHERE room_id = ?').run(roomId);
db.prepare('DELETE FROM purchase_attachments WHERE purchase_id IN (SELECT id FROM purchases WHERE room_id = ?)').run(roomId);
db.prepare('DELETE FROM purchase_history WHERE purchase_id IN (SELECT id FROM purchases WHERE room_id = ?)').run(roomId);
db.prepare('DELETE FROM purchases WHERE room_id = ?').run(roomId);
db.prepare('DELETE FROM rooms WHERE id = ?').run(roomId);
broadcast({ type: 'room_deleted', roomId });
res.json({ success: true });
});
module.exports = router;