98 lines
2.7 KiB
TypeScript
98 lines
2.7 KiB
TypeScript
import * as fs from 'fs';
|
|
|
|
interface IPendingWrite {
|
|
lastSize: number;
|
|
lastChange: number;
|
|
timeoutId: ReturnType<typeof setTimeout>;
|
|
resolve: (stats: fs.Stats) => void;
|
|
reject: (error: Error) => void;
|
|
}
|
|
|
|
/**
|
|
* Implements awaitWriteFinish functionality by polling file size until stable.
|
|
* This replaces chokidar's built-in write stabilization.
|
|
*/
|
|
export class WriteStabilizer {
|
|
private pendingWrites = new Map<string, IPendingWrite>();
|
|
|
|
constructor(
|
|
private stabilityThreshold: number = 300,
|
|
private pollInterval: number = 100
|
|
) {}
|
|
|
|
/**
|
|
* Wait for a file write to complete by polling until size is stable
|
|
*/
|
|
async waitForWriteFinish(filePath: string): Promise<fs.Stats> {
|
|
// Cancel any existing pending check for this file
|
|
this.cancel(filePath);
|
|
|
|
return new Promise((resolve, reject) => {
|
|
const poll = async () => {
|
|
try {
|
|
const stats = await fs.promises.stat(filePath);
|
|
const pending = this.pendingWrites.get(filePath);
|
|
|
|
if (!pending) {
|
|
// Was cancelled
|
|
return;
|
|
}
|
|
|
|
const now = Date.now();
|
|
|
|
if (stats.size !== pending.lastSize) {
|
|
// Size changed - file is still being written, reset timer
|
|
pending.lastSize = stats.size;
|
|
pending.lastChange = now;
|
|
pending.timeoutId = setTimeout(poll, this.pollInterval);
|
|
} else if (now - pending.lastChange >= this.stabilityThreshold) {
|
|
// Size has been stable for the threshold duration
|
|
this.pendingWrites.delete(filePath);
|
|
resolve(stats);
|
|
} else {
|
|
// Size is the same but not yet past threshold
|
|
pending.timeoutId = setTimeout(poll, this.pollInterval);
|
|
}
|
|
} catch (error: any) {
|
|
this.pendingWrites.delete(filePath);
|
|
if (error.code === 'ENOENT') {
|
|
// File was deleted during polling
|
|
reject(new Error(`File was deleted: ${filePath}`));
|
|
} else {
|
|
reject(error);
|
|
}
|
|
}
|
|
};
|
|
|
|
this.pendingWrites.set(filePath, {
|
|
lastSize: -1,
|
|
lastChange: Date.now(),
|
|
timeoutId: setTimeout(poll, this.pollInterval),
|
|
resolve,
|
|
reject
|
|
});
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Cancel any pending write stabilization for a file
|
|
*/
|
|
cancel(filePath: string): void {
|
|
const pending = this.pendingWrites.get(filePath);
|
|
if (pending) {
|
|
clearTimeout(pending.timeoutId);
|
|
this.pendingWrites.delete(filePath);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Cancel all pending write stabilizations
|
|
*/
|
|
cancelAll(): void {
|
|
for (const [filePath, pending] of this.pendingWrites) {
|
|
clearTimeout(pending.timeoutId);
|
|
}
|
|
this.pendingWrites.clear();
|
|
}
|
|
}
|