// @vitest-environment jsdom
import { afterEach, describe, expect, it, vi } from 'vitest'
import { cleanup, fireEvent, render, screen, waitFor } from '@testing-library/react'
import type {
  ConversationSnapshot, SessionId, SessionListState,
} from '@deepseek-ai/dsh-client-runtime/client'
import { createSnapshotStore } from '@deepseek-ai/dsh-client-runtime/client'
import { bindSnapshotSelector } from '@deepseek-ai/dsh-client-web-react'
import type { SnapshotSelectorHook } from '@deepseek-ai/dsh-client-ui-slots'
import type { PlanModeState } from '@deepseek-ai/dsh-client-connection/client'
import { PlanModeControl } from '../src/client/PlanModeControl.tsx'

afterEach(cleanup)

const SID = 's-plan' as SessionId

function setup(
  planMode: PlanModeState | null,
  setPlanMode = vi.fn(() => Promise.resolve<string | null>(null)),
  running = false,
) {
  const store = createSnapshotStore({ planMode, running })
  const useSession = bindSnapshotSelector(store) as unknown as SnapshotSelectorHook<ConversationSnapshot>
  const useSessions = (() => { throw new Error('unused') }) as unknown as SnapshotSelectorHook<SessionListState>
  const view = render(
    <PlanModeControl
      sessionId={SID}
      useSession={useSession}
      useSessions={useSessions}
      setPlanMode={setPlanMode}
    />,
  )
  return { store, setPlanMode, view }
}

describe('PlanModeControl', () => {
  it('hides an unavailable capability and renders committed modes', () => {
    const unavailable = setup(null)
    expect(unavailable.view.container.innerHTML).toBe('')
    cleanup()
    setup({ active: false })
    expect(screen.getByTitle('当前为默认模式')).toBeTruthy()
    const select = screen.getByRole('combobox', { name: '协作模式' }) as HTMLSelectElement
    expect(select.value).toBe('default')
    expect(document.getElementById(select.getAttribute('aria-describedby') ?? '')?.textContent)
      .toBe('当前为默认模式')
  })

  it('treats pending field presence as the target, including pending false', () => {
    setup({ active: false, pending: true })
    expect(screen.getByText('计划 · 待生效')).toBeTruthy()
    expect(screen.getByTitle(/当前为默认模式/)).toBeTruthy()
    const planSelect = screen.getByRole('combobox')
    expect(document.getElementById(planSelect.getAttribute('aria-describedby') ?? '')?.textContent)
      .toBe('当前为默认模式；计划模式将在下一次模型请求时生效')
    cleanup()
    setup({ active: true, pending: false })
    expect(screen.getByText('默认 · 待生效')).toBeTruthy()
    const defaultSelect = screen.getByRole('combobox') as HTMLSelectElement
    expect(defaultSelect.value).toBe('default')
    expect(document.getElementById(defaultSelect.getAttribute('aria-describedby') ?? '')?.textContent)
      .toBe('当前为计划模式；默认模式将在下一次模型请求时生效')
  })

  it('switches from the effective target and remains available while a turn runs', async () => {
    let resolve!: (value: string | null) => void
    const setPlanMode = vi.fn(() => new Promise<string | null>((done) => { resolve = done }))
    const { store } = setup({ active: false }, setPlanMode, true)
    const select = screen.getByRole('combobox', { name: '协作模式' }) as HTMLSelectElement
    expect(select.disabled).toBe(false)
    fireEvent.change(select, { target: { value: 'plan' } })
    expect(setPlanMode).toHaveBeenCalledWith(true)
    expect(select.disabled).toBe(true)

    store.set({ planMode: { active: false, pending: true }, running: true })
    resolve(null)
    await waitFor(() => {
      expect((screen.getByRole('combobox') as HTMLSelectElement).disabled).toBe(false)
    })
    expect(screen.getByText('计划 · 待生效')).toBeTruthy()
    fireEvent.change(screen.getByRole('combobox'), { target: { value: 'plan' } })
    expect(setPlanMode).toHaveBeenCalledTimes(1)
  })

  it('surfaces host and transport failures without changing the confirmed mode', async () => {
    const setPlanMode = vi.fn()
      .mockResolvedValueOnce('host said no')
      .mockRejectedValueOnce(new Error('network down'))
      .mockRejectedValueOnce('socket closed')
    setup({ active: false }, setPlanMode)
    fireEvent.change(screen.getByRole('combobox'), { target: { value: 'plan' } })
    expect((await screen.findByText('模式切换失败')).getAttribute('title')).toBe('host said no')
    expect(screen.getByTitle('当前为默认模式')).toBeTruthy()

    fireEvent.change(screen.getByRole('combobox'), { target: { value: 'plan' } })
    expect(await screen.findByTitle('network down')).toBeTruthy()
    expect((screen.getByRole('combobox') as HTMLSelectElement).disabled).toBe(false)

    fireEvent.change(screen.getByRole('combobox'), { target: { value: 'plan' } })
    expect(await screen.findByTitle('socket closed')).toBeTruthy()
  })

  it('ignores in-flight fulfillment and rejection after unmount', () => {
    let resolve!: (value: string | null) => void
    const successful = setup(
      { active: false },
      vi.fn(() => new Promise<string | null>((done) => { resolve = done })),
    )
    fireEvent.change(screen.getByRole('combobox'), { target: { value: 'plan' } })
    successful.view.unmount()
    expect(() => { resolve(null) }).not.toThrow()

    let reject!: (reason: unknown) => void
    const setPlanMode = vi.fn(() => new Promise<string | null>((_done, fail) => { reject = fail }))
    const { view } = setup({ active: false }, setPlanMode)
    fireEvent.change(screen.getByRole('combobox'), { target: { value: 'plan' } })
    view.unmount()
    expect(() => { reject(new Error('late')) }).not.toThrow()
  })
})