dsh-qq-tools 0.10.1 → 0.10.2

package/lib/prompts.js

@@ -0,0 +1,56 @@
+// 唤醒提示词与回复风格。只放文案和拼接，不碰网络和状态。
+
+import { clean } from './utils.js'
+import { hasActiveTool } from './session.js'
+
+// 唤醒提示里的消息行：和戳一戳一致，固定带上昵称和 QQ 号。
+// 有 message_id 时附在末尾，模型可以把它传给 reply_to 做引用。
+export function formatIncomingLine(name, userId, text, messageId) {
+  const who = clean(name) || 'QQ'
+  const qq = String(userId || '?')
+  return `${who}(${qq})：${text || '（空）'}` + (messageId != null ? `（消息 #${messageId}）` : '')
+}
+
+// 回复风格：群聊要像随手打字，短、分条、不刷屏地 @ 和引用。
+export const CHAT_STYLE = '回复像群友随手打字：想连发几条就**多次调用同一个工具**（一次调用只发一条），每条一句短话（十几二十个字），最多三次调用；'
+  + '不要句号收尾，不要写长段落，不要 Markdown，不要把昵称或 QQ 号写进正文，不要每句都 @ 或引用——'
+  + '只有确实要指代某条消息时才传 reply_to，只有必须提醒某个人时才传 at_user_ids。'
+
+// 主动发表情：平时列表只给备注，真要发的时候才 preview 一张图，省 token。
+export const FACE_STYLE = '想回表情时，约有 25% 的概率主动发一个收藏表情：'
+  + '先用 qq_list_faces 按备注挑一个，再用 qq_send_face 传 preview=true 看一眼这张图，确认合适再正式发；'
+
+// 收到带图消息时给一个低概率的收藏引导；index 和 qq_favorite_face 的取图索引一致。
+// desc 直接描述画面内容，否则 qq_list_faces 只有备注，后面根本不知道用哪张。
+export function favoriteHint(text, messageId) {
+  const m = String(text || '').match(/\[(?:图片|表情包)#(\d+)\]/)
+  if (!m || messageId == null || messageId === '') return ''
+  return '这条消息里有图片，约有 25% 的概率可以顺手收藏一个你觉得好玩的'
+    + '（qq_favorite_face(message_id=' + String(messageId) + ', index=' + m[1] + ', desc=表情内容)）；'
+    + 'desc 直接简短描述画面内容，比如「闭眼安详」「锅铲敲地」「穿冬装的肥鱼」。'
+}
+
+const CHAT_BUBBLE_CHARS = 80
+
+// 一条消息的清理：压掉复制来的对齐空格和多行，去掉正文开头重复的 @昵称。
+function cleanBubble(value) {
+  return String(value || '')
+    .replace(/\r/g, '')
+    .replace(/[\u3000\t ]+/g, ' ')
+    .replace(/\s*\n\s*/g, ' ')
+    .trim()
+    .replace(/^@[^\s@]{1,40}\s+/, '')
+    .slice(0, CHAT_BUBBLE_CHARS)
+}
+
+// 一次工具调用只发一条：想分几条说就让模型多次调用工具。
+export function chatText(args) {
+  return cleanBubble(args && args.message)
+}
+
+export function lookHint() {
+  const parts = []
+  if (hasActiveTool('qq_list_groups')) parts.push('先用 qq_list_groups 看看群')
+  if (hasActiveTool('qq_get_group_history')) parts.push('再用 qq_get_group_history 看 2~3 个群最近 30 条')
+  return parts.length ? '你可以' + parts.join('，') + '，' : ''
+}

package/lib/qq-state.js

@@ -0,0 +1,51 @@
+// QQ 运行时状态：登录号缓存 + 群名缓存。
+import { onebot } from './onebot.js'
+import { clean } from './utils.js'
+
+// 这两份数据以前混在 @ 轮询里，被私聊/戳一戳/私聊记录共用；拆出来避免模块之间互相 import。
+let selfId = ''
+let names = new Map()
+let namesAt = 0
+
+// 当前登录的 QQ 号。第一次用到时向 SnowLuma 查一次，之后复用。
+export async function getSelfId(cfg) {
+  if (selfId) return selfId
+  try {
+    const info = await onebot('get_login_info', {}, cfg)
+    selfId = String((info && info.user_id) || '')
+  } catch (error) {}
+  return selfId
+}
+
+// 只读缓存，不发请求；戳一戳事件本身带 self_id，缺省时保持原来的行为（不额外查登录号）。
+export function cachedSelfId() {
+  return selfId
+}
+
+// 群名缓存拿不到时退回群号，和原来的 `names.get(gid) || gid` 一致。
+export function groupName(groupId) {
+  const id = String(groupId)
+  return names.get(id) || id
+}
+
+async function refreshNames(cfg) {
+  if (names.size && Date.now() - namesAt < 300000) return
+  try {
+    const data = await onebot('get_group_list', {}, cfg)
+    const next = new Map()
+    for (const g of Array.isArray(data) ? data : []) next.set(String(g.group_id), clean(g.group_name || g.group_id))
+    if (next.size) {
+      names = next
+      namesAt = Date.now()
+    }
+  } catch (error) {}
+}
+
+export { refreshNames }
+
+// 组件重新加载时从头开始，避免把停用期间的旧状态带进新实例。
+export function resetQqState() {
+  selfId = ''
+  names = new Map()
+  namesAt = 0
+}

package/lib/render.js

@@ -0,0 +1,49 @@
+// 工具结果的文本渲染，以及常见的列表取值。
+
+export function renderTool(_args, v) {
+  if (!v || !v.ok) return [{ type: 'text', text: 'QQ 工具失败：' + String((v && v.error) || '未知错误') }]
+  if (Array.isArray(v.groups)) {
+    if (!v.groups.length) return [{ type: 'text', text: '现在没有允许查看的群。' }]
+    return [{ type: 'text', text: '可以去的群：\n' + v.groups.map((g) => `- ${g.name || g.id}（群号 ${g.id}，${g.member_count} 人）`).join('\n') }]
+  }
+  if (Array.isArray(v.friends)) {
+    if (!v.friends.length) return [{ type: 'text', text: v.query ? '没有匹配的好友。' : '好友列表是空的。' }]
+    const lines = v.friends.map((f) => `- ${f.remark || f.nickname || f.user_id}（${f.nickname || ''}，QQ ${f.user_id}）`)
+    return [{ type: 'text', text: `你的好友（${v.total} 个）：\n` + lines.join('\n') }]
+  }
+  if (Array.isArray(v.messages)) {
+    const where = v.group_id != null ? `群 ${v.group_id}` : `QQ ${v.user_id}`
+    if (!v.messages.length) return [{ type: 'text', text: `${where} 最近没有消息。` }]
+    const lines = v.messages.map((m) => `${m.t || ''} [#${m.message_id || '?'}] ${m.sender}(${m.user_id})：${m.text || '（空）'}`)
+    const blocks = [{ type: 'text', text: `${where} 最近 ${v.count} 条：\n` + lines.join('\n') }]
+    if (Array.isArray(v.images) && v.images.length) {
+      blocks.push({ type: 'text', text: '下面是这些图片（按消息顺序）：' })
+      for (const img of v.images) {
+        if (img && img.caption) blocks.push({ type: 'text', text: img.caption })
+        if (img && img.block) blocks.push(img.block)
+      }
+    }
+    return blocks
+  }
+  if (Array.isArray(v.members)) {
+    const lines = v.members.map((m) => `- ${m.card || m.nickname || m.user_id}（QQ ${m.user_id}，${m.role || 'member'}）`)
+    return [{ type: 'text', text: `群 ${v.group_id} 成员：\n` + (lines.join('\n') || '（空）') }]
+  }
+  if (v.poke) {
+    return [{ type: 'text', text: `戳回去了：${v.group_id ? '群 ' + v.group_id : '私聊'} QQ ${v.user_id}。` }]
+  }
+  if (v.group_id != null && v.message_id != null) {
+    return [{ type: 'text', text: `已发到群 ${v.group_id}（消息 ${v.message_id || '?'}）。` }]
+  }
+  if (v.user_id != null && v.message_id != null) {
+    return [{ type: 'text', text: `已发给 QQ ${v.user_id}（消息 ${v.message_id || '?'}）。` }]
+  }
+  return [{ type: 'text', text: JSON.stringify(v).slice(0, 2000) }]
+}
+
+
+export function listOf(data, key) {
+  if (Array.isArray(data)) return data
+  if (data && Array.isArray(data[key])) return data[key]
+  return []
+}

package/lib/session.js

@@ -0,0 +1,201 @@
+// 目标会话解析、工具挂载、挂载会话 HTTP 接口，以及当前活着的工具表。
+
+import { clean } from './utils.js'
+import { ensureCfg, loadCfg, saveCfg, diag } from './config.js'
+
+export function targetSessionId() {
+  const cfg = loadCfg()
+  return String(cfg.targetSession || '')
+}
+
+function sessionIdOf(agent) {
+  try {
+    const s = agent && agent.session
+    return String((s && (s.id || s.sessionId)) || (agent && agent.id) || '')
+  } catch (error) {
+    return ''
+  }
+}
+
+export function findAgent(agents, id) {
+  const raw = String(id || '').trim()
+  if (!raw) return undefined
+  const candidates = [raw]
+  if (raw.startsWith('session-')) candidates.push(raw.slice('session-'.length))
+  else candidates.push('session-' + raw)
+  for (const one of candidates) {
+    try {
+      const a = agents.get(one)
+      if (a !== undefined) return a
+    } catch (error) {}
+  }
+  try {
+    for (const a of agents.list() || []) {
+      const sid = sessionIdOf(a)
+      if (sid === raw || 'session-' + sid === raw || sid === raw.replace(/^session-/, '')) return a
+    }
+  } catch (error) {}
+  return undefined
+}
+
+
+// ---------- 目标会话 UI 接口（由「挂载会话」组件启动） ----------
+// 浏览器半侧只读写 targetSession：GET 返回当前值和活着的会话，POST 写回配置文件；
+// 其他组件仍然每 3 秒 loadCfg()，所以保存后不需要重启。
+const TARGET_SESSION_ROUTE = '/api/dsh-qq-tools/target-session'
+
+function replyJson(res, status, body) {
+  const text = JSON.stringify(body)
+  try {
+    res.writeHead(status, {
+      'content-type': 'application/json; charset=utf-8',
+      'cache-control': 'no-store',
+    })
+    res.end(text)
+  } catch (error) {}
+}
+
+async function readRequestJson(req, max = 64 * 1024) {
+  const chunks = []
+  let size = 0
+  for await (const chunk of req) {
+    const buf = Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk)
+    size += buf.length
+    if (size > max) throw new Error('body too large')
+    chunks.push(buf)
+  }
+  if (!chunks.length) return {}
+  return JSON.parse(Buffer.concat(chunks).toString('utf8'))
+}
+
+function sameOriginRequest(req) {
+  const origin = String((req.headers && req.headers.origin) || '')
+  if (!origin) return true
+  try {
+    return new URL(origin).host === String((req.headers && req.headers.host) || '')
+  } catch (error) {
+    return false
+  }
+}
+
+export function mountTargetSessionRoute(ctx) {
+  const webServer = ctx.get('webServer')
+  if (!webServer || typeof webServer.register !== 'function') {
+    diag('挂载会话：宿主没有 webServer，设置页接口不可用')
+    return
+  }
+  ctx.effect(() => {
+    const dispose = webServer.register({
+      kind: 'exact',
+      path: TARGET_SESSION_ROUTE,
+      handler: async (req, res) => {
+        if (!sameOriginRequest(req)) {
+          replyJson(res, 403, { ok: false, error: 'forbidden' })
+          return
+        }
+        const method = String(req.method || 'GET').toUpperCase()
+        if (method === 'GET') {
+          const liveSessionIds = []
+          try {
+            const agents = ctx.get('agents') || ctx.agents
+            for (const agent of (agents && agents.list()) || []) {
+              const sid = sessionIdOf(agent)
+              if (sid) liveSessionIds.push(sid)
+            }
+          } catch (error) {}
+          replyJson(res, 200, { ok: true, targetSession: targetSessionId(), liveSessionIds })
+          return
+        }
+        if (method !== 'POST') {
+          replyJson(res, 405, { ok: false, error: 'method not allowed' })
+          return
+        }
+        let body
+        try {
+          body = await readRequestJson(req)
+        } catch (error) {
+          replyJson(res, 400, { ok: false, error: 'invalid body' })
+          return
+        }
+        const raw = body && typeof body.targetSession === 'string' ? body.targetSession : ''
+        const targetSession = clean(raw).slice(0, 200)
+        try {
+          saveCfg({ targetSession })
+        } catch (error) {
+          replyJson(res, 500, { ok: false, error: 'save failed' })
+          return
+        }
+        diag('目标会话已更新：' + (targetSession || '（不挂载）'))
+        replyJson(res, 200, { ok: true, targetSession })
+      },
+    })
+    return () => {
+      try { dispose() } catch (error) {}
+    }
+  }, 'qq-tools: target session route')
+}
+
+// ---------- 当前活着的工具组件 ----------
+// @ 提醒的提示词只提还开着的工具，免得模型去调一个被关掉的组件。
+const activeCounts = new Map()
+
+function markActive(name, delta) {
+  const next = Math.max(0, (activeCounts.get(name) || 0) + delta)
+  if (next === 0) activeCounts.delete(name)
+  else activeCounts.set(name, next)
+}
+
+export function hasActiveTool(name) {
+  return (activeCounts.get(name) || 0) > 0
+}
+
+
+// ---------- 把一个工具挂到目标会话 ----------
+// 每个工具组件各自调用一次：自己找目标会话、自己挂、自己卸，
+// 所以关掉某个组件只会影响它自己的工具。
+export function mountOneTool(ctx, tool) {
+  ensureCfg()
+  markActive(tool.name, 1)
+  const mounted = new Map()
+
+  function unmountAll(reason) {
+    for (const [sid, v] of mounted) {
+      try {
+        if (typeof v.off === 'function') v.off()
+      } catch (error) {}
+      diag('工具 ' + tool.name + ' 已从会话 ' + sid + ' 卸下：' + reason)
+    }
+    mounted.clear()
+  }
+
+  function tick() {
+    const target = targetSessionId()
+    if (!target) {
+      if (mounted.size) unmountAll('没有目标会话')
+      return
+    }
+    const agent = findAgent(ctx.agents, target)
+    if (!agent) {
+      if (mounted.size) unmountAll('目标会话没有活着')
+      return
+    }
+    const sid = sessionIdOf(agent)
+    if (mounted.has(sid)) return
+    unmountAll('目标会话变了')
+    try {
+      const off = agent.ctx.tools.register(tool)
+      mounted.set(sid, { agent, off })
+      diag('工具 ' + tool.name + ' 已挂到会话 ' + sid)
+    } catch (error) {
+      diag('工具挂载失败 ' + tool.name + '：' + String((error && error.message) || error))
+    }
+  }
+
+  tick()
+  ctx.effect(() => ctx.interval(tick, 3000))
+  ctx.effect(() => () => {
+    markActive(tool.name, -1)
+    unmountAll('插件卸载')
+  })
+  return { tool, tick, unmountAll, mounted }
+}