-
Notifications
You must be signed in to change notification settings - Fork 51
Expand file tree
/
Copy pathshutdown.mjs
More file actions
102 lines (95 loc) · 2.25 KB
/
Copy pathshutdown.mjs
File metadata and controls
102 lines (95 loc) · 2.25 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
function closeServerOnce(closeServer) {
return new Promise((resolve) => {
try {
let settled = false;
const done = () => {
if (settled) return;
settled = true;
resolve();
};
const maybe = closeServer?.(done);
if (maybe && typeof maybe.then === 'function') {
maybe.then(() => done()).catch(() => done());
return;
}
if (typeof closeServer !== 'function' || closeServer.length === 0) {
done();
}
} catch {
resolve();
}
});
}
export async function cleanupRuntimeResources({
closeServer,
stopWatchFolders,
disposeBrowserBackend
} = {}) {
await Promise.allSettled([
closeServerOnce(closeServer),
Promise.resolve().then(() => stopWatchFolders?.()),
Promise.resolve().then(() => disposeBrowserBackend?.())
]);
}
export function createGracefulShutdown({
closeServer,
stopWatchFolders,
disposeBrowserBackend,
stopOrchestrators,
setTabsQuitting,
markQuitting,
quitApp
} = {}) {
let cleanupPromise = null;
let allowingQuit = false;
async function cleanup() {
if (cleanupPromise) return cleanupPromise;
cleanupPromise = (async () => {
try {
markQuitting?.();
} catch {}
try {
stopOrchestrators?.();
} catch {}
try {
setTabsQuitting?.();
} catch {}
await cleanupRuntimeResources({ closeServer, stopWatchFolders, disposeBrowserBackend });
})();
return cleanupPromise;
}
async function requestQuit() {
await cleanup();
if (allowingQuit) return;
allowingQuit = true;
quitApp?.();
}
function handleBeforeQuit(event) {
if (allowingQuit) return;
try {
event?.preventDefault?.();
} catch {}
void requestQuit();
}
return {
cleanup,
requestQuit,
handleBeforeQuit,
isAllowingQuit: () => allowingQuit
};
}
export function registerShutdownSignals({ processLike = process, requestQuit } = {}) {
const handler = () => {
void requestQuit?.();
};
processLike?.on?.('SIGINT', handler);
processLike?.on?.('SIGTERM', handler);
return () => {
try {
processLike?.removeListener?.('SIGINT', handler);
} catch {}
try {
processLike?.removeListener?.('SIGTERM', handler);
} catch {}
};
}