import { strict as assert } from 'node:assert'
import { mkdir, mkdtemp, readFile, rm } from 'node:fs/promises'
import { tmpdir } from 'node:os'
import { dirname, join, resolve } from 'node:path'
import { setTimeout as delay } from 'node:timers/promises'
import { fileURLToPath } from 'node:url'
import { _electron as electron } from 'playwright'

const scriptDir = dirname(fileURLToPath(import.meta.url))
const packageRoot = resolve(scriptDir, '..')
const repoRoot = resolve(packageRoot, '../..')
const screenshotPath = String(process.env.OPENSQUILLA_DESKTOP_ONBOARDING_SCREENSHOT || '').trim()

async function waitFor(check, label, timeoutMs = 60_000) {
  const startedAt = Date.now()
  let lastError
  while (Date.now() - startedAt < timeoutMs) {
    try {
      const value = await check()
      if (value) return value
    } catch (error) {
      lastError = error
    }
    await delay(250)
  }
  const suffix = lastError ? ` Last error: ${lastError.message || lastError}` : ''
  throw new Error(`Timed out waiting for ${label}.${suffix}`)
}

async function setupWindow(app) {
  return await waitFor(async () => {
    for (const page of app.windows()) {
      if (page.isClosed()) continue
      await page.waitForLoadState('domcontentloaded', { timeout: 5_000 }).catch(() => {})
      const hasSetupForm = await page.locator('#setup-form').count().catch(() => 0)
      if (hasSetupForm > 0) return page
    }
    return null
  }, 'desktop onboarding window')
}

const userDataRoot = await mkdtemp(join(tmpdir(), 'opensquilla-electron-onboarding-test-'))
const userDataDir = join(userDataRoot, 'chromium-user-data')
const isolatedHome = join(userDataRoot, 'home')
await mkdir(isolatedHome, { recursive: true })
const app = await electron.launch({
  args: [
    '--use-mock-keychain',
    `--user-data-dir=${userDataDir}`,
    packageRoot,
  ],
  env: {
    ...process.env,
    HOME: isolatedHome,
    USERPROFILE: isolatedHome,
    OPENSQUILLA_DESKTOP_REPO_ROOT: repoRoot,
    OPENSQUILLA_DESKTOP_SECRET_STORAGE: 'plain',
    OPENSQUILLA_DESKTOP_GATEWAY_PORT: '18897',
    OPENSQUILLA_DESKTOP_DISABLE_AUTO_UPDATE: '1',
    OPENSQUILLA_DESKTOP_MOCK_UPDATE_VERSION: '',
    LANG: 'en_US.UTF-8',
    LC_ALL: 'en_US.UTF-8',
  },
})

