@electron-memory/monitor 0.1.0 → 0.2.1

package/dist/index.mjs

@@ -1,6 +1,6 @@
 // src/core/monitor.ts
-import { app as app2 } from "electron";
-import * as path3 from "path";
+import { app as app3 } from "electron";
+import * as path4 from "path";
 import * as v82 from "v8";
 import { EventEmitter as EventEmitter3 } from "events";
 
@@ -1070,7 +1070,149 @@ var Analyzer = class {
 
 // src/core/dashboard.ts
 import { BrowserWindow as BrowserWindow2 } from "electron";
+import * as path3 from "path";
+
+// src/core/dashboard-protocol.ts
+import { app as app2, protocol } from "electron";
+import { readFile } from "fs/promises";
 import * as path2 from "path";
+var SCHEME = "emm-dashboard";
+var privilegedRegistered = false;
+var handlerRegistered = false;
+var MIME_BY_EXT = {
+  ".html": "text/html; charset=utf-8",
+  ".js": "text/javascript; charset=utf-8",
+  ".mjs": "text/javascript; charset=utf-8",
+  ".css": "text/css; charset=utf-8",
+  ".json": "application/json",
+  ".map": "application/json",
+  ".svg": "image/svg+xml",
+  ".png": "image/png",
+  ".jpg": "image/jpeg",
+  ".jpeg": "image/jpeg",
+  ".gif": "image/gif",
+  ".webp": "image/webp",
+  ".ico": "image/x-icon",
+  ".woff": "font/woff",
+  ".woff2": "font/woff2",
+  ".ttf": "font/ttf",
+  ".txt": "text/plain; charset=utf-8"
+};
+function isPathInsideRoot(filePath, root) {
+  const f = path2.resolve(filePath);
+  const r = path2.resolve(root);
+  if (f === r) {
+    return true;
+  }
+  const sep2 = r.endsWith(path2.sep) ? r : `${r}${path2.sep}`;
+  return f.startsWith(sep2);
+}
+function registerDashboardSchemePrivileged() {
+  if (privilegedRegistered) {
+    return;
+  }
+  privilegedRegistered = true;
+  protocol.registerSchemesAsPrivileged([
+    {
+      scheme: SCHEME,
+      privileges: {
+        standard: true,
+        secure: true,
+        supportFetchAPI: true,
+        corsEnabled: true,
+        stream: true
+      }
+    }
+  ]);
+}
+function ensureDashboardProtocolHandler(uiRoot) {
+  if (handlerRegistered) {
+    return;
+  }
+  if (!app2.isReady()) {
+    throw new Error(
+      "[@electron-memory/monitor] ensureDashboardProtocolHandler: app must be ready before opening dashboard"
+    );
+  }
+  handlerRegistered = true;
+  const base = path2.resolve(uiRoot);
+  protocol.handle(SCHEME, async (request) => {
+    try {
+      const { pathname } = new URL(request.url);
+      let rel = pathname.startsWith("/") ? pathname.slice(1) : pathname;
+      if (!rel) {
+        rel = "index.html";
+      }
+      const filePath = path2.resolve(path2.join(base, rel));
+      if (!isPathInsideRoot(filePath, base)) {
+        return new Response("Forbidden", { status: 403 });
+      }
+      const body = await readFile(filePath);
+      const ext = path2.extname(filePath).toLowerCase();
+      const mime = MIME_BY_EXT[ext] || "application/octet-stream";
+      return new Response(body, {
+        status: 200,
+        headers: {
+          "Content-Type": mime,
+          "Cache-Control": "no-store"
+        }
+      });
+    } catch (err) {
+      const msg = err instanceof Error ? err.message : String(err);
+      return new Response(msg, { status: 404 });
+    }
+  });
+}
+function getDashboardPageURL() {
+  return `${SCHEME}://electron/index.html`;
+}
+
+// src/core/dashboard-window-hooks.ts
+function shouldOpenDevToolsShortcut(input) {
+  if (input.type !== "keyDown") {
+    return false;
+  }
+  if (input.key === "F12") {
+    return true;
+  }
+  const k = input.key.toLowerCase();
+  if (k !== "i") {
+    return false;
+  }
+  if (input.control && input.shift) {
+    return true;
+  }
+  if (process.platform === "darwin" && input.meta && input.alt) {
+    return true;
+  }
+  return false;
+}
+function attachDashboardWindowHooks(win, options) {
+  const wc = win.webContents;
+  wc.on("before-input-event", (event, input) => {
+    if (!shouldOpenDevToolsShortcut(input)) {
+      return;
+    }
+    event.preventDefault();
+    if (wc.isDevToolsOpened()) {
+      wc.closeDevTools();
+    } else {
+      wc.openDevTools({ mode: "detach" });
+    }
+  });
+  wc.on("did-fail-load", (_e, code, desc, url) => {
+    console.error("[@electron-memory/monitor] dashboard did-fail-load", code, desc, url);
+  });
+  let autoOpened = false;
+  wc.on("did-finish-load", () => {
+    if (options.openDevToolsOnStart && !autoOpened) {
+      autoOpened = true;
+      wc.openDevTools({ mode: "detach" });
+    }
+  });
+}
+
+// src/core/dashboard.ts
 var DashboardManager = class {
   constructor(config) {
     this.window = null;
@@ -1093,7 +1235,7 @@ var DashboardManager = class {
       this.window.focus();
       return;
     }
-    const preloadPath = path2.join(__dirname, "dashboard-preload.js");
+    const preloadPath = path3.join(__dirname, "dashboard-preload.js");
     this.window = new BrowserWindow2({
       width: this.config.dashboard.width,
       height: this.config.dashboard.height,
@@ -1102,11 +1244,16 @@ var DashboardManager = class {
       webPreferences: {
         preload: preloadPath,
         contextIsolation: true,
-        nodeIntegration: false
+        nodeIntegration: false,
+        devTools: true
       }
     });
-    const uiPath = path2.join(__dirname, "ui", "index.html");
-    this.window.loadFile(uiPath);
+    attachDashboardWindowHooks(this.window, {
+      openDevToolsOnStart: this.config.dashboard.openDevToolsOnStart
+    });
+    const uiRoot = path3.join(__dirname, "ui");
+    ensureDashboardProtocolHandler(uiRoot);
+    this.window.loadURL(getDashboardPageURL());
     this.window.on("closed", () => {
       this.window = null;
     });
@@ -1257,7 +1404,8 @@ var DEFAULT_CONFIG = {
   dashboard: {
     width: 1400,
     height: 900,
-    alwaysOnTop: false
+    alwaysOnTop: false,
+    openDevToolsOnStart: false
   },
   processLabels: {}
 };
@@ -1276,6 +1424,7 @@ var ElectronMemoryMonitor = class extends EventEmitter3 {
       this.dashboard = null;
       return;
     }
+    registerDashboardSchemePrivileged();
     this.collector = new MemoryCollector(this.config);
     this.anomalyDetector = new AnomalyDetector(this.config);
     this.analyzer = new Analyzer();
@@ -1288,10 +1437,10 @@ var ElectronMemoryMonitor = class extends EventEmitter3 {
   /** 启动监控 */
   async start() {
     if (!this.config.enabled || this.started) return;
-    if (!app2.isReady()) {
-      await app2.whenReady();
+    if (!app3.isReady()) {
+      await app3.whenReady();
     }
-    const storageDir = this.config.storage.directory || path3.join(app2.getPath("userData"), "memory-monitor");
+    const storageDir = this.config.storage.directory || path4.join(app3.getPath("userData"), "memory-monitor");
     this.persister = new DataPersister(this.config, storageDir);
     this.sessionManager = new SessionManager(this.persister);
     this.ipcHandler = new IPCMainHandler(this);
@@ -1363,7 +1512,7 @@ var ElectronMemoryMonitor = class extends EventEmitter3 {
       anomalies,
       completedSession.dataFile
     );
-    const reportPath = path3.join(this.persister.getStorageDir(), completedSession.id, "report.json");
+    const reportPath = path4.join(this.persister.getStorageDir(), completedSession.id, "report.json");
     const fs2 = await import("fs");
     fs2.writeFileSync(reportPath, JSON.stringify(report, null, 2), "utf-8");
     this.emit("session-end", report);
@@ -1393,7 +1542,7 @@ var ElectronMemoryMonitor = class extends EventEmitter3 {
   /** 获取指定会话报告 */
   async getSessionReport(sessionId) {
     const fs2 = await import("fs");
-    const reportPath = path3.join(this.persister.getStorageDir(), sessionId, "report.json");
+    const reportPath = path4.join(this.persister.getStorageDir(), sessionId, "report.json");
     try {
       const content = fs2.readFileSync(reportPath, "utf-8");
       return JSON.parse(content);
@@ -1524,7 +1673,7 @@ var ElectronMemoryMonitor = class extends EventEmitter3 {
       } catch {
       }
     }
-    await new Promise((resolve) => setTimeout(resolve, 100));
+    await new Promise((resolve2) => setTimeout(resolve2, 100));
     const afterMem = process.memoryUsage();
     const freed = beforeMem.heapUsed - afterMem.heapUsed;
     return {
@@ -1537,7 +1686,7 @@ var ElectronMemoryMonitor = class extends EventEmitter3 {
   }
   /** 导出堆快照 */
   async takeHeapSnapshot(filePath) {
-    const snapshotPath = filePath || path3.join(
+    const snapshotPath = filePath || path4.join(
       this.persister.getStorageDir(),
       `heap-${Date.now()}.heapsnapshot`
     );
@@ -1572,7 +1721,15 @@ var ElectronMemoryMonitor = class extends EventEmitter3 {
     this.emit("snapshot", snapshot);
   }
   mergeConfig(userConfig) {
-    if (!userConfig) return { ...DEFAULT_CONFIG };
+    if (!userConfig) {
+      return {
+        ...DEFAULT_CONFIG,
+        dashboard: {
+          ...DEFAULT_CONFIG.dashboard,
+          openDevToolsOnStart: process.env.LAUNCHER_MEMORY_MONITOR_DEVTOOLS === "1"
+        }
+      };
+    }
     return {
       ...DEFAULT_CONFIG,
       ...userConfig,
@@ -1586,7 +1743,8 @@ var ElectronMemoryMonitor = class extends EventEmitter3 {
       },
       dashboard: {
         ...DEFAULT_CONFIG.dashboard,
-        ...userConfig.dashboard || {}
+        ...userConfig.dashboard || {},
+        openDevToolsOnStart: userConfig.dashboard?.openDevToolsOnStart !== void 0 ? userConfig.dashboard.openDevToolsOnStart : process.env.LAUNCHER_MEMORY_MONITOR_DEVTOOLS === "1"
       },
       processLabels: {
         ...DEFAULT_CONFIG.processLabels,
@@ -1597,6 +1755,7 @@ var ElectronMemoryMonitor = class extends EventEmitter3 {
 };
 export {
   ElectronMemoryMonitor,
-  IPC_CHANNELS
+  IPC_CHANNELS,
+  registerDashboardSchemePrivileged
 };
 //# sourceMappingURL=index.mjs.map