dsh-pocket 1.4.8 → 1.6.0

package/lib/index.js

@@ -148,6 +148,8 @@ export function apply(ctx, config = {}, internals = {}) {
   // 代理随插件自动启动（局域网二维码开箱即用，零配置）
   void service.startProxy().then((proxy) => {
     logger.info('dsh-pocket: proxy ready on :%d | 局域网代理已就绪', proxy.port);
+    // 自动恢复上次开启的公网隧道（DSH 重启后 cloudflared 子进程被杀，issue #11）
+    void service.restoreTunnelIfNeeded?.().catch(() => {});
   }).catch((err) => {
     logger.error('dsh-pocket: proxy start failed | 代理启动失败: %s', err?.message ?? err);
   });

package/lib/proxy.mjs

@@ -13,6 +13,7 @@
 
 import { createServer } from 'node:http';
 import { request as httpRequest } from 'node:http';
+import { createGzip, createBrotliCompress } from 'node:zlib';
 
 const DEFAULT_UPSTREAM = { host: '127.0.0.1', port: 3080 };
 
@@ -98,6 +99,37 @@ export function createPocketProxy({ port = 3081, host = '0.0.0.0', upstream = DE
           proxyRes.on('error', () => res.destroy());
           return;
         }
+        // 大 JSON/text 响应**流式压缩**（issue #12）：长会话历史一次返回 17MB+，
+        // 局域网直连与隧道段都吃满带宽；压缩到 ~1.3MB。跳过已压缩、SSE 流
+        // （/api/events.* 原样透传）、HTML（走上面的注入分支）。
+        const acceptEncoding = String(req.headers['accept-encoding'] ?? '');
+        const canGzip = /\bgzip\b/.test(acceptEncoding);
+        const canBr = /\bbr\b/.test(acceptEncoding);
+        const isEventStream = contentType.includes('text/event-stream');
+        const knownLen = Number(proxyRes.headers['content-length'] || 0);
+        const shouldCompress = (canGzip || canBr)
+          && !isCompressed(proxyRes.headers)
+          && !isEventStream
+          && (contentType.includes('application/json') || contentType.startsWith('text/'))
+          && (knownLen === 0 || knownLen >= 1024);
+        if (shouldCompress) {
+          const enc = canBr ? 'br' : 'gzip';
+          const outHeaders = { ...proxyRes.headers };
+          delete outHeaders['content-length'];
+          delete outHeaders['transfer-encoding'];
+          outHeaders['content-encoding'] = enc;
+          res.writeHead(proxyRes.statusCode ?? 200, outHeaders);
+          const z = enc === 'br' ? createBrotliCompress() : createGzip();
+          proxyRes.pipe(z).pipe(res);
+          // 任一端断开都要清理（含压缩流）。注意：不能用 proxyRes 的 'close'
+          // 来掐 res——正常结束后 close 也会触发，此时压缩流可能还没写完，
+          // 会误杀连接；异常中止用 'aborted'。
+          res.on('close', () => { proxyRes.destroy(); z.destroy(); });
+          proxyRes.on('error', () => { z.destroy(); res.destroy(); });
+          proxyRes.on('aborted', () => { z.destroy(); res.destroy(); });
+          z.on('error', () => res.destroy());
+          return;
+        }
         res.writeHead(proxyRes.statusCode ?? 502, proxyRes.headers);
         proxyRes.pipe(res);
         // 任一端断开都要清理另一端：客户端断连销毁上游流（不留僵尸），

package/lib/service.mjs

@@ -8,6 +8,8 @@
 
 import { networkInterfaces } from 'node:os';
 import { createRequire } from 'node:module';
+import { mkdir, readFile, rm, writeFile } from 'node:fs/promises';
+import { join, dirname } from 'node:path';
 import { createPocketProxy } from './proxy.mjs';
 import { startQuickTunnel } from './tunnel.mjs';
 
@@ -112,6 +114,21 @@ export function createPocketService({
     return qrCache.get(text);
   }
 
+  // 公网隧道自动恢复（issue #11）：DSH 重启后 cloudflared 子进程被杀、隧道消失，
+  // 插件无从知晓。启动时检查持久化的「隧道开启中」标记，自动重新拉起。
+  const autoStatePath = home ? join(home, 'dsh-pocket', 'tunnel-auto.json') : null;
+  async function persistAutoTunnel() {
+    if (!autoStatePath) return;
+    try {
+      await mkdir(dirname(autoStatePath), { recursive: true });
+      await writeFile(autoStatePath, JSON.stringify({ at: Date.now() }), 'utf8');
+    } catch { /* 忽略 */ }
+  }
+  async function clearAutoTunnel() {
+    if (!autoStatePath) return;
+    try { await rm(autoStatePath, { force: true }); } catch { /* 忽略 */ }
+  }
+
   return {
     dshPort,
     /** 启动局域网代理（幂等）。端口被占（EADDRINUSE，如桌面版与普通环境同时运行）时自动尝试下一个端口。 */
@@ -166,6 +183,8 @@ export function createPocketService({
             tunnelState.phase = 'error';
             tunnelState.detail = `隧道进程退出（code=${code}）| tunnel process exited`;
           });
+          // 记录「隧道开启中」，供重启后自动恢复（issue #11）
+          void persistAutoTunnel();
           return tunnel.url;
         } catch (err) {
           // stopTunnel 触发的 abort 不算错误：保持 idle，别把状态刷成 error
@@ -195,6 +214,25 @@ export function createPocketService({
       tunnelState.phase = 'idle';
       tunnelState.detail = '';
       tunnelState.startedAt = null;
+      void clearAutoTunnel(); // 手动关闭后不再自动恢复
+    },
+
+    /** 启动时自动恢复上次开启的公网隧道（DSH 重启后 cloudflared 子进程被杀，issue #11）。 */
+    async restoreTunnelIfNeeded() {
+      if (!autoStatePath || tunnel || tunnelPromise) return;
+      let has = false;
+      try {
+        const raw = await readFile(autoStatePath, 'utf8');
+        has = /"at"\s*:/.test(raw);
+      } catch { return; } // 无标记 → 不恢复
+      if (!has) return;
+      try {
+        await this.startTunnel();
+        console.log('dsh-pocket: public tunnel auto-restored | 已自动恢复公网隧道');
+      } catch (err) {
+        // 恢复失败保留标记（下次启动再试）；网络问题见 README 排障
+        console.warn('dsh-pocket: tunnel auto-restore failed | 自动恢复隧道失败: %s', err?.message ?? err);
+      }
     },
 
     /** 状态快照（RPC 返回，不含敏感信息；二维码 data URL 本地生成 + 缓存）。 */

package/package.json

@@ -76,5 +76,5 @@
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
-  "version": "1.4.8"
+  "version": "1.6.0"
 }