try {
  const page = await setupWindow(app)

  await page.locator('#onboardingLocale').selectOption('zh-Hans')
  assert.equal(await page.evaluate(() => document.documentElement.lang), 'zh-Hans')
  assert.equal(await page.locator('[data-screen="0"] h2').innerText(), '选择设置深度')
  assert.equal(await page.title(), '设置 OpenSquilla')
  assert.doesNotMatch(await page.locator('[data-setup-mode="advanced"]').innerText(), /Smart Router mode/)
  await page.locator('[data-setup-mode="advanced"]').click()

  await page.locator('[data-screen="0"].active .next-button').click()
  await page.locator('[data-screen="1"].active').waitFor({ state: 'visible', timeout: 10_000 })
  assert.equal(await page.locator('[data-step-label="2"]').count(), 1, 'advanced setup should expose the routing-mode progress step')

  assert.equal(await page.locator('#provider').inputValue(), 'tokenrhythm')
  assert.equal(await page.locator('#baseUrl').inputValue(), 'https://tokenrhythm.studio/v1')
  assert.equal(await page.locator('#model').inputValue(), 'deepseek-v4-pro')
  assert.equal(await page.locator('#modelRoutingMode').inputValue(), 'squilla_router')
  assert.equal(await page.locator('#routerMode').inputValue(), 'recommended')

  const tokenRhythmFeature = page.locator('[data-provider-feature="tokenrhythm"]')
  assert.equal(await tokenRhythmFeature.count(), 1)
  assert.equal(await tokenRhythmFeature.locator('[data-tokenrhythm-title]').innerText(), '推荐使用 TokenRhythm')
  assert.equal(
    await tokenRhythmFeature.locator('[data-tokenrhythm-value]').innerText(),
    'TokenRhythm API 调用限时免费。',
  )
  assert.equal(
    await tokenRhythmFeature.locator('[data-tokenrhythm-registration]').innerText(),
    '活动期间，注册并获取 API Key，即可免费调用 DeepSeek、GLM、MiniMax、Kimi 等主流模型。',
  )
  const tokenRhythmCta = tokenRhythmFeature.locator('#tokenrhythmRegister')
  assert.equal(await tokenRhythmCta.innerText(), '注册并获取 API Key')
  assert.equal(await tokenRhythmCta.getAttribute('href'), 'https://tokenrhythm.studio/register')
  assert.equal(await tokenRhythmCta.getAttribute('target'), '_blank')
  assert.equal(await tokenRhythmCta.getAttribute('rel'), 'noopener noreferrer')
  assert.equal(
    await tokenRhythmCta.getAttribute('aria-label'),
    '注册并获取 API Key — TokenRhythm（在外部浏览器中打开）',
  )
  assert.equal(await tokenRhythmFeature.locator('img, svg, canvas').count(), 0)
  assert.equal(await tokenRhythmFeature.locator('[data-provider="tokenrhythm"]').getAttribute('aria-pressed'), 'true')

  const providerMoreToggle = page.locator('#providerMoreToggle')
  const providerMorePanel = page.locator('#providerMorePanel')
  assert.equal(await providerMoreToggle.getAttribute('aria-expanded'), 'false')
  assert.equal(await providerMoreToggle.getAttribute('aria-controls'), 'providerMorePanel')
  assert.equal(await providerMorePanel.isHidden(), true)

  await page.locator('#onboardingLocale').selectOption('en')
  assert.equal(await page.evaluate(() => document.documentElement.lang), 'en')
  assert.equal(await page.locator('[data-screen="1"] h2').innerText(), 'Connect a provider')
  assert.equal(await page.locator('#provider').inputValue(), 'tokenrhythm', 'locale changes should preserve the selected provider')
  assert.equal(await tokenRhythmFeature.locator('[data-tokenrhythm-title]').innerText(), 'Recommended: TokenRhythm')
  assert.equal(
    await tokenRhythmFeature.locator('[data-tokenrhythm-value]').innerText(),
    'TokenRhythm API calls are free for a limited time.',
  )
  assert.equal(
    await tokenRhythmFeature.locator('[data-tokenrhythm-registration]').innerText(),
    'During the promotion, register and get an API key to call DeepSeek, GLM, MiniMax, Kimi, and other leading models for free.',
  )
  assert.equal(await tokenRhythmCta.innerText(), 'Register and get an API key')
  assert.equal(
    await tokenRhythmCta.getAttribute('aria-label'),
    'Register and get an API key — TokenRhythm (opens in external browser)',
  )

  await providerMoreToggle.click()
  assert.equal(await providerMoreToggle.getAttribute('aria-expanded'), 'true')
  assert.equal(await providerMorePanel.isVisible(), true)
  const openRouterProvider = page.locator('#providerGrid [data-provider="openrouter"]')
  await openRouterProvider.click()
  assert.equal(await page.locator('#provider').inputValue(), 'openrouter')
  assert.equal(await openRouterProvider.getAttribute('aria-pressed'), 'true')
  assert.equal(await tokenRhythmFeature.locator('[data-provider="tokenrhythm"]').getAttribute('aria-pressed'), 'false')
  await page.locator('#onboardingLocale').selectOption('zh-Hans')
  assert.equal(await page.locator('#provider').inputValue(), 'openrouter', 'locale changes should preserve another provider selection')
  await page.locator('#onboardingLocale').selectOption('en')
  assert.equal(await page.locator('#provider').inputValue(), 'openrouter')
  await tokenRhythmFeature.locator('[data-provider="tokenrhythm"]').click()
  assert.equal(await page.locator('#provider').inputValue(), 'tokenrhythm', 'TokenRhythm should remain re-selectable')
  assert.equal(await page.locator('#modelRoutingMode').inputValue(), 'squilla_router')
  assert.equal(await tokenRhythmFeature.locator('[data-provider="tokenrhythm"]').getAttribute('aria-pressed'), 'true')
  await page.locator('#onboardingLocale').selectOption('zh-Hans')
  await page.locator('#apiKey').fill('synthetic-tokenrhythm-key')
  await page.locator('[data-screen="1"].active .next-button').click()
  await page.locator('[data-screen="2"].active').waitFor({ state: 'visible', timeout: 10_000 })
  await page.waitForTimeout(300)
  assert.equal(await page.locator('[data-screen="2"] h2').innerText(), '选择路由模式')
  assert.equal(await page.locator('[data-model-routing-mode="squilla_router"]').isEnabled(), true)
  assert.equal(await page.locator('[data-model-routing-mode="direct"]').isEnabled(), true)
  assert.equal(await page.locator('[data-model-routing-mode="llm_ensemble"]').isEnabled(), true)
  assert.equal(await page.locator('#modelRoutingMode').inputValue(), 'squilla_router')
  assert.match(
    await page.locator('[data-model-routing-mode="squilla_router"] small').innerText(),
    /此提供商现有的 Squilla Router 层级默认值/,
  )
  assert.match(
    await page.locator('[data-model-routing-mode="llm_ensemble"] small').innerText(),
    /当前提供商的 static B5 Ensemble/,
  )
  if (screenshotPath) {
    await mkdir(dirname(screenshotPath), { recursive: true })
    await page.screenshot({ path: screenshotPath })
  }
  await page.locator('[data-screen="2"].active .next-button').click()
  await page.locator('[data-screen="3"].active').waitFor({ state: 'visible', timeout: 10_000 })
  assert.equal(await page.locator('[data-screen="3"] h2').innerText(), '候选模型池')
  const tokenRhythmTierText = await page.locator('#tierBody').innerText()
  for (const modelId of ['deepseek-v4-flash', 'deepseek-v4-pro', 'kimi-k2.7-code', 'glm-5.2', 'kimi-k2.6']) {
    assert.match(tokenRhythmTierText, new RegExp(modelId.replaceAll('.', '\\.')))
  }
  await page.locator('[data-screen="3"].active .back-button').click()
  await page.locator('[data-screen="2"].active').waitFor({ state: 'visible', timeout: 5_000 })
  await page.locator('[data-model-routing-mode="direct"]').click()
  assert.equal(await page.locator('#modelRoutingMode').inputValue(), 'direct')
  assert.equal(await page.locator('#directModelRoute').inputValue(), 'deepseek-v4-pro')
  await page.locator('#directModelRoute').fill('glm-5.2')
  assert.equal(await page.locator('#model').inputValue(), 'glm-5.2')
  await page.locator('[data-model-routing-mode="llm_ensemble"]').click()
  assert.equal(await page.locator('#modelRoutingMode').inputValue(), 'llm_ensemble')
  await page.locator('[data-screen="2"].active .next-button').click()
  await page.locator('[data-screen="4"].active').waitFor({ state: 'visible', timeout: 10_000 })
  await page.locator('[data-screen="4"].active .back-button').click()
  await page.locator('[data-screen="2"].active').waitFor({ state: 'visible', timeout: 5_000 })
  await page.locator('[data-screen="2"].active .back-button').click()
  await page.locator('[data-screen="1"].active').waitFor({ state: 'visible', timeout: 5_000 })
  await page.locator('#onboardingLocale').selectOption('en')

  await page.locator('#providerGrid [data-provider="ollama"]').click()
  assert.equal(await page.locator('#modelRoutingMode').inputValue(), 'direct')
  assert.equal(await page.locator('#routerMode').inputValue(), 'disabled')
  assert.equal(await page.locator('#model').inputValue(), '', 'direct-only providers without a default model should not inherit the previous provider model')
  assert.equal(await page.locator('#endpointToggle').getAttribute('aria-expanded'), 'true', 'direct-only providers that need a model should open the endpoint panel')
  await page.locator('[data-screen="1"].active .next-button').click()
  await page.locator('[data-screen="2"].active').waitFor({ state: 'visible', timeout: 5_000 })
  assert.equal(await page.locator('[data-model-routing-mode="squilla_router"]').isDisabled(), true)
  assert.equal(await page.locator('[data-model-routing-mode="direct"]').isEnabled(), true)
  assert.equal(await page.locator('[data-model-routing-mode="llm_ensemble"]').isDisabled(), true)
  assert.equal(await page.locator('[data-step-label="3"]').isVisible(), false, 'route-excluded tier step should be hidden from the progress rail')
  await page.locator('[data-screen="2"].active .next-button').click()
  await page.locator('[data-screen="2"].active').waitFor({ state: 'visible', timeout: 5_000 })
  assert.match(await page.locator('#error').innerText(), /Direct model is required/)
  await page.locator('[data-screen="2"].active .back-button').click()
  await page.locator('[data-screen="1"].active').waitFor({ state: 'visible', timeout: 5_000 })

  await page.locator('#providerGrid [data-provider="openai"]').click()

  assert.equal(await page.locator('#provider').inputValue(), 'openai')
  assert.equal(await page.locator('#baseUrl').inputValue(), 'https://api.openai.com/v1')
  assert.equal(await page.locator('#model').inputValue(), 'gpt-5.4-mini')
  await page.locator('#providerGrid [data-provider="openai"].active').waitFor({ state: 'visible', timeout: 5_000 })
  const openAiHint = await page.locator('#providerHint').innerText()
  assert.match(openAiHint, /OpenAI-only tier profile/)
  assert.doesNotMatch(openAiHint, /OPENAI_API_KEY/)
  await page.locator('#apiKey').fill('test-openai-key')
  await page.locator('[data-screen="1"].active .next-button').click()
  await page.locator('[data-screen="2"].active').waitFor({ state: 'visible', timeout: 10_000 })
  assert.equal(await page.locator('#modelRoutingMode').inputValue(), 'squilla_router')
  assert.equal(await page.locator('[data-model-routing-mode="squilla_router"]').isEnabled(), true)
  assert.equal(await page.locator('[data-model-routing-mode="direct"]').isEnabled(), true)
  assert.equal(await page.locator('[data-model-routing-mode="llm_ensemble"]').isDisabled(), true)
  await page.locator('[data-screen="2"].active .next-button').click()
  await page.locator('[data-screen="3"].active').waitFor({ state: 'visible', timeout: 10_000 })
  assert.match(await page.locator('[data-screen="3"] .eyebrow').innerText(), /step 04/i)
  assert.equal(await page.locator('[data-screen="3"] h2').innerText(), 'Review tier models')

  await page.locator('[data-screen="3"].active .back-button').click()
  await page.locator('[data-screen="2"].active').waitFor({ state: 'visible', timeout: 5_000 })
  await page.locator('[data-screen="2"].active .back-button').click()
  await page.locator('[data-screen="1"].active').waitFor({ state: 'visible', timeout: 5_000 })
  await tokenRhythmFeature.locator('[data-provider="tokenrhythm"]').click()
  await page.locator('#apiKey').fill('synthetic-tokenrhythm-key')
  await page.locator('[data-screen="1"].active .next-button').click()
  await page.locator('[data-screen="2"].active').waitFor({ state: 'visible', timeout: 5_000 })
  await page.locator('[data-model-routing-mode="llm_ensemble"]').click()
  await page.locator('[data-screen="2"].active .next-button').click()
  await page.locator('[data-screen="4"].active').waitFor({ state: 'visible', timeout: 5_000 })
  await page.locator('#finish').click()
  const saved = await waitFor(async () => {
    const credential = JSON.parse(await readFile(join(userDataDir, 'desktop-credential.json'), 'utf8'))
    if (credential.modelRoutingMode !== 'llm_ensemble') return null
    const config = await readFile(join(userDataDir, 'opensquilla', 'config.toml'), 'utf8')
    return { credential, config }
  }, 'saved ensemble credential and config')
  const { credential, config } = saved
  assert.equal(credential.provider, 'tokenrhythm')
  assert.equal(credential.modelRoutingMode, 'llm_ensemble')
  assert.equal(credential.routerMode, 'recommended')
  assert.equal(credential.routerTiers.c0.model, 'deepseek-v4-flash')
  assert.equal(credential.routerTiers.c1.model, 'deepseek-v4-pro')
  assert.equal(credential.routerTiers.c2.model, 'kimi-k2.7-code')
  assert.equal(credential.routerTiers.c3.model, 'glm-5.2')
  assert.equal(credential.routerTiers.image_model.model, 'kimi-k2.6')
  assert.match(config, /\[squilla_router\]\nenabled = true/)
  assert.doesNotMatch(config, /tier_profile = "tokenrhythm"/)
  assert.match(config, /\[squilla_router\.tiers\.c0\]\nprovider = "tokenrhythm"\nmodel = "deepseek-v4-flash"/)
  assert.match(config, /\[squilla_router\.tiers\.c3\]\nprovider = "tokenrhythm"\nmodel = "glm-5\.2"/)
  assert.match(config, /\[llm_ensemble\]\nenabled = true\nselection_mode = "static_tokenrhythm_b5"/)

  console.log(JSON.stringify({
    ok: true,
    provider: credential.provider,
    modelRoutingMode: credential.modelRoutingMode,
    routerMode: credential.routerMode,
    model: credential.model,
    screenshotPath: screenshotPath || null,
  }, null, 2))
} finally {
  await app.close().catch(() => {})
  await rm(userDataRoot, { recursive: true, force: true }).catch(() => {})
}