@ian2018cs/agenthub 0.1.87 → 0.1.89

package/server/routes/share.js

@@ -0,0 +1,116 @@
+import express from 'express';
+import { shareDb, userDb } from '../database/db.js';
+import { getSessionMessages, loadProjectConfig } from '../projects.js';
+import { extractShareMessages, saveStaticSnapshot, readStaticSnapshot } from '../services/share-renderer.js';
+import { authenticateToken } from '../middleware/auth.js';
+
+const router = express.Router();
+
+// GET /api/share/info/:sessionId — 获取当前用户的分享状态（需要鉴权，必须在 /:userUuid/:sessionId 前定义）
+router.get('/info/:sessionId', authenticateToken, (req, res) => {
+  const { sessionId } = req.params;
+  const userUuid = req.user.uuid;
+
+  const share = shareDb.getShare(userUuid, sessionId);
+  if (!share) {
+    return res.status(404).json({ error: 'not found' });
+  }
+  res.json({
+    mode: share.share_mode,
+    status: share.status,
+    shareUrl: `/share/${userUuid}/${sessionId}`,
+  });
+});
+
+// GET /api/share/:userUuid/:sessionId — 公开访问，返回分享数据
+router.get('/:userUuid/:sessionId', async (req, res) => {
+  const { userUuid, sessionId } = req.params;
+
+  const share = shareDb.getShare(userUuid, sessionId);
+  if (!share || share.status !== 'active') {
+    return res.status(404).json({ error: '分享不存在或已关闭' });
+  }
+
+  const user = userDb.getUserByUuid(userUuid);
+  const sharerName = user?.email || user?.username || '';
+
+  const projectConfig = await loadProjectConfig(userUuid);
+  const projectDisplayName = projectConfig[share.project_name]?.displayName || share.project_name;
+
+  if (share.share_mode === 'realtime') {
+    try {
+      const rawMessages = await getSessionMessages(share.project_name, sessionId, null, 0, userUuid);
+      const messages = extractShareMessages(rawMessages);
+      return res.json({
+        mode: 'realtime',
+        title: share.session_title || '对话分享',
+        projectName: projectDisplayName,
+        messages,
+        sharerName,
+      });
+    } catch (err) {
+      console.error('[share] Failed to read realtime messages:', err);
+      return res.status(500).json({ error: '读取消息失败' });
+    }
+  }
+
+  // 静态模式：读取 snapshot.json
+  const snapshot = readStaticSnapshot(userUuid, sessionId);
+  if (!snapshot) {
+    return res.status(500).json({ error: '读取快照失败' });
+  }
+  return res.json({
+    mode: 'static',
+    title: share.session_title || snapshot.title || '对话分享',
+    projectName: share.project_name,
+    messages: snapshot.messages || [],
+    generatedAt: snapshot.generatedAt,
+    sharerName,
+  });
+});
+
+// POST /api/share — 创建或更新分享（需要鉴权）
+router.post('/', authenticateToken, async (req, res) => {
+  const { sessionId, projectName, mode, title } = req.body;
+  const userUuid = req.user.uuid;
+
+  if (!sessionId || !projectName || !mode) {
+    return res.status(400).json({ error: '缺少必要参数' });
+  }
+  if (mode !== 'realtime' && mode !== 'static') {
+    return res.status(400).json({ error: '无效的分享模式' });
+  }
+
+  if (mode === 'static') {
+    try {
+      const rawMessages = await getSessionMessages(projectName, sessionId, null, 0, userUuid);
+      const messages = extractShareMessages(rawMessages);
+      saveStaticSnapshot(messages, userUuid, sessionId, title);
+    } catch (err) {
+      console.error('[share] Failed to save static snapshot:', err);
+      return res.status(500).json({ error: '保存快照失败' });
+    }
+  }
+
+  shareDb.upsertShare({
+    userUuid,
+    sessionId,
+    projectName,
+    shareMode: mode,
+    sessionTitle: title || null,
+  });
+
+  const shareUrl = `/share/${userUuid}/${sessionId}`;
+  res.json({ shareUrl, mode });
+});
+
+// DELETE /api/share/:sessionId — 停止分享（需要鉴权）
+router.delete('/:sessionId', authenticateToken, (req, res) => {
+  const { sessionId } = req.params;
+  const userUuid = req.user.uuid;
+
+  shareDb.stopShare(userUuid, sessionId);
+  res.json({ ok: true });
+});
+
+export default router;

package/server/services/builtin-tools/index.js

@@ -178,6 +178,9 @@ registry.register(sdkCronCreateRedirect);
 registry.register(sdkCronDeleteRedirect);
 registry.register(sdkCronListRedirect);
 
