koishi-plugin-chat-patch 1.3.0 → 2.0.1

package/client/vue/composables/useChatActions.ts

@@ -0,0 +1,61 @@
+import { ref } from 'vue'
+import { send } from '@koishijs/client'
+import type { MessageInfo } from '../types'
+
+export function useChatActions() {
+  const isSending = ref(false)
+
+  async function sendMessage(botId: string, channelId: string, content: string, images: any[]) {
+    if (isSending.value) return
+    isSending.value = true
+    try {
+      const result = await (send as any)('send-message', {
+        selfId: botId,
+        channelId: channelId,
+        content: content,
+        images: images.map(img => ({
+          tempId: img.tempId,
+          filename: img.filename
+        }))
+      })
+      return result
+    } finally {
+      isSending.value = false
+    }
+  }
+
+  async function deleteBotData(botId: string) {
+    return await (send as any)('delete-bot-data', { selfId: botId })
+  }
+
+  async function deleteChannelData(botId: string, channelId: string) {
+    return await (send as any)('delete-channel-data', { selfId: botId, channelId: channelId })
+  }
+
+  async function clearHistory(botId: string, channelId: string) {
+    return await (send as any)('clear-channel-history', { selfId: botId, channelId: channelId })
+  }
+
+  async function togglePinBot(botId: string, pinned: boolean, pinnedBots: Set<string>) {
+    if (pinned) pinnedBots.delete(botId)
+    else pinnedBots.add(botId)
+    await (send as any)('set-pinned-bots', { pinnedBots: Array.from(pinnedBots) })
+  }
+
+  async function togglePinChannel(botId: string, channelId: string, pinned: boolean, pinnedChannels: Set<string>) {
+    const key = `${botId}:${channelId}`
+    if (pinned) pinnedChannels.delete(key)
+    else pinnedChannels.add(key)
+    await (send as any)('set-pinned-channels', { pinnedChannels: Array.from(pinnedChannels) })
+  }
+
+  return {
+    isSending,
+    sendMessage,
+    deleteBotData,
+    deleteChannelData,
+    clearHistory,
+    togglePinBot,
+    togglePinChannel
+  }
+}

package/client/vue/composables/useChatData.ts

