Memoh/apps/web/src/store/chat-list.ts

import { defineStore, storeToRefs } from 'pinia'
import { computed, reactive, ref, watch } from 'vue'
import { useRetryingStream } from '@/composables/useRetryingStream'
import { useUserStore } from '@/store/user'
import { useChatSelectionStore } from '@/store/chat-selection'
import {
  createSession,
  deleteSession as requestDeleteSession,
  fetchSessions,
  type Bot,
  type SessionSummary,
  type Message,
  type StreamEvent,
  type MessageStreamEvent,
  fetchBots,
  fetchMessages,
  extractMessageText,
  extractToolCalls,
  extractAllToolResults,
  extractMessageReasoning,
  sendLocalChannelMessage,
  streamLocalChannel,
  streamMessageEvents,
  connectWebSocket,
  type ChatAttachment,
  type ChatWebSocket,
} from '@/composables/api/useChat'

// ---- Message model (blocks-based, aligned with main branch) ----

export interface TextBlock {
  type: 'text'
  content: string
}

export interface ThinkingBlock {
  type: 'thinking'
  content: string
  done: boolean
}

export interface ToolCallBlock {
  type: 'tool_call'
  toolCallId: string
  toolName: string
  input: unknown
  result: unknown | null
  done: boolean
}

export interface AttachmentItem {
  type: string
  path?: string
  name?: string
  url?: string
  base64?: string
  content_hash?: string
  bot_id?: string
  mime?: string
  size?: number
  storage_key?: string
  [key: string]: unknown
}

export interface AttachmentBlock {
  type: 'attachment'
  attachments: AttachmentItem[]
}

export type ContentBlock = TextBlock | ThinkingBlock | ToolCallBlock | AttachmentBlock

export interface ChatMessage {
  id: string
  role: 'user' | 'assistant'
  blocks: ContentBlock[]
  timestamp: Date
  streaming: boolean
  platform?: string
  senderDisplayName?: string
  senderAvatarUrl?: string
  isSelf?: boolean
}

interface PendingAssistantStream {
  assistantMsg: ChatMessage
  textBlockIdx: number
  thinkingBlockIdx: number
  done: boolean
  resolve: () => void
  reject: (err: Error) => void
}

// ---- Store ----

