fix rerun

Signed-off-by: Roman Nikitenko <rnikiten@redhat.com>

Author: RomanNikitenko

code/src/vs/workbench/contrib/tasks/browser/terminalTaskSystem.ts

@@ -153,6 +153,7 @@ export class TerminalTaskSystem extends Disposable implements ITaskSystem {
 
 	private _activeTasks: IStringDictionary<IActiveTerminalData>;
 	private _busyTasks: IStringDictionary<Task>;
+	private _taskByMapKey: Map<string, Task>;
 	private _taskErrors: IStringDictionary<boolean>; // Tracks which tasks had errors from problem matchers
 	private _taskDependencies: IStringDictionary<string[]>; // Tracks which tasks depend on which other tasks
 	private _terminals: IStringDictionary<ITerminalData>;
@@ -215,6 +216,7 @@ export class TerminalTaskSystem extends Disposable implements ITaskSystem {
 
 		this._activeTasks = Object.create(null);
 		this._busyTasks = Object.create(null);
+		this._taskByMapKey = new Map();
 		this._taskErrors = Object.create(null);
 		this._taskDependencies = Object.create(null);
 		this._terminals = Object.create(null);
@@ -523,6 +525,7 @@ export class TerminalTaskSystem extends Disposable implements ITaskSystem {
 		this._showTaskLoadErrors(task);
 
 		const mapKey = task.getMapKey();
+		this._taskByMapKey.set(mapKey, task);
 
 		// It's important that we add this task's entry to _activeTasks before
 		// any of the code in the then runs (see #180541 and #180578). Wrapping
@@ -1931,6 +1934,16 @@ export class TerminalTaskSystem extends Disposable implements ITaskSystem {
 				return activeTask.task;
 			}
 		}
+		for (const key in this._terminals) {
+			const terminalData = this._terminals[key];
+			if (terminalData.terminal?.instanceId === instanceId) {
+				const task = this._taskByMapKey.get(terminalData.lastTask);
+				if (task) {
+					this._log(`TaskSystem.getTaskForTerminal found cached task (instanceId=${instanceId}, mapKey=${terminalData.lastTask}, label=${task._label})`);
+					return task;
+				}
+			}
+		}
 		this._log(`TaskSystem.getTaskForTerminal found no active task (instanceId=${instanceId})`);
 		return undefined;
 	}