@@ -0,0 +1,184 @@
+import { ref, computed } from 'vue'
+import { receive, send } from '@koishijs/client'
+import type { ChatData, BotInfo, ChannelInfo, MessageInfo, PluginConfig } from '../types'
+
+export function useChatData() {
+  const chatData = ref<ChatData>({
+    bots: {},
+    channels: {},
+    messages: {}
+  })
+
+  const pluginConfig = ref<PluginConfig>({
+    maxMessagesPerChannel: 1000,
+    keepMessagesOnClear: 50,
+    loggerinfo: false,
+    blockedPlatforms: [],
+    chatContainerHeight: 80,
+    clearIndexedDBOnStart: true
+  })
+
+  const pinnedBots = ref<Set<string>>(new Set())
+  const pinnedChannels = ref<Set<string>>(new Set())
+
+  // 记录每个频道的分页状态
+  const channelPagination = ref<Record<string, { offset: number, hasMore: boolean }>>({})
+
+  // 计算属性
+  const bots = computed(() => {
+    return Object.values(chatData.value.bots).sort((a, b) => {
+      const aPinned = pinnedBots.value.has(a.selfId)
+      const bPinned = pinnedBots.value.has(b.selfId)
+      return aPinned === bPinned ? 0 : aPinned ? -1 : 1
+    })
+  })
+
+  const getChannels = (botId: string) => {
+    if (!botId || !chatData.value.channels[botId]) return []
+    return Object.values(chatData.value.channels[botId]).sort((a, b) => {
+      const aPinned = pinnedChannels.value.has(`${botId}:${a.id}`)
+      const bPinned = pinnedChannels.value.has(`${botId}:${b.id}`)
+      return aPinned === bPinned ? 0 : aPinned ? -1 : 1
+    })
+  }
+
+  const getMessages = (botId: string, channelId: string) => {
+    const key = `${botId}:${channelId}`
+    return chatData.value.messages[key] || []
+  }
+
+  const getPagination = (botId: string, channelId: string) => {
+    const key = `${botId}:${channelId}`
+    return channelPagination.value[key] || { offset: 0, hasMore: true }
+  }
+
+  // 加载数据
+  async function loadInitialData() {
+    const result = await (send as any)('get-chat-data')
+    if (result.success && result.data) {
+      chatData.value = {
+        bots: result.data.bots || {},
+        channels: result.data.channels || {},
+        messages: result.data.messages || {}
+      }
+      pinnedBots.value = new Set(result.data.pinnedBots || [])
+      pinnedChannels.value = new Set(result.data.pinnedChannels || [])
+    }
+  }
+
+  async function loadConfig() {
+    const result = await (send as any)('get-plugin-config')
+    if (result.success && result.config) {
+      pluginConfig.value = result.config
+    }
+  }
+
+  // 消息处理
+  function addMessage(msg: any) {
+    const selfId = msg.selfId
+    const channelId = msg.channelId
+    const key = `${selfId}:${channelId}`
+
+    // 1. 确保机器人存在
+    if (!chatData.value.bots[selfId]) {
+      chatData.value.bots[selfId] = {
+        selfId,
+        platform: msg.platform || 'unknown',
+        username: msg.bot?.name || `Bot-${selfId}`,
+        avatar: msg.bot?.avatar,
+        status: 'online'
+      }
+    }
+
+    // 2. 确保频道存在
+    if (!chatData.value.channels[selfId]) {
+      chatData.value.channels[selfId] = {}
+    }
+    if (!chatData.value.channels[selfId][channelId]) {
+      chatData.value.channels[selfId][channelId] = {
+        id: channelId,
+        name: msg.guildName || msg.username || channelId,
+        type: msg.channelType || 0
+      }
+    }
+
+    // 3. 消息去重与添加
+    if (!chatData.value.messages[key]) chatData.value.messages[key] = []
+
+    const messages = chatData.value.messages[key]
+    // 严格去重：检查 ID，如果是机器人发送的临时消息，则通过内容和时间戳近似匹配
+    const isDuplicate = messages.some(m => {
+      if (m.id === msg.messageId || m.id === msg.id) return true
+      // 针对机器人发送消息的特殊去重逻辑
+      if ((msg.type === 'bot-message' || msg.type === 'bot-message-sent' || msg.type === 'bot') && m.isBot && Math.abs(m.timestamp - msg.timestamp) < 2000 && m.content === msg.content) return true
+      return false
+    })
+
+    if (!isDuplicate) {
+      const newMsg: MessageInfo = {
+        id: msg.messageId || msg.id,
+        content: msg.content,
+        userId: msg.userId,
+        username: msg.username,
+        avatar: msg.avatar,
+        timestamp: msg.timestamp,
+        channelId: channelId,
+        selfId: selfId,
+        elements: msg.elements,
+        isBot: msg.isBot || msg.type === 'bot-message' || msg.type === 'bot',
+        quote: msg.quote
+      }
+      messages.push(newMsg)
+      messages.sort((a, b) => a.timestamp - b.timestamp)
+    }
+
+    // 强制触发响应式更新，确保列表实时刷新
+    chatData.value = { ...chatData.value }
+  }
+
+  async function loadHistory(botId: string, channelId: string, limit = 50) {
+    const key = `${botId}:${channelId}`
+    const pagination = getPagination(botId, channelId)
+    if (!pagination.hasMore) return
+
+    const result = await (send as any)('get-history-messages', {
+      selfId: botId,
+      channelId: channelId,
+      limit,
+      offset: pagination.offset
+    })
+
+    if (result.success && result.messages) {
+      if (!chatData.value.messages[key]) chatData.value.messages[key] = []
+      const existing = chatData.value.messages[key]
+
+      // 合并并去重
+      const newMsgs = result.messages.filter((m: any) => !existing.find(e => e.id === m.id))
+      chatData.value.messages[key] = [...newMsgs, ...existing].sort((a, b) => a.timestamp - b.timestamp)
+
+      channelPagination.value[key] = {
+        offset: pagination.offset + result.messages.length,
+        hasMore: result.messages.length >= limit && result.total > (pagination.offset + result.messages.length)
+      }
+
+      chatData.value = { ...chatData.value }
+      return result.messages.length
+    }
+    return 0
+  }
+
+  return {
+    chatData,
+    pluginConfig,
+    pinnedBots,
+    pinnedChannels,
+    bots,
+    getChannels,
+    getMessages,
+    getPagination,
+    loadInitialData,
+    loadConfig,
+    addMessage,
+    loadHistory
+  }
+}

package/client/vue/composables/useImageCache.ts