+import userInfoTool from './user-info.js';
+registry.register(userInfoTool);
+
 // 重新导出后台任务相关模块，供 server/index.js 使用
 export {
   backgroundTaskPool,

package/server/services/builtin-tools/user-info.js

@@ -0,0 +1,32 @@
+import { userDb } from '../../database/db.js';
+
+export default {
+  name: '__get_user_info__',
+
+  systemPrompt: `
+## 获取当前用户信息
+当你需要了解当前登录用户的基本信息时，执行：
+Bash(__get_user_info__)
+返回当前用户的 username、email、uuid 三个字段（JSON 格式）。无需任何参数。
+`.trim(),
+
+  match(hookInput) {
+    return (
+      hookInput.tool_name === 'Bash' &&
+      hookInput.tool_input?.command?.trim() === '__get_user_info__'
+    );
+  },
+
+  async execute(hookInput, { userUuid }) {
+    const user = userDb.getUserByUuid(userUuid);
+    if (!user) {
+      return { decision: 'deny', reason: '无法获取用户信息：用户不存在' };
+    }
+    const info = JSON.stringify(
+      { username: user.username, email: user.email, uuid: user.uuid },
+      null,
+      2
+    );
+    return { decision: 'deny', reason: info };
+  },
+};

package/server/services/share-renderer.js

@@ -0,0 +1,105 @@
+import path from 'path';
+import fs from 'fs';
+
+const DATA_DIR = process.env.DATA_DIR || path.join(process.cwd(), 'data');
+const SHARE_DIR = path.join(DATA_DIR, 'sharepage');
+
+const SKIP_USER_PREFIXES = [
+  '<command-args>',
+  '<local-command-stdout>',
+  '<system-reminder>',
+  '[Background Task Result]',
+  '<task-notification>',
+  'Caveat:',
+  'This session is being continued from a previous',
+  '[Request interrupted',
+];
+
+/**
+ * 从包含 <command-message> 标签的文本中提取命令消息内容
+ * e.g. "<command-message>foo</command-message><command-name>/foo</command-name>" → "foo"
+ */
+function extractCommandMessage(text) {
+  const m = text.match(/<command-message>([\s\S]*?)<\/command-message>/);
+  return m ? m[1].trim() : null;
+}
+
+/**
+ * 从原始 JSONL 消息数组中提取纯文本消息（过滤工具调用、系统消息等）
+ * @param {Array} rawMessages
+ * @returns {Array<{role, content, timestamp, uuid}>}
+ */
+export function extractShareMessages(rawMessages) {
+  const results = [];
+
+  for (const msg of rawMessages) {
+    if (msg.isMeta) continue;
+
+    const role = msg.message?.role;
+    if (role !== 'user' && role !== 'assistant') continue;
+
+    const content = msg.message?.content;
+
+    if (role === 'user') {
+      let text = '';
+      if (typeof content === 'string') {
+        text = content;
+      } else if (Array.isArray(content)) {
+        text = content.filter(p => p.type === 'text').map(p => p.text).join('\n');
+      }
+      if (!text) continue;
+      if (SKIP_USER_PREFIXES.some(prefix => text.startsWith(prefix))) continue;
+      if (text.startsWith('<command-message>') || text.startsWith('<command-name>')) {
+        const cmd = extractCommandMessage(text);
+        if (!cmd) continue;
+        text = `/${cmd}`;
+      }
+
+      results.push({ role: 'user', content: text, timestamp: msg.timestamp, uuid: msg.uuid });
+
+    } else if (role === 'assistant') {
+      let text = '';
+      if (typeof content === 'string') {
+        text = content;
+      } else if (Array.isArray(content)) {
+        text = content.filter(p => p.type === 'text').map(p => p.text).join('\n');
+      }
+      if (!text.trim()) continue;
+
+      results.push({ role: 'assistant', content: text, timestamp: msg.timestamp, uuid: msg.uuid });
+    }
+  }
+
+  return results;
+}
+
+/**
+ * 保存静态快照 JSON（供前端渲染）
+ * @returns {string} JSON 文件绝对路径
+ */
+export function saveStaticSnapshot(messages, userUuid, sessionId, title) {
+  const sessionDir = path.join(SHARE_DIR, userUuid, sessionId);
+  fs.mkdirSync(sessionDir, { recursive: true });
+
+  const jsonPath = path.join(sessionDir, 'snapshot.json');
+  fs.writeFileSync(jsonPath, JSON.stringify({
+    messages,
+    title,
+    generatedAt: new Date().toISOString(),
+  }, null, 2), 'utf8');
+
+  return jsonPath;
+}
+
+/**
+ * 读取静态快照 JSON
+ * @returns {{ messages, title, generatedAt } | null}
+ */
+export function readStaticSnapshot(userUuid, sessionId) {
+  const jsonPath = path.join(SHARE_DIR, userUuid, sessionId, 'snapshot.json');
+  try {
+    return JSON.parse(fs.readFileSync(jsonPath, 'utf8'));
+  } catch {
+    return null;
+  }
+}