koishi-plugin-chat-patch 3.0.2 → 5.0.0

package/client/web/src/function/satori.ts

@@ -0,0 +1,325 @@
+// Satori 协议客户端：负责 WebSocket 事件与 HTTP API 请求。
+
+export interface SatoriBootstrap {
+  endpoint: string
+  token: string
+  basePath: string
+}
+
+export interface SatoriLogin {
+  platform: string
+  selfId: string
+  name: string
+  avatar?: string
+  status: number
+  features?: string[]
+}
+
+export interface SatoriEvent {
+  type: string
+  platform: string
+  selfId: string
+  timestamp: number
+  sn: number
+  body: Record<string, unknown>
+}
+
+interface PendingConsoleRequest {
+  resolve: (value: unknown) => void
+  reject: (reason: unknown) => void
+}
+
+let bootstrap: SatoriBootstrap | null = null
+let bootstrapPromise: Promise<SatoriBootstrap> | null = null
+let logins: SatoriLogin[] = []
+let activeBot: { platform: string; selfId: string } | null = null
+let sequence = Number(localStorage.getItem('chat-patch:sn') ?? 0)
+const pendingConsole = new Map<string, PendingConsoleRequest>()
+let consoleRequestId = 0
+let resolveBootstrapCallback: ((value: SatoriBootstrap) => void) | null = null
+let bootstrapTimeout = 0
+
+function getString(value: unknown): string {
+  return typeof value === 'string' ? value : ''
+}
+
+function getObject(value: unknown): Record<string, unknown> {
+  return typeof value === 'object' && value !== null ? value as Record<string, unknown> : {}
+}
+
+async function parseJsonResponse(response: Response): Promise<unknown> {
+  const text = await response.text()
+  if (!response.ok) {
+    throw new Error(`Satori API 返回 ${response.status}`)
+  }
+  try {
+    return JSON.parse(text) as unknown
+  } catch {
+    throw new Error(`Satori API 返回了非 JSON 内容: ${text.slice(0, 120)}`)
+  }
+}
+
+function normalizeEndpoint(endpoint: string): string {
+  const raw = endpoint.startsWith('undefined') ? endpoint.slice(9) : endpoint
+  if (import.meta.env.DEV) {
+    const url = /^https?:\/\//i.test(raw)
+      ? new URL(raw)
+      : new URL(raw, location.origin)
+    return `${location.origin}${url.pathname}`.replace(/\/+$/, '')
+  }
+  if (/^https?:\/\//i.test(raw)) return raw.replace(/\/+$/, '')
+  return `${location.origin}${raw.startsWith('/') ? raw : `/${raw}`}`.replace(/\/+$/, '')
+}
+
+function resolveBootstrap(): Promise<SatoriBootstrap> {
+  if (bootstrap) return Promise.resolve(bootstrap)
+  if (bootstrapPromise) return bootstrapPromise
+
+  bootstrapPromise = new Promise<SatoriBootstrap>((resolve, reject) => {
+    resolveBootstrapCallback = resolve
+    bootstrapTimeout = window.setTimeout(() => {
+      resolveBootstrapCallback = null
+      bootstrapPromise = null
+      reject(new Error('等待 Satori bootstrap 超时'))
+    }, 10000)
+  })
+
+  window.parent.postMessage({ source: 'chat-patch-ready' }, '*')
+  return bootstrapPromise
+}
+
+function receiveBootstrap(data: Record<string, unknown>) {
+  const payload = getObject(data.payload)
+  bootstrap = {
+    endpoint: getString(payload.endpoint) || '/satori',
+    token: getString(payload.token),
+    basePath: getString(payload.basePath) || '/chat-patch',
+  }
+  window.clearTimeout(bootstrapTimeout)
+  resolveBootstrapCallback?.(bootstrap)
+  resolveBootstrapCallback = null
+}
+
+window.addEventListener('message', (event) => {
+  const data = getObject(event.data)
+  if (getString(data.source) === 'chat-patch-bootstrap') {
+    receiveBootstrap(data)
+  }
+})
+
+function wsUrl(): string {
+  const endpoint = normalizeEndpoint(bootstrap?.endpoint || '/satori')
+  const protocol = location.protocol === 'https:' ? 'wss:' : 'ws:'
+  const url = new URL(endpoint)
+  return `${protocol}//${url.host}${url.pathname}/v1/events`
+}
+
+export function getBootstrap(): Promise<SatoriBootstrap> {
+  return resolveBootstrap()
+}
+
+export function getLogins(): SatoriLogin[] {
+  return logins
+}
+
+export function getActiveBot(): { platform: string; selfId: string } | null {
+  return activeBot
+}
+
+export function setActiveBot(platform: string, selfId: string) {
+  activeBot = { platform, selfId }
+}
+
+export function getSequence(): number {
+  return sequence
+}
+
+export async function request(
+  method: string,
+  params: Record<string, unknown> = {},
+  bot: { platform?: string; selfId?: string } | null = activeBot,
+): Promise<unknown> {
+  const info = await resolveBootstrap()
+  const headers: Record<string, string> = {
+    'Content-Type': 'application/json',
+  }
+  if (info.token) headers.Authorization = `Bearer ${info.token}`
+  if (bot?.platform) headers['Satori-Platform'] = bot.platform
+  if (bot?.selfId) headers['Satori-User-ID'] = bot.selfId
+
+  const response = await fetch(`${normalizeEndpoint(info.endpoint)}/v1/${method}`, {
+    method: 'POST',
+    headers,
+    body: JSON.stringify(params),
+  })
+  return parseJsonResponse(response)
+}
+
+export async function upload(files: File[], bot: { platform: string; selfId: string }): Promise<unknown> {
+  const info = await resolveBootstrap()
+  const form = new FormData()
+  for (const file of files) form.append('file', file, file.name)
+  const headers: Record<string, string> = {}
+  if (info.token) headers.Authorization = `Bearer ${info.token}`
+  headers['Satori-Platform'] = bot.platform
+  headers['Satori-User-ID'] = bot.selfId
+
+  const response = await fetch(`${normalizeEndpoint(info.endpoint)}/v1/upload.create`, {
+    method: 'POST',
+    headers,
+    body: form,
+  })
+  return parseJsonResponse(response)
+}
+
+export function requestConsole(method: string, params: Record<string, unknown> = {}): Promise<unknown> {
+  const id = ++consoleRequestId
+  return new Promise((resolve, reject) => {
+    pendingConsole.set(String(id), { resolve, reject })
+    window.parent.postMessage({
+      source: 'chat-patch-request',
+      id,
+      method,
+      params,
+    }, '*')
+  })
+}
+
+export function handleConsoleResponse(data: Record<string, unknown>) {
+  const id = getString(data.id)
+  const pending = pendingConsole.get(id)
+  if (!pending) return
+  pendingConsole.delete(id)
+  if (data.ok === false) {
+    pending.reject(new Error(getString(data.error) || 'console 请求失败'))
+  } else {
+    pending.resolve(data.payload)
+  }
+}
+
+export function connect(
+  onEvent: (event: SatoriEvent) => void,
+  onReady: (logins: SatoriLogin[]) => void,
+  onStatus: (online: boolean) => void,
+): () => void {
+  let socket: WebSocket | null = null
+  let pingTimer = 0
+  let reconnectTimer = 0
+  let retry = 0
+  let disposed = false
+
+  const handleMessage = (event: MessageEvent) => {
+    if (getString(event.data.source) === 'chat-patch-response') {
+      handleConsoleResponse(getObject(event.data))
+      return
+    }
+    if (getString(event.data.source) === 'chat-patch-bootstrap') {
+      const payload = getObject(event.data.payload)
+      bootstrap = {
+        endpoint: getString(payload.endpoint) || '/satori',
+        token: getString(payload.token),
+        basePath: getString(payload.basePath) || '/chat-patch',
+      }
+    }
+  }
+
+  const open = () => {
+    void resolveBootstrap().then((info) => {
+      if (disposed) return
+      bootstrap = info
+      socket = new WebSocket(wsUrl())
+      socket.addEventListener('open', () => {
+        retry = 0
+        socket?.send(JSON.stringify({
+          op: 3,
+          body: {
+            token: info.token || undefined,
+            sn: sequence || undefined,
+          },
+        }))
+        pingTimer = window.setInterval(() => {
+          socket?.send(JSON.stringify({ op: 1, body: {} }))
+        }, 10000)
+        onStatus(true)
+      })
+
+      socket.addEventListener('message', (event) => {
+        try {
+          const raw = JSON.parse(String(event.data)) as Record<string, unknown>
+          const op = Number(raw.op)
+          const body = getObject(raw.body)
+          if (op === 4) {
+            const loginList = Array.isArray(body.logins) ? body.logins : []
+            const seen = new Set<string>()
+            const nextLogins: SatoriLogin[] = []
+            for (const item of loginList) {
+              const login = getObject(item)
+              const user = getObject(login.user)
+              const platform = getString(login.platform) || getString(user.platform)
+              const selfId = getString(user.id) || getString(login.selfId)
+              const key = [platform, selfId].map((value) => encodeURIComponent(value)).join(':')
+              if (!platform || !selfId || seen.has(key)) continue
+              seen.add(key)
+              nextLogins.push({
+                platform: getString(login.platform) || getString(user.platform),
+                selfId,
+                name: getString(user.name) || getString(user.nick) || getString(login.selfId),
+                avatar: getString(user.avatar) || undefined,
+                status: Number(login.status ?? 1),
+                features: Array.isArray(login.features) ? login.features.map(String) : [],
+              })
+            }
+            logins = nextLogins
+            if (!activeBot && logins[0]) {
+              activeBot = { platform: logins[0].platform, selfId: logins[0].selfId }
+            }
+            onReady(logins)
+            return
+          }
+          if (op === 5) return
+          if (op === 0) {
+            sequence = Number(getObject(body).sn ?? sequence)
+            localStorage.setItem('chat-patch:sn', String(sequence))
+            onEvent({
+              type: getString(body.type),
+              platform: getString(body.platform),
+              selfId: getString(body.selfId) || getString(getObject(getObject(body.login).user).id),
+              timestamp: Number(body.timestamp ?? Date.now()),
+              sn: sequence,
+              body,
+            })
+          }
+        } catch {
+          // 忽略无法解析的信令
+        }
+      })
+
+      socket.addEventListener('close', () => {
+        window.clearInterval(pingTimer)
+        onStatus(false)
+        if (disposed) return
+        retry += 1
+        const delay = Math.min(30000, 1000 * 2 ** retry)
+        reconnectTimer = window.setTimeout(open, delay)
+      })
+
+      socket.addEventListener('error', () => {
+        socket?.close()
+      })
+    }).catch(() => {
+      onStatus(false)
+    })
+  }
+
+  window.addEventListener('message', handleMessage)
+  open()
+
+  return () => {
+    disposed = true
+    window.clearInterval(pingTimer)
+    window.clearTimeout(reconnectTimer)
+    window.removeEventListener('message', handleMessage)
+    socket?.close()
+    socket = null
+  }
+}

package/client/web/src/function/sender.ts

@@ -0,0 +1,178 @@
+// 将 UI 消息段转换为 Satori 元素字符串。
+
+import type { MsgItemElem } from './elements/information'
+
+function escapeText(value: string): string {
+  return value
+    .replaceAll('&', '&')
+    .replaceAll('<', '&lt;')
+    .replaceAll('>', '&gt;')
+}
+
+function escapeAttr(value: string): string {
+  return escapeText(value).replaceAll('"', '&quot;')
+}
+
+const satoriMarkupPattern = /<(\/?)(quote|at|img|image|audio|record|video|file|mface|face|sharp|p|br|message|text|json|xml|markdown|keyboard)(?:\s[^>]*)?\/?>/gi
+const sqTokenPattern = /(\[SQ:\d+\]|\[图片\]|\[@[^\]]+\])/g
+
+function escapeTextPreservingMarkup(value: string): string {
+  const decoded = value
+    .replaceAll('&lt;', '<')
+    .replaceAll('&gt;', '>')
+    .replaceAll('&quot;', '"')
+    .replaceAll('&#39;', "'")
+    .replaceAll('&amp;', '&')
+  const saved: string[] = []
+  const marker = (index: number) => `\u0000sq${index}\u0000`
+  const masked = decoded.replace(satoriMarkupPattern, (tag) => {
+    saved.push(tag)
+    return marker(saved.length - 1)
+  })
+  return masked
+    .replaceAll('&', '&amp;')
+    .replaceAll('<', '&lt;')
+    .replaceAll('>', '&gt;')
+    .replace(/\u0000sq(\d+)\u0000/g, (_, index: string) => saved[Number(index)] ?? '')
+}
+
+function segmentToSatori(segment: MsgItemElem): string {
+  const type = String(segment.type ?? '')
+  if (type === 'text') return escapeText(String(segment.text ?? ''))
+  if (type === 'reply' || type === 'quote') {
+    const id = String(segment.id ?? segment.data?.id ?? '')
+    return id ? `<quote id="${escapeAttr(id)}"/>` : ''
+  }
+  if (type === 'at') {
+    const id = String(segment.qq ?? segment.id ?? '')
+    const name = String(segment.text ?? segment.name ?? '')
+    return `<at id="${escapeAttr(id)}"${name ? ` name="${escapeAttr(name)}"` : ''}/>`
+  }
+  if (type === 'image') {
+    const src = String(segment.file ?? segment.url ?? segment.src ?? '')
+    return src ? `<img src="${escapeAttr(src)}"/>` : ''
+  }
+  if (type === 'face' || type === 'bface') {
+    const id = String(segment.id ?? segment.data?.id ?? '')
+    if (/^\d+$/.test(id)) {
+      const codePoint = Number(id)
+      if (codePoint >= 128000) return String.fromCodePoint(codePoint)
+    }
+    return String(segment.text ?? '[表情]')
+  }
+  if (type === 'record' || type === 'audio') {
+    const src = String(segment.file ?? segment.url ?? segment.src ?? '')
+    const name = String(segment.fileName ?? segment.name ?? '')
+    return src ? `<audio src="${escapeAttr(src)}"${name ? ` name="${escapeAttr(name)}"` : ''}/>` : ''
+  }
+  if (type === 'video') {
+    const src = String(segment.file ?? segment.url ?? segment.src ?? '')
+    return src ? `<video src="${escapeAttr(src)}"/>` : ''
+  }
+  if (type === 'file') {
+    const src = String(segment.file ?? segment.url ?? segment.src ?? '')
+    const name = String(segment.name ?? '')
+    return src ? `<file src="${escapeAttr(src)}"${name ? ` name="${escapeAttr(name)}"` : ''}/>` : ''
+  }
+  if (type === 'json') {
+    const data = String(segment.data ?? segment.content ?? '')
+    return data ? `<json data="${escapeAttr(data)}"/>` : ''
+  }
+  if (type === 'xml') {
+    const data = String(segment.data ?? segment.content ?? '')
+    return data ? `<xml data="${escapeAttr(data)}"/>` : ''
+  }
+  if (type === 'markdown') {
+    const content = String(segment.content ?? '')
+    return content ? `<markdown content="${escapeAttr(content)}"/>` : ''
+  }
+  if (type === 'forward') {
+    const id = String(segment.id ?? '')
+    return id ? `<forward id="${escapeAttr(id)}"/>` : ''
+  }
+  return escapeText(String(segment.text ?? `[${type}]`))
+}
+
+/**
+ * 将一组消息节点构建为 Satori 合并转发元素
+ */
+export function buildForwardMessage(nodes: unknown[]): string {
+  const messages = nodes.map((raw) => {
+    const item = typeof raw === 'object' && raw !== null
+      ? raw as Record<string, unknown>
+      : {}
+    const nodeData = typeof item.data === 'object' && item.data !== null
+      ? item.data as Record<string, unknown>
+      : item
+    const userId = String(item.user_id ?? nodeData.user_id ?? nodeData.uin ?? '')
+    const nickname = String(item.nickname ?? nodeData.nickname ?? nodeData.name ?? '')
+    const displayName = nickname || userId
+    const time = String(item.time ?? nodeData.time ?? '')
+    const content = Array.isArray(item.content)
+      ? item.content as unknown[]
+      : Array.isArray(nodeData.content)
+        ? nodeData.content as unknown[]
+        : []
+    const inner = content.map((segment) => segmentToSatori(segment as MsgItemElem)).join('')
+    const author = `<author id="${escapeAttr(userId)}" name="${escapeAttr(displayName)}"${time ? ` time="${escapeAttr(time)}"` : ''}/>`
+    return `<message>${author}${inner}</message>`
+  }).join('')
+  return `<figure>${messages}</figure>`
+}
+
+export function parseMsg(msg: string, cache: MsgItemElem[], _img: string[]): string {
+  let output = ''
+  const pattern = sqTokenPattern
+  const consumed = new Set<number>()
+  const replyPrefix = cache
+    .filter((segment) => segment?.type === 'reply' || segment?.type === 'quote')
+    .map((segment) => segmentToSatori(segment))
+    .join('')
+  let match: RegExpExecArray | null
+  let cursor = 0
+
+  while ((match = pattern.exec(msg)) !== null) {
+    output += escapeTextPreservingMarkup(msg.slice(cursor, match.index))
+    const token = match[0]
+    if (token.startsWith('[SQ:')) {
+      const index = Number(token.slice(4, -1))
+      const segment = cache[index]
+      if (segment) {
+        output += segmentToSatori(segment)
+        consumed.add(index)
+      }
+    } else if (token.startsWith('[@')) {
+      const atIndex = cache.findIndex((segment, index) => {
+        return !consumed.has(index) && segment?.type === 'at'
+      })
+      if (atIndex >= 0) {
+        output += segmentToSatori(cache[atIndex])
+        consumed.add(atIndex)
+      } else {
+        output += escapeTextPreservingMarkup(token)
+      }
+    } else {
+      const imageIndex = cache.findIndex((segment, index) => {
+        return !consumed.has(index) && segment?.type === 'image'
+      })
+      if (imageIndex >= 0) {
+        output += segmentToSatori(cache[imageIndex])
+        consumed.add(imageIndex)
+      } else {
+        output += '[图片]'
+      }
+    }
+    cursor = match.index + match[0].length
+  }
+  output += escapeTextPreservingMarkup(msg.slice(cursor))
+  return replyPrefix + output
+}
+
+export function getSQList(msg: string): RegExpMatchArray | null {
+  return msg.match(sqTokenPattern)
+}
+
+export default {
+  parseMsg,
+  getSQList,
+}

