tt-help-cli-ycl 1.4.6 → 1.4.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tt-help-cli-ycl",
-  "version": "1.4.6",
+  "version": "1.4.7",
   "description": "TikTok user & video data scraper - extract ttSeller, verified, locationCreated from HTML source",
   "type": "module",
   "bin": {

package/src/cli/tag.js

@@ -1,7 +1,8 @@
-import { writeFileSync } from "fs";
+import { writeFileSync, unlinkSync } from "fs";
 import { randomUUID } from "crypto";
 import { execSync } from "child_process";
 import os from "os";
+import path from "path";
 import readline from "readline";
 import { fetchTagData, enrichVideosWithLocation } from "../lib/tag-fetcher.js";
 import { killEdgeProcesses, ensureBrowserReady } from "../lib/browser/cdp.js";
@@ -68,26 +69,50 @@ function getBrowserMemory(port = null) {
       }
     } else if (platform === "win32") {
       // Windows: 用 PowerShell 查 msedge 进程内存（WorkingSet64 → MB）
-      // 按端口筛选时匹配命令行参数 --remote-debugging-port=PORT
-      if (port) {
-        cmd = `powershell -NoProfile -Command "Get-CimInstance Win32_Process -Filter 'Name LIKE \\\"msedge%\\\"' | Where-Object { $_.CommandLine -match 'remote-debugging-port=${port}' } | ForEach-Object { $_.WorkingSet64 } | Measure-Object -Sum | Select-Object -ExpandProperty Sum"`;
-      } else {
-        cmd = `powershell -NoProfile -Command "Get-CimInstance Win32_Process -Filter 'Name LIKE \\\"msedge%\\\"' | ForEach-Object { $_.WorkingSet64 } | Measure-Object -Sum | Select-Object -ExpandProperty Sum"`;
+      // 用临时 .ps1 文件避免命令行转义问题
+      const ps1Path = path.join(os.tmpdir(), `edge-mem-${Date.now()}.ps1`);
+      try {
+        let ps1Content;
+        if (port) {
+          ps1Content =
+            `$procs = Get-CimInstance Win32_Process -Filter "Name LIKE 'msedge%'" -ErrorAction SilentlyContinue; ` +
+            `$sum = 0; ` +
+            `foreach ($p in $procs) { ` +
+            `  if ($p.CommandLine -and $p.CommandLine -like "*--remote-debugging-port=${port}*") { ` +
+            `    $sum += $p.WorkingSetSize ` +
+            `  } ` +
+            `}; ` +
+            `Write-Output $sum`;
+        } else {
+          ps1Content =
+            `$procs = Get-CimInstance Win32_Process -Filter "Name LIKE 'msedge%'" -ErrorAction SilentlyContinue; ` +
+            `$sum = 0; ` +
+            `foreach ($p in $procs) { $sum += $p.WorkingSetSize }; ` +
+            `Write-Output $sum`;
+        }
+        writeFileSync(ps1Path, ps1Content);
+        cmd = `powershell.exe -NoProfile -ExecutionPolicy Bypass -File "${ps1Path}"`;
+        const result = execSync(cmd, { timeout: 5000 }).toString().trim();
+        try {
+          unlinkSync(ps1Path);
+        } catch {}
+        // PowerShell 返回 WorkingSetSize（字节），需除以 1048576 得 MB
+        const bytes = parseInt(result, 10);
+        const mb = bytes > 0 ? Math.round(bytes / 1048576) : 0;
+        return mb > 0 ? mb : null;
+      } catch {
+        try {
+          unlinkSync(ps1Path);
+        } catch {}
+        return null;
       }
     } else {
       return null;
     }
+    // macOS/Linux: ps RSS 是 KB，awk 已除以 1024 得 MB
     const result = execSync(cmd, { timeout: 5000 }).toString().trim();
-    if (platform === "win32") {
-      // PowerShell 返回 WorkingSet64（字节），需除以 1048576 得 MB
-      const bytes = parseInt(result, 10);
-      const mb = bytes > 0 ? Math.round(bytes / 1048576) : 0;
-      return mb > 0 ? mb : null;
-    } else {
-      // macOS/Linux: ps RSS 是 KB，awk 已除以 1024 得 MB
-      const mb = parseInt(result, 10);
-      return mb > 0 ? mb : null;
-    }
+    const mb = parseInt(result, 10);
+    return mb > 0 ? mb : null;
   } catch {
     return null;
   }