refactor: v2 重构 — 模块化拆分 + AI Prompt模板化 + 前端状态管理
P0 - AI交互层:
- ai/prompt.js: 模块化 Prompt 模板
- ai/validator.js: JSON Schema 校验 + normalize
- ai/client.js: DeepSeek API 封装
- 金额计算唯一化,时间格式归一化
P1 - 后端架构:
- routes/: auth/rooms/messages/purchases 独立路由
- middleware/auth.js: 认证授权中间件
- ws/index.js: WebSocket 连接管理
- db.js: 启用 foreign_keys
- server.js: 从588行精简到40行入口
P2 - 前端架构:
- public/css/style.css: CSS 独立
- public/js/store.js: 全局状态 Store
- public/js/{chat,ws,purchases,auth}.js: 功能模块拆分
- index.html: 纯 HTML 结构, v2.6
This commit is contained in:
27
server/routes/auth.js
Normal file
27
server/routes/auth.js
Normal file
@@ -0,0 +1,27 @@
|
||||
// 用户认证路由
|
||||
const express = require('express');
|
||||
const jwt = require('jsonwebtoken');
|
||||
const { JWT_SECRET } = require('../middleware/auth');
|
||||
const router = express.Router();
|
||||
|
||||
const USERS = {};
|
||||
(process.env.USERS || '').split(',').forEach(u => {
|
||||
const [name, pass] = u.split(':').map(s => s.trim());
|
||||
if (name && pass) USERS[name] = pass;
|
||||
});
|
||||
const ADMINS = (process.env.ADMINS || '').split(',').map(s => s.trim()).filter(Boolean);
|
||||
|
||||
router.post('/login', (req, res) => {
|
||||
const { username, password } = req.body;
|
||||
if (USERS[username] && USERS[username] === password) {
|
||||
const token = jwt.sign({ username, isAdmin: ADMINS.includes(username) }, JWT_SECRET, { expiresIn: '7d' });
|
||||
return res.json({ token, username, isAdmin: ADMINS.includes(username) });
|
||||
}
|
||||
res.status(401).json({ error: '用户名或密码错误' });
|
||||
});
|
||||
|
||||
router.get('/user', require('../middleware/auth').authMiddleware, (req, res) => {
|
||||
res.json({ username: req.user.username, isAdmin: req.user.isAdmin });
|
||||
});
|
||||
|
||||
module.exports = router;
|
||||
70
server/routes/messages.js
Normal file
70
server/routes/messages.js
Normal file
@@ -0,0 +1,70 @@
|
||||
// 消息路由 + AI 分析
|
||||
const express = require('express');
|
||||
const { authMiddleware } = require('../middleware/auth');
|
||||
const { broadcastToRoomExcludeSelf } = require('../ws');
|
||||
const { buildSystemPrompt } = require('../ai/prompt');
|
||||
const { validate, normalize } = require('../ai/validator');
|
||||
const { callDeepSeek } = require('../ai/client');
|
||||
const { getHistory, addToHistory, hasPendingAction, handleAIResult, executePendingAction } = require('../handlers');
|
||||
const db = require('../db');
|
||||
const { timestamp } = require('../utils');
|
||||
|
||||
const router = express.Router();
|
||||
|
||||
router.get('/:roomId/messages', authMiddleware, (req, res) => {
|
||||
const msgs = db.prepare('SELECT * FROM messages WHERE room_id = ? ORDER BY id ASC').all(req.params.roomId);
|
||||
res.json(msgs);
|
||||
});
|
||||
|
||||
router.post('/:roomId/messages', authMiddleware, async (req, res) => {
|
||||
const { text, attachments } = req.body;
|
||||
console.log('📨 收到消息, text:', (text||'').substring(0,30), 'attachments:', JSON.stringify(attachments));
|
||||
const roomId = req.params.roomId;
|
||||
const username = req.user.username;
|
||||
const now = timestamp();
|
||||
|
||||
const result = db.prepare('INSERT INTO messages (room_id, user, text, attachments, timestamp) VALUES (?,?,?,?,?)').run(
|
||||
roomId, username, text || '', attachments ? JSON.stringify(attachments) : null, now
|
||||
);
|
||||
const msg = { id: result.lastInsertRowid, room_id: roomId, user: username, text: text || '', attachments: attachments || [], timestamp: now };
|
||||
|
||||
const lastMsg = db.prepare('SELECT text, timestamp FROM messages WHERE room_id = ? ORDER BY id DESC LIMIT 1').get(roomId);
|
||||
broadcastToRoomExcludeSelf(roomId, username, {
|
||||
type: 'new_message', message: msg,
|
||||
room_preview: { room_id: roomId, last_message: lastMsg ? lastMsg.text : '', last_time: lastMsg ? lastMsg.timestamp : '' }
|
||||
});
|
||||
addToHistory(roomId, 'user', `${username}: ${text || '图片'}`);
|
||||
res.json(msg);
|
||||
|
||||
if (text && text.trim() === '确认' && hasPendingAction(roomId, username)) {
|
||||
executePendingAction(roomId, username);
|
||||
return;
|
||||
}
|
||||
|
||||
try {
|
||||
const history = getHistory(roomId);
|
||||
const historyMessages = history.map(e => ({ role: e.role, content: e.content }));
|
||||
const aiResponse = await analyzeWithDeepSeek(text || '', historyMessages, username, req.user.isAdmin);
|
||||
if (aiResponse && aiResponse.action !== 'ignore') {
|
||||
handleAIResult(aiResponse, roomId, username, text || '', attachments || []);
|
||||
}
|
||||
} catch (e) { console.error('AI 分析失败:', e); }
|
||||
});
|
||||
|
||||
async function analyzeWithDeepSeek(text, historyMessages, username, isAdmin) {
|
||||
const systemPrompt = buildSystemPrompt(username, isAdmin);
|
||||
const messages = [
|
||||
{ role: 'system', content: systemPrompt },
|
||||
...historyMessages.slice(-30),
|
||||
{ role: 'user', content: `${username}: ${text}` }
|
||||
];
|
||||
const content = await callDeepSeek(messages, 0.1);
|
||||
console.log('🤖 AI 返回:', content);
|
||||
const parsed = JSON.parse(content.replace(/```json|```/g, '').trim());
|
||||
normalize(parsed);
|
||||
const v = validate(parsed);
|
||||
if (!v.valid) console.warn('⚠️ AI 返回校验失败:', v.error);
|
||||
return parsed;
|
||||
}
|
||||
|
||||
module.exports = router;
|
||||
48
server/routes/purchases.js
Normal file
48
server/routes/purchases.js
Normal file
@@ -0,0 +1,48 @@
|
||||
// 采购相关路由
|
||||
const express = require('express');
|
||||
const { authMiddleware } = require('../middleware/auth');
|
||||
const db = require('../db');
|
||||
|
||||
const router = express.Router();
|
||||
|
||||
router.get('/rooms/:roomId/purchases', authMiddleware, (req, res) => {
|
||||
const purchases = db.prepare('SELECT * FROM purchases WHERE room_id = ? ORDER BY created_at DESC').all(req.params.roomId);
|
||||
res.json(purchases);
|
||||
});
|
||||
|
||||
router.get('/purchases/:id', authMiddleware, (req, res) => {
|
||||
const pur = db.prepare('SELECT * FROM purchases WHERE id = ?').get(req.params.id);
|
||||
if (!pur) return res.status(404).json({ error: '未找到' });
|
||||
const history = db.prepare('SELECT * FROM purchase_history WHERE purchase_id = ? ORDER BY id ASC').all(req.params.id);
|
||||
const attachments = db.prepare('SELECT * FROM purchase_attachments WHERE purchase_id = ?').all(req.params.id);
|
||||
res.json({ ...pur, history, attachments });
|
||||
});
|
||||
|
||||
router.get('/summary', authMiddleware, (req, res) => {
|
||||
const username = req.user.username;
|
||||
const isAdmin = req.user.isAdmin;
|
||||
let rooms;
|
||||
if (isAdmin) {
|
||||
rooms = db.prepare('SELECT id, name FROM rooms').all();
|
||||
} else {
|
||||
rooms = db.prepare("SELECT id, name FROM rooms WHERE white_list = '' OR (',' || white_list || ',' LIKE ?)").all(`%,${username},%`);
|
||||
}
|
||||
const summary = rooms.map(room => {
|
||||
const stats = db.prepare('SELECT COUNT(*) as count, SUM(amount) as total FROM purchases WHERE room_id = ?').get(room.id);
|
||||
return { room_id: room.id, room_name: room.name, purchase_count: stats.count || 0, total_amount: stats.total || 0 };
|
||||
});
|
||||
res.json(summary);
|
||||
});
|
||||
|
||||
router.get('/rooms/:roomId/purchases/export', authMiddleware, (req, res) => {
|
||||
const purchases = db.prepare('SELECT * FROM purchases WHERE room_id = ?').all(req.params.roomId);
|
||||
let csv = '时间,事项,数量,单价,金额,邮费,付款方式,发票类型,状态,申请人,备注\n';
|
||||
purchases.forEach(p => {
|
||||
csv += `${p.created_at},"${p.item||''}",${p.quantity},${p.unit_price},${p.amount},${p.freight},"${p.payment_method||''}","${p.invoice_type||''}","${p.status}","${p.applicant}","${p.remarks||''}"\n`;
|
||||
});
|
||||
res.setHeader('Content-Type', 'text/csv; charset=utf-8');
|
||||
res.setHeader('Content-Disposition', `attachment; filename="purchases-${req.params.roomId}.csv"`);
|
||||
res.send('\uFEFF' + csv);
|
||||
});
|
||||
|
||||
module.exports = router;
|
||||
57
server/routes/rooms.js
Normal file
57
server/routes/rooms.js
Normal file
@@ -0,0 +1,57 @@
|
||||
// 群聊管理路由
|
||||
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 : '' };
|
||||
});
|
||||
res.json(result);
|
||||
});
|
||||
|
||||
router.post('/', authMiddleware, adminOnly, (req, res) => {
|
||||
const { name, whiteList } = req.body;
|
||||
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, whiteList || '', now);
|
||||
broadcast({ type: 'room_created', room: { id, name, white_list: whiteList || '' } });
|
||||
res.json({ id, name });
|
||||
});
|
||||
|
||||
router.put('/:roomId', authMiddleware, adminOnly, (req, res) => {
|
||||
const { name, whiteList } = req.body;
|
||||
const roomId = req.params.roomId;
|
||||
const room = db.prepare('SELECT * FROM rooms WHERE id = ?').get(roomId);
|
||||
if (!room) return res.status(404).json({ error: '群聊不存在' });
|
||||
db.prepare('UPDATE rooms SET name = ?, white_list = ? WHERE id = ?').run(name, whiteList || '', roomId);
|
||||
broadcast({ type: 'room_updated', room: { id: roomId, name, white_list: whiteList || '' } });
|
||||
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;
|
||||
Reference in New Issue
Block a user