[Init] Initial commit - NetMesh terminal manager

electron/mainQuitGuardSource.test.cjs

@@ -0,0 +1,118 @@
+const test = require("node:test");
+const assert = require("node:assert/strict");
+const { readFileSync } = require("node:fs");
+const path = require("node:path");
+
+test("before-quit dirty editor guard queries only registered editor-owner windows", () => {
+  const source = readFileSync(path.join(__dirname, "main.cjs"), "utf8");
+  const lifecycleSource = readFileSync(path.join(__dirname, "main", "appLockLifecycle.cjs"), "utf8");
+  const beforeQuitIndex = source.indexOf('app.on("before-quit"');
+  const dirtyEditorWindowsIndex = source.indexOf("getDirtyEditorWindows", beforeQuitIndex);
+  const handleBeforeQuitIndex = source.indexOf("handleBeforeQuit", beforeQuitIndex);
+  const handleBeforeQuitEndIndex = source.indexOf("}).catch", handleBeforeQuitIndex);
+  const queryableIndex = lifecycleSource.indexOf("const queryableWebContents");
+  const queryCallIndex = lifecycleSource.indexOf("queryDirtyEditors", queryableIndex);
+  const lifecycleBeforeQuitIndex = lifecycleSource.indexOf("async function handleBeforeQuit");
+  const lifecycleGuardSetup = lifecycleSource.slice(lifecycleBeforeQuitIndex, queryCallIndex);
+
+  assert.notEqual(beforeQuitIndex, -1);
+  assert.notEqual(dirtyEditorWindowsIndex, -1);
+  assert.notEqual(handleBeforeQuitIndex, -1);
+  assert.notEqual(handleBeforeQuitEndIndex, -1);
+  assert.notEqual(lifecycleBeforeQuitIndex, -1);
+  assert.notEqual(queryableIndex, -1);
+  assert.ok(dirtyEditorWindowsIndex < handleBeforeQuitIndex);
+  assert.match(source.slice(beforeQuitIndex, handleBeforeQuitIndex), /const dirtyEditorWindows = typeof getWindowManager\(\)\.getDirtyEditorWindows === "function"/);
+  assert.match(source.slice(handleBeforeQuitIndex, handleBeforeQuitEndIndex), /mainWindows,\s*\n\s*queryDirtyEditors/);
+  assert.match(lifecycleGuardSetup, /const reachableMainWindows = \(Array\.isArray\(mainWindows\) \? mainWindows : \[\]\)\.filter/);
+  // Prefer queryableWindows (reachable + non-crashed webContents) over a raw
+  // reachableMainWindows map so dirty checks and focus targets stay aligned.
+  assert.match(
+    lifecycleSource.slice(queryableIndex, queryCallIndex),
+    /queryableWindows\s*\n?\s*\.map\(\(candidate\) => candidate\.webContents\)/,
+  );
+  assert.doesNotMatch(lifecycleGuardSetup, /isVisible|isMinimized/);
+});
+
+test("macOS reopen after last window re-applies app lock for a fresh session", () => {
+  const source = readFileSync(path.join(__dirname, "main.cjs"), "utf8");
+  const lifecycleSource = readFileSync(path.join(__dirname, "main", "appLockLifecycle.cjs"), "utf8");
+
+  assert.match(lifecycleSource, /function ensureAppLockForFreshSession/);
+  assert.match(lifecycleSource, /function hasNoUsableAppContentWindows/);
+  assert.match(source, /ensureAppLockForFreshSession/);
+  assert.match(source, /hasNoUsableAppContentWindows/);
+
+  const createIndex = source.indexOf("async function createAndShowMainWindow");
+  const createBodyEnd = source.indexOf("async function deliverSshDeepLink", createIndex);
+  assert.notEqual(createIndex, -1);
+  assert.notEqual(createBodyEnd, -1);
+  const createBody = source.slice(createIndex, createBodyEnd);
+  assert.match(createBody, /hasNoUsableAppContentWindows\(getAppLockReopenWindows\(\)\)/);
+  assert.match(createBody, /ensureAppLockForFreshSession\(appLockController,\s*"startup"\)/);
+
+  const allClosedIndex = source.indexOf('app.on("window-all-closed"');
+  assert.notEqual(allClosedIndex, -1);
+  const allClosedBody = source.slice(allClosedIndex, allClosedIndex + 350);
+  assert.match(allClosedBody, /process\.platform !== "darwin"/);
+  assert.match(allClosedBody, /ensureAppLockForFreshSession\(appLockController,\s*"startup"\)/);
+});
+
+test("before-quit dirty editor guard foregrounds dirty windows through the focus recovery helper", () => {
+  const lifecycleSource = readFileSync(path.join(__dirname, "main", "appLockLifecycle.cjs"), "utf8");
+  const beforeQuitIndex = lifecycleSource.indexOf("async function handleBeforeQuit");
+  const loopIndex = lifecycleSource.indexOf("for (const win of dirtyWindows)", beforeQuitIndex);
+  const dialogIndex = lifecycleSource.indexOf("// App Lock overlay sits above renderer toasts", loopIndex);
+  const foregroundBlock = lifecycleSource.slice(loopIndex, dialogIndex);
+
+  assert.notEqual(beforeQuitIndex, -1);
+  assert.notEqual(loopIndex, -1);
+  assert.notEqual(dialogIndex, -1);
+  assert.match(foregroundBlock, /windowManager\.showAndFocusMainWindow\(win\)/);
+  assert.doesNotMatch(foregroundBlock, /commitQuit\(\)/);
+});
+
+test("before-quit keeps the original event cancelled while plugin shutdown runs without a renderer", () => {
+  const lifecycleSource = readFileSync(path.join(__dirname, "main", "appLockLifecycle.cjs"), "utf8");
+  const commitIndex = lifecycleSource.indexOf("const commit = () => {");
+  const commitEnd = lifecycleSource.indexOf("};", commitIndex);
+  const commitBlock = lifecycleSource.slice(commitIndex, commitEnd);
+
+  assert.notEqual(commitIndex, -1);
+  assert.match(commitBlock, /event\?\.preventDefault\?\.\(\)/);
+  assert.match(commitBlock, /commitQuit\(\)/);
+  assert.ok(
+    commitBlock.indexOf("event?.preventDefault?.()") < commitBlock.indexOf("commitQuit()"),
+    "the original quit must be cancelled before asynchronous plugin shutdown starts",
+  );
+
+  const source = readFileSync(path.join(__dirname, "main.cjs"), "utf8");
+  const beforeQuitIndex = source.indexOf('app.on("before-quit"');
+  const handleIndex = source.indexOf("handleBeforeQuit", beforeQuitIndex);
+  const handleEnd = source.indexOf("}).catch", handleIndex);
+  assert.notEqual(handleIndex, -1);
+  assert.match(source.slice(handleIndex, handleEnd), /commitQuit,/);
+});
+
+test("all app content windows use the WindowManager-level last-window close handler", () => {
+  const mainSource = readFileSync(path.join(__dirname, "main.cjs"), "utf8");
+  const windowSource = readFileSync(path.join(
+    __dirname,
+    "bridges",
+    "windowManager",
+    "mainWindow.cjs",
+  ), "utf8");
+  const createWindowIndex = mainSource.indexOf("async function createWindow()");
+  const setHandlerIndex = mainSource.indexOf("setAppContentWindowClosedHandler", createWindowIndex);
+  const managerCreateIndex = mainSource.indexOf("windowManager.createWindow", createWindowIndex);
+
+  assert.notEqual(setHandlerIndex, -1);
+  assert.notEqual(managerCreateIndex, -1);
+  assert.ok(setHandlerIndex < managerCreateIndex);
+  const closedIndex = windowSource.indexOf("win.on('closed'");
+  const appContentBranchIndex = windowSource.indexOf("if (registerAsAppContentWindow)", closedIndex);
+  const notifyIndex = windowSource.indexOf("notifyAppContentWindowClosed(win)", appContentBranchIndex);
+  assert.notEqual(closedIndex, -1);
+  assert.notEqual(appContentBranchIndex, -1);
+  assert.notEqual(notifyIndex, -1);
+});