package/client/web/src/function/tooltip.ts

@@ -0,0 +1,49 @@
+import { Component, DirectiveBinding, markRaw, ObjectDirective, shallowReactive } from 'vue'
+import { VueCompData } from './elements/vueComp'
+import { v4 as uuid } from 'uuid'
+import UserInfoTooltip from '../components/tooltip/UserInfoTooltip.vue'
+import { vTooltip } from './utils/appUtil'
+
+export const tooltipList: TooltipInfo<Component>[] = shallowReactive([])
+
+export type TooltipInfo<T extends Component> = {
+    compData: VueCompData<T>
+    pos: { x: number, y: number }
+    id: string
+}
+
+export type TooltipController = {
+    id: string
+    close: () => void
+}
+
+export function addTooltip<T extends Component>(compData: VueCompData<T>, pos: { x: number, y: number }): TooltipController {
+    const id = uuid()
+    tooltipList.push({ compData, pos, id })
+    return {
+        id,
+        close: () => closeTooltip(id)
+    }
+}
+
+export function closeTooltip(id: string) {
+    const index = tooltipList.findIndex(t => t.id === id)
+    if (index !== -1) {
+        tooltipList.splice(index, 1)
+    }
+}
+
+type IUser = any
+
+type VUserTooltipBinding = IUser | number | (() => IUser | number)
+
+export const vUserTooltip: ObjectDirective<HTMLElement, VUserTooltipBinding> = {
+    mounted(el: HTMLElement, binding: DirectiveBinding<VUserTooltipBinding>) {
+        (vTooltip as any).mounted(el, {
+            value: { comp: markRaw(UserInfoTooltip), props: { user: binding.value } }
+        })
+    },
+    unmounted(el: HTMLElement) {
+        (vTooltip as any).unmounted(el)
+    }
+}