nastech-tui 0.0.1

package/src/domain/providers.ts

@@ -0,0 +1,11 @@
+export const providerDisplayNames = (providers: readonly { name: string; slug: string }[]): string[] => {
+  const counts = new Map<string, number>()
+
+  for (const p of providers) {
+    counts.set(p.name, (counts.get(p.name) ?? 0) + 1)
+  }
+
+  return providers.map(p =>
+    (counts.get(p.name) ?? 0) > 1 && p.slug && p.slug !== p.name ? `${p.name} (${p.slug})` : p.name
+  )
+}

package/src/domain/roles.ts

@@ -0,0 +1,9 @@
+import type { Theme } from '../theme.js'
+import type { Role } from '../types.js'
+
+export const ROLE: Record<Role, (t: Theme) => { body: string; glyph: string; prefix: string }> = {
+  assistant: t => ({ body: t.color.text, glyph: t.brand.tool, prefix: t.color.border }),
+  system: t => ({ body: '', glyph: '·', prefix: t.color.muted }),
+  tool: t => ({ body: t.color.muted, glyph: '⚡', prefix: t.color.muted }),
+  user: t => ({ body: t.color.label, glyph: t.brand.prompt, prefix: t.color.label })
+}

package/src/domain/slash.ts

@@ -0,0 +1,10 @@
+/** Appended to `/model` args from the TUI picker for session scope; stripped in `session` slash before `config.set`. */
+export const TUI_SESSION_MODEL_FLAG = '--tui-session'
+
+export const looksLikeSlashCommand = (text: string) => /^\/[^\s/]*(?:\s|$)/.test(text)
+
+export const parseSlashCommand = (cmd: string) => {
+  const [name = '', ...rest] = cmd.slice(1).split(/\s+/)
+
+  return { arg: rest.join(' '), cmd, name: name.toLowerCase() }
+}

package/src/domain/usage.ts

@@ -0,0 +1,3 @@
+import type { Usage } from '../types.js'
+
+export const ZERO: Usage = { calls: 0, input: 0, output: 0, total: 0 }

package/src/domain/viewport.ts

@@ -0,0 +1,51 @@
+import type { Msg } from '../types.js'
+
+import { userDisplay } from './messages.js'
+
+const upperBound = (offsets: ArrayLike<number>, target: number) => {
+  let lo = 0
+  let hi = offsets.length
+
+  while (lo < hi) {
+    const mid = (lo + hi) >> 1
+
+    offsets[mid]! <= target ? (lo = mid + 1) : (hi = mid)
+  }
+
+  return lo
+}
+
+export const stickyPromptFromViewport = (
+  messages: readonly Msg[],
+  offsets: ArrayLike<number>,
+  top: number,
+  bottom: number,
+  sticky: boolean
+) => {
+  if (sticky || !messages.length) {
+    return ''
+  }
+
+  const first = Math.max(0, upperBound(offsets, top) - 1)
+  const last = Math.max(first, upperBound(offsets, bottom) - 1)
+  const visibleStart = Math.min(messages.length, first)
+  const visibleEnd = Math.min(messages.length - 1, last)
+
+  for (let i = visibleStart; i <= visibleEnd; i++) {
+    if (messages[i]?.role === 'user') {
+      return ''
+    }
+  }
+
+  for (let i = Math.min(messages.length - 1, visibleStart - 1); i >= 0; i--) {
+    if (messages[i]?.role !== 'user') {
+      continue
+    }
+
+    return (offsets[i + 1] ?? (offsets[i] ?? 0) + 1) <= top
+      ? userDisplay(messages[i]!.text.trim()).replace(/\s+/g, ' ').trim()
+      : ''
+  }
+
+  return ''
+}

package/src/entry.tsx

@@ -0,0 +1,104 @@
+#!/usr/bin/env -S node --max-old-space-size=8192 --expose-gc
+// Must be first import. If the user explicitly opts into truecolor, this
+// nudges chalk / supports-color before either package is initialized.
+import './lib/forceTruecolor.js'
+
+import type { FrameEvent } from '@nastechai/ink'
+
+import { TERMUX_TUI_MODE } from './config/env.js'
+import { GatewayClient } from './gatewayClient.js'
+import { setupGracefulExit } from './lib/gracefulExit.js'
+import { formatBytes, type HeapDumpResult, performHeapDump } from './lib/memory.js'
+import { type MemorySnapshot, startMemoryMonitor } from './lib/memoryMonitor.js'
+import { openExternalUrl } from './lib/openExternalUrl.js'
+import { resetTerminalModes } from './lib/terminalModes.js'
+
+if (!process.stdin.isTTY) {
+  console.log('nastech-tui: no TTY')
+  process.exit(0)
+}
+
+// Start from a clean slate. If a previous TUI crashed or was kill -9'd, the
+// terminal tab can still have mouse/focus/paste modes enabled.
+resetTerminalModes()
+
+// Desktop terminals benefit from a clean startup slate because the TUI usually
+// runs in AlternateScreen. On Termux we keep prior output intact so users can
+// review/copy earlier assistant replies after reopening the app.
+if (TERMUX_TUI_MODE) {
+  process.stdout.write('\n')
+} else {
+  process.stdout.write('\x1b[2J\x1b[H\x1b[3J')
+}
+
+const gw = new GatewayClient()
+
+gw.start()
+
+const dumpNotice = (snap: MemorySnapshot, dump: HeapDumpResult | null) =>
+  `nastech-tui: ${snap.level} memory (${formatBytes(snap.heapUsed)}) — auto heap dump → ${dump?.heapPath ?? '(failed)'}\n`
+
+setupGracefulExit({
+  cleanups: [
+    () => {
+      resetTerminalModes()
+
+      return gw.kill('graceful-exit-cleanup')
+    }
+  ],
+  onError: (scope, err) => {
+    const message = err instanceof Error ? `${err.name}: ${err.message}\n${err.stack ?? ''}` : String(err)
+
+    process.stderr.write(`nastech-tui lifecycle ${scope}: ${message.slice(0, 2000)}\n`)
+  },
+  onSignal: signal => {
+    resetTerminalModes()
+    process.stderr.write(`nastech-tui lifecycle: received ${signal}\n`)
+  }
+})
+
+const stopMemoryMonitor = startMemoryMonitor({
+  onCritical: (snap, dump) => {
+    resetTerminalModes()
+    process.stderr.write(`nastech-tui lifecycle: memory critical exit heap=${formatBytes(snap.heapUsed)} rss=${formatBytes(snap.rss)}\n`)
+    process.stderr.write(dumpNotice(snap, dump))
+    process.stderr.write('nastech-tui: exiting to avoid OOM; restart to recover\n')
+    process.exit(137)
+  },
+  onHigh: (snap, dump) => process.stderr.write(dumpNotice(snap, dump))
+})
+
+if (process.env.NASTECH_HEAPDUMP_ON_START === '1') {
+  void performHeapDump('manual')
+}
+
+process.on('beforeExit', () => stopMemoryMonitor())
+
+const [ink, { App }, { logFrameEvent }, { trackFrame }] = await Promise.all([
+  import('@nastechai/ink'),
+  import('./app.js'),
+  import('./lib/perfPane.js'),
+  import('./lib/fpsStore.js')
+])
+
+// Both consumers are undefined when their env flags are off; only attach
+// onFrame when at least one is on so ink skips timing in the default case.
+const onFrame =
+  logFrameEvent || trackFrame
+    ? (event: FrameEvent) => {
+        logFrameEvent?.(event)
+        trackFrame?.(event.durationMs)
+      }
+    : undefined
+
+ink.render(<App gw={gw} />, {
+  exitOnCtrlC: false,
+  onFrame,
+  // Open URLs in the user's default browser when a link cell is clicked.
+  // The TUI's mouse tracking captures click events before Terminal.app's
+  // own URL detection can fire, so without this hook clicks on `<Link>`
+  // do nothing in any terminal where mouseTracking is on.
+  onHyperlinkClick: url => {
+    openExternalUrl(url)
+  }
+})