@tina_summer/codepocket 2.0.0

package/bin/codepocket.js

@@ -0,0 +1,453 @@
+#!/usr/bin/env node
+
+/**
+ * CodePocket — 你的全能 AI 随身控制台
+ *
+ *   npx codepocket
+ *
+ * Powered by Local Claude Code.
+ */
+
+const { WebSocket } = require('ws')
+const { spawn, execSync } = require('child_process')
+const fs = require('fs')
+const path = require('path')
+const os = require('os')
+const readline = require('readline')
+const { startMcpServer, handlePhoneActionResponse: onPhoneActionResponse } = require('../src/mcp-server')
+const { runTestCli } = require('../src/mcp-client')
+
+// ── Config ──
+
+const CONFIG_DIR = path.join(os.homedir(), '.codepocket')
+const CONFIG_FILE = path.join(CONFIG_DIR, 'config.json')
+const SESSION_FILE = path.join(CONFIG_DIR, 'session.json')
+
+function loadConfig() {
+  try { return JSON.parse(fs.readFileSync(CONFIG_FILE, 'utf-8')) } catch { return {} }
+}
+
+function saveConfig(cfg) {
+  fs.mkdirSync(CONFIG_DIR, { recursive: true })
+  fs.writeFileSync(CONFIG_FILE, JSON.stringify(cfg, null, 2))
+}
+
+function injectMcpConfig(port) {
+  const settingsPath = path.join(os.homedir(), '.claude', 'settings.json')
+  let settings = {}
+  try { settings = JSON.parse(fs.readFileSync(settingsPath, 'utf-8')) } catch {}
+  if (!settings.mcpServers) settings.mcpServers = {}
+  if (settings.mcpServers['codepocket-phone']) return // already configured
+  settings.mcpServers['codepocket-phone'] = {
+    type: 'sse',
+    url: `http://127.0.0.1:${port}/sse`,
+  }
+  fs.mkdirSync(path.dirname(settingsPath), { recursive: true })
+  fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2))
+  console.log(`  [✓] MCP 配置已写入 ${settingsPath}`)
+}
+
+// ── Parse args ──
+
+const args = process.argv.slice(2)
+
+let hubUrl = process.env.CPKT_HUB_URL || process.env.HAPI_HUB_URL || 'http://127.0.0.1:3006'
+let relayUrl = process.env.RELAY_WS_URL || ''
+let relayKey = process.env.RELAY_KEY || ''
+let noHub = false
+
+// Handle 'connect' subcommand (connector mode: hub + relay + MCP)
+const CONNECT_ARGS = ['connect', 'c']
+
+// Handle 'test' subcommand (doesn't need hub/relay)
+if (args[0] === 'test') {
+  runTestCli(args.slice(1))
+  return
+}
+
+if (!CONNECT_ARGS.includes(args[0])) {
+  // Default or any other subcommand → pass through to hub binary (runs Claude)
+  const hubBin = findHubBin()
+  if (hubBin) {
+    const child = spawn(hubBin, args, {
+      stdio: 'inherit',
+      env: {
+        ...process.env,
+        HAPI_HOME: path.join(os.homedir(), '.codepocket', 'hub'),
+      },
+    })
+    child.on('exit', (code) => process.exit(code || 0))
+  } else {
+    console.error('[!] 找不到 Hub 二进制')
+    process.exit(1)
+  }
+  return
+}
+
+const connectArgs = args[0] === 'connect' || args[0] === 'c' ? args.slice(1) : args
+
+for (let i = 0; i < connectArgs.length; i++) {
+  const arg = connectArgs[i]
+  if (arg.startsWith('--hub=')) hubUrl = arg.slice(6).replace(/\/+$/, '')
+  if (arg === '--hub' && connectArgs[i + 1]) { hubUrl = connectArgs[++i].replace(/\/+$/, '') }
+  if (arg.startsWith('--relay=')) relayUrl = arg.slice(8)
+  if (arg === '--relay' && connectArgs[i + 1]) relayUrl = connectArgs[++i]
+  if (arg.startsWith('--key=')) relayKey = arg.slice(6)
+  if (arg === '--key' && connectArgs[i + 1]) relayKey = connectArgs[++i]
+  if (arg === '--no-hub') noHub = true
+}
+
+const saved = loadConfig()
+if (relayKey) {
+  saveConfig({ ...saved, relayKey, relayUrl: relayUrl || saved.relayUrl || 'wss://bkstory.cn/ws' })
+} else {
+  relayKey = saved.relayKey || ''
+}
+if (!relayUrl) relayUrl = saved.relayUrl || 'wss://bkstory.cn/ws'
+
+let jwt = ''
+let relayWs = null
+
+// ── Step 1: Start hub if needed ──
+
+function startHub() {
+  return new Promise((resolve, reject) => {
+    if (noHub) { resolve(); return }
+
+    // Check if hub already running
+    try {
+      const result = execSync(`curl -s -o /dev/null -w "%{http_code}" ${hubUrl}/ 2>/dev/null`).toString().trim()
+      if (result === '200') {
+        console.log('  [✓] Hub 已在运行')
+        resolve()
+        return
+      }
+    } catch {}
+
+    console.log('  启动 Hub...')
+    const hubBin = findHubBin()
+    if (!hubBin) {
+      console.error('  [!] 找不到 Hub 二进制')
+      reject(new Error('Hub binary not found'))
+      return
+    }
+
+    const hub = spawn(hubBin, ['hub', 'start'], {
+      stdio: ['ignore', 'pipe', 'pipe'],
+      detached: false,
+      env: {
+        ...process.env,
+        HAPI_HOME: path.join(os.homedir(), '.codepocket', 'hub'),
+      },
+    })
+
+    let started = false
+    hub.stdout.on('data', (data) => {
+      const text = data.toString()
+      if (!started) process.stdout.write('  ')
+      process.stdout.write(text)
+    })
+    hub.stderr.on('data', (data) => {
+      const text = data.toString()
+      if (text.includes('listening') || text.includes('ready') || text.includes('started')) {
+        if (!started) { started = true; resolve() }
+      }
+    })
+
+    // Wait up to 10s for hub
+    setTimeout(() => {
+      if (!started) {
+        // Try connecting
+        try {
+          execSync(`curl -s -o /dev/null -w "%{http_code}" ${hubUrl}/ 2>/dev/null`)
+          started = true
+          resolve()
+        } catch {
+          console.log('  [i] Hub 启动中，继续连接...')
+          resolve()
+        }
+      }
+    }, 8000)
+
+    hub.on('exit', (code) => {
+      if (!started && code !== 0) {
+        reject(new Error(`Hub exited with code ${code}`))
+      }
+    })
+  })
+}
+
+function findHubBin() {
+  // 1. Bundled binary in vendor/
+  const bundledBin = path.join(__dirname, '..', 'vendor', process.platform === 'win32' ? 'cpkt-hub.exe' : 'cpkt-hub')
+  if (fs.existsSync(bundledBin)) return bundledBin
+
+  // 2. Global install
+  try { const r = execSync('which cpkt-hub 2>/dev/null').toString().trim(); if (r) return r } catch {}
+
+  return null
+}
+
+// ── Step 2: Authenticate ──
+
+async function findHubToken() {
+  const hubHome = process.env.HAPI_HOME || path.join(os.homedir(), '.codepocket', 'hub')
+  const settingsPath = path.join(hubHome, 'settings.json')
+  try {
+    const settings = JSON.parse(fs.readFileSync(settingsPath, 'utf-8'))
+    if (settings.cliApiToken) return settings.cliApiToken
+  } catch {}
+  if (process.env.CLI_API_TOKEN) return process.env.CLI_API_TOKEN
+  throw new Error('找不到 Hub token')
+}
+
+async function authenticateHub(token) {
+  const res = await fetch(`${hubUrl}/api/auth`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({ accessToken: token }),
+  })
+  if (!res.ok) throw new Error(`Hub auth failed: ${res.status}`)
+  return (await res.json()).token
+}
+
+// ── Step 3: SSE stream ──
+
+async function startSSE() {
+  const url = `${hubUrl}/api/events?token=${encodeURIComponent(jwt)}&all=true`
+  try {
+    const res = await fetch(url, { headers: { Accept: 'text/event-stream' } })
+    if (!res.ok || !res.body) { setTimeout(startSSE, 5000); return }
+    console.log('  [✓] SSE 已连接，监听事件...')
+    const reader = res.body.getReader()
+    const decoder = new TextDecoder()
+    let buffer = ''
+    async function pump() {
+      while (true) {
+        const { done, value } = await reader.read()
+        if (done) break
+        buffer += decoder.decode(value, { stream: true })
+        const lines = buffer.split('\n')
+        buffer = lines.pop() || ''
+        for (const line of lines) {
+          if (line.startsWith('data:')) {
+            try {
+              const event = JSON.parse(line.slice(5).trim())
+              console.log(`[SSE] ${event.type || 'unknown'}`)
+              sendToRelay({ type: 'sse-event', event })
+            } catch {}
+          }
+        }
+      }
+      setTimeout(startSSE, 5000)
+    }
+    pump().catch(() => { setTimeout(startSSE, 5000) })
+  } catch { setTimeout(startSSE, 5000) }
+}
+
+// ── Step 4: Relay connection ──
+
+let relayPingInterval = null
+
+function connectRelay() {
+  let savedSession = null
+  try { savedSession = JSON.parse(fs.readFileSync(SESSION_FILE, 'utf-8')) } catch {}
+
+  // Clean up old connection
+  if (relayPingInterval) { clearInterval(relayPingInterval); relayPingInterval = null }
+  if (relayWs) {
+    try { relayWs.removeAllListeners() } catch {}
+    try { relayWs.close() } catch {}
+    relayWs = null
+  }
+
+  relayWs = new WebSocket(relayUrl)
+
+  relayWs.on('open', () => {
+    relayWs.send(JSON.stringify({
+      type: 'register',
+      key: relayKey,
+      sessionToken: savedSession?.sessionToken || undefined,
+    }))
+    // Heartbeat every 30s to detect stale connections
+    relayPingInterval = setInterval(() => {
+      if (relayWs?.readyState === WebSocket.OPEN) {
+        relayWs.ping()
+      }
+    }, 30000)
+  })
+
+  relayWs.on('message', (raw) => {
+    let msg
+    try { msg = JSON.parse(raw.toString()) } catch { return }
+    switch (msg.type) {
+      case 'registered': {
+        try {
+          fs.mkdirSync(path.dirname(SESSION_FILE), { recursive: true })
+          fs.writeFileSync(SESSION_FILE, JSON.stringify({ sessionToken: msg.sessionToken, hubUrl, relayUrl }))
+        } catch {}
+        if (msg.reconnected) {
+          console.log('\n  [✓] 已重新连接\n')
+        } else {
+          console.log('  [✓] 已连接中继，等待小程序...\n')
+        }
+        break
+      }
+      case 'client-connected': {
+        console.log('  [✓] 小程序已连接')
+        break
+      }
+      case 'rest-request': {
+        forwardToHub(msg.requestId, msg.method, msg.path, msg.body)
+        break
+      }
+      case 'auth-request': {
+        handleAuthRequest(msg.requestId, msg.accessToken)
+        break
+      }
+      case 'register-error': {
+        console.error(`  [!] ${msg.error}`)
+        break
+      }
+      case 'phone-action-response': {
+        onPhoneActionResponse(msg)
+        break
+      }
+    }
+  })
+
+  relayWs.on('close', () => {
+    if (relayPingInterval) { clearInterval(relayPingInterval); relayPingInterval = null }
+    relayWs = null
+    console.log('[relay] 连接断开，3秒后重连...')
+    setTimeout(connectRelay, 3000)
+  })
+  relayWs.on('error', (err) => {
+    console.error('[relay] 连接错误:', err.message)
+    relayWs?.close()
+  })
+}
+
+function sendToRelay(msg) {
+  if (relayWs?.readyState === WebSocket.OPEN) relayWs.send(JSON.stringify(msg))
+}
+
+async function doForwardToHub(requestId, method, p, body) {
+  const init = {
+    method,
+    headers: { 'Content-Type': 'application/json', ...(jwt ? { Authorization: `Bearer ${jwt}` } : {}) },
+  }
+  if (body && method !== 'GET') init.body = JSON.stringify(body)
+  return fetch(`${hubUrl}${p}`, init)
+}
+
+async function forwardToHub(requestId, method, p, body) {
+  try {
+    let res = await doForwardToHub(requestId, method, p, body)
+    // JWT expired — re-authenticate and retry
+    if (res.status === 401) {
+      console.log('[REST] JWT expired, re-authenticating...')
+      try {
+        const token = await findHubToken()
+        jwt = await authenticateHub(token)
+        res = await doForwardToHub(requestId, method, p, body)
+      } catch (authErr) {
+        console.error('[REST] Re-auth failed:', authErr.message)
+      }
+    }
+    const data = await res.json().catch(() => ({}))
+    sendToRelay({ type: 'rest-response', requestId, status: res.status, data })
+  } catch (err) {
+    console.error(`[REST] ${method} ${p} failed:`, err.message, err.cause?.message || '')
+    sendToRelay({ type: 'rest-response', requestId, status: 502, data: { error: err.message } })
+  }
+}
+
+async function handleAuthRequest(requestId, accessToken) {
+  try {
+    const res = await fetch(`${hubUrl}/api/auth`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ accessToken }),
+    })
+    sendToRelay({ type: 'auth-response', requestId, ok: res.ok, data: await res.json() })
+  } catch (err) {
+    sendToRelay({ type: 'auth-response', requestId, ok: false, data: { error: err.message } })
+  }
+}
+
+// ── Main ──
+
+async function main() {
+  console.log()
+  console.log('  ╔══════════════════════════════════════╗')
+  console.log('  ║  CodePocket                           ║')
+  console.log('  ╚══════════════════════════════════════╝')
+  console.log()
+
+  if (!relayKey) {
+    console.log('  [!] 未找到配置，请输入 Key')
+    console.log('  在 CodePocket 小程序中获取 Key →')
+    relayKey = await new Promise((resolve) => {
+      const rl = readline.createInterface({ input: process.stdin, output: process.stdout })
+      rl.question('  Key: ', (answer) => {
+        rl.close()
+        resolve(answer.trim())
+      })
+    })
+    if (!relayKey) {
+      console.error('')
+      console.error('  [!] Key 不能为空')
+      process.exit(1)
+    }
+    saveConfig({ ...loadConfig(), relayKey, relayUrl })
+    console.log('')
+    console.log('  [✓] 配置已保存')
+    console.log('')
+  }
+
+  console.log(`  Relay: ${relayUrl}`)
+  console.log(`  Key:   ${relayKey.slice(0, 8)}...`)
+  console.log()
+
+  // Step 1: Start hub
+  console.log('[1/3] 检查 Hub...')
+  try {
+    await startHub()
+  } catch (e) {
+    console.error(`  [!] ${e.message}`)
+    console.error('  你可以用 --no-hub 跳过自动启动')
+    process.exit(1)
+  }
+
+  // Step 1.5: Start MCP server
+  console.log('[1.5/3] 启动 MCP server...')
+  try {
+    const mcpPort = await startMcpServer(sendToRelay)
+    injectMcpConfig(mcpPort)
+  } catch (e) {
+    console.error(`  [!] MCP server 启动失败: ${e.message}`)
+    console.error('  手机工具功能将不可用')
+  }
+
+  // Step 2: Authenticate
+  console.log('[2/3] 认证...')
+  try {
+    const token = await findHubToken()
+    jwt = await authenticateHub(token)
+    console.log('  [✓] 认证成功')
+  } catch (e) {
+    console.error(`  [!] ${e.message}`)
+    process.exit(1)
+  }
+
+  // Step 3: Connect relay
+  console.log('[3/3] 连接中继...')
+  startSSE()
+  connectRelay()
+}
+
+main().catch((err) => {
+  console.error('[!] 启动失败:', err.message)
+  process.exit(1)
+})