[FEATURE] TaskUtil: Add 'force' flag to cleanup task callback

lib/build/ProjectBuilder.js

@@ -388,17 +388,18 @@ class ProjectBuilder {
 		}));
 	}
 
-	async _executeCleanupTasks() {
+	async _executeCleanupTasks(force) {
 		this.#log.info("Executing cleanup tasks...");
-		await this._buildContext.executeCleanupTasks();
+
+		await this._buildContext.executeCleanupTasks(force);
 	}
 
 	_registerCleanupSigHooks() {
 		const that = this;
 		function createListener(exitCode) {
 			return function() {
 				// Asynchronously cleanup resources, then exit
-				that._executeCleanupTasks().then(() => {
+				that._executeCleanupTasks(true).then(() => {
 					process.exit(exitCode);
 				});
 			};

lib/build/helpers/BuildContext.js

@@ -80,9 +80,9 @@ class BuildContext {
 		return projectBuildContext;
 	}
 
-	async executeCleanupTasks() {
+	async executeCleanupTasks(force) {
 		await Promise.all(this._projectBuildContexts.map((ctx) => {
-			return ctx.executeCleanupTasks();
+			return ctx.executeCleanupTasks(force);
 		}));
 	}
 }

lib/build/helpers/ProjectBuildContext.js

@@ -47,9 +47,9 @@ class ProjectBuildContext {
 		this._queues.cleanup.push(callback);
 	}
 
-	async executeCleanupTasks() {
+	async executeCleanupTasks(force) {
 		await Promise.all(this._queues.cleanup.map((callback) => {
-			return callback();
+			return callback(force);
 		}));
 	}
 

test/lib/build/ProjectBuilder.js

@@ -727,8 +727,16 @@ test("_executeCleanupTasks", async (t) => {
 	const executeCleanupTasksStub = sinon.stub(builder._buildContext, "executeCleanupTasks");
 	await builder._executeCleanupTasks();
 	t.is(executeCleanupTasksStub.callCount, 1, "BuildContext#executeCleanupTasks got called once");
-	t.deepEqual(executeCleanupTasksStub.getCall(0).args, [],
+	t.deepEqual(executeCleanupTasksStub.getCall(0).args, [undefined],
 		"BuildContext#executeCleanupTasks got called with no arguments");
+
+	// reset stub
+	executeCleanupTasksStub.reset();
+	// Call with enforcement flag
+	await builder._executeCleanupTasks(true);
+	t.is(executeCleanupTasksStub.callCount, 1, "BuildContext#executeCleanupTasks got called once");
+	t.deepEqual(executeCleanupTasksStub.getCall(0).args, [true],
+		"BuildContext#executeCleanupTasks got called with force flag");
 });
 
 test("instantiate new logger for every ProjectBuilder", async (t) => {