@simonyea/holysheep-cli 1.7.41 → 1.7.43

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@simonyea/holysheep-cli",
-  "version": "1.7.41",
+  "version": "1.7.43",
   "description": "Claude Code/Cursor/Cline API relay for China — ¥1=$1, WeChat/Alipay payment, no credit card, no VPN. One command setup for all AI coding tools.",
   "keywords": [
     "openai-china",

package/src/tools/claude-process-proxy.js

@@ -58,12 +58,8 @@ async function readJsonResponse(response) {
   }
 }
 
-async function requestSessionLease(config, sessionId) {
-  const cached = leaseCache.get(sessionId)
-  if (cached?.expiresAt && new Date(cached.expiresAt).getTime() - Date.now() > 30_000) {
-    return cached
-  }
-
+// 向 relay 申请新 lease（启动时 + CONNECT 失败时被动重试）
+async function fetchFreshLease(config, sessionId) {
   const controlPlaneUrl = getControlPlaneUrl(config)
   if (!controlPlaneUrl) throw new Error('Claude relay control plane is not configured')
 
@@ -87,6 +83,13 @@ async function requestSessionLease(config, sessionId) {
   return payload.data
 }
 
+// 请求路径：只读缓存，不检查过期时间（续约由失败触发，不由时间触发）
+function getCachedLease(sessionId) {
+  const cached = leaseCache.get(sessionId)
+  if (!cached) throw new Error('No session lease available')
+  return cached
+}
+
 function buildAuthHeaders(config, lease) {
   return {
     'x-hs-bridge-id': config.bridgeId || 'local-bridge',
@@ -149,84 +152,100 @@ function pipeWithCleanup(a, b) {
 
 function createProcessProxyServer({ sessionId, configPath = CONFIG_PATH }) {
   const server = http.createServer(async (clientReq, clientRes) => {
-    try {
+    const doForward = async (lease) => {
       const config = readConfig(configPath)
-      const lease = await requestSessionLease(config, sessionId)
       const nodeProxyUrl = deriveNodeProxyUrl(lease)
       const headers = {
         ...buildAuthHeaders(config, lease),
         host: new URL(clientReq.url).host,
       }
-
       const upstream = new URL(nodeProxyUrl)
-      const forwardReq = http.request({
-        host: upstream.hostname,
-        port: Number(upstream.port || 80),
-        method: clientReq.method,
-        path: clientReq.url,
-        headers: {
-          ...clientReq.headers,
-          ...headers,
-          connection: 'close',
-        },
-      }, (forwardRes) => {
-        clientRes.writeHead(forwardRes.statusCode || 502, forwardRes.headers)
-        forwardRes.pipe(clientRes)
+      return new Promise((resolve, reject) => {
+        const forwardReq = http.request({
+          host: upstream.hostname,
+          port: Number(upstream.port || 80),
+          method: clientReq.method,
+          path: clientReq.url,
+          headers: { ...clientReq.headers, ...headers, connection: 'close' },
+        }, (forwardRes) => {
+          clientRes.writeHead(forwardRes.statusCode || 502, forwardRes.headers)
+          forwardRes.pipe(clientRes)
+          resolve()
+        })
+        forwardReq.once('error', reject)
+        clientReq.pipe(forwardReq)
       })
+    }
 
-      forwardReq.once('error', (error) => {
+    try {
+      await doForward(getCachedLease(sessionId))
+    } catch {
+      // lease 失效，拿新 lease 重试一次
+      try {
+        const config = readConfig(configPath)
+        leaseCache.delete(sessionId)
+        const freshLease = await fetchFreshLease(config, sessionId)
+        await doForward(freshLease)
+      } catch (retryError) {
         clientRes.writeHead(502, { 'content-type': 'text/plain; charset=utf-8' })
-        clientRes.end(error.message || 'Proxy error')
-      })
-
-      clientReq.pipe(forwardReq)
-    } catch (error) {
-      clientRes.writeHead(500, { 'content-type': 'text/plain; charset=utf-8' })
-      clientRes.end(error.message || 'Proxy error')
+        clientRes.end(retryError.message || 'Proxy error')
+      }
     }
   })
 
   server.on('connect', async (req, clientSocket, head) => {
-    try {
-      const config = readConfig(configPath)
-      const lease = await requestSessionLease(config, sessionId)
-      const target = String(req.url || '').trim()
-      const [host, rawPort] = target.split(':')
-      const port = Number(rawPort || 443)
-      if (!host || !Number.isInteger(port) || ![80, 443].includes(port)) {
-        clientSocket.write('HTTP/1.1 403 Forbidden\r\n\r\n')
-        return clientSocket.destroy()
-      }
+    const target = String(req.url || '').trim()
+    const [host, rawPort] = target.split(':')
+    const port = Number(rawPort || 443)
+    if (!host || !Number.isInteger(port) || ![80, 443].includes(port)) {
+      clientSocket.write('HTTP/1.1 403 Forbidden\r\n\r\n')
+      return clientSocket.destroy()
+    }
 
+    const doConnect = async (lease) => {
       const upstreamSocket = await createConnectTunnel(
         deriveNodeProxyUrl(lease),
         target,
-        buildAuthHeaders(config, lease)
+        buildAuthHeaders(readConfig(configPath), lease)
       )
-
       clientSocket.write('HTTP/1.1 200 Connection Established\r\n\r\n')
       if (head?.length) upstreamSocket.write(head)
       pipeWithCleanup(clientSocket, upstreamSocket)
-    } catch (error) {
-      clientSocket.write(`HTTP/1.1 502 Bad Gateway\r\ncontent-type: text/plain; charset=utf-8\r\n\r\n${error.message}`)
-      clientSocket.destroy()
+    }
+
+    try {
+      await doConnect(getCachedLease(sessionId))
+    } catch {
+      // lease 失效，拿新 lease 重试一次
+      try {
+        const config = readConfig(configPath)
+        leaseCache.delete(sessionId)
+        const freshLease = await fetchFreshLease(config, sessionId)
+        await doConnect(freshLease)
+      } catch (retryError) {
+        clientSocket.write(`HTTP/1.1 502 Bad Gateway\r\ncontent-type: text/plain; charset=utf-8\r\n\r\n${retryError.message}`)
+        clientSocket.destroy()
+      }
     }
   })
 
   return server
 }
 
-function startProcessProxy({ port = null, sessionId = null, configPath = CONFIG_PATH } = {}) {
+async function startProcessProxy({ port = null, sessionId = null, configPath = CONFIG_PATH } = {}) {
   const config = readConfig(configPath)
   const preferredPort = port || getProcessProxyPort(config)
   const effectiveSessionId = sessionId || crypto.randomUUID()
+
+  // 启动时拿一次 lease，之后靠被动重试维持，不再主动续约
+  await fetchFreshLease(config, effectiveSessionId)
+
   const server = createProcessProxyServer({ sessionId: effectiveSessionId, configPath })
 
   return new Promise((resolve, reject) => {
     const tryListen = (p) => {
       server.once('error', (err) => {
         if (err.code === 'EADDRINUSE') {
-          // 端口被占用，让 OS 分配一个随机可用端口
           server.once('error', reject)
           server.listen(0, '127.0.0.1')
         } else {
@@ -264,7 +283,6 @@ module.exports = {
   getProcessProxyPort,
   getControlPlaneUrl,
   readConfig,
-  requestSessionLease,
   startProcessProxy,
   writeConfig,
 }