@lvce-editor/main-process 6.13.0 → 6.14.0

package/dist/mainProcessMain.js

@@ -5,7 +5,7 @@ import { fileURLToPath } from 'node:url';
 import { inspect } from 'node:util';
 import { spawn } from 'node:child_process';
 import { Console } from 'node:console';
-import { mkdirSync, createWriteStream, readFileSync, openAsBlob, existsSync, openSync, writeSync, closeSync, rmSync } from 'node:fs';
+import { mkdirSync, createWriteStream, readFileSync, openAsBlob, existsSync, writeFileSync, rmSync, openSync, writeSync, closeSync } from 'node:fs';
 import * as NodePath from 'node:path';
 import { dirname, join as join$1 } from 'node:path';
 import { homedir, tmpdir } from 'node:os';
@@ -4888,6 +4888,106 @@ const openContextMenu = async (menuItems, x, y) => {
   }
 };
 
+const profileDuration = 10_000;
+const getFileName$1 = targetName => {
+  const safeTargetName = targetName.replaceAll(/[^a-zA-Z0-9._-]+/g, '-').replace(/^-/, '').replace(/-$/, '') || 'extension-host';
+  return `${safeTargetName}-${Date.now()}.cpuprofile`;
+};
+const wait = async () => {
+  await new Promise(resolve => {
+    setTimeout(resolve, profileDuration);
+  });
+};
+const takeCpuProfile = async (windowId, workerName) => {
+  const browserWindow = Electron.BrowserWindow.fromId(windowId);
+  if (!browserWindow) {
+    throw new Error(`Browser window not found: ${windowId}`);
+  }
+  const electronDebugger = browserWindow.webContents.debugger;
+  const wasAttached = electronDebugger.isAttached();
+  let filePath = '';
+  let profileStarted = false;
+  let profileStopped = false;
+  let sessionId;
+  let success = false;
+  if (!wasAttached) {
+    electronDebugger.attach();
+  }
+  try {
+    if (workerName) {
+      const {
+        frameTree
+      } = await electronDebugger.sendCommand('Page.getFrameTree');
+      const {
+        targetInfos
+      } = await electronDebugger.sendCommand('Target.getTargets');
+      const target = targetInfos.find(targetInfo => targetInfo.type === 'worker' && targetInfo.title === workerName && targetInfo.parentFrameId === frameTree.frame.id);
+      if (!target) {
+        throw new Error(`Worker not found: ${workerName}`);
+      }
+      const attachResult = await electronDebugger.sendCommand('Target.attachToTarget', {
+        flatten: true,
+        targetId: target.targetId
+      });
+      sessionId = attachResult.sessionId;
+    }
+    await electronDebugger.sendCommand('Profiler.enable', undefined, sessionId);
+    await electronDebugger.sendCommand('Profiler.start', undefined, sessionId);
+    profileStarted = true;
+    await wait();
+    const {
+      profile
+    } = await electronDebugger.sendCommand('Profiler.stop', undefined, sessionId);
+    profileStopped = true;
+    const downloadsPath = Electron.app.getPath('downloads');
+    mkdirSync(downloadsPath, {
+      recursive: true
+    });
+    const targetName = workerName || `Extension Host Window ${windowId}`;
+    filePath = join$1(downloadsPath, getFileName$1(targetName));
+    writeFileSync(filePath, JSON.stringify(profile), {
+      flag: 'wx'
+    });
+    success = true;
+    return filePath;
+  } finally {
+    try {
+      if (profileStarted && !profileStopped && electronDebugger.isAttached()) {
+        await electronDebugger.sendCommand('Profiler.stop', undefined, sessionId);
+      }
+      if (electronDebugger.isAttached()) {
+        await electronDebugger.sendCommand('Profiler.disable', undefined, sessionId);
+      }
+    } finally {
+      try {
+        if (!success && filePath) {
+          rmSync(filePath, {
+            force: true
+          });
+        }
+        if (sessionId && electronDebugger.isAttached()) {
+          await electronDebugger.sendCommand('Target.detachFromTarget', {
+            sessionId
+          });
+        }
+      } finally {
+        if (!wasAttached && electronDebugger.isAttached()) {
+          electronDebugger.detach();
+        }
+      }
+    }
+  }
+};
+const takeWindowCpuProfile = async windowId => {
+  number(windowId);
+  return takeCpuProfile(windowId, '');
+};
+const takeWorkerCpuProfile = async (windowId, workerName) => {
+  number(windowId);
+  string(workerName);
+  return takeCpuProfile(windowId, workerName);
+};
+
 const getFileName = workerName => {
   const safeWorkerName = workerName.replaceAll(/[^a-zA-Z0-9._-]+/g, '-').replace(/^-/, '').replace(/-$/, '') || 'worker';
   return `${safeWorkerName}-${Date.now()}.heapsnapshot`;
@@ -6601,6 +6701,8 @@ const commandMap = {
   'ElectronContextMenu.openContextMenu': openContextMenu,
   'ElectronDeveloper.crashMainProcess': crashMainProcess,
   'ElectronDeveloper.getPerformanceEntries': getPerformanceEntries,
+  'ElectronDeveloper.takeWindowCpuProfile': takeWindowCpuProfile,
+  'ElectronDeveloper.takeWorkerCpuProfile': takeWorkerCpuProfile,
   'ElectronDeveloper.takeWorkerHeapSnapshot': takeWorkerHeapSnapshot,
   'ElectronDialog.showMessageBox': showMessageBox,
   'ElectronDialog.showOpenDialog': showOpenDialog,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lvce-editor/main-process",
-  "version": "6.13.0",
+  "version": "6.14.0",
   "keywords": [
     "lvce-editor",
     "electron"