@newbeebox/newbeebox-app-engine-cli 1.5.0 → 1.7.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@newbeebox/newbeebox-app-engine-cli",
-  "version": "1.5.0",
+  "version": "1.7.0",
   "description": "NewBee App Engine 命令行客户端（nae）——参数直达、默认人类可读（加 -o json 出 JSON），便于在终端使用与脚本/管道里解析。",
   "type": "module",
   "bin": {
@@ -24,7 +24,8 @@
   ],
   "homepage": "https://workshop.newbeebox.com/app_engine/documents/nae-cli/",
   "dependencies": {
-    "commander": "^12.1.0"
+    "commander": "^12.1.0",
+    "ws": "^8.21.0"
   },
   "publishConfig": {
     "access": "public"

package/src/forward.js

@@ -0,0 +1,275 @@
+// 转发：把内网资源转发到本地调试。两条命令同一套机制——本地 TCP server，每条连接桥到平台一条 WS 隧道、
+// 双向对拷字节；都要求当前登录态有效（启动先验活），都前台常驻、Ctrl-C 即停。
+//   - app 端口转发：隧道到指定应用的 Service:port。
+//   - LLM 转发：隧道到平台为本人开的 LLM 反代；写全局 NAE_LLM_GATEWAY(=本地址)/NAE_LLM_API_KEY(=本地占位令牌)，退出即清除。
+//     真实网关令牌只在集群内注入、不出本机；隧道随本命令退出即断，地址是 127.0.0.1——key 拿到别处连不上集群。
+import net from 'node:net'
+import { randomBytes } from 'node:crypto'
+import { spawnSync } from 'node:child_process'
+import { homedir, platform } from 'node:os'
+import { join } from 'node:path'
+import { readFileSync, writeFileSync } from 'node:fs'
+import WebSocket from 'ws'
+import { apiBase } from './config.js'
+import { CLI_VERSION } from './version.js'
+import { request } from './http.js'
+import * as out from './output.js'
+
+const isWin = platform() === 'win32'
+
+// ensureLoggedIn 启动前验活：登录态无效（无令牌/过期）直接抛 AuthError，由 run() 提示 nae login，绝不开隧道。
+async function ensureLoggedIn(cfg) {
+  await request(cfg, 'GET', '/me')
+}
+
+// --- app 端口转发 ---
+
+// forwardApp 在本地起 TCP server，把每条连接桥到平台隧道。前台常驻，Ctrl-C 退出。
+export async function forwardApp(cfg, appid, opts) {
+  await ensureLoggedIn(cfg)
+  const remotePort = opts.port // undefined → 平台取应用容器端口
+  const path = `/apps/${appid}/forward` + (remotePort ? `?port=${remotePort}` : '')
+  let active = 0
+  let up = 0
+  let down = 0
+
+  return new Promise((resolve, reject) => {
+    const server = net.createServer((sock) => {
+      active++
+      bridge(cfg, path, sock, {
+        onUp: (n) => (up += n),
+        onDown: (n) => (down += n),
+        onClose: () => (active = Math.max(0, active - 1)),
+      })
+    })
+
+    server.on('error', reject)
+    server.listen(opts.local || 0, '127.0.0.1', () => {
+      const port = server.address().port
+      const dst = remotePort ? `${appid}:${remotePort}` : `${appid}（容器端口）`
+      out.info(`本地 127.0.0.1:${port} → 应用 ${dst}`)
+      out.info('前台常驻转发中，Ctrl-C 退出。')
+    })
+
+    const timer = liveLine(() => `活动连接 ${active} | ↑ ${humanBytes(up)} | ↓ ${humanBytes(down)}`)
+    onShutdown(() => {
+      clearInterval(timer)
+      process.stderr.write('\n已停止转发。\n')
+      server.close(() => resolve())
+      process.exit(0)
+    })
+  })
+}
+
+// --- LLM 转发 ---
+
+// forwardLLM 在本地起 TCP server，每条连接桥到平台为本人开的 LLM 反代隧道。前台常驻，Ctrl-C 退出。
+// 写全局 env：NAE_LLM_GATEWAY=本地址、NAE_LLM_API_KEY=本地占位令牌；退出即清除。
+// 安全：真实网关令牌只在集群内注入、不出本机；隧道随本命令退出即断，地址 127.0.0.1——key 拿到别处连不上集群。
+export async function forwardLLM(cfg, opts) {
+  await ensureLoggedIn(cfg)
+  const localKey = 'naelocal_' + randomBytes(18).toString('hex') // 本地占位，真实鉴权在集群内完成
+  let active = 0
+  let up = 0
+  let down = 0
+
+  return new Promise((resolve, reject) => {
+    const server = net.createServer((sock) => {
+      active++
+      bridge(cfg, '/me/llm/forward', sock, {
+        onUp: (n) => (up += n),
+        onDown: (n) => (down += n),
+        onClose: () => (active = Math.max(0, active - 1)),
+      })
+    })
+
+    server.on('error', reject)
+    server.listen(opts.local || 0, '127.0.0.1', () => {
+      const port = server.address().port
+      const gateway = `http://127.0.0.1:${port}`
+      applyEnv({ NAE_LLM_GATEWAY: gateway, NAE_LLM_API_KEY: localKey })
+      out.info('本地 LLM 转发已就绪（前台常驻，Ctrl-C 退出即停止并清除变量）：')
+      out.info(`  本地网关  ${gateway}`)
+      out.info(`  NAE_LLM_GATEWAY = ${gateway}`)
+      out.info(`  NAE_LLM_API_KEY = ${localKey}`)
+      out.info('OpenAI 客户端 base=${NAE_LLM_GATEWAY}/v1；Anthropic SDK base=${NAE_LLM_GATEWAY}。')
+      out.info(reloadHint())
+    })
+
+    const timer = liveLine(() => `活动连接 ${active} | ↑ ${humanBytes(up)} | ↓ ${humanBytes(down)}`)
+    onShutdown(() => {
+      clearInterval(timer)
+      clearEnv(['NAE_LLM_GATEWAY', 'NAE_LLM_API_KEY'])
+      process.stderr.write('\n已停止 LLM 转发，全局环境变量已清除。\n')
+      server.close(() => resolve())
+      process.exit(0)
+    })
+  })
+}
+
+// --- 隧道 ---
+
+// bridge 一条本地 TCP 连接 ↔ 一条平台 WS 隧道：本地数据缓冲到隧道就绪后发，回包写回本地。
+function bridge(cfg, path, sock, cb) {
+  const ws = new WebSocket(wsBase(cfg) + path, {
+    headers: { Authorization: `Bearer ${cfg.token}`, 'X-NAE-CLI-Version': CLI_VERSION },
+  })
+  ws.binaryType = 'nodebuffer'
+
+  let open = false
+  const pending = []
+  sock.on('data', (d) => {
+    cb.onUp(d.length)
+    if (open) ws.send(d)
+    else pending.push(d)
+  })
+  ws.on('open', () => {
+    open = true
+    for (const d of pending) ws.send(d)
+    pending.length = 0
+  })
+  ws.on('message', (d) => {
+    cb.onDown(d.length)
+    sock.write(d)
+  })
+  ws.on('unexpected-response', (_req, res) => {
+    out.info(`转发连接失败：HTTP ${res.statusCode}（检查 appid / 登录态 / 应用是否在运行）`)
+    sock.destroy()
+  })
+  ws.on('error', () => sock.destroy())
+  ws.on('close', () => sock.destroy())
+  let closed = false
+  const finish = () => {
+    if (closed) return
+    closed = true
+    cb.onClose()
+  }
+  sock.on('close', () => {
+    ws.close()
+    finish()
+  })
+  sock.on('error', () => {
+    ws.close()
+    finish()
+  })
+}
+
+// onShutdown 在所有可捕获的退出信号上跑一次清理（Ctrl-C=SIGINT、Ctrl-Break=SIGBREAK、kill=SIGTERM、挂断=SIGHUP）。
+// 多注册一次只跑一次（once 兜底），尽量不给用户留下指向死端口的残留环境变量。SIGKILL/强杀无法捕获，认栽。
+function onShutdown(fn) {
+  let ran = false
+  const once = () => {
+    if (ran) return
+    ran = true
+    fn()
+  }
+  for (const sig of ['SIGINT', 'SIGTERM', 'SIGBREAK', 'SIGHUP']) {
+    try {
+      process.on(sig, once)
+    } catch {
+      // 个别平台不支持某信号，忽略。
+    }
+  }
+}
+
+// liveLine 实时流量行（覆盖式单行，写 stderr 不污染可能被重定向的 stdout）。
+function liveLine(text) {
+  return setInterval(() => {
+    process.stderr.write(`\r${text()}    `)
+  }, 1000)
+}
+
+// wsBase 把 apiBase 的 http(s):// 换成 ws(s)://。
+function wsBase(cfg) {
+  return apiBase(cfg).replace(/^http/, 'ws')
+}
+
+// humanBytes 字节数转人类可读。
+function humanBytes(n) {
+  const u = ['B', 'KB', 'MB', 'GB', 'TB']
+  let i = 0
+  let v = n
+  while (v >= 1024 && i < u.length - 1) {
+    v /= 1024
+    i++
+  }
+  return (i === 0 ? v : v.toFixed(1)) + u[i]
+}
+
+// --- 全局环境变量读写（跨平台）---
+
+// applyEnv 写用户级全局环境变量（新开终端生效）。
+// Windows 用 reg add 直写 HKCU\Environment——不用 setx：setx 会同步广播 WM_SETTINGCHANGE 给所有窗口，
+// 某个窗口不响应就把本进程的事件循环卡死，转发随之假死。reg add 不广播、瞬时返回，与 clearEnv 的 reg delete 对称。
+function applyEnv(pairs) {
+  if (isWin) {
+    for (const [k, v] of Object.entries(pairs)) {
+      const r = spawnSync('reg', ['add', 'HKCU\\Environment', '/v', k, '/t', 'REG_SZ', '/d', v, '/f'], { encoding: 'utf8' })
+      if (r.status !== 0) {
+        throw new Error(`写入环境变量 ${k} 失败：${(r.stderr || r.stdout || r.error?.message || '').trim()}`)
+      }
+    }
+  } else {
+    writeUnixBlock(pairs)
+  }
+}
+
+// clearEnv 清除用户级全局环境变量。Windows: reg delete；Unix: 清空 profile 托管块。
+function clearEnv(names) {
+  if (isWin) {
+    for (const k of names) {
+      spawnSync('reg', ['delete', 'HKCU\\Environment', '/v', k, '/f'], { encoding: 'utf8' })
+    }
+  } else {
+    writeUnixBlock({})
+  }
+}
+
+const MARK_BEGIN = '# >>> nae llm forward >>>'
+const MARK_END = '# <<< nae llm forward <<<'
+
+// writeUnixBlock 把托管块（marker 之间）重写为 pairs 的 export；pairs 空即删块。幂等。
+function writeUnixBlock(pairs) {
+  const file = unixProfile()
+  let content = ''
+  try {
+    content = readFileSync(file, 'utf8')
+  } catch {
+    // 无 profile，新建。
+  }
+  content = stripBlock(content)
+  const keys = Object.keys(pairs)
+  if (keys.length > 0) {
+    const lines = keys.map((k) => `export ${k}=${shellQuote(pairs[k])}`)
+    content = content.replace(/\n*$/, '\n') + MARK_BEGIN + '\n' + lines.join('\n') + '\n' + MARK_END + '\n'
+  }
+  writeFileSync(file, content)
+}
+
+// stripBlock 删掉 marker 间的旧托管块（含 marker 行）。
+function stripBlock(content) {
+  const b = content.indexOf(MARK_BEGIN)
+  const e = content.indexOf(MARK_END)
+  if (b === -1 || e === -1 || e < b) return content
+  return (content.slice(0, b) + content.slice(e + MARK_END.length)).replace(/\n{3,}/g, '\n\n')
+}
+
+// unixProfile 按 $SHELL 选 profile 文件。
+function unixProfile() {
+  const sh = process.env.SHELL || ''
+  if (sh.includes('zsh')) return join(homedir(), '.zshrc')
+  if (sh.includes('bash')) return join(homedir(), '.bashrc')
+  return join(homedir(), '.profile')
+}
+
+// shellQuote 单引号包裹，转义内部单引号。
+function shellQuote(s) {
+  return `'${String(s).replace(/'/g, `'\\''`)}'`
+}
+
+// reloadHint 提示新值在新终端才生效。
+function reloadHint() {
+  return isWin
+    ? '提示：已写入用户级环境变量，在另开的新终端里生效（已打开的终端不会自动刷新）。'
+    : `提示：已写入 ${unixProfile()}，在另开的新终端或 source 后生效。`
+}

package/src/index.js

@@ -9,6 +9,7 @@ import { CLI_VERSION } from './version.js'
 import { request, AuthError, NetworkError } from './http.js'
 import { runLogin } from './login.js'
 import { streamLogs } from './logs.js'
+import { forwardApp, forwardLLM } from './forward.js'
 import * as out from './output.js'
 
 const program = new Command()
@@ -426,6 +427,52 @@ program
     })
   )
 
+// --- 转发：把内网资源转发到本地调试 ---
+
+const forward = program
+  .command('forward')
+  .description('把内网资源转发到本地调试：app 端口转发 / 内置 LLM 网关转发')
+
+forward
+  .command('app <appid>')
+  .description('把应用监听端口转发到本地可用端口（前台常驻，类 kubectl port-forward，Ctrl-C 退出，实时显示流量）')
+  .option('--port <n>', '应用侧端口（缺省取应用容器端口）', (v) => parseInt(v, 10))
+  .option('--local <n>', '本地监听端口（缺省自动选空闲端口）', (v) => parseInt(v, 10))
+  .addHelpText(
+    'after',
+    `
+示例：
+  # 把应用容器端口转发到本地随机空闲端口
+  nae forward app myredis
+
+  # 指定应用侧端口 6379、本地固定 16379
+  nae forward app myredis --port 6379 --local 16379`
+  )
+  .action(
+    run(async (appid, opts) => {
+      await forwardApp(cfg(), appid, opts)
+    })
+  )
+
+forward
+  .command('llm')
+  .description('把内置 LLM 网关转发到本地（前台常驻，类 forward app）：写全局 env(NAE_LLM_GATEWAY/NAE_LLM_API_KEY)，Ctrl-C 退出即停止并清除')
+  .option('--local <n>', '本地监听端口（缺省自动选空闲端口）', (v) => parseInt(v, 10))
+  .addHelpText(
+    'after',
+    `
+示例：
+  nae forward llm                 # 前台常驻，另开终端里本地代码即可经环境变量调试内置网关
+
+说明：转发到 127.0.0.1，仅本命令运行期间有效，Ctrl-C 退出即失效；
+NAE_LLM_API_KEY 是本次会话的临时本地令牌，只能连本地端点，拿到别处连不上集群；需先登录。`
+  )
+  .action(
+    run(async (opts) => {
+      await forwardLLM(cfg(), opts)
+    })
+  )
+
 // --- 模板目录 ---
 
 program