import type { SessionListRow } from '@hermes/shared/gateway-events'
import { describe, expect, it } from 'vitest'

import {
  canTypeOrchestratorPrompt,
  clampOrchestratorSelection,
  closeFallbackAfterClose,
  currentSessionSelectionIndex,
  draftModelArgFromPickerValue,
  draftModelDisplayLabel,
  draftTitleFromPrompt,
  isNewSessionRow,
  newSessionRowIndex,
  orchestratorGlobalHotkeyHint,
  orchestratorRowClickAction,
  orchestratorVisibleRowIndexes,
  relativeSessionAge,
  resumableHistory,
  sessionRowKindAt,
  sessionStatusLabel
} from '../components/activeSessionSwitcher.js'
import type { SessionActiveItem } from '../gatewayTypes.js'
import { applyLocale, messages, resetLocale } from '../i18n/runtime.js'

describe('session orchestrator helpers', () => {
  it('turns model picker values into session-scoped draft model args', () => {
    expect(draftModelArgFromPickerValue('kimi-k2.6 --provider ollama-cloud --tui-session')).toBe(
      'kimi-k2.6 --provider ollama-cloud --session'
    )
    expect(draftModelArgFromPickerValue('openai/gpt-5.5 --provider openai-codex --global')).toBe(
      'openai/gpt-5.5 --provider openai-codex --session'
    )
  })

  it('highlights the current live session when the picker opens', () => {
    const sessions = [
      { id: 'first', status: 'idle' },
      { id: 'second', status: 'working', current: true },
      { id: 'third', status: 'idle' }
    ] satisfies SessionActiveItem[]

    expect(currentSessionSelectionIndex(sessions, 'second')).toBe(1)
    expect(
      currentSessionSelectionIndex(
        [
          { id: 'first', status: 'idle' },
          { id: 'third', status: 'idle' }
        ],
        'third'
      )
    ).toBe(1)
    expect(currentSessionSelectionIndex(sessions, 'missing')).toBe(1)
    expect(currentSessionSelectionIndex([], 'missing')).toBe(0)
  })

  it('adds a selectable New row after the live sessions and gates prompt typing to it', () => {
    expect(newSessionRowIndex(0)).toBe(0)
    expect(newSessionRowIndex(3)).toBe(3)
    expect(clampOrchestratorSelection(-5, 2)).toBe(0)
    expect(clampOrchestratorSelection(99, 2)).toBe(2)
    expect(isNewSessionRow(0, 0)).toBe(true)
    expect(isNewSessionRow(1, 2)).toBe(false)
    expect(isNewSessionRow(2, 2)).toBe(true)
    expect(canTypeOrchestratorPrompt(1, 2)).toBe(false)
    expect(canTypeOrchestratorPrompt(2, 2)).toBe(true)
    expect(orchestratorVisibleRowIndexes(3, 3, 12)).toEqual([0, 1, 2, 3])
    expect(orchestratorVisibleRowIndexes(13, 13, 12)).toContain(13)
  })

  it('selects a safe fallback after closing the current live session', () => {
    const remaining = [
      { id: 'next', status: 'idle' },
      { id: 'other', status: 'working' }
    ] satisfies SessionActiveItem[]

    expect(closeFallbackAfterClose('other', 'current', remaining)).toEqual({ action: 'stay' })
    expect(closeFallbackAfterClose('current', 'current', remaining)).toEqual({ action: 'activate', sessionId: 'next' })
    expect(closeFallbackAfterClose('current', 'current', [])).toEqual({ action: 'new' })
  })

  it('shows clean draft model labels without picker flags or provider params', () => {
    expect(draftModelDisplayLabel('kimi-k2.6 --provider ollama-cloud --tui-session')).toBe('kimi-k2.6')
    expect(draftModelDisplayLabel('openai/gpt-5.5 --provider openai-codex --global')).toBe('gpt-5.5')
    expect(draftModelDisplayLabel('')).toBe(messages().pickers.session.currentOrDefault)
  })

  it('maps row clicks to existing-session activation or New-row focus', () => {
    const sessions = [
      { id: 'a', status: 'idle' },
      { id: 'b', status: 'idle' }
    ] satisfies SessionActiveItem[]

    expect(orchestratorRowClickAction(1, sessions)).toEqual({ action: 'activate', sessionId: 'b' })
    expect(orchestratorRowClickAction(2, sessions)).toEqual({ action: 'select-new' })
    expect(orchestratorRowClickAction(99, sessions)).toEqual({ action: 'select-new' })
  })

  it('builds a compact title from the orchestrator prompt', () => {
    expect(draftTitleFromPrompt('  Build the websocket orchestrator panel and make it robust.  ', 24)).toBe(
      'Build the websocket orc…'
    )
  })
})

describe('unified Sessions overlay helpers', () => {
  it('orders rows as [new][live…][history…]', () => {
    // 2 live sessions, any number of history rows after them.
    expect(sessionRowKindAt(0, 2)).toBe('new')
    expect(sessionRowKindAt(1, 2)).toBe('live')
    expect(sessionRowKindAt(2, 2)).toBe('live')
    expect(sessionRowKindAt(3, 2)).toBe('history')
    expect(sessionRowKindAt(9, 2)).toBe('history')
    // No live sessions: row 0 is new, everything after is history.
    expect(sessionRowKindAt(0, 0)).toBe('new')
    expect(sessionRowKindAt(1, 0)).toBe('history')
  })

  it('drops already-live sessions from the resumable history (dedupe by id)', () => {
    const history = [
      { id: 'a', message_count: 1, preview: '', started_at: 0, title: 'A' },
      { id: 'b', message_count: 2, preview: '', started_at: 0, title: 'B' },
      { id: 'c', message_count: 3, preview: '', started_at: 0, title: 'C' }
    ] satisfies SessionListRow[]

    const live = [{ id: 'b', status: 'idle' }] satisfies SessionActiveItem[]

    expect(resumableHistory(history, live).map(h => h.id)).toEqual(['a', 'c'])
    expect(resumableHistory(history, []).map(h => h.id)).toEqual(['a', 'b', 'c'])
  })

  it('renders relative session age, blank when unknown', () => {
    const nowSec = Math.floor(Date.now() / 1000)

    expect(relativeSessionAge(nowSec)).toBe(messages().pickers.session.age.today)
    expect(relativeSessionAge(nowSec - 36 * 3600)).toBe(messages().pickers.session.age.yesterday)
    expect(relativeSessionAge(nowSec - 3 * 86400)).toBe(messages().pickers.session.age.daysAgo(3))
    expect(relativeSessionAge(undefined)).toBe('')
    expect(relativeSessionAge(0)).toBe('')
  })
  it('resolves status labels and hint fragments against the active language at call time', () => {
    expect(sessionStatusLabel('working')).toBe(messages().pickers.session.status.working)
    expect(sessionStatusLabel('mystery')).toBe('mystery')
    expect(orchestratorGlobalHotkeyHint()).toBe('↑↓ move · Ctrl+N new · Ctrl+R refresh · Esc close')

    applyLocale('pl', {
      lang: 'pl',
      surface: 'tui',
      messages: { 'pickers.session.status.working': 'pracuje', 'pickers.session.hint.close': ' zamknij' }
    })

    try {
      expect(sessionStatusLabel('working')).toBe('pracuje')
      expect(orchestratorGlobalHotkeyHint()).toBe('↑↓ move · Ctrl+N new · Ctrl+R refresh · Esc zamknij')
    } finally {
      resetLocale()
    }

    expect(sessionStatusLabel('working')).toBe('working')
  })
})