subtasks alpha version (still in development)

Author: ShayBC

src/activate/registerCommands.ts

@@ -20,7 +20,7 @@ export const registerCommands = (options: RegisterCommandOptions) => {
 const getCommandsMap = ({ context, outputChannel, provider }: RegisterCommandOptions) => {
 	return {
 		"roo-cline.plusButtonClicked": async () => {
-			await provider.clearTask()
+			await provider.removeClineFromStack()
 			await provider.postStateToWebview()
 			await provider.postMessageToWebview({ type: "action", action: "chatButtonClicked" })
 		},

src/core/Cline.ts

@@ -75,6 +75,10 @@ type UserContent = Array<
 
 export class Cline {
 	readonly taskId: string
+	// a flag that indicated if this Cline instance is a subtask (on finish return control to parent task)
+	private isSubTask: boolean = false
+	// a flag that indicated if this Cline instance is paused (waiting for provider to resume it after subtask completion)
+	private isPaused: boolean = false
 	api: ApiHandler
 	private terminalManager: TerminalManager
 	private urlContentFetcher: UrlContentFetcher
@@ -160,6 +164,12 @@ export class Cline {
 		}
 	}
 
+	// a helper function to set the private member isSubTask to true
+	// and by that set this Cline instance to be a subtask (on finish return control to parent task)
+	setSubTask() {
+		this.isSubTask = true
+	}
+
 	// Add method to update diffStrategy
 	async updateDiffStrategy(experimentalDiffStrategy?: boolean) {
 		// If not provided, get from current state
@@ -480,6 +490,43 @@ export class Cline {
 		])
 	}
 
+	async resumePausedTask() {
+		// release this Cline instance from paused state
+		this.isPaused = false
+
+		// Clear any existing ask state and simulate a completed ask response
+		// this.askResponse = "messageResponse";
+		// this.askResponseText = "Sub Task finished Successfully!\nthere is no need to perform this task again, please continue to the next task.";
+		// this.askResponseImages = undefined;
+		// this.lastMessageTs = Date.now();
+
+		// This adds the completion message to conversation history
+		await this.say(
+			"text",
+			"Sub Task finished Successfully!\nthere is no need to perform this task again, please continue to the next task.",
+		)
+
+		// this.userMessageContent.push({
+		// 	type: "text",
+		// 	text: `${"Result:\\n\\nSub Task finished Successfully!\nthere is no need to perform this task again, please continue to the next task."}`,
+		// })
+
+		try {
+			// Resume parent task
+			await this.ask("resume_task")
+		} catch (error) {
+			if (error.message === "Current ask promise was ignored") {
+				// ignore the ignored promise, since it was performed by launching a subtask and it probably took more then 1 sec,
+				// also set the didAlreadyUseTool flag to indicate that the tool was already used, and there is no need to relaunch it
+				this.didAlreadyUseTool = true
+			} else {
+				// Handle error appropriately
+				console.error("Failed to resume task:", error)
+				throw error
+			}
+		}
+	}
+
 	private async resumeTaskFromHistory() {
 		const modifiedClineMessages = await this.getSavedClineMessages()
 
@@ -2553,10 +2600,12 @@ export class Cline {
 								const provider = this.providerRef.deref()
 								if (provider) {
 									await provider.handleModeSwitch(mode)
-									await provider.initClineWithTask(message)
+									await provider.initClineWithSubTask(message)
 									pushToolResult(
 										`Successfully created new task in ${targetMode.name} mode with message: ${message}`,
 									)
+									// pasue the current task and start the new task
+									this.isPaused = true
 								} else {
 									pushToolResult(
 										formatResponse.toolError("Failed to create new task: provider not available"),
@@ -2648,6 +2697,10 @@ export class Cline {
 									if (lastMessage && lastMessage.ask !== "command") {
 										// havent sent a command message yet so first send completion_result then command
 										await this.say("completion_result", result, undefined, false)
+										if (this.isSubTask) {
+											// tell the provider to remove the current subtask and resume the previous task in the stack
+											this.providerRef.deref()?.finishSubTask()
+										}
 									}
 
 									// complete command message
@@ -2665,6 +2718,10 @@ export class Cline {
 									commandResult = execCommandResult
 								} else {
 									await this.say("completion_result", result, undefined, false)
+									if (this.isSubTask) {
+										// tell the provider to remove the current subtask and resume the previous task in the stack
+										this.providerRef.deref()?.finishSubTask()
+									}
 								}
 
 								// we already sent completion_result says, an empty string asks relinquishes control over button and field
@@ -2740,6 +2797,20 @@ export class Cline {
 		}
 	}
 
+	// this function checks if this Cline instance is set to pause state and wait for being resumed,
+	// this is used when a sub-task is launched and the parent task is waiting for it to finish
+	async waitForResume() {
+		// wait until isPaused is false
+		await new Promise<void>((resolve) => {
+			const interval = setInterval(() => {
+				if (!this.isPaused) {
+					clearInterval(interval)
+					resolve()
+				}
+			}, 1000) // TBD: the 1 sec should be added to the settings, also should add a timeout to prevent infinit wait
+		})
+	}
+
 	async recursivelyMakeClineRequests(
 		userContent: UserContent,
 		includeFileDetails: boolean = false,
@@ -2779,6 +2850,12 @@ export class Cline {
 			await this.checkpointSave({ isFirst: true })
 		}
 
+		// in this Cline request loop, we need to check if this cline (Task) instance has been asked to wait
+		// for a sub-task (it has launched) to finish before continuing
+		if (this.isPaused) {
+			await this.waitForResume()
+		}
+
 		// getting verbose details is an expensive operation, it uses globby to top-down build file structure of project which for large projects can take a few seconds
 		// for the best UX we show a placeholder api_req_started message with a loading spinner as this happens
 		await this.say(