@puku/puku-cli 1.8.21 → 1.8.23

package/bin/puku-cli

@@ -0,0 +1,101 @@
+#!/usr/bin/env node
+
+import { existsSync } from 'fs'
+import { join, dirname } from 'path'
+import { fileURLToPath, pathToFileURL } from 'url'
+import { spawnSync } from 'child_process'
+
+const __dirname = dirname(fileURLToPath(import.meta.url))
+const distPath = join(__dirname, '..', 'dist', 'cli.mjs')
+
+const HEAP_RELAUNCHED_ENV = 'PUKU_CLI_HEAP_RELAUNCHED'
+const DISABLE_HEAP_RELAUNCH_ENV = 'PUKU_CLI_DISABLE_HEAP_RELAUNCH'
+const HEAP_SIZE_ENV = 'PUKU_CLI_NODE_MAX_OLD_SPACE_SIZE_MB'
+const MAX_MEMORY_ENV = 'PUKU_CLI_MAX_MEMORY_MB'
+const DEFAULT_HEAP_SIZE_MB = 8192
+
+function hasNodeFlag(args, flag) {
+  return args.some(arg => arg === flag || arg.startsWith(`${flag}=`))
+}
+
+function hasNodeOptionFlag(flag) {
+  return hasNodeFlag(
+    [
+      ...process.execArgv,
+      ...(process.env.NODE_OPTIONS || '').split(/\s+/).filter(Boolean),
+    ],
+    flag,
+  )
+}
+
+function getHeapSizeMb() {
+  // --max-memory=N flag overrides env var
+  const maxMemArg = process.argv.find(a => a.startsWith('--max-memory='))
+  if (maxMemArg) {
+    const mb = Number.parseInt(maxMemArg.split('=')[1] || '0', 10)
+    if (Number.isSafeInteger(mb) && mb > 0) {
+      process.env[HEAP_SIZE_ENV] = String(mb)
+      process.env[MAX_MEMORY_ENV] = String(mb)
+      return mb
+    }
+  }
+
+  const raw = process.env[HEAP_SIZE_ENV]
+  if (!raw) return DEFAULT_HEAP_SIZE_MB
+  const parsed = Number.parseInt(raw, 10)
+  return Number.isSafeInteger(parsed) && parsed > 0 ? parsed : DEFAULT_HEAP_SIZE_MB
+}
+
+function relaunchWithLongSessionHeapIfNeeded() {
+  if (process.env[DISABLE_HEAP_RELAUNCH_ENV] === '1') return
+  if (process.env[HEAP_RELAUNCHED_ENV] === '1') return
+  const hasHeapLimit = hasNodeOptionFlag('--max-old-space-size')
+  const hasExplicitGc = hasNodeOptionFlag('--expose-gc')
+  if (hasHeapLimit && hasExplicitGc) return
+
+  const execArgv = [...process.execArgv]
+  if (!hasHeapLimit) {
+    execArgv.push(`--max-old-space-size=${getHeapSizeMb()}`)
+  }
+
+  // --expose-gc enables global.gc() for periodic forced GC in long sessions.
+  // NODE_OPTIONS cannot carry --expose-gc, so the launcher must inject it here.
+  if (!hasExplicitGc) {
+    execArgv.push('--expose-gc')
+  }
+
+  // Strip --max-memory before forwarding — it's a launcher-only arg that
+  // Commander in the built CLI would reject as unknown.
+  const childArgs = process.argv
+    .slice(2)
+    .filter(arg => !arg.startsWith('--max-memory=') && arg !== '--max-memory')
+
+  const result = spawnSync(
+    process.execPath,
+    [...execArgv, fileURLToPath(import.meta.url), ...childArgs],
+    {
+      stdio: 'inherit',
+      env: {
+        ...process.env,
+        [HEAP_RELAUNCHED_ENV]: '1',
+      },
+    },
+  )
+
+  if (result.error) {
+    console.error(
+      `puku-cli: failed to restart with long-session heap: ${result.error.message}`,
+    )
+    process.exit(1)
+  }
+
+  process.exit(result.status ?? 1)
+}
+
+if (existsSync(distPath)) {
+  relaunchWithLongSessionHeapIfNeeded()
+  await import(pathToFileURL(distPath).href)
+} else {
+  console.error('puku-cli: build not found. Please reinstall the package.')
+  process.exit(1)
+}