koishi-plugin-chat-patch 5.6.1 → 6.1.0

package/src/web.ts

@@ -1,11 +1,12 @@
 import { Context } from 'koishi'
 import {} from '@koishijs/plugin-server'
-import { createReadStream, existsSync, promises as fs, statSync } from 'node:fs'
+import { createReadStream, createWriteStream, existsSync, promises as fs, statSync } from 'node:fs'
 import { createHash } from 'node:crypto'
 import path from 'node:path'
 import { pathToFileURL } from 'node:url'
-import send from 'koa-send'
-import type { DefaultContext, DefaultState, ParameterizedContext } from 'koa'
+import send from 'koa-send'
+import type Router from '@koa/router'
+import type { DefaultContext, DefaultState, ParameterizedContext } from 'koa'
 import FileType from 'file-type'
 
 import { Config } from './config'
@@ -13,7 +14,7 @@ import { ContactCacheService } from './cache'
 import { ChatDatabase } from './database'
 import { MediaManager } from './media'
 import { PluginLogger } from './logger'
-import { ContactCacheItem, SelfMessagePayload, SelfMessageRecord } from './types'
+import { ContactCacheItem, MessageRecord, SelfMessagePayload, SelfMessageRecord } from './types'
 
 interface ViteConsoleServer {
   config?: {
@@ -39,10 +40,11 @@ export function registerWeb(
   const webRoot = path.resolve(__dirname, '..', 'client', 'web', 'dist')
   const webSource = path.resolve(__dirname, '..', 'client', 'web', 'src')
   const webPublic = path.resolve(__dirname, '..', 'client', 'web', 'public')
-  const webIndex = path.resolve(__dirname, '..', 'client', 'web', 'index.html')
-  const uploadDir = path.resolve(ctx.baseDir, 'data', 'chat-patch', 'upload-media')
+  const webIndex = path.resolve(__dirname, '..', 'client', 'web', 'index.html')
+  const uploadDir = path.resolve(ctx.baseDir, 'data', 'chat-patch', 'upload-media')
+  const mediaDir = path.resolve(ctx.baseDir, 'data', 'chat-patch', 'media')
 
-  type WebContext = ParameterizedContext<DefaultState, DefaultContext>
+  type WebContext = Router.RouterContext<DefaultState, DefaultContext>
   const cacheType = (type: string) => type === 'user' ? 'friend' : type
 
   const getVite = (): ViteConsoleServer | undefined => {
@@ -105,6 +107,22 @@ export function registerWeb(
   const historyChannelCandidates = (channelId: string): string[] => {
     return channelId ? [channelId] : []
   }
+
+  const messageSortTime = (item: MessageRecord): number => {
+    const timestampMs = Number(item.timestampMs ?? 0)
+    if (timestampMs > 0) return timestampMs > 1e12 ? timestampMs : timestampMs * 1000
+    const timestamp = Number(item.timestamp ?? 0)
+    if (timestamp > 0) return timestamp * 1000
+    return Number(item.receivedAt ?? 0)
+  }
+
+  const selfMessageSortTime = (item: SelfMessageRecord): number => {
+    const timestampMs = Number(item.timestampMs ?? item.sentAt ?? 0)
+    if (timestampMs > 0) return timestampMs > 1e12 ? timestampMs : timestampMs * 1000
+    const timestamp = Number(item.timestamp ?? 0)
+    if (timestamp > 0) return timestamp * 1000
+    return item.sentAt
+  }
 
   const mimeToExt: Record<string, string> = {
     'audio/mpeg': '.mp3',
@@ -133,16 +151,21 @@ export function registerWeb(
   }
   const MAX_UPLOAD_BYTES = 500 * 1024 * 1024
 
-  ctx.server.post(`${config.basePath}/api/upload-media`, async (koa) => {
-    const requestBody = (koa.request as unknown as { body?: unknown }).body
-    const body = (requestBody ?? {}) as Record<string, unknown>
-    let source = String(body.dataUrl ?? body.data ?? '')
-    let name = String(body.name ?? '')
-    let mime = ''
-    let buffer: Buffer | null = null
-    if (source) {
-      if (source.startsWith('base64://')) source = source.slice(9)
-      const comma = source.indexOf('base64,')
+  ctx.server.post(`${config.basePath}/api/upload-media`, async (koa) => {
+    const requestBody = (koa.request as unknown as { body?: unknown }).body
+    const body = (requestBody ?? {}) as Record<string, unknown>
+    let source = String(body.dataUrl ?? body.data ?? '')
+    let name = String(body.name ?? '')
+    let mime = ''
+    let buffer: Buffer | null = null
+    let tempPath = ''
+    const rawHash = createHash('md5')
+    const cleanupTemp = async () => {
+      if (tempPath) await fs.unlink(tempPath).catch(() => undefined)
+    }
+    if (source) {
+      if (source.startsWith('base64://')) source = source.slice(9)
+      const comma = source.indexOf('base64,')
       const base64 = comma >= 0 ? source.slice(comma + 7) : source
       const normalizedBase64 = base64.replace(/-/g, '+').replace(/_/g, '/')
       try {
@@ -152,124 +175,152 @@ export function registerWeb(
         koa.body = { error: 'invalid media data' }
         return
       }
-      mime = source.startsWith('data:')
-        ? source.slice(5, source.indexOf(';')).toLowerCase()
-        : ''
-    } else {
-      const chunks: Buffer[] = []
-      for await (const chunk of koa.req as AsyncIterable<Buffer | string>) {
-        chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk))
-      }
-      buffer = Buffer.concat(chunks)
-      name = String(koa.query.name ?? koa.get('x-file-name') ?? '')
-      try {
-        name = decodeURIComponent(name)
+      mime = source.startsWith('data:')
+        ? source.slice(5, source.indexOf(';')).toLowerCase()
+        : ''
+    } else {
+      tempPath = path.join(uploadDir, `.upload-${Date.now()}-${Math.random().toString(36).slice(2)}.tmp`)
+      const write = createWriteStream(tempPath)
+      try {
+        for await (const chunk of koa.req as AsyncIterable<Buffer | string>) {
+          const part = Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk)
+          rawHash.update(part)
+          if (!write.write(part)) {
+            await new Promise<void>((resolve) => write.once('drain', resolve))
+          }
+        }
+        await new Promise<void>((resolve, reject) => {
+          write.once('error', reject)
+          write.end(() => resolve())
+        })
+      } catch (error) {
+        write.destroy()
+        await cleanupTemp()
+        throw error
+      }
+      name = String(koa.query.name ?? koa.get('x-file-name') ?? '')
+      try {
+        name = decodeURIComponent(name)
       } catch {
         // 保持原始名称即可
       }
-      mime = String(koa.get('content-type') ?? '').split(';')[0].trim().toLowerCase()
-    }
-    if (!buffer || !buffer.length) {
-      koa.status = 400
-      koa.body = { error: 'missing media data' }
-      return
-    }
-    if (buffer.length > MAX_UPLOAD_BYTES) {
-      koa.status = 413
-      koa.body = { error: 'media file too large' }
-      return
-    }
-    if (!buffer.length) {
-      koa.status = 400
-      koa.body = { error: 'invalid media data' }
-      return
-    }
-
-    await fs.mkdir(uploadDir, { recursive: true })
-    let detected: { ext?: string } | undefined
-    try {
-      const result = await FileType.fromBuffer(buffer)
-      detected = result ?? undefined
-    } catch {
-      // 部分音频/文件无法识别时继续用 MIME 或文件名兜底
-      detected = undefined
+      mime = String(koa.get('content-type') ?? '').split(';')[0].trim().toLowerCase()
+    }
+    const fileSize = buffer ? buffer.length : existsSync(tempPath) ? statSync(tempPath).size : 0
+    if (!fileSize) {
+      await cleanupTemp()
+      koa.status = 400
+      koa.body = { error: 'missing media data' }
+      return
+    }
+    if (fileSize > MAX_UPLOAD_BYTES) {
+      await cleanupTemp()
+      koa.status = 413
+      koa.body = { error: 'media file too large' }
+      return
+    }
+
+    await fs.mkdir(uploadDir, { recursive: true })
+    let detected: { ext?: string } | undefined
+    try {
+      const result = buffer
+        ? await FileType.fromBuffer(buffer)
+        : await FileType.fromStream(createReadStream(tempPath))
+      detected = result ?? undefined
+    } catch {
+      // 部分音频/文件无法识别时继续用 MIME 或文件名兜底
+      detected = undefined
     }
     const mimeExt = mimeToExt[mime] || ''
     const nameExt = path.extname(name).toLowerCase()
-    const ext = detected?.ext
-      ? toExtension(detected.ext)
-      : mimeExt || nameExt || '.bin'
-    const filename = `${createHash('md5').update(buffer).digest('hex')}${ext}`
-    const filePath = path.join(uploadDir, filename)
-    if (!existsSync(filePath)) {
-      await fs.writeFile(filePath, buffer)
-    }
-    koa.body = {
-      path: pathToFileURL(filePath).href,
+    const ext = detected?.ext
+      ? toExtension(detected.ext)
+      : mimeExt || nameExt || '.bin'
+    const digest = buffer ? createHash('md5').update(buffer).digest('hex') : rawHash.digest('hex')
+    const filename = `${digest}${ext}`
+    const filePath = path.join(uploadDir, filename)
+    try {
+      if (!existsSync(filePath)) {
+        if (buffer) {
+          await fs.writeFile(filePath, buffer)
+        } else {
+          await fs.rename(tempPath, filePath)
+          tempPath = ''
+        }
+      }
+    } catch (error) {
+      await cleanupTemp()
+      throw error
+    }
+    await cleanupTemp()
+    koa.body = {
+      path: pathToFileURL(filePath).href,
       localPath: filePath,
     }
   })
 
-  ctx.server.get(`${config.basePath}/api/media`, async (koa) => {
-    const fileName = path.basename(String(koa.query.file ?? koa.query.name ?? ''))
-    if (!fileName || fileName === '.' || fileName === '..') {
-      koa.status = 400
-      koa.body = { error: 'missing media file' }
-      return
-    }
-    const filePath = path.resolve(uploadDir, fileName)
-    const root = path.resolve(uploadDir)
-    if (filePath !== root && !filePath.startsWith(`${root}${path.sep}`)) {
-      koa.status = 400
-      koa.body = { error: 'invalid media file' }
-      return
-    }
-    if (!existsSync(filePath) || !statSync(filePath).isFile()) {
-      koa.status = 404
-      koa.body = { error: 'media file not found' }
-      return
-    }
-    const size = statSync(filePath).size
-    koa.set('Accept-Ranges', 'bytes')
-    if (fileName.toLowerCase().endsWith('.webm')) {
-      koa.type = 'audio/webm'
-    } else {
-      koa.type = fileName
-    }
-    const range = koa.headers.range
-    if (range) {
-      const match = /^bytes=(\d*)-(\d*)$/.exec(String(range))
-      if (!match) {
-        koa.status = 416
-        koa.set('Content-Range', `bytes */${size}`)
-        koa.body = ''
-        return
-      }
-      let start = 0
-      let end = size - 1
-      if (match[1] === '' && match[2]) {
-        start = Math.max(0, size - Number(match[2]))
-      } else if (match[1]) {
-        start = Number(match[1])
-        end = match[2] ? Number(match[2]) : end
-      }
-      if (!Number.isFinite(start) || !Number.isFinite(end) || start < 0 || start >= size) {
-        koa.status = 416
-        koa.set('Content-Range', `bytes */${size}`)
-        koa.body = ''
-        return
-      }
-      if (end >= size) end = size - 1
-      if (end < start) end = start
-      koa.status = 206
-      koa.set('Content-Range', `bytes ${start}-${end}/${size}`)
-      koa.set('Content-Length', String(end - start + 1))
-      koa.body = createReadStream(filePath, { start, end })
-      return
-    }
-    koa.set('Content-Length', String(size))
-    koa.body = createReadStream(filePath)
-  })
+  const sendLocalFile = async (koa: WebContext, filePath: string) => {
+    const size = statSync(filePath).size
+    const fileName = path.basename(filePath)
+    koa.set('Accept-Ranges', 'bytes')
+    if (fileName.toLowerCase().endsWith('.webm')) {
+      koa.type = 'audio/webm'
+    } else {
+      koa.type = fileName
+    }
+    const range = koa.headers.range
+    if (range) {
+      const match = /^bytes=(\d*)-(\d*)$/.exec(String(range))
+      if (!match) {
+        koa.status = 416
+        koa.set('Content-Range', `bytes */${size}`)
+        koa.body = ''
+        return
+      }
+      let start = 0
+      let end = size - 1
+      if (match[1] === '' && match[2]) {
+        start = Math.max(0, size - Number(match[2]))
+      } else if (match[1]) {
+        start = Number(match[1])
+        end = match[2] ? Number(match[2]) : end
+      }
+      if (!Number.isFinite(start) || !Number.isFinite(end) || start < 0 || start >= size) {
+        koa.status = 416
+        koa.set('Content-Range', `bytes */${size}`)
+        koa.body = ''
+        return
+      }
+      if (end >= size) end = size - 1
+      if (end < start) end = start
+      koa.status = 206
+      koa.set('Content-Range', `bytes ${start}-${end}/${size}`)
+      koa.set('Content-Length', String(end - start + 1))
+      koa.body = createReadStream(filePath, { start, end })
+      return
+    }
+    koa.set('Content-Length', String(size))
+    koa.body = createReadStream(filePath)
+  }
+
+  ctx.server.get(`${config.basePath}/api/media`, async (koa) => {
+    const fileName = path.basename(String(koa.query.file ?? koa.query.name ?? ''))
+    if (!fileName || fileName === '.' || fileName === '..') {
+      koa.status = 400
+      koa.body = { error: 'missing media file' }
+      return
+    }
+    for (const root of [uploadDir, mediaDir]) {
+      const filePath = path.resolve(root, fileName)
+      if (filePath !== root && filePath.startsWith(`${root}${path.sep}`)
+        && existsSync(filePath) && statSync(filePath).isFile()) {
+        await sendLocalFile(koa, filePath)
+        return
+      }
+    }
+    koa.status = 404
+    koa.body = { error: 'media file not found' }
+  })
 
   ctx.server.get(`${config.basePath}/api/cache/all`, async (koa) => {
     const entries = await database.getAllContacts()
@@ -303,8 +354,12 @@ export function registerWeb(
       koa.body = { error: 'missing history params' }
       return
     }
-    const parsedLimit = Number(koa.query.limit ?? config.maxMessagesPerChannel)
-    const limit = Number.isFinite(parsedLimit) && parsedLimit > 0 ? parsedLimit : config.maxMessagesPerChannel
+    const defaultLimit = Math.min(config.maxMessagesPerChannel, 100)
+    const parsedLimit = Number(koa.query.limit ?? defaultLimit)
+    const limit = Math.min(
+      Math.max(1, Number.isFinite(parsedLimit) && parsedLimit > 0 ? Math.floor(parsedLimit) : defaultLimit),
+      config.maxMessagesPerChannel,
+    )
     const beforeTime = Number(koa.query.beforeTime ?? 0)
     const queryMessages = async (cid: string) => {
       return Number.isFinite(beforeTime) && beforeTime > 0
@@ -326,15 +381,40 @@ export function registerWeb(
       selfMessages = await querySelfMessages(candidate)
       if (selfMessages.length) break
     }
-    // 统一按本地收到时间排序，避免平台时间不准导致顺序颠倒
-    messages.sort((a, b) => {
-      const timeA = Number(a?.receivedAt ?? a?.timestampMs ?? a?.timestamp ?? 0)
-      const timeB = Number(b?.receivedAt ?? b?.timestampMs ?? b?.timestamp ?? 0)
-      return timeA - timeB
-    })
-    selfMessages.sort((a, b) => a.sentAt - b.sentAt)
-    koa.body = { messages, selfMessages }
-  })
+    // 统一按本地收到时间排序，避免平台时间不准导致顺序颠倒
+    messages.sort((a, b) => {
+      const timeA = messageSortTime(a)
+      const timeB = messageSortTime(b)
+      return timeA - timeB
+    })
+    selfMessages.sort((a, b) => selfMessageSortTime(a) - selfMessageSortTime(b))
+    // 合并两条记录流后只返回 limit 条，保证前端分页/结束判断不会因自消息被放大
+    const combined: Array<{
+      kind: 'message' | 'self'
+      time: number
+      record: MessageRecord | SelfMessageRecord
+    }> = []
+    for (const item of messages) {
+      combined.push({
+        kind: 'message',
+        time: messageSortTime(item),
+        record: item,
+      })
+    }
+    for (const item of selfMessages) {
+      combined.push({ kind: 'self', time: selfMessageSortTime(item), record: item })
+    }
+    combined.sort((a, b) => a.time - b.time)
+    const capped = combined.slice(-limit)
+    koa.body = {
+      messages: capped
+        .filter((item) => item.kind === 'message')
+        .map((item) => item.record as MessageRecord),
+      selfMessages: capped
+        .filter((item) => item.kind === 'self')
+        .map((item) => item.record as SelfMessageRecord),
+    }
+  })
 
   ctx.server.get(`${config.basePath}/api/self-messages`, async (koa) => {
     const platform = String(koa.query.platform ?? '')
@@ -355,7 +435,7 @@ export function registerWeb(
         : await database.listSelfMessages(platform, selfId, candidate, limit)
       if (messages.length) break
     }
-    messages.sort((a, b) => a.sentAt - b.sentAt)
+    messages.sort((a, b) => selfMessageSortTime(a) - selfMessageSortTime(b))
     koa.body = { messages }
   })
 
@@ -377,9 +457,15 @@ export function registerWeb(
       message: Array.isArray(body.message) ? body.message as unknown[] : undefined,
       forwardId: typeof body.forwardId === 'string' ? body.forwardId : undefined,
       forwardContent: Array.isArray(body.forwardContent) ? body.forwardContent as unknown[] : undefined,
-      sentAt: typeof body.sentAt === 'number' && Number.isFinite(body.sentAt)
-        ? body.sentAt
-        : Date.now(),
+      sentAt: typeof body.sentAt === 'number' && Number.isFinite(body.sentAt)
+        ? body.sentAt
+        : Date.now(),
+      timestamp: typeof body.timestamp === 'number' && Number.isFinite(body.timestamp)
+        ? body.timestamp
+        : undefined,
+      timestampMs: typeof body.timestampMs === 'number' && Number.isFinite(body.timestampMs)
+        ? body.timestampMs
+        : undefined,
       sequence: typeof body.sequence === 'number' && Number.isFinite(body.sequence)
         ? body.sequence
         : 0,
@@ -395,7 +481,8 @@ export function registerWeb(
       koa.body = { error: 'missing self-message params' }
       return
     }
-    const record: SelfMessageRecord = {
+    const sentAt = payload.sentAt ?? Date.now()
+    const record: SelfMessageRecord = {
       id: payload.id || `web-${Date.now()}-${Math.random().toString(36).slice(2)}`,
       platform: payload.platform,
       selfId: payload.selfId,
@@ -408,7 +495,9 @@ export function registerWeb(
       message: payload.message,
       forwardId: payload.forwardId,
       forwardContent: payload.forwardContent,
-      sentAt: payload.sentAt ?? Date.now(),
+      sentAt,
+      timestamp: payload.timestamp ?? Math.floor(sentAt / 1000),
+      timestampMs: payload.timestampMs ?? sentAt,
       sequence: payload.sequence ?? 0,
       source: payload.source ?? 'webui',
       kind: payload.kind ?? 'text',
@@ -685,15 +774,20 @@ export function registerWeb(
   })
 
   const serveFile = async (koa: WebContext, fileName: string) => {
-    const fullPath = path.join(webRoot, fileName)
-    if (existsSync(fullPath) && statSync(fullPath).isFile()) {
-      await send(koa, path.relative(webRoot, fullPath), { root: webRoot })
-      return
-    }
-    koa.status = 200
-    koa.body = ''
-    await send(koa, 'index.html', { root: webRoot })
-  }
+    const fullPath = path.join(webRoot, fileName)
+    if (existsSync(fullPath) && statSync(fullPath).isFile()) {
+      const rootContext = koa as unknown as ParameterizedContext<DefaultState, DefaultContext>
+      await send(rootContext, path.relative(webRoot, fullPath), { root: webRoot })
+      return
+    }
+    koa.status = 200
+    koa.body = ''
+    await send(
+      koa as unknown as ParameterizedContext<DefaultState, DefaultContext>,
+      'index.html',
+      { root: webRoot },
+    )
+  }
 
   const serveDevWeb = async (koa: WebContext): Promise<boolean> => {
     const vite = getVite()