export const useChatStore = defineStore('chat', () => {
  const selectionStore = useChatSelectionStore()
  const { currentBotId, sessionId } = storeToRefs(selectionStore)

  const messages = reactive<ChatMessage[]>([])
  const streamingSessionId = ref<string | null>(null)
  const streaming = computed(() => streamingSessionId.value !== null && streamingSessionId.value === sessionId.value)
  const sessions = ref<SessionSummary[]>([])
  const loading = ref(false)
  const loadingChats = ref(false)
  const loadingOlder = ref(false)
  const hasMoreOlder = ref(true)
  const initializing = ref(false)
  const bots = ref<Bot[]>([])
  const overrideModelId = ref<string>('')
  const overrideReasoningEffort = ref<string>('')

  let abortFn: (() => void) | null = null
  let messageEventsSince = ''
  let pendingAssistantStream: PendingAssistantStream | null = null
  const knownServerMessageIds = new Set<string>()
  const messageEventsStream = useRetryingStream()
  const localStream = useRetryingStream()
  let activeWs: ChatWebSocket | null = null

  const activeSession = computed(() =>
    sessions.value.find((s) => s.id === sessionId.value) ?? null,
  )

  const activeChatReadOnly = computed(() => false)

  watch(currentBotId, (newId) => {
    if (newId) {
      void initialize()
    } else {
      stopMessageEvents()
      stopLocalStream()
      stopWebSocket()
      rejectPendingAssistantStream(new Error('Bot stream stopped'))
      messageEventsSince = ''
      sessions.value = []
      sessionId.value = null
      replaceMessages([])
    }
  })

  const nextId = () => `${Date.now()}-${Math.floor(Math.random() * 1000)}`

  const isPendingBot = (bot: Bot | null | undefined) =>
    bot?.status === 'creating' || bot?.status === 'deleting'

  // ---- Message adapter: convert server Message to ChatMessage ----

  function mediaTypeFromMime(mime: string): string {
    const m = (mime || '').toLowerCase().trim()
    if (m.startsWith('image/')) return 'image'
    if (m.startsWith('audio/')) return 'audio'
    if (m.startsWith('video/')) return 'video'
    return 'file'
  }

  function buildAssetBlocks(raw: Message): AttachmentBlock[] {
    if (!raw.assets?.length) return []
    const items: Array<Record<string, unknown>> = raw.assets.map((a) => ({
      type: mediaTypeFromMime(a.mime),
      content_hash: a.content_hash,
      bot_id: raw.bot_id,
      mime: a.mime,
      size: a.size_bytes,
      storage_key: a.storage_key,
      name: a.name || undefined,
      metadata: a.metadata || undefined,
    }))
    return [{ type: 'attachment', attachments: items }]
  }

  function messageToChat(raw: Message): ChatMessage | null {
    if (raw.role !== 'user' && raw.role !== 'assistant') return null

    const text = extractMessageText(raw)
    const assetBlocks = buildAssetBlocks(raw)
    const reasoningTexts = extractMessageReasoning(raw)

    if (!text && assetBlocks.length === 0 && reasoningTexts.length === 0) return null

    const blocks: ContentBlock[] = []
    for (const r of reasoningTexts) {
      blocks.push({ type: 'thinking', content: r, done: true })
    }
    if (text) blocks.push({ type: 'text', content: text })
    blocks.push(...assetBlocks)

    const createdAt = raw.created_at ? new Date(raw.created_at) : new Date()
    const timestamp = Number.isNaN(createdAt.getTime()) ? new Date() : createdAt
    const platform = (raw.platform ?? '').trim().toLowerCase()
    const channelTag = platform && platform !== 'web' ? platform : undefined

    if (raw.role === 'user') {
      const isSelf = resolveIsSelf(raw)
      const senderName = (raw.sender_display_name ?? '').trim() || undefined
      const senderAvatar = (raw.sender_avatar_url ?? '').trim() || undefined
      return {
        id: raw.id || nextId(),
        role: 'user',
        blocks,
        timestamp,
        streaming: false,
        isSelf,
        ...(channelTag && { platform: channelTag }),
        ...(channelTag && { senderDisplayName: senderName, senderAvatarUrl: senderAvatar }),
      }
    }

    return {
      id: raw.id || nextId(),
      role: 'assistant',
      blocks,
      timestamp,
      streaming: false,
      ...(channelTag && { platform: channelTag }),
    }
  }

  function convertMessagesToChats(rows: Message[]): ChatMessage[] {
    const result: ChatMessage[] = []
    let pendingAssistant: ChatMessage | null = null
    const pendingToolCallMap = new Map<string, ToolCallBlock>()

    function flushPending() {
      if (!pendingAssistant) return
      for (const block of pendingAssistant.blocks) {
        if (block.type === 'tool_call' && !block.done) block.done = true
      }
      result.push(pendingAssistant)
      pendingAssistant = null
      pendingToolCallMap.clear()
    }

    function makeTimestamp(raw: Message): Date {
      const d = raw.created_at ? new Date(raw.created_at) : new Date()
      return Number.isNaN(d.getTime()) ? new Date() : d
    }

    for (const raw of rows) {
      if (raw.role === 'user') {
        flushPending()
        const chat = messageToChat(raw)
        if (chat) result.push(chat)
        continue
      }

      if (raw.role === 'assistant') {
        const toolCalls = extractToolCalls(raw)
        const text = extractMessageText(raw)
        const reasoningTexts = extractMessageReasoning(raw)

        if (toolCalls.length > 0) {
          if (!pendingAssistant) {
            const platform = (raw.platform ?? '').trim().toLowerCase()
            const channelTag = platform && platform !== 'web' ? platform : undefined
            pendingAssistant = {
              id: raw.id || nextId(),
              role: 'assistant',
              blocks: [],
              timestamp: makeTimestamp(raw),
              streaming: false,
              ...(channelTag && { platform: channelTag }),
            }
          }
          for (const r of reasoningTexts) {
            pendingAssistant.blocks.push({ type: 'thinking', content: r, done: true })
          }
          if (text) {
            pendingAssistant.blocks.push({ type: 'text', content: text })
          }
          for (const tc of toolCalls) {
            const block: ToolCallBlock = {
              type: 'tool_call',
              toolCallId: tc.id ?? '',
              toolName: tc.name,
              input: tc.input,
              result: null,
              done: false,
            }
            pendingAssistant.blocks.push(block)
            if (tc.id) pendingToolCallMap.set(tc.id, block)
          }
          pendingAssistant.blocks.push(...buildAssetBlocks(raw))
          continue
        }

        if (pendingAssistant && text) {
          for (const r of reasoningTexts) {
            pendingAssistant.blocks.push({ type: 'thinking', content: r, done: true })
          }
          pendingAssistant.blocks.push({ type: 'text', content: text })
          pendingAssistant.blocks.push(...buildAssetBlocks(raw))
          flushPending()
          continue
        }

        flushPending()
        const chat = messageToChat(raw)
        if (chat) result.push(chat)
        continue
      }

      if (raw.role === 'tool') {
        const results = extractAllToolResults(raw)
        for (const r of results) {
          if (r.toolCallId && pendingToolCallMap.has(r.toolCallId)) {
            const block = pendingToolCallMap.get(r.toolCallId)!
            block.result = r.output
            block.done = true
          }
        }
        continue
      }
    }

    flushPending()
    return result
  }

  function resolveIsSelf(raw: Message): boolean {
    const platform = (raw.platform ?? '').trim().toLowerCase()
    if (!platform || platform === 'web') return true
    const senderUserId = (raw.sender_user_id ?? '').trim()
    if (!senderUserId) return false
    const userStore = useUserStore()
    const currentUserId = (userStore.userInfo.id ?? '').trim()
    if (!currentUserId) return false
    return senderUserId === currentUserId
  }

  // ---- Abort ----

  function abort() {
    if (activeWs) {
      activeWs.abort()
    }
    abortFn?.()
    abortFn = null
    for (const msg of messages) {
      if (msg.streaming) msg.streaming = false
    }
    streamingSessionId.value = null
  }

  // ---- Message list management ----

  function replaceMessages(items: ChatMessage[], serverRowIds?: string[]) {
    messages.splice(0, messages.length, ...items)
    knownServerMessageIds.clear()
    for (const item of items) {
      const tid = String(item.id ?? '').trim()
      if (tid) knownServerMessageIds.add(tid)
    }
    if (serverRowIds) {
      for (const id of serverRowIds) {
        const tid = id.trim()
        if (tid) knownServerMessageIds.add(tid)
      }
    }
  }

  // ---- SSE real-time events ----

  function stopMessageEvents() {
    messageEventsStream.stop()
  }

  function stopLocalStream() {
    localStream.stop()
  }

  function stopWebSocket() {
    if (activeWs) {
      activeWs.close()
      activeWs = null
    }
  }

  function startWebSocket(targetBotId: string) {
    const bid = targetBotId.trim()
    stopWebSocket()
    if (!bid) return

    activeWs = connectWebSocket(
      bid,
      handleLocalStreamEvent,
      (e) => handleStreamEvent(bid, e),
    )
  }

  function pushAssistantBlock(session: PendingAssistantStream, block: ContentBlock): number {
    session.assistantMsg.blocks.push(block)
    return session.assistantMsg.blocks.length - 1
  }

  function appendAssistantError(session: PendingAssistantStream, errorMessage: string) {
    const message = errorMessage.trim()
    if (!message) return
    ensureAssistantTextBlock(session).content += `\n\n**Error:** ${message}`
  }

  function ensureAssistantTextBlock(session: PendingAssistantStream): TextBlock {
    if (session.textBlockIdx < 0 || session.assistantMsg.blocks[session.textBlockIdx]?.type !== 'text') {
      session.textBlockIdx = pushAssistantBlock(session, { type: 'text', content: '' })
    }
    return session.assistantMsg.blocks[session.textBlockIdx] as TextBlock
  }

  function resolveStreamErrorMessage(event: StreamEvent, fallback: string): string {
    if (typeof event.message === 'string') return event.message
    if (typeof event.error === 'string') return event.error
    return fallback
  }

  function resolvePendingAssistantStream() {
    if (!pendingAssistantStream || pendingAssistantStream.done) return
    const session = pendingAssistantStream
    session.done = true
    pendingAssistantStream = null
    session.resolve()
  }

  function rejectPendingAssistantStream(err: Error) {
    if (!pendingAssistantStream || pendingAssistantStream.done) return
    const session = pendingAssistantStream
    session.done = true
    pendingAssistantStream = null
    session.reject(err)
  }

  function handleLocalStreamEvent(event: StreamEvent) {
    const meta = event.metadata as Record<string, unknown> | undefined
    const sourceChannel = meta?.source_channel as string | undefined
    const isCrossChannel = !!sourceChannel

    // Cross-channel events (Telegram, Discord, etc.) don't carry session_id,
    // so we can't determine which session they belong to. Skip them here;
    // persisted messages will arrive through the SSE message events stream
    // with proper session_id filtering via appendRealtimeMessage.
    if (isCrossChannel) return

    const session = pendingAssistantStream
    if (!session || session.done) return

    const type = (event.type ?? '').toLowerCase()
    switch (type) {
      case 'text_start':
        session.textBlockIdx = pushAssistantBlock(session, { type: 'text', content: '' })
        break
      case 'text_delta':
        if (typeof event.delta === 'string') {
          ensureAssistantTextBlock(session).content += event.delta
        }
        break
      case 'text_end':
        session.textBlockIdx = -1
        break
      case 'reasoning_start':
        session.thinkingBlockIdx = pushAssistantBlock(session, { type: 'thinking', content: '', done: false })
        break
      case 'reasoning_delta':
        if (typeof event.delta === 'string') {
          if (session.thinkingBlockIdx < 0 || session.assistantMsg.blocks[session.thinkingBlockIdx]?.type !== 'thinking') {
            session.thinkingBlockIdx = pushAssistantBlock(session, { type: 'thinking', content: '', done: false })
          }
          ;(session.assistantMsg.blocks[session.thinkingBlockIdx] as ThinkingBlock).content += event.delta
        }
        break
      case 'reasoning_end':
        if (session.thinkingBlockIdx >= 0 && session.assistantMsg.blocks[session.thinkingBlockIdx]?.type === 'thinking') {
          const tb = session.assistantMsg.blocks[session.thinkingBlockIdx] as ThinkingBlock
          if (tb.content.trim() === '') {
            session.assistantMsg.blocks.splice(session.thinkingBlockIdx, 1)
          } else {
            tb.done = true
          }
        }
        session.thinkingBlockIdx = -1
        break
      case 'tool_call_start':
        pushAssistantBlock(session, {
          type: 'tool_call',
          toolCallId: (event.toolCallId as string) ?? '',
          toolName: (event.toolName as string) ?? 'unknown',
          input: event.input ?? null,
          result: null,
          done: false,
        })
        session.textBlockIdx = -1
        break
      case 'tool_call_end': {
        const callId = (event.toolCallId as string) ?? ''
        let matched = false
        if (callId) {
          for (let i = 0; i < session.assistantMsg.blocks.length; i++) {
            const block = session.assistantMsg.blocks[i]
            if (block && block.type === 'tool_call' && block.toolCallId === callId && !block.done) {
              block.result = event.result ?? null
              block.input = event.input ?? block.input
              block.done = true
              matched = true
              break
            }
          }
        }
        if (!matched) {
          for (let i = 0; i < session.assistantMsg.blocks.length; i++) {
            const block = session.assistantMsg.blocks[i]
            if (block && block.type === 'tool_call' && block.toolName === event.toolName && !block.done) {
              block.result = event.result ?? null
              block.input = event.input ?? block.input
              block.done = true
              break
            }
          }
        }
        break
      }
      case 'attachment_delta': {
        const items = event.attachments
        if (Array.isArray(items) && items.length > 0) {
          const lastBlock = session.assistantMsg.blocks[session.assistantMsg.blocks.length - 1]
          if (lastBlock && lastBlock.type === 'attachment') {
            lastBlock.attachments.push(...items)
          } else {
            pushAssistantBlock(session, { type: 'attachment', attachments: [...items] })
          }
        }
        break
      }
      case 'final':
        break
      case 'processing_started':
        if (session.assistantMsg.blocks.length === 0) {
          session.textBlockIdx = pushAssistantBlock(session, { type: 'text', content: '' })
        }
        break
      case 'processing_completed':
        resolvePendingAssistantStream()
        break
      case 'processing_failed': {
        const message = resolveStreamErrorMessage(event, 'processing failed')
        appendAssistantError(session, message)
        rejectPendingAssistantStream(new Error(message))
        break
      }
      case 'error': {
        const message = resolveStreamErrorMessage(event, 'stream error')
        appendAssistantError(session, message)
        rejectPendingAssistantStream(new Error(message))
        break
      }
      case 'agent_abort':
      case 'agent_end':
        resolvePendingAssistantStream()
        break
      case 'agent_start':
      default: {
        const fallback = extractFallbackText(event)
        if (fallback) {
          ensureAssistantTextBlock(session).content += fallback
        }
        break
      }
    }
  }

  function startLocalStream(targetBotId: string) {
    const bid = targetBotId.trim()
    stopLocalStream()
    if (!bid) return

    localStream.start(async (signal) => {
      await streamLocalChannel(bid, signal, handleLocalStreamEvent)
    })
  }

  function updateSince(createdAt?: string) {
    const v = (createdAt ?? '').trim()
    if (!v) return
    if (!messageEventsSince) { messageEventsSince = v; return }
    const cur = Date.parse(messageEventsSince)
    const next = Date.parse(v)
    if (!Number.isNaN(next) && (Number.isNaN(cur) || next > cur)) {
      messageEventsSince = v
    }
  }

  function updateSinceFromRows(rows: Message[]) {
    messageEventsSince = ''
    for (const row of rows) updateSince(row.created_at)
  }

  function hasMessageWithId(id: string) {
    const tid = id.trim()
    if (!tid) return false
    if (knownServerMessageIds.has(tid)) return true
    return messages.some((m) => String(m.id).trim() === tid)
  }

  function resolveMessagePlatform(raw: Message): string {
    const direct = (raw.platform ?? '').trim().toLowerCase()
    if (direct) return direct
    const fromMeta = raw.metadata?.platform
    return typeof fromMeta === 'string' ? fromMeta.trim().toLowerCase() : ''
  }

  function appendRealtimeMessage(raw: Message) {
    updateSince(raw.created_at)
    const msgSessionId = (raw.session_id ?? '').trim()
    if (msgSessionId && sessionId.value && msgSessionId !== sessionId.value) return
    const platform = resolveMessagePlatform(raw)
    if (platform === 'web') return
    const mid = String(raw.id ?? '').trim()
    if (mid && hasMessageWithId(mid)) return
    const item = messageToChat(raw)
    if (!item) return
    messages.push(item)
    messages.sort((a, b) => a.timestamp.getTime() - b.timestamp.getTime())
    if (sessionId.value) touchSession(sessionId.value)
  }

  function handleStreamEvent(targetBotId: string, event: MessageStreamEvent) {
    const eventType = (event.type ?? '').toLowerCase()
    const eBotId = (event.bot_id ?? '').trim()
    if (eBotId && eBotId !== targetBotId) return

    if (eventType === 'message_created') {
      const raw = event.message
      if (!raw) return
      const pBotId = (raw.bot_id ?? '').trim()
      if (pBotId && pBotId !== targetBotId) return
      appendRealtimeMessage(raw)
    } else if (eventType === 'session_title_updated') {
      const sid = (event.session_id ?? '').trim()
      const title = (event.title ?? '').trim()
      if (!sid || !title) return
      const target = sessions.value.find((s) => s.id === sid)
      if (target) target.title = title
    }
  }

  function startMessageEvents(targetBotId: string) {
    const bid = targetBotId.trim()
    stopMessageEvents()
    if (!bid) return

    messageEventsStream.start(async (signal) => {
      await streamMessageEvents(
        bid,
        signal,
        (e) => handleStreamEvent(bid, e),
        messageEventsSince || undefined,
      )
    })
  }

  // ---- Bot management ----

  async function ensureBot(): Promise<string | null> {
    try {
      const list = await fetchBots()
      bots.value = list
      if (!list.length) { currentBotId.value = null; return null }
      if (currentBotId.value) {
        const found = list.find((b) => b.id === currentBotId.value)
        if (found && !isPendingBot(found)) return currentBotId.value
      }
      const ready = list.find((b) => !isPendingBot(b))
      currentBotId.value = ready ? ready.id : list[0]!.id
      return currentBotId.value
    } catch (err) {
      console.error('Failed to fetch bots:', err)
      return currentBotId.value
    }
  }

  // ---- Pagination ----

  const PAGE_SIZE = 30

  async function loadMessages(botId: string, sid: string) {
    const rows = await fetchMessages(botId, sid, { limit: PAGE_SIZE })
    const items = convertMessagesToChats(rows)
    const serverRowIds = rows.map((r) => r.id).filter(Boolean)
    replaceMessages(items, serverRowIds)
    hasMoreOlder.value = true
    updateSinceFromRows(rows)
  }

  async function loadOlderMessages(): Promise<number> {
    const bid = currentBotId.value ?? ''
    const sid = sessionId.value ?? ''
    if (!bid || !sid || loadingOlder.value || !hasMoreOlder.value) return 0
    const first = messages[0]
    if (!first?.timestamp) return 0

    const before = first.timestamp.toISOString()
    loadingOlder.value = true
    try {
      const rows = await fetchMessages(bid, sid, { limit: PAGE_SIZE, before })
      const items = convertMessagesToChats(rows)
      if (rows.length < PAGE_SIZE) hasMoreOlder.value = false
      for (const r of rows) {
        const tid = (r.id ?? '').trim()
        if (tid) knownServerMessageIds.add(tid)
      }
      for (const item of items) {
        const tid = String(item.id ?? '').trim()
        if (tid) knownServerMessageIds.add(tid)
      }
      messages.unshift(...items)
      return items.length
    } finally {
      loadingOlder.value = false
    }
  }

  // ---- Session CRUD ----

  function touchSession(targetSessionId: string) {
    const idx = sessions.value.findIndex((s) => s.id === targetSessionId)
    if (idx < 0) return
    const [target] = sessions.value.splice(idx, 1)
    if (!target) return
    target.updated_at = new Date().toISOString()
    sessions.value.unshift(target)
  }

  async function ensureActiveSession() {
    if (sessionId.value) return
    const bid = currentBotId.value ?? await ensureBot()
    if (!bid) throw new Error('Bot not ready')
    const created = await createSession(bid)
    sessions.value = [created, ...sessions.value.filter((s) => s.id !== created.id)]
    sessionId.value = created.id
    replaceMessages([])
  }

  // ---- Initialize ----

  async function initialize() {
    if (initializing.value) return
    initializing.value = true
    loadingChats.value = true
    stopMessageEvents()
    stopLocalStream()
    stopWebSocket()
    try {
      const bid = await ensureBot()
      if (!bid) {
        messageEventsSince = ''
        sessions.value = []
        sessionId.value = null
        replaceMessages([])
        return
      }
      const visible = await fetchSessions(bid)
      sessions.value = visible
      if (!visible.length) {
        messageEventsSince = ''
        sessionId.value = null
        replaceMessages([])
        return
      }
      const activeSessionId = sessionId.value && visible.some((s) => s.id === sessionId.value)
        ? sessionId.value
        : visible[0]!.id
      sessionId.value = activeSessionId
      await loadMessages(bid, activeSessionId)
      startWebSocket(bid)
      startMessageEvents(bid)
      startLocalStream(bid)
    } finally {
      loadingChats.value = false
      initializing.value = false
    }
  }

  async function selectBot(targetBotId: string) {
    if (currentBotId.value === targetBotId) return
    abort()
    currentBotId.value = targetBotId
    sessionId.value = null
    await initialize()
  }

  async function selectSession(targetSessionId: string) {
    const sid = targetSessionId.trim()
    if (!sid || sid === sessionId.value) return
    sessionId.value = sid
    loadingChats.value = true
    try {
      const bid = currentBotId.value ?? ''
      if (!bid) throw new Error('Bot not selected')
      await loadMessages(bid, sid)
    } finally {
      loadingChats.value = false
    }
  }

  async function createNewSession() {
    const bid = await ensureBot()
    if (!bid) return
    sessionId.value = null
    replaceMessages([])
  }

  async function removeSession(targetSessionId: string) {
    const delId = targetSessionId.trim()
    if (!delId) return
    loadingChats.value = true
    try {
      const bid = currentBotId.value ?? ''
      if (!bid) throw new Error('Bot not selected')
      await requestDeleteSession(bid, delId)
      const remaining = sessions.value.filter((s) => s.id !== delId)
      sessions.value = remaining
      if (sessionId.value !== delId) return
      if (!remaining.length) {
        sessionId.value = null
        replaceMessages([])
        return
      }
      sessionId.value = remaining[0]!.id
      await loadMessages(bid, remaining[0]!.id)
    } finally {
      loadingChats.value = false
    }
  }

  // ---- Send message (blocks-based streaming) ----

  async function sendMessage(text: string, attachments?: ChatAttachment[]) {
    const trimmed = text.trim()
    if ((!trimmed && !attachments?.length) || streaming.value || !currentBotId.value) return

    loading.value = true

    try {
      await ensureActiveSession()

      const bid = currentBotId.value!
      const sid = sessionId.value!
      streamingSessionId.value = sid

      const userBlocks: ContentBlock[] = []
      if (trimmed) userBlocks.push({ type: 'text', content: trimmed })
      if (attachments?.length) {
        userBlocks.push({
          type: 'attachment',
          attachments: attachments.map((a) => ({
            type: a.type,
            name: a.name ?? '',
            mime: a.mime ?? '',
            url: a.base64,
          })),
        })
      }
      messages.push({
        id: nextId(),
        role: 'user',
        blocks: userBlocks,
        timestamp: new Date(),
        streaming: false,
      })

      messages.push({
        id: nextId(),
        role: 'assistant',
        blocks: [],
        timestamp: new Date(),
        streaming: true,
      })
      const assistantMsg = messages[messages.length - 1]!
      const completion = new Promise<void>((resolve, reject) => {
        pendingAssistantStream = {
          assistantMsg,
          textBlockIdx: -1,
          thinkingBlockIdx: -1,
          done: false,
          resolve,
          reject,
        }
      })

      abortFn = () => {
        const abortError = new Error('aborted')
        abortError.name = 'AbortError'
        if (activeWs) {
          activeWs.abort()
        }
        rejectPendingAssistantStream(abortError)
      }

      const modelId = overrideModelId.value || undefined
      const re = overrideReasoningEffort.value
      const reasoningEffort = (re && re !== 'off') ? re : undefined
      if (activeWs?.connected) {
        activeWs.send({ type: 'message', text: trimmed, session_id: sid, attachments, model_id: modelId, reasoning_effort: reasoningEffort })
      } else {
        await sendLocalChannelMessage(bid, trimmed, attachments, { modelId, reasoningEffort })
      }
      await completion

      assistantMsg.streaming = false
      streamingSessionId.value = null
      loading.value = false
      abortFn = null
      touchSession(sid)
    } catch (err) {
      const isAbort = err instanceof Error && err.name === 'AbortError'
      const reason = err instanceof Error ? err.message : 'Unknown error'
      const last = messages[messages.length - 1]
      if (!isAbort && last?.role === 'assistant' && last.streaming) {
        last.blocks = [{ type: 'text', content: `Failed to send message: ${reason}` }]
        last.streaming = false
      } else if (!isAbort) {
        messages.push({
          id: nextId(),
          role: 'assistant',
          blocks: [{ type: 'text', content: `Failed to send message: ${reason}` }],
          timestamp: new Date(),
          streaming: false,
        })
      }
      pendingAssistantStream = null
      streamingSessionId.value = null
      loading.value = false
      abortFn = null
    }
  }

  function clearMessages() {
    abort()
    replaceMessages([])
  }

  // Backward-compatible aliases
  const chats = sessions
  const chatId = sessionId

  return {
    messages,
    streaming,
    sessions,
    chats,
    chatId,
    sessionId,
    currentBotId,
    bots,
    activeSession,
    activeChatReadOnly,
    loading,
    loadingChats,
    loadingOlder,
    hasMoreOlder,
    initializing,
    overrideModelId,
    overrideReasoningEffort,
    initialize,
    selectBot,
    selectSession,
    selectChat: selectSession,
    createNewSession,
    createNewChat: createNewSession,
    removeSession,
    removeChat: removeSession,
    deleteChat: removeSession,
    sendMessage,
    clearMessages,
    loadOlderMessages,
    abort,
  }
})

function extractFallbackText(event: StreamEvent): string | null {
  if (typeof event.delta === 'string') return event.delta
  if (typeof (event as Record<string, unknown>).text === 'string') return (event as Record<string, unknown>).text as string
  if (typeof (event as Record<string, unknown>).content === 'string') return (event as Record<string, unknown>).content as string
  return null
}