@tina_summer/codepocket 2.0.0

package/bin/run.js

@@ -0,0 +1,289 @@
+#!/usr/bin/env node
+
+/**
+ * CodePocket Connector — 一行命令连接小程序
+ *
+ * 用法:
+ *   首次: npx codepocket --key YOUR_KEY
+ *   之后: npx codepocket
+ *
+ * 也可通过环境变量配置:
+ *   RELAY_KEY=xxx codepocket
+ */
+
+const { WebSocket } = require('ws')
+const fs = require('fs')
+const path = require('path')
+const os = require('os')
+const qrcode = require('qrcode-terminal')
+
+// ── Config ──
+
+const CONFIG_DIR = path.join(os.homedir(), '.codepocket', 'hub')
+const CONFIG_FILE = path.join(path.join(os.homedir(), '.codepocket'), 'config.json')
+const SESSION_FILE = path.join(CONFIG_DIR, 'wechat-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))
+}
+
+// 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 || ''
+
+for (let i = 0; i < args.length; i++) {
+  const arg = args[i]
+  if (arg.startsWith('--hub=')) hubUrl = arg.slice(6).replace(/\/+$/, '')
+  if (arg === '--hub' && args[i + 1]) { hubUrl = args[++i].replace(/\/+$/, '') }
+  if (arg.startsWith('--relay=')) relayUrl = arg.slice(8)
+  if (arg === '--relay' && args[i + 1]) { relayUrl = args[++i] }
+  if (arg.startsWith('--key=')) relayKey = arg.slice(6)
+  if (arg === '--key' && args[i + 1]) { relayKey = args[++i] }
+}
+
+// Load saved config, merge with args
+const saved = loadConfig()
+if (relayKey) {
+  // New key provided — save it
+  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
+let relaySessionToken = ''
+let sseResponse = null
+
+// ── Find CLI token ──
+
+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
+  console.error('[!] 找不到 Hub token，请先运行: cpkt')
+  process.exit(1)
+}
+
+// ── Authenticate ──
+
+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) {
+    const data = await res.json().catch(() => ({}))
+    throw new Error(`Hub auth failed: ${data.error || res.status}`)
+  }
+  return (await res.json()).token
+}
+
+// ── 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) {
+      console.error(`[SSE] failed: ${res.status}, retrying in 5s...`)
+      setTimeout(startSSE, 5000)
+      return
+    }
+    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:')) {
+            const raw = line.slice(5).trim()
+            if (!raw) continue
+            try { sendToRelay({ type: 'sse-event', event: JSON.parse(raw) }) } catch {}
+          }
+        }
+      }
+      console.error('[SSE] stream ended, reconnecting in 5s...')
+      setTimeout(startSSE, 5000)
+    }
+    sseResponse = res
+    pump().catch(() => { setTimeout(startSSE, 5000) })
+  } catch {
+    console.error('[SSE] connection error, retrying in 5s...')
+    setTimeout(startSSE, 5000)
+  }
+}
+
+// ── Relay connection ──
+
+function connectRelay() {
+  let savedSession = null
+  try { savedSession = JSON.parse(fs.readFileSync(SESSION_FILE, 'utf-8')) } catch {}
+
+  relayWs = new WebSocket(relayUrl)
+
+  relayWs.on('open', () => {
+    console.log('[Relay] connected')
+    relayWs.send(JSON.stringify({
+      type: 'register',
+      key: relayKey,
+      sessionToken: savedSession?.sessionToken || undefined,
+    }))
+  })
+
+  relayWs.on('message', (raw) => {
+    let msg
+    try { msg = JSON.parse(raw.toString()) } catch { return }
+    handleRelayMessage(msg)
+  })
+
+  relayWs.on('close', () => {
+    console.log('[Relay] disconnected, reconnecting in 3s...')
+    relayWs = null
+    setTimeout(connectRelay, 3000)
+  })
+
+  relayWs.on('error', () => { relayWs?.close() })
+}
+
+function handleRelayMessage(msg) {
+  switch (msg.type) {
+    case 'registered': {
+      relaySessionToken = msg.sessionToken
+      try {
+        fs.mkdirSync(path.dirname(SESSION_FILE), { recursive: true })
+        fs.writeFileSync(SESSION_FILE, JSON.stringify({ sessionToken: relaySessionToken, hubUrl, relayUrl }))
+      } catch {}
+      if (msg.reconnected) {
+        console.log('\n  [✓] 已重新连接\n')
+      } else {
+        const qrContent = `codepocket://pair?token=${msg.token}`
+        console.log()
+        qrcode.generate(qrContent, { small: true }, (qrStr) => {
+          console.log(qrStr)
+        })
+        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}`)
+      console.error('  请检查 key 是否正确')
+      console.error('  获取 key: curl -H "Authorization: Bearer ADMIN_KEY" http://127.0.0.1:3010/admin/keys')
+      break
+    }
+  }
+}
+
+function sendToRelay(msg) {
+  if (relayWs?.readyState === WebSocket.OPEN) {
+    relayWs.send(JSON.stringify(msg))
+  }
+}
+
+async function forwardToHub(requestId, method, p, body) {
+  try {
+    const init = {
+      method,
+      headers: {
+        'Content-Type': 'application/json',
+        ...(jwt ? { Authorization: `Bearer ${jwt}` } : {}),
+      },
+    }
+    if (body && method !== 'GET') init.body = JSON.stringify(body)
+    const res = await fetch(`${hubUrl}${p}`, init)
+    const data = await res.json().catch(() => ({}))
+    sendToRelay({ type: 'rest-response', requestId, status: res.status, data })
+  } catch (err) {
+    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 }),
+    })
+    const data = await res.json()
+    sendToRelay({ type: 'auth-response', requestId, ok: res.ok, data })
+  } catch (err) {
+    sendToRelay({ type: 'auth-response', requestId, ok: false, data: { error: err.message } })
+  }
+}
+
+// ── Main ──
+
+async function main() {
+  console.log('╔══════════════════════════════════════╗')
+  console.log('║    CodePocket Connect                 ║')
+  console.log('╚══════════════════════════════════════╝')
+  console.log()
+
+  if (!relayKey) {
+    console.error('[!] 需要 relay 密钥（首次使用只需设置一次）')
+    console.error()
+    console.error('  用法:')
+    console.error('    codepocket --key YOUR_KEY')
+    console.error()
+    console.error('  获取 key:')
+    console.error('    curl -H "Authorization: Bearer ADMIN_KEY" http://YOUR_SERVER:3010/admin/keys')
+    console.error()
+    console.error('  或设置环境变量:')
+    console.error('    export RELAY_KEY=xxx')
+    console.error()
+    process.exit(1)
+  }
+
+  console.log(`  Hub:   ${hubUrl}`)
+  console.log(`  Relay: ${relayUrl}`)
+  console.log(`  Key:   ${relayKey.slice(0, 8)}...`)
+  console.log()
+
+  console.log('[1/3] 连接 Hub...')
+  const token = await findHubToken()
+
+  console.log('[2/3] 认证中...')
+  jwt = await authenticateHub(token)
+  console.log('[✓] Hub 认证成功')
+
+  console.log('[3/3] 连接中继服务器...')
+  startSSE()
+  connectRelay()
+}
+
+main().catch((err) => {
+  console.error('[!] 启动失败:', err.message)
+  process.exit(1)
+})

package/package.json

@@ -0,0 +1,31 @@
+{
+  "name": "@tina_summer/codepocket",
+  "version": "2.0.0",
+  "description": "CodePocket — 你的全能 AI 随身控制台",
+  "bin": {
+    "codepocket": "bin/codepocket.js",
+    "cpkt": "bin/codepocket.js"
+  },
+  "files": [
+    "bin",
+    "src",
+    "vendor"
+  ],
+  "scripts": {
+    "start": "node bin/codepocket.js"
+  },
+  "dependencies": {
+    "qrcode-terminal": "^0.12.0",
+    "ws": "^8.18.0"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "keywords": [
+    "codepocket",
+    "claude",
+    "wechat",
+    "ai"
+  ],
+  "license": "MIT"
+}

package/src/index.ts

@@ -0,0 +1,304 @@
+/**
+ * CodePocket Connector — 在用户机器上运行，桥接本地 Hapi Hub 和远程 Relay
+ *
+ * 用法:
+ *   bun run src/index.ts                          # 自动发现本地 hub
+ *   bun run src/index.ts --hub http://localhost:3006 --relay wss://relay.example.com/ws
+ *
+ * 工作流程:
+ *   1. 连接本地 Hapi Hub REST API，获取 CLI_API_TOKEN
+ *   2. 用 token 调用 /api/auth 获取 JWT
+ *   3. 开启 SSE 流，接收实时事件
+ *   4. 连接远程 Relay WebSocket
+ *   5. 注册并获取配对码，显示给用户
+ *   6. 双向转发: SSE→Relay, Relay REST→Hub REST
+ */
+
+const HUB_URL = (process.env.CPKT_HUB_URL || process.env.HAPI_HUB_URL || 'http://127.0.0.1:3006').replace(/\/+$/, '')
+const RELAY_WS = process.env.RELAY_WS_URL || 'ws://127.0.0.1:3010/ws'
+const HUB_HOME = process.env.HAPI_HOME || `${process.env.HOME}/.codepocket/hub`
+const SESSION_FILE = `${HUB_HOME}/wechat-session.json`
+
+let hubToken = ''       // CLI_API_TOKEN
+let jwt = ''            // JWT from hub
+let relayWs: any = null
+let relaySessionToken = ''
+let sseController: AbortController | null = null
+
+// --- Parse args ---
+for (const arg of process.argv.slice(2)) {
+  if (arg.startsWith('--hub=')) process.env.HAPI_HUB_URL = arg.slice(6)
+  if (arg.startsWith('--relay=')) process.env.RELAY_WS_URL = arg.slice(8)
+}
+
+const hubUrl = (process.env.HAPI_HUB_URL || HUB_URL).replace(/\/+$/, '')
+const relayUrl = process.env.RELAY_WS_URL || RELAY_WS
+
+// --- Step 1: Find CLI token ---
+
+async function findHubToken(): Promise<string> {
+  // Try settings file
+  const settingsPath = `${HUB_HOME}/settings.json`
+  try {
+    const settings = await Bun.file(settingsPath).json()
+    if (settings.cliApiToken) return settings.cliApiToken
+  } catch {}
+
+  // Try env
+  if (process.env.CLI_API_TOKEN) return process.env.CLI_API_TOKEN
+
+  console.error('[!] 无法找到 CLI_API_TOKEN')
+  console.error('    请确保 Hub 正在运行，或设置 CLI_API_TOKEN 环境变量')
+  process.exit(1)
+}
+
+// --- Step 2: Authenticate with hub ---
+
+async function authenticateHub(token: string): Promise<string> {
+  const res = await fetch(`${hubUrl}/api/auth`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({ accessToken: token }),
+  })
+
+  if (!res.ok) {
+    const data = await res.json().catch(() => ({}))
+    throw new Error(`Hub auth failed: ${data.error || res.status}`)
+  }
+
+  const data = await res.json()
+  return data.token
+}
+
+// --- Step 3: Start SSE from hub ---
+
+function startSSE() {
+  sseController?.abort()
+  sseController = new AbortController()
+
+  const url = `${hubUrl}/api/events?token=${encodeURIComponent(jwt)}&all=true`
+
+  ;(async () => {
+    try {
+      const res = await fetch(url, {
+        signal: sseController!.signal,
+        headers: { Accept: 'text/event-stream' },
+      })
+
+      if (!res.ok || !res.body) {
+        console.error(`[SSE] connection failed: ${res.status}`)
+        return
+      }
+
+      const reader = res.body.getReader()
+      const decoder = new TextDecoder()
+      let buffer = ''
+
+      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:')) {
+            const raw = line.slice(5).trim()
+            if (!raw) continue
+            try {
+              const event = JSON.parse(raw)
+              forwardToRelay({ type: 'sse-event', event })
+            } catch {}
+          }
+        }
+      }
+    } catch (err: any) {
+      if (err.name !== 'AbortError') {
+        console.error('[SSE] stream ended, reconnecting in 5s...')
+        setTimeout(startSSE, 5000)
+      }
+    }
+  })()
+}
+
+// --- Step 4: Connect to relay ---
+
+function connectRelay() {
+  try {
+    relayWs = new WebSocket(relayUrl)
+  } catch (err) {
+    console.error(`[Relay] connect failed, retrying in 5s...`)
+    setTimeout(connectRelay, 5000)
+    return
+  }
+
+  relayWs.onopen = () => {
+    console.log('[Relay] connected')
+
+    // Try to reconnect with existing session
+    let savedSession: any = null
+    try { savedSession = JSON.parse(require('fs').readFileSync(SESSION_FILE, 'utf-8')) } catch {}
+
+    relayWs.send(JSON.stringify({
+      type: 'register',
+      sessionToken: savedSession?.sessionToken || undefined,
+    }))
+  }
+
+  relayWs.onmessage = (event: MessageEvent) => {
+    let msg: any
+    try { msg = JSON.parse(event.data as string) } catch { return }
+    handleRelayMessage(msg)
+  }
+
+  relayWs.onclose = () => {
+    console.log('[Relay] disconnected, reconnecting in 3s...')
+    relayWs = null
+    setTimeout(connectRelay, 3000)
+  }
+
+  relayWs.onerror = () => {
+    relayWs?.close()
+  }
+}
+
+function handleRelayMessage(msg: any) {
+  switch (msg.type) {
+    case 'registered': {
+      relaySessionToken = msg.sessionToken
+
+      // Save session for reconnection
+      try {
+        require('fs').mkdirSync(require('path').dirname(SESSION_FILE), { recursive: true })
+        require('fs').writeFileSync(SESSION_FILE, JSON.stringify({
+          sessionToken: relaySessionToken,
+          hubUrl,
+          relayUrl,
+        }))
+      } catch {}
+
+      if (msg.reconnected) {
+        console.log(`\n[✓] 已重新连接\n`)
+      } else {
+        console.log(`\n${'═'.repeat(40)}`)
+        console.log(`  微信小程序配对码:  ${msg.code}`)
+        console.log(`${'═'.repeat(40)}\n`)
+        console.log('  打开小程序 → 输入配对码 → 即可连接')
+        console.log('  配对码 10 分钟内有效\n')
+      }
+      break
+    }
+
+    case 'client-connected': {
+      console.log('[✓] 微信小程序已连接')
+      break
+    }
+
+    // Client REST request → forward to local hub
+    case 'rest-request': {
+      forwardToHub(msg.requestId, msg.method, msg.path, msg.body)
+      break
+    }
+
+    // Client auth request → forward to local hub
+    case 'auth-request': {
+      handleAuthRequest(msg.requestId, msg.accessToken)
+      break
+    }
+  }
+}
+
+// --- Forwarding ---
+
+function forwardToRelay(msg: any) {
+  if (relayWs && relayWs.readyState === WebSocket.OPEN) {
+    relayWs.send(JSON.stringify(msg))
+  }
+}
+
+async function forwardToHub(requestId: string, method: string, path: string, body?: any) {
+  try {
+    const init: RequestInit = {
+      method,
+      headers: {
+        'Content-Type': 'application/json',
+        ...(jwt ? { Authorization: `Bearer ${jwt}` } : {}),
+      },
+    }
+    if (body && method !== 'GET') init.body = JSON.stringify(body)
+
+    const res = await fetch(`${hubUrl}${path}`, init)
+    const data = await res.json().catch(() => ({}))
+
+    forwardToRelay({
+      type: 'rest-response',
+      requestId,
+      status: res.status,
+      data,
+    })
+  } catch (err: any) {
+    console.error(`[REST] ${method} ${path} failed:`, err.message)
+    forwardToRelay({
+      type: 'rest-response',
+      requestId,
+      status: 502,
+      data: { error: err.message },
+    })
+  }
+}
+
+async function handleAuthRequest(requestId: string, accessToken: string) {
+  try {
+    const res = await fetch(`${hubUrl}/api/auth`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ accessToken }),
+    })
+    const data = await res.json()
+    forwardToRelay({
+      type: 'auth-response',
+      requestId,
+      ok: res.ok,
+      data,
+    })
+  } catch (err: any) {
+    forwardToRelay({
+      type: 'auth-response',
+      requestId,
+      ok: false,
+      data: { error: err.message },
+    })
+  }
+}
+
+// --- Main ---
+
+async function main() {
+  console.log('╔══════════════════════════════════════╗')
+  console.log('║    CodePocket Connector v1.0          ║')
+  console.log('╚══════════════════════════════════════╝')
+  console.log()
+  console.log(`Hub:   ${hubUrl}`)
+  console.log(`Relay: ${relayUrl}`)
+  console.log()
+
+  // 1. Find token
+  console.log('[1/3] 连接 Hub...')
+  hubToken = await findHubToken()
+
+  // 2. Authenticate
+  console.log('[2/3] 认证中...')
+  jwt = await authenticateHub(hubToken)
+  console.log('[✓] Hub 认证成功')
+
+  // 3. Start SSE + connect relay
+  console.log('[3/3] 连接中继服务器...')
+  startSSE()
+  connectRelay()
+}
+
+main().catch((err) => {
+  console.error('[!] 启动失败:', err.message)
+  process.exit(1)
+})