argus-eye 0.2.1 → 0.4.0

package/lib/compress.d.ts

@@ -0,0 +1,19 @@
+export interface CompressOptions {
+    /** 目标体积上限（字节）。 */
+    targetBytes: number;
+    /** 起始 jpeg 质量。 */
+    initialQuality?: number;
+    /** 最低 jpeg 质量。 */
+    minQuality?: number;
+}
+/**
+ * 把图片压到 `targetBytes` 以内的 JPEG。
+ *
+ * 性能优先（photon WASM 实现）：
+ *  - 输入已经 ≤ target → 直接返回（0 ms）。
+ *  - 否则估算 quality 编一遍。命中就完事；不行再 resize 一次。
+ *  - photon 的 resize 偏慢（~180ms / 2560x1440），所以放在最后兜底。
+ *
+ * 典型路径 ≤ 200ms：decode (~65ms) + 1 次 encode (~110ms) = ~175ms。
+ */
+export declare function compressToBudget(input: Buffer, options: CompressOptions): Buffer;

package/lib/config.d.ts

@@ -4,6 +4,8 @@ export interface ResolvedConfig {
     name: string;
     display?: number | string;
     format: 'png' | 'jpg';
+    /** 客户端预压缩目标体积（KB）。默认 500。0 = 不预压缩。 */
+    maxKB: number;
     reconnect: boolean;
     backoff: number;
     color: boolean;

package/lib/index.mjs

@@ -71,6 +71,51 @@ async function capture(options = {}) {
 }
 __name(capture, "capture");
 
+// src/compress.ts
+import {
+  PhotonImage,
+  resize,
+  SamplingFilter
+} from "@cf-wasm/photon/node";
+function compressToBudget(input, options) {
+  const target = Math.max(8 * 1024, options.targetBytes);
+  if (input.length <= target) return input;
+  const initialQuality = options.initialQuality ?? 80;
+  const minQuality = options.minQuality ?? 40;
+  const img = PhotonImage.new_from_byteslice(new Uint8Array(input));
+  try {
+    const ratioByteWise = target / input.length;
+    const estQ = Math.max(
+      minQuality,
+      Math.min(initialQuality, Math.round(initialQuality * ratioByteWise))
+    );
+    let buf = Buffer.from(img.get_bytes_jpeg(estQ));
+    if (buf.length <= target) return buf;
+    const q2 = Math.max(
+      minQuality,
+      Math.round(estQ * (target / buf.length) * 0.95)
+    );
+    if (q2 < estQ) {
+      buf = Buffer.from(img.get_bytes_jpeg(q2));
+      if (buf.length <= target) return buf;
+    }
+    const w = img.get_width();
+    const h = img.get_height();
+    const scale = Math.sqrt(target / buf.length) * 0.9;
+    const newW = Math.max(64, Math.round(w * scale));
+    const newH = Math.max(64, Math.round(h * scale));
+    const small = resize(img, newW, newH, SamplingFilter.Triangle);
+    try {
+      return Buffer.from(small.get_bytes_jpeg(q2));
+    } finally {
+      small.free();
+    }
+  } finally {
+    img.free();
+  }
+}
+__name(compressToBudget, "compressToBudget");
+
 // src/fullscreen.ts
 var DEFAULT_ALLOW_APPS = [
   // browsers
@@ -516,19 +561,33 @@ var ArgusClient = class {
         display: target,
         format: this.config.format
       });
-      const payload = encryptBuffer(result.buffer, this.config.token);
+      const compressStart = Date.now();
+      let final = result.buffer;
+      let mime = result.mime;
+      const rawSize = result.buffer.length;
+      const budgetBytes = this.config.maxKB * 1024;
+      if (budgetBytes > 0 && rawSize > budgetBytes) {
+        final = compressToBudget(result.buffer, {
+          targetBytes: budgetBytes
+        });
+        mime = "image/jpeg";
+      }
+      const compressMs = Date.now() - compressStart;
+      const payload = encryptBuffer(final, this.config.token);
       this.send({
         type: "peek_result",
         id: frame.id,
         image: payload,
         enc: ENC_ALGO,
-        mime: result.mime,
+        mime,
         display: frame.display
       });
       const elapsed = Date.now() - start2;
-      const kb = (result.buffer.length / 1024).toFixed(1);
+      const rawKb = (rawSize / 1024).toFixed(1);
+      const finalKb = (final.length / 1024).toFixed(1);
+      const same = final === result.buffer;
       logger.info(
-        `peek #${frame.id} → display ${frame.display ?? "default"} (${kb} KB ${this.config.format} encrypted, ${elapsed}ms)`
+        `peek #${frame.id} → display ${frame.display ?? "default"} (${same ? finalKb : `${rawKb}→${finalKb}`} KB ${same ? mime.replace("image/", "") : "jpg"} encrypted, total=${elapsed}ms compress=${compressMs}ms)`
       );
     } catch (err) {
       const message = err instanceof Error ? err.message : String(err);
@@ -617,7 +676,10 @@ var HELP_TEXT = `argus-eye [options]
   -n, --name   <name>      上报给服务端的名字（默认: os.hostname()）
   -d, --display <id>       默认截哪块屏（数字 id，缺省=主屏）
       --list-displays      列出本机显示器后退出
-      --format <png|jpg>   传输格式（默认 jpg，省流量）
+      --format <png|jpg>   原始捕获格式（默认 jpg）。
+      --max-kb <n>         预压缩目标体积上限（KB，默认 600）。
+                           客户端会用降质量 + 缩小图片把 buffer 压到这个上限以内
+                           再加密发出，省带宽。0 = 关闭预压缩，按原始截图发。
       --no-reconnect       禁用断线重连
       --backoff <ms>       重连最大间隔（默认 30000）
       --no-detect-fullscreen
@@ -646,7 +708,7 @@ function parseArgs(argv) {
     },
     string: ["server", "token", "name", "config", "format", "display"],
     array: ["allowApp", "busyApp"],
-    number: ["backoff"],
+    number: ["backoff", "maxKb"],
     boolean: [
       "help",
       "version",
@@ -679,6 +741,7 @@ function parseArgs(argv) {
     name: asString(parsed.name) ?? env.ARGUS_NAME ?? asString(fileConfig.name) ?? os.hostname(),
     display: asDisplay(parsed.display) ?? asDisplay(fileConfig.display) ?? void 0,
     format: asString(parsed.format) ?? asString(fileConfig.format) ?? "jpg",
+    maxKB: asNumber(parsed.maxKb) ?? asNumber(fileConfig.maxKB) ?? asNumber(fileConfig.maxKb) ?? 600,
     reconnect: asBoolean(parsed.reconnect) ?? asBoolean(fileConfig.reconnect) ?? true,
     backoff: asNumber(parsed.backoff) ?? asNumber(fileConfig.backoff) ?? 3e4,
     color: asBoolean(parsed.color) ?? true,
@@ -695,6 +758,7 @@ function parseArgs(argv) {
         name: merged.name ?? os.hostname(),
         display: merged.display,
         format: merged.format ?? "jpg",
+        maxKB: merged.maxKB ?? 600,
         reconnect: merged.reconnect ?? true,
         backoff: merged.backoff ?? 3e4,
         color: merged.color ?? true,
@@ -720,6 +784,7 @@ function parseArgs(argv) {
       name: merged.name ?? os.hostname(),
       display: merged.display,
       format: merged.format,
+      maxKB: merged.maxKB ?? 600,
       reconnect: merged.reconnect ?? true,
       backoff: merged.backoff ?? 3e4,
       color: merged.color ?? true,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "argus-eye",
-  "version": "0.2.1",
+  "version": "0.4.0",
   "type": "module",
   "description": "Argus eye —— koishi-plugin-argus 的截图客户端 CLI。",
   "author": "dingyi222666 <dingyi222666@foxmail.com>",
@@ -22,6 +22,7 @@
     "lint-fix": "yarn eslint src --ext=ts --fix"
   },
   "dependencies": {
+    "@cf-wasm/photon": "^0.3.5",
     "kleur": "^4.1.5",
     "screenshot-desktop": "^1.15.0",
     "ws": "^8.18.0",