@@ -0,0 +1,140 @@
+import { ref, onMounted, onUnmounted } from 'vue'
+import { send } from '@koishijs/client'
+
+// 图片缓存 - IndexedDB
+interface ImageCacheItem {
+  url: string
+  blob: Blob
+  timestamp: number
+  size: number
+  channelKey: string
+}
+
+export function useImageCache() {
+  const imageBlobUrls = ref<Record<string, string>>({})
+  const loadingImages = new Map<string, Promise<string | null>>()
+
+  // 内存管理配置
+  const MAX_MEMORY_USAGE = 100 * 1024 * 1024 // 100MB
+  const MAX_BLOB_COUNT = 50
+  let currentMemoryUsage = 0
+
+  // IndexedDB 配置
+  let imageDB: IDBDatabase | null = null
+  const DB_NAME = 'ChatImageCache'
+  const DB_VERSION = 2
+  const STORE_NAME = 'images'
+  const MAX_DB_SIZE = 50 * 1024 * 1024 // 50MB
+  const MAX_TOTAL_IMAGES = 500
+  const MAX_IMAGE_SIZE = 12 * 1024 * 1024 // 12MB
+
+  // 初始化数据库
+  async function initDB(): Promise<boolean> {
+    return new Promise((resolve) => {
+      const request = indexedDB.open(DB_NAME, DB_VERSION)
+      request.onupgradeneeded = (event) => {
+        const db = (event.target as IDBOpenDBRequest).result
+        if (!db.objectStoreNames.contains(STORE_NAME)) {
+          const store = db.createObjectStore(STORE_NAME, { keyPath: 'url' })
+          store.createIndex('channelKey', 'channelKey', { unique: false })
+          store.createIndex('timestamp', 'timestamp', { unique: false })
+        }
+      }
+      request.onsuccess = () => {
+        imageDB = request.result
+        resolve(true)
+      }
+      request.onerror = () => resolve(false)
+    })
+  }
+
+  // 获取缓存图片
+  async function getCachedImageUrl(channelKey: string, url: string): Promise<string | null> {
+    if (imageBlobUrls.value[url]) return imageBlobUrls.value[url]
+    if (loadingImages.has(url)) return loadingImages.get(url)!
+
+    const promise = (async () => {
+      try {
+        if (!imageDB) await initDB()
+        const item = await getImageFromDB(url)
+        if (item) {
+          const blobUrl = URL.createObjectURL(item.blob)
+          imageBlobUrls.value[url] = blobUrl
+          return blobUrl
+        }
+        return null
+      } finally {
+        loadingImages.delete(url)
+      }
+    })()
+
+    loadingImages.set(url, promise)
+    return promise
+  }
+
+  async function getImageFromDB(url: string): Promise<ImageCacheItem | null> {
+    if (!imageDB) return null
+    return new Promise((resolve) => {
+      const transaction = imageDB!.transaction([STORE_NAME], 'readonly')
+      const store = transaction.objectStore(STORE_NAME)
+      const request = store.get(url)
+      request.onsuccess = () => resolve(request.result || null)
+      request.onerror = () => resolve(null)
+    })
+  }
+
+  async function cacheImage(channelKey: string, url: string): Promise<string | null> {
+    const existing = await getCachedImageUrl(channelKey, url)
+    if (existing) return existing
+
+    try {
+      const result = await (send as any)('fetch-image', { url })
+      if (!result.success) return null
+
+      const response = await fetch(result.dataUrl)
+      const blob = await response.blob()
+
+      if (blob.size > MAX_IMAGE_SIZE) return null
+
+      const item: ImageCacheItem = {
+        url,
+        blob,
+        timestamp: Date.now(),
+        size: blob.size,
+        channelKey
+      }
+
+      await saveToDB(item)
+      const blobUrl = URL.createObjectURL(blob)
+      imageBlobUrls.value[url] = blobUrl
+      return blobUrl
+    } catch (e) {
+      return null
+    }
+  }
+
+  async function saveToDB(item: ImageCacheItem) {
+    if (!imageDB) return
+    const transaction = imageDB.transaction([STORE_NAME], 'readwrite')
+    transaction.objectStore(STORE_NAME).put(item)
+  }
+
+  async function clearChannelCache(channelKey: string) {
+    // 简化实现
+    Object.keys(imageBlobUrls.value).forEach(url => {
+      URL.revokeObjectURL(imageBlobUrls.value[url])
+      delete imageBlobUrls.value[url]
+    })
+  }
+
+  onUnmounted(() => {
+    Object.values(imageBlobUrls.value).forEach(URL.revokeObjectURL)
+    if (imageDB) imageDB.close()
+  })
+
+  return {
+    getCachedImageUrl,
+    cacheImage,
+    clearChannelCache
+  }
+}