koishi-plugin-chat-patch 3.0.2 → 5.0.0

package/client/web/src/function/utils/pinyin.ts

@@ -0,0 +1,141 @@
+import { isDebugMode } from '../debug'
+
+export type PinYinData = {
+    main: string[]
+    short: string[]
+}
+
+/* eslint-disable no-console */
+
+const PINYIN_SCRIPT_SRC = 'https://lib.stapxs.cn/modules/pinyin.min.js'
+
+let pinyinLoadPromise: Promise<boolean> | null = null
+
+function createEmptyPinyinData(): PinYinData {
+    return {
+        main: [],
+        short: []
+    }
+}
+
+function hasPinyinLib() {
+    return typeof window !== 'undefined' && typeof window.pinyin !== 'undefined'
+}
+
+function scheduleIdleTask(task: () => void) {
+    if (typeof window === 'undefined') return
+
+    const idleWindow = window as Window & {
+        requestIdleCallback?: (
+            callback: IdleRequestCallback,
+            options?: IdleRequestOptions
+        ) => number
+    }
+
+    if (typeof idleWindow.requestIdleCallback === 'function') {
+        idleWindow.requestIdleCallback(() => task(), { timeout: 1500 })
+        return
+    }
+
+    window.setTimeout(task, 300)
+}
+
+export function isPinyinReady() {
+    return hasPinyinLib()
+}
+
+export function ensurePinyinLoaded(): Promise<boolean> {
+    if (hasPinyinLib()) return Promise.resolve(true)
+    if (typeof window === 'undefined' || typeof document === 'undefined') {
+        return Promise.resolve(false)
+    }
+    if (pinyinLoadPromise !== null) return pinyinLoadPromise
+
+    pinyinLoadPromise = new Promise((resolve) => {
+        let script = document.querySelector(
+            'script[data-ssqq-pinyin-lib="true"]',
+        ) as HTMLScriptElement | null
+
+        const finish = (success: boolean) => {
+            if (!success) {
+                pinyinLoadPromise = null
+            }
+            resolve(success)
+        }
+
+        const handleLoad = () => {
+            if (script) {
+                script.dataset.loaded = 'true'
+            }
+            finish(hasPinyinLib())
+        }
+
+        const handleError = () => {
+            if (isDebugMode()) console.warn('拼音库加载失败')
+            script?.remove()
+            finish(false)
+        }
+
+        if (script?.dataset.loaded === 'true') {
+            finish(hasPinyinLib())
+            return
+        }
+
+        if (!script) {
+            script = document.createElement('script')
+            script.src = PINYIN_SCRIPT_SRC
+            script.async = true
+            script.dataset.ssqqPinyinLib = 'true'
+            document.body.appendChild(script)
+        }
+
+        script.addEventListener('load', handleLoad, { once: true })
+        script.addEventListener('error', handleError, { once: true })
+    })
+
+    return pinyinLoadPromise
+}
+
+export function preloadPinyin() {
+    scheduleIdleTask(() => {
+        void ensurePinyinLoaded()
+    })
+}
+
+export function getPinyin(name: string): PinYinData {
+    if (!hasPinyinLib()) return createEmptyPinyinData()
+
+    try {
+        const pinyinLib = window.pinyin
+        if (!pinyinLib) return createEmptyPinyinData()
+        return {
+            main: pinyinLib.pinyin(name, {
+                heteronym: true,
+                compact: true,
+                style: 'normal',
+            }).map((item: string[]) => item.join('').toLowerCase()),
+            short: pinyinLib.pinyin(name, {
+                heteronym: true,
+                compact: true,
+                style: 'first_letter',
+            }).map((item: string[]) => item.join('').toLowerCase()),
+        }
+    } catch (error) {
+        if (isDebugMode()) console.warn('拼音转换失败:', error)
+        return createEmptyPinyinData()
+    }
+}
+
+export function matchPinyin(
+    pinyinData: PinYinData,
+    matchStr: string
+): boolean {
+    const str = matchStr.toLowerCase()
+    for (const py of pinyinData.main) {
+        if (py.includes(str)) return true
+    }
+    for (const pyShort of pinyinData.short) {
+        if (pyShort.includes(str)) return true
+    }
+    return false
+}

package/client/web/src/function/utils/recordUtil.ts

@@ -0,0 +1,187 @@
+/*
+ * @FileDescription: 语音消息处理工具
+ * @Author: FuQuan233
+ * @Date: 2026/06/04
+ * @Version: 1.0
+ * @Description: 通过 OneBot get_record API 加载语音消息。
+ *   以 out_format=mp3 请求格式转换，响应 base64 数据直接用于播放。
+ */
+
+import { Logger } from '../../function/base'
+import { Connector } from '../../function/connect'
+import { getBootstrap } from '../../function/satori'
+
+const logger = new Logger()
+
+/** 语音消息原始数据（来自 OneBot record 消息段） */
+export interface RecordMsgData {
+    file?: string
+    url?: string
+    path?: string
+    webUrl?: string
+    magic?: boolean
+    duration?: number
+}
+
+/** 语音加载结果 */
+export interface RecordLoadResult {
+    src: string
+    duration: number
+}
+
+/** 加载缓存 */
+const loadCache = new Map<string, RecordLoadResult>()
+
+/**
+ * base64 字符串转 Blob URL
+ */
+function base64ToBlobUrl(base64: string): string {
+    if (base64.startsWith('base64://')) base64 = base64.substring(9)
+    const binary = atob(base64)
+    const bytes = new Uint8Array(binary.length)
+    for (let i = 0; i < binary.length; i++) bytes[i] = binary.charCodeAt(i)
+    return URL.createObjectURL(new Blob([bytes], { type: 'audio/mpeg' }))
+}
+
+/**
+ * 获取音频时长
+ */
+function getAudioDuration(url: string): Promise<number> {
+    return new Promise((resolve) => {
+        const audio = new Audio()
+        let resolved = false
+        const done = (val: number) => {
+            if (!resolved) {
+                resolved = true
+                audio.removeEventListener('loadedmetadata', onMeta)
+                audio.removeEventListener('error', onErr)
+                audio.src = ''
+                resolve(val)
+            }
+        }
+        const onMeta = () => done(audio.duration || 0)
+        const onErr = () => done(0)
+        audio.addEventListener('loadedmetadata', onMeta)
+        audio.addEventListener('error', onErr)
+        setTimeout(() => done(0), 3000)
+        audio.src = url
+    })
+}
+
+function isDirectAudioSource(value: string): boolean {
+  return /^(https?:|data:|blob:)/i.test(value) || value.startsWith('/')
+}
+
+function isLocalAudioPath(value: string): boolean {
+  return value.startsWith('file:') ||
+    /^[a-z]:[\\/]/i.test(value) ||
+    value.startsWith('\\\\')
+}
+
+async function resolveLocalAudioUrl(value: string): Promise<string> {
+  if (isDirectAudioSource(value)) return value
+  if (!isLocalAudioPath(value)) return ''
+  const info = await getBootstrap()
+  const basePath = info.basePath || '/chat-patch'
+  const fileName = value
+    .replace(/^file:\/\//i, '')
+    .split(/[\\/]/)
+    .pop() || value
+  return `${location.origin}${basePath}/api/media?file=${encodeURIComponent(fileName)}`
+}
+
+// ============================================================================
+// 公开 API
+// ============================================================================
+
+/**
+ * 加载语音消息为可播放的音频 URL
+ *
+ * 通过 OneBot get_record API (out_format=mp3) 获取语音，
+ * 响应中的 base64 字段直接包含 MP3 数据。
+ *
+ * @param data   语音消息数据
+ * @param msgId  消息 ID（用于缓存键）
+ * @returns      加载结果
+ * @throws       无法加载时抛出错误
+ */
+export async function loadRecord(data: RecordMsgData, msgId?: string): Promise<RecordLoadResult> {
+    const cacheKey = msgId || data.file!
+
+    // 缓存检查
+    const cached = loadCache.get(cacheKey)
+    if (cached) return cached
+
+    if (!data.file) {
+        throw new Error('语音消息缺少 file 字段')
+    }
+
+    const directSource = String(data.webUrl || data.url || data.file || data.path || '')
+    if (directSource) {
+        const directUrl = await resolveLocalAudioUrl(directSource)
+        if (directUrl) {
+            const duration = data.duration || (await getAudioDuration(directUrl))
+            const loadResult: RecordLoadResult = { src: directUrl, duration }
+            loadCache.set(cacheKey, loadResult)
+            return loadResult
+        }
+    }
+
+    // 调用 get_record API
+    const raw = await Connector.callApi('get_record', {
+        file: data.file,
+        out_format: 'mp3',
+    })
+
+    if (!raw) throw new Error('get_record API 返回空')
+
+    // result 兼容数组和单个对象
+    const result: any = Array.isArray(raw) ? raw[0] : raw
+
+    if (!result?.base64 || typeof result.base64 !== 'string') {
+        logger.debug('get_record 响应中没有 base64: ' + JSON.stringify(result).substring(0, 200))
+        throw new Error('OneBot 端不支持语音格式转换')
+    }
+
+    const blobUrl = base64ToBlobUrl(result.base64)
+    const duration = data.duration || (await getAudioDuration(blobUrl))
+
+    logger.debug(`语音加载完成: ${duration.toFixed(1)}s`)
+
+    const loadResult: RecordLoadResult = { src: blobUrl, duration }
+    loadCache.set(cacheKey, loadResult)
+    return loadResult
+}
+
+/**
+ * 释放语音资源
+ */
+export function revokeRecord(msgId?: string): void {
+    if (msgId) {
+        const cached = loadCache.get(msgId)
+        if (cached?.src.startsWith('blob:')) URL.revokeObjectURL(cached.src)
+        loadCache.delete(msgId)
+    }
+}
+
+/**
+ * 清除所有缓存
+ */
+export function clearRecordCache(): void {
+    for (const [, r] of loadCache) {
+        if (r.src.startsWith('blob:')) URL.revokeObjectURL(r.src)
+    }
+    loadCache.clear()
+}
+
+/**
+ * 格式化时长显示
+ * @param seconds 秒数
+ * @returns 格式化的时长字符串 (m:ss)
+ */
+export function formatRecordDuration(seconds: number): string {
+    if (!seconds || seconds <= 0) return '0:00'
+    const m = Math.floor(seconds / 60)
+    const s = Math.floor(seconds % 60)
+    return `${m}:${s.toString().padStart(2, '0')}`
+}

package/client/web/src/function/utils/sessionUtil.ts

@@ -0,0 +1,114 @@
+import type { Session } from '../elements/information'
+
+export function normalizeSessionId(value: number | string): string {
+  return String(value)
+}
+
+export function getSessionId(item: Session): number | string {
+    return item.user_id ?? item.group_id ?? 0
+}
+
+export function findSessionContact(
+    contacts: Session[],
+    sessionId: number | string,
+) {
+    return contacts.find((item) => {
+        return normalizeSessionId(getSessionId(item)) === normalizeSessionId(sessionId)
+    })
+}
+
+export function getMissingGroupPreviewSessions(
+    contacts: Session[],
+    knownSessions: ReadonlyMap<number | string, Session>,
+) {
+    return contacts.filter((item) => {
+        const sessionId = getSessionId(item)
+        return Boolean(item.group_id) &&
+            sessionId !== 0 &&
+            sessionId !== '' &&
+            !item.time &&
+            !item.raw_msg &&
+            !knownSessions.has(normalizeSessionId(sessionId))
+    })
+}
+
+export function resolveIncomingSession(
+    contacts: Session[],
+    sessionId: number | string,
+    isGroup: boolean,
+    senderName?: string,
+) {
+    const contact = findSessionContact(contacts, sessionId)
+    if (contact) return contact
+
+    const normalizedId = normalizeSessionId(sessionId)
+
+    // 消息事件可能早于好友/群列表返回。先保留会话动态状态，列表加载后再合并真实资料。
+    if (isGroup) {
+        return {
+            group_id: normalizedId,
+            group_name: normalizedId,
+        } as Session
+    }
+    return {
+        user_id: normalizedId,
+        nickname: senderName || normalizedId,
+        remark: '',
+    } as Session
+}
+
+const SESSION_STATE_KEYS = [
+    'new_msg',
+    'raw_msg',
+    'raw_msg_base',
+    'time',
+    'always_top',
+    'message_id',
+    'highlight',
+] as const
+
+type SessionStateKey = (typeof SESSION_STATE_KEYS)[number]
+
+function copyDefinedSessionState<K extends SessionStateKey>(
+    contact: Session,
+    currentSession: Session,
+    key: K,
+) {
+    const value = currentSession[key]
+    if (value !== undefined) {
+        contact[key] = value
+    }
+}
+
+export function mergeSessionState(
+    contact: Session,
+    currentSession: Session,
+) {
+    SESSION_STATE_KEYS.forEach((key) =>
+        copyDefinedSessionState(contact, currentSession, key),
+    )
+    return contact
+}
+
+/**
+ * 让真实联系人接管消息早到时创建的占位会话。
+ * 返回 true 表示 Map 中的对象引用已替换，调用方需要重建派生会话列表。
+ */
+export function mergeEarlySessionContacts(
+    contacts: Session[],
+    sessions: Map<number | string, Session>,
+) {
+    let didMerge = false
+    contacts.forEach((contact) => {
+        const sessionId = normalizeSessionId(getSessionId(contact))
+        const currentSession = sessions.get(sessionId)
+        if (currentSession && currentSession !== contact) {
+            sessions.set(
+                sessionId,
+                mergeSessionState(contact, currentSession),
+            )
+            didMerge = true
+        }
+    })
+    return didMerge
+}