Teknium
b1d34cf6e2
Some hosts (notably WSL) report a junk window size such as 131072 columns by 1 row. Both the Ink fork and our components only guard against 0/null/undefined/NaN (stdout.columns || 80), so a positive-but-absurd width sails through into createScreen(width*height), allocating tens to hundreds of MB per frame and tripping the TUI memory monitor's hard exit. Add clampStdoutDimensions(), installed in entry.tsx before ink.render: it patches process.stdout.columns/rows with clamping getters (cols 1-2000, rows 1-1000; out-of-range -> 80x24). One install point fixes the renderer, its resize handler, and every component read. Live resizes still propagate through the original descriptor, just clamped.
112 lines
3.9 KiB
TypeScript
112 lines
3.9 KiB
TypeScript
#!/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 '@hermes/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 { clampStdoutDimensions } from './lib/terminalDimensions.js'
|
|
import { resetTerminalModes } from './lib/terminalModes.js'
|
|
|
|
if (!process.stdin.isTTY) {
|
|
console.log('hermes-tui: no TTY')
|
|
process.exit(0)
|
|
}
|
|
|
|
// Some hosts (notably WSL) report bogus window sizes such as 131072x1. Clamp
|
|
// `process.stdout.columns`/`rows` at the source so the Ink renderer, its
|
|
// resize handler, and every component read see sane values. Must run before
|
|
// `ink.render` constructs the renderer.
|
|
clampStdoutDimensions()
|
|
|
|
// 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) =>
|
|
`hermes-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(`hermes-tui lifecycle ${scope}: ${message.slice(0, 2000)}\n`)
|
|
},
|
|
onSignal: signal => {
|
|
resetTerminalModes()
|
|
process.stderr.write(`hermes-tui lifecycle: received ${signal}\n`)
|
|
}
|
|
})
|
|
|
|
const stopMemoryMonitor = startMemoryMonitor({
|
|
onCritical: (snap, dump) => {
|
|
resetTerminalModes()
|
|
process.stderr.write(`hermes-tui lifecycle: memory critical exit heap=${formatBytes(snap.heapUsed)} rss=${formatBytes(snap.rss)}\n`)
|
|
process.stderr.write(dumpNotice(snap, dump))
|
|
process.stderr.write('hermes-tui: exiting to avoid OOM; restart to recover\n')
|
|
process.exit(137)
|
|
},
|
|
onHigh: (snap, dump) => process.stderr.write(dumpNotice(snap, dump))
|
|
})
|
|
|
|
if (process.env.HERMES_HEAPDUMP_ON_START === '1') {
|
|
void performHeapDump('manual')
|
|
}
|
|
|
|
process.on('beforeExit', () => stopMemoryMonitor())
|
|
|
|
const [ink, { App }, { logFrameEvent }, { trackFrame }] = await Promise.all([
|
|
import('@hermes/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)
|
|
}
|
|
})
|