|
| 1 | +import { rmSync } from 'node:fs' |
| 2 | +import process from 'node:process' |
| 3 | + |
| 4 | +// Tracks the project folder currently being scaffolded so an interrupt can remove the partial |
| 5 | +// directory. Only ever holds a folder the installer created this run (callers validate that the |
| 6 | +// directory did not exist before starting), so removing it on abort never touches user data. |
| 7 | +let activeProjectFolder: string | undefined |
| 8 | +let signalHandlersRegistered = false |
| 9 | + |
| 10 | +type RemoveDirectory = (path: string, options: { recursive: boolean; force: boolean }) => void |
| 11 | + |
| 12 | +// Removes the in-progress project folder, if any, then clears the active reference so a finished |
| 13 | +// or already-removed project is never deleted. `rm` is injectable for testing. |
| 14 | +export function removeActiveProject(rm: RemoveDirectory = rmSync): void { |
| 15 | + if (activeProjectFolder === undefined) { |
| 16 | + return |
| 17 | + } |
| 18 | + |
| 19 | + const folder = activeProjectFolder |
| 20 | + activeProjectFolder = undefined |
| 21 | + rm(folder, { recursive: true, force: true }) |
| 22 | +} |
| 23 | + |
| 24 | +function handleAbort(signal: NodeJS.Signals): void { |
| 25 | + removeActiveProject() |
| 26 | + // Conventional exit code for a signal is 128 + signal number (SIGINT 2 → 130, SIGTERM 15 → 143). |
| 27 | + process.exit(signal === 'SIGTERM' ? 143 : 130) |
| 28 | +} |
| 29 | + |
| 30 | +// Marks the start of disk-writing work. Registers interrupt handlers on first use. |
| 31 | +export function beginInstall(projectFolder: string): void { |
| 32 | + activeProjectFolder = projectFolder |
| 33 | + |
| 34 | + if (!signalHandlersRegistered) { |
| 35 | + process.on('SIGINT', handleAbort) |
| 36 | + process.on('SIGTERM', handleAbort) |
| 37 | + signalHandlersRegistered = true |
| 38 | + } |
| 39 | +} |
| 40 | + |
| 41 | +// Marks the scaffold complete; an interrupt after this point leaves the finished project intact. |
| 42 | +export function completeInstall(): void { |
| 43 | + activeProjectFolder = undefined |
| 44 | +} |
0 commit comments