@electron-memory/monitor 0.1.0 → 0.2.0

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,75 @@ 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, net, protocol } from "electron";
 import * as path2 from "path";
+import { pathToFileURL } from "url";
+var SCHEME = "emm-dashboard";
+var privilegedRegistered = false;
+var handlerRegistered = false;
+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, (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 });
+      }
+      return net.fetch(pathToFileURL(filePath).href);
+    } catch {
+      return new Response("Bad Request", { status: 400 });
+    }
+  });
+}
+function getDashboardPageURL() {
+  return `${SCHEME}://electron/index.html`;
+}
+
+// src/core/dashboard.ts
 var DashboardManager = class {
   constructor(config) {
     this.window = null;
@@ -1093,7 +1161,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,
@@ -1105,8 +1173,9 @@ var DashboardManager = class {
         nodeIntegration: false
       }
     });
-    const uiPath = path2.join(__dirname, "ui", "index.html");
-    this.window.loadFile(uiPath);
+    const uiRoot = path3.join(__dirname, "ui");
+    ensureDashboardProtocolHandler(uiRoot);
+    this.window.loadURL(getDashboardPageURL());
     this.window.on("closed", () => {
       this.window = null;
     });
@@ -1276,6 +1345,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 +1358,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 +1433,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 +1463,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 +1594,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 +1607,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`
     );
@@ -1597,6 +1667,7 @@ var ElectronMemoryMonitor = class extends EventEmitter3 {
 };
 export {
   ElectronMemoryMonitor,
-  IPC_CHANNELS
+  IPC_CHANNELS,
+  registerDashboardSchemePrivileged
 };
 //# sourceMappingURL=index.mjs.map