@myassis/gateway 1.0.0

package/dist/routes/agent.js

@@ -0,0 +1,513 @@
+import { Router } from 'express';
+import { requireAuth } from '@/middleware/auth.js';
+import { sessionManager } from '@/services/session';
+import { initAgentManager, agentManager } from '@/services/agent/AgentManager';
+import { AgentStore } from '@/services/agent/AgentStore';
+import { getLogger } from '@pocketclaw/shared';
+const logger = getLogger('AgentRoutes');
+import { sessionStore } from '@/services/session/SessionStore';
+const router = Router();
+// 所有路由需要认证
+router.use(requireAuth);
+// Middleware: ensure agentManager is initialized
+function ensureAgentManager(req, res, next) {
+    try {
+        // Get or create AgentStore using SessionStore's db instance
+        const db = sessionStore.getDb();
+        const agentStore = new AgentStore(db);
+        // Initialize AgentManager if not already done
+        if (!agentManager) {
+            initAgentManager(agentStore);
+        }
+        agentManager.initialize();
+        req.agentManager = agentManager;
+        next();
+    }
+    catch (error) {
+        logger.error('Failed to initialize agentManager:', error);
+        res.status(500).json({ success: false, error: 'Failed to initialize agent manager' });
+    }
+}
+// ============ Agent CRUD ============
+/**
+ * GET /api/agent/list
+ * Get all agents for current user
+ */
+router.get('/list', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const agentList = agentManager.getAgentList();
+        res.json({ success: true, data: agentList });
+    }
+    catch (error) {
+        logger.error(`Get agent list error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to get agent list' });
+    }
+});
+/**
+ * GET /api/agent/:id
+ * Get agent details
+ */
+router.get('/:id', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const agent = agentManager.getAgent(req.params.id);
+        if (!agent) {
+            return res.status(404).json({ success: false, error: 'Agent not found' });
+        }
+        res.json({
+            success: true,
+            data: {
+                id: agent.id,
+                name: agent.name,
+                description: agent.description,
+                systemPrompt: agent.systemPrompt,
+                avatar: agent.avatar,
+                config: agent.config,
+                createdAt: agent.createdAt,
+                updatedAt: agent.updatedAt,
+            }
+        });
+    }
+    catch (error) {
+        logger.error(`Get agent error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to get agent' });
+    }
+});
+/**
+ * POST /api/agent/create
+ * Create new agent
+ */
+router.post('/create', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const { name, description, systemPrompt, avatar, config } = req.body;
+        if (!name) {
+            return res.status(400).json({ success: false, error: 'Name is required' });
+        }
+        const agent = agentManager.createAgent({
+            name,
+            description,
+            systemPrompt,
+            avatar,
+            config,
+        });
+        res.json({
+            success: true,
+            data: {
+                id: agent.id,
+                name: agent.name,
+                description: agent.description,
+                systemPrompt: agent.systemPrompt,
+                avatar: agent.avatar,
+                createdAt: agent.createdAt,
+                updatedAt: agent.updatedAt,
+            }
+        });
+    }
+    catch (error) {
+        logger.error(`Create agent error: ${error}`);
+        const message = error instanceof Error ? error.message : 'Failed to create agent';
+        res.status(400).json({ success: false, error: message });
+    }
+});
+/**
+ * PUT /api/agent/:id
+ * Update agent
+ */
+router.put('/:id', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        logger.debug('update agent', req.body);
+        const { name, description, systemPrompt, avatar, config } = req.body;
+        const agent = agentManager.updateAgent(req.params.id, {
+            name,
+            description,
+            systemPrompt,
+            avatar,
+            config,
+        });
+        if (!agent) {
+            return res.status(404).json({ success: false, error: 'Agent not found' });
+        }
+        res.json({
+            success: true,
+            data: {
+                id: agent.id,
+                name: agent.name,
+                description: agent.description,
+                systemPrompt: agent.systemPrompt,
+                avatar: agent.avatar,
+                updatedAt: agent.updatedAt,
+            }
+        });
+    }
+    catch (error) {
+        logger.error(`Update agent error: ${error}`);
+        const message = error instanceof Error ? error.message : 'Failed to update agent';
+        res.status(400).json({ success: false, error: message });
+    }
+});
+/**
+ * DELETE /api/agent/:id
+ * Delete agent
+ */
+router.delete('/:id', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const deleted = agentManager.deleteAgent(req.params.id);
+        if (!deleted) {
+            return res.status(404).json({ success: false, error: 'Agent not found' });
+        }
+        res.json({ success: true });
+    }
+    catch (error) {
+        logger.error(`Delete agent error: ${error}`);
+        const message = error instanceof Error ? error.message : 'Failed to delete agent';
+        res.status(400).json({ success: false, error: message });
+    }
+});
+// ============ Current Agent ============
+/**
+ * GET /api/agent/current
+ * Get current agent
+ */
+router.get('/current', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const current = agentManager.getCurrentAgent();
+        res.json({
+            success: true, data: current ? {
+                id: current.id,
+                userId: current.userId,
+                name: current.name,
+                description: current.description,
+                systemPrompt: current.systemPrompt,
+                avatar: current.avatar,
+                config: current.config,
+                createdAt: current.createdAt,
+                updatedAt: current.updatedAt,
+            } : null
+        });
+    }
+    catch (error) {
+        logger.error(`Get current agent error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to get current agent' });
+    }
+});
+/**
+ * POST /api/agent/set-current
+ * Set current agent
+ */
+router.post('/set-current', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const { agentId } = req.body;
+        agentManager.setCurrentAgent(agentId || null);
+        res.json({ success: true });
+    }
+    catch (error) {
+        logger.error(`Set current agent error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to set current agent' });
+    }
+});
+// ============ Session Management under Agent ============
+/**
+ * GET /api/agent/:id/sessions
+ * Get all sessions under an agent
+ */
+router.get('/:id/sessions', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const agent = agentManager.getAgent(req.params.id);
+        if (!agent) {
+            return res.status(404).json({ success: false, error: 'Agent not found' });
+        }
+        const sessions = agentManager.getAgentSessions(req.params.id);
+        res.json({ success: true, data: sessions });
+    }
+    catch (error) {
+        logger.error(`Get agent sessions error: `, error);
+        res.status(500).json({ success: false, error: 'Failed to get sessions' });
+    }
+});
+/**
+ * POST /api/agent/:id/sessions
+ * Create new session under agent
+ */
+router.post('/:id/sessions', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const agent = agentManager.getAgent(req.params.id);
+        if (!agent) {
+            return res.status(404).json({ success: false, error: 'Agent not found' });
+        }
+        const { title, selectModelId } = req.body;
+        const session = agent.createSession({
+            title: title || '新会话',
+            selectModelId,
+        });
+        res.json({
+            success: true,
+            data: {
+                id: session.id,
+                title: session.title,
+                selectModelId: session.selectModelId,
+                createdAt: session.createdAt,
+                updatedAt: session.updatedAt,
+            }
+        });
+    }
+    catch (error) {
+        logger.error(`Create session error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to create session' });
+    }
+});
+/**
+ * PUT /api/agent/:id/sessions/:sessionId
+ * Update session under agent
+ */
+router.put('/:id/sessions/:sessionId', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const agent = agentManager.getAgent(req.params.id);
+        if (!agent) {
+            return res.status(404).json({ success: false, error: 'Agent not found' });
+        }
+        const { title, selectModelId, unreadCount } = req.body;
+        const session = agent.updateSession(req.params.sessionId, { title, selectModelId, unreadCount });
+        res.json({
+            success: true,
+            data: {
+                id: session.id,
+                title: session.title,
+                selectModelId: session.selectModelId,
+                createdAt: session.createdAt,
+                updatedAt: session.updatedAt,
+            }
+        });
+    }
+    catch (error) {
+        logger.error(`Update session error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to update session' });
+    }
+});
+/**
+ * DELETE /api/agent/:id/sessions/:sessionId
+ * Delete session under agent
+ */
+router.delete('/:id/sessions/:sessionId', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const agent = agentManager.getAgent(req.params.id);
+        if (!agent) {
+            return res.status(404).json({ success: false, error: 'Agent not found' });
+        }
+        const deleted = agent.deleteSession(req.params.sessionId);
+        res.json({ success: deleted });
+    }
+    catch (error) {
+        logger.error(`Delete session error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to delete session' });
+    }
+});
+// ============ Session Messages ============
+/**
+ * GET /api/agent/sessions/:sessionId/messages
+ * Get messages for a session
+ */
+router.get('/sessions/:sessionId/messages', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const { sessionId } = req.params;
+        const page = parseInt(req.query.page) || 1;
+        const pageSize = parseInt(req.query.pageSize) || 20;
+        // Find session directly
+        const session = sessionManager.getSession(sessionId);
+        if (session) {
+            session.loadMessages(); // Ensure messages are loaded
+            const messages = session.getMessagesByPage(page, pageSize);
+            return res.json({
+                success: true,
+                data: messages.messages.reverse(), // Return in chronological order
+                pagination: {
+                    page,
+                    pageSize,
+                    total: messages.total,
+                    hasMore: messages.hasMore,
+                },
+            });
+        }
+        res.status(404).json({ success: false, error: 'Session not found' });
+    }
+    catch (error) {
+        logger.error(`Get messages error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to get messages' });
+    }
+});
+/**
+ * DELETE /api/agent/sessions/:sessionId/messages/:messageId
+ * Delete a specific message
+ */
+router.delete('/sessions/:sessionId/messages/:messageId', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const { sessionId, messageId } = req.params;
+        // Find session directly
+        const session = sessionManager.getSession(sessionId);
+        if (session) {
+            session.deleteMessage(messageId);
+            return res.json({ success: true });
+        }
+        res.status(404).json({ success: false, error: 'Session or message not found' });
+    }
+    catch (error) {
+        logger.error(`Delete message error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to delete message' });
+    }
+});
+/**
+ * PUT /api/agent/sessions/:sessionId/messages/:messageId/feedback
+ * Update message feedback (like/dislike)
+ */
+router.put('/sessions/:sessionId/messages/:messageId/feedback', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const { sessionId, messageId } = req.params;
+        const { feedback } = req.body; // 'like' | 'dislike' | null
+        if (feedback !== null && feedback !== 'like' && feedback !== 'dislike') {
+            return res.status(400).json({ success: false, error: 'Invalid feedback value. Must be "like", "dislike", or null' });
+        }
+        const session = sessionManager.getSession(sessionId);
+        if (session) {
+            session.updateMessageFeedback(messageId, feedback);
+            return res.json({ success: true, data: { messageId, feedback } });
+        }
+        res.status(404).json({ success: false, error: 'Session or message not found' });
+    }
+    catch (error) {
+        logger.error(`Update message feedback error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to update message feedback' });
+    }
+});
+/**
+ * DELETE /api/agent/sessions/:sessionId/messages
+ * Clear all messages in a session
+ */
+router.delete('/sessions/:sessionId/messages', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const { sessionId } = req.params;
+        // Find session directly
+        const session = sessionManager.getSession(sessionId);
+        if (session) {
+            session.clearMessages();
+            return res.json({ success: true });
+        }
+        res.status(404).json({ success: false, error: 'Session not found' });
+    }
+    catch (error) {
+        logger.error(`Clear messages error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to clear messages' });
+    }
+});
+/**
+ * POST /api/agent/sessions/:sessionId/stream
+ * Stream chat response
+ */
+router.post('/sessions/:sessionId/stream', ensureAgentManager, async (req, res) => {
+    try {
+        const userId = req.userId;
+        const { sessionId } = req.params;
+        const { content, attachments, userMessageId, assistantMessageId, taskId } = req.body;
+        if (!content) {
+            return res.status(400).json({ success: false, error: 'Content is required' });
+        }
+        // Find session directly
+        const session = sessionManager.getSession(sessionId);
+        if (session) {
+            // Set SSE headers
+            res.setHeader('Content-Type', 'text/event-stream');
+            res.setHeader('Cache-Control', 'no-cache');
+            res.setHeader('Connection', 'keep-alive');
+            res.flushHeaders();
+            // Stream response
+            await session.streamChat(content, attachments || [], res, userMessageId, assistantMessageId);
+            return;
+        }
+        res.status(404).json({ success: false, error: 'Session not found' });
+    }
+    catch (error) {
+        logger.error(`Stream chat error: ${error}`);
+        if (error.message.includes('未提供认证 Token')) {
+            res.status(401).json({ success: false, error: 'Token失效' });
+        }
+        if (!res.headersSent) {
+            res.status(500).json({ success: false, error: 'Failed to stream chat' });
+        }
+        else {
+            sendSSE(res, { type: 'error', error: error.message });
+            res.end();
+        }
+    }
+});
+// SSE 辅助方法：发送数据并立即 flush
+const sendSSE = (res, data) => {
+    res.write(`data: ${JSON.stringify(data)}\n\n`);
+    res.flush?.();
+};
+/**
+ * POST /api/agent/sessions/:sessionId/reset
+ * Stop session generation
+ */
+router.post('/sessions/:sessionId/reset', async (req, res) => {
+    try {
+        const userId = req.userId;
+        const { sessionId } = req.params;
+        const stopped = sessionManager.stopSession(sessionId);
+        res.json({ success: true, data: { stopped } });
+    }
+    catch (error) {
+        logger.error(`Reset session error: ${error}`);
+        const message = error instanceof Error ? error.message : 'Failed to reset session';
+        res.status(500).json({ success: false, error: message });
+    }
+});
+/**
+ * POST /api/agent/sessions/:sessionId/set-current
+ * Mark session as current (clears unread count)
+ */
+router.post('/sessions/:sessionId/set-current', async (req, res) => {
+    try {
+        const userId = req.userId;
+        const { sessionId } = req.params;
+        sessionManager.setCurrentSession(sessionId);
+        // 同步清除 session 的未读数
+        const session = sessionManager.getSession(sessionId);
+        if (session) {
+            session.clearUnreadCount();
+        }
+        res.json({ success: true });
+    }
+    catch (error) {
+        logger.error(`Set current session error: ${error}`);
+        const message = error instanceof Error ? error.message : 'Failed to set current session';
+        res.status(500).json({ success: false, error: message });
+    }
+});
+/**
+ * GET /api/agent/sessions/unread-count
+ * Get total unread count across all sessions
+ */
+router.get('/sessions/unread-count', async (req, res) => {
+    try {
+        const userId = req.userId;
+        const total = sessionManager.getTotalUnreadCount();
+        res.json({ success: true, data: { total } });
+    }
+    catch (error) {
+        logger.error(`Get unread count error: ${error}`);
+        res.status(500).json({ success: false, error: 'Failed to get unread count' });
+    }
+});
+export default router;

package/dist/routes/auth.js

@@ -0,0 +1,172 @@
+import { Router } from 'express';
+import { authApi } from '../api/index.js';
+import { authStore } from '../stores/index.js';
+import { optionalAuth, requireAuth } from '../middleware/auth.js';
+import { getLogger } from '@pocketclaw/shared';
+import { getPlatform } from '@pocketclaw/shared/dist/utils/system.js';
+const logger = getLogger('auth');
+const router = Router();
+// Check auth status - 解码请求中的 token 返回认证状态
+router.get('/isAuth', optionalAuth, async (req, res) => {
+    const userId = req.userId;
+    res.json({
+        authenticated: !!userId,
+        user: userId ? { id: userId } : null,
+    });
+});
+// Get publish token - 获取发布技能库所需的 token
+router.get('/publish-token', requireAuth, async (req, res) => {
+    res.json({
+        success: true,
+        token: req.token,
+        user: { id: req.userId },
+    });
+});
+// Register - 使用 authApi 注册
+router.post('/register', async (req, res) => {
+    try {
+        const { username, password } = req.body;
+        if (!username || !password) {
+            res.status(400).json({ success: false, error: 'Missing required fields' });
+            return;
+        }
+        // 使用 authApi.register（服务端路由期望 account 字段）
+        const response = await authApi.register({
+            account: username,
+            password,
+            nickname: username,
+            platform: getPlatform(),
+        });
+        if (response.success && response.data) {
+            const expiresIn = response.data.expiresIn || 86400;
+            authStore.save({
+                token: response.data.accessToken,
+                refreshToken: response.data.refreshToken,
+                expiresIn,
+                expiresAt: Date.now() + expiresIn * 1000,
+                user: response.data.user,
+            });
+            res.status(201).json({
+                success: true,
+                user: response.data.user,
+            });
+        }
+        else {
+            res.status(400).json({ success: false, error: response.error || 'Registration failed' });
+        }
+    }
+    catch (error) {
+        logger.error('Register error:', error);
+        res.status(error.status || 500).json({
+            success: false,
+            error: error.message || 'Registration failed',
+        });
+    }
+});
+// Login - 使用 authApi 登录
+router.post('/login', async (req, res) => {
+    try {
+        const { username, password } = req.body;
+        logger.debug('Login attempt for:', username);
+        if (!username || !password) {
+            res.status(400).json({ success: false, error: 'Missing required fields' });
+            return;
+        }
+        // 使用 authApi.login（服务端路由期望 email 字段）
+        const response = await authApi.login({
+            account: username,
+            password,
+            platform: getPlatform(),
+        });
+        if (response.success && response.data) {
+            const expiresIn = response.data.expiresIn || 86400;
+            authStore.save({
+                token: response.data.accessToken,
+                refreshToken: response.data.refreshToken,
+                expiresIn,
+                expiresAt: Date.now() + expiresIn * 1000,
+                user: response.data.user,
+            });
+            res.json({
+                success: true,
+                user: response.data.user,
+            });
+        }
+        else {
+            res.status(401).json({ success: false, error: response.error || 'Invalid credentials' });
+        }
+    }
+    catch (error) {
+        logger.error('Login error:', error);
+        res.status(error.status || 500).json({
+            success: false,
+            error: error.message || 'Login failed',
+        });
+    }
+});
+// Refresh token - 使用 authApi 刷新
+router.post('/refresh', async (req, res) => {
+    try {
+        const { refreshToken } = req.body;
+        if (!refreshToken) {
+            res.status(400).json({ success: false, error: 'Missing refresh token' });
+            return;
+        }
+        const response = await authApi.refresh(refreshToken);
+        if (response.success && response.accessToken) {
+            authStore.updateToken(response.accessToken, response.refreshToken);
+            res.json({ success: true });
+        }
+        else {
+            res.status(401).json({ success: false, error: response.error || 'Invalid refresh token' });
+        }
+    }
+    catch (error) {
+        logger.error('Refresh error:', error);
+        res.status(error.status || 500).json({
+            success: false,
+            error: error.message || 'Refresh failed',
+        });
+    }
+});
+// Logout - 清除本地存储
+router.post('/logout', requireAuth, async (req, res) => {
+    try {
+        // refreshToken 在请求 body 中
+        const { refreshToken } = req.body;
+        // 尝试转发登出请求到 Server（忽略错误）
+        if (refreshToken) {
+            authApi.logout().catch(() => { });
+        }
+        authStore.clear();
+        res.json({ success: true });
+    }
+    catch (error) {
+        authStore.clear();
+        res.json({ success: true });
+    }
+});
+// Delete account - 删除账号
+router.post('/account/delete', requireAuth, async (req, res) => {
+    try {
+        const { password } = req.body;
+        if (!password) {
+            res.status(400).json({ success: false, error: 'Missing password' });
+            return;
+        }
+        // 转发删除账号请求到 Server（带上 token）
+        const response = await authApi.deleteAccount({ password }, req.token);
+        if (response.success) {
+            authStore.clear();
+        }
+        res.json(response);
+    }
+    catch (error) {
+        logger.error('Delete account error:', error);
+        res.status(error.status || 500).json({
+            success: false,
+            error: error.message || 'Delete account failed',
+        });
+    }
+});
+export default router;