feat: make task mode sticky across sessions

- Add mode field to HistoryItem schema to persist selected mode
- Update taskMetadata to save current mode when creating history items
- Update handleModeSwitch to save mode changes to task history for active tasks
- Restore saved mode when reopening tasks from history

This ensures that when a task is reopened, it automatically switches to the last mode that was active on that task.

Author: roomote

packages/types/src/history.ts

@@ -16,6 +16,7 @@ export const historyItemSchema = z.object({
 	totalCost: z.number(),
 	size: z.number().optional(),
 	workspace: z.string().optional(),
+	mode: z.string().optional(),
 })
 
 export type HistoryItem = z.infer<typeof historyItemSchema>

src/core/task-persistence/taskMetadata.ts

@@ -18,6 +18,7 @@ export type TaskMetadataOptions = {
 	taskNumber: number
 	globalStoragePath: string
 	workspace: string
+	mode?: string
 }
 
 export async function taskMetadata({
@@ -26,6 +27,7 @@ export async function taskMetadata({
 	taskNumber,
 	globalStoragePath,
 	workspace,
+	mode,
 }: TaskMetadataOptions) {
 	const taskDir = await getTaskDirectoryPath(globalStoragePath, taskId)
 
@@ -92,6 +94,7 @@ export async function taskMetadata({
 		totalCost: tokenUsage.totalCost,
 		size: taskDirSize,
 		workspace,
+		mode,
 	}
 
 	return { historyItem, tokenUsage }

src/core/task/Task.ts

@@ -405,12 +405,17 @@ export class Task extends EventEmitter<ClineEvents> {
 				globalStoragePath: this.globalStoragePath,
 			})
 
+			const provider = this.providerRef.deref()
+			const state = await provider?.getState()
+			const currentMode = state?.mode
+
 			const { historyItem, tokenUsage } = await taskMetadata({
 				messages: this.clineMessages,
 				taskId: this.taskId,
 				taskNumber: this.taskNumber,
 				globalStoragePath: this.globalStoragePath,
 				workspace: this.cwd,
+				mode: currentMode,
 			})
 
 			this.emit("taskTokenUsageUpdated", this.taskId, tokenUsage)

src/core/webview/ClineProvider.ts

@@ -578,6 +578,11 @@ export class ClineProvider
 	public async initClineWithHistoryItem(historyItem: HistoryItem & { rootTask?: Task; parentTask?: Task }) {
 		await this.removeClineFromStack()
 
+		// If the history item has a saved mode, restore it
+		if (historyItem.mode) {
+			await this.updateGlobalState("mode", historyItem.mode)
+		}
+
 		const {
 			apiConfiguration,
 			diffEnabled: enableDiff,
@@ -807,6 +812,14 @@ export class ClineProvider
 		if (cline) {
 			TelemetryService.instance.captureModeSwitch(cline.taskId, newMode)
 			cline.emit("taskModeSwitched", cline.taskId, newMode)
+
+			// Update the task history with the new mode
+			const history = this.getGlobalState("taskHistory") ?? []
+			const taskHistoryItem = history.find((item) => item.id === cline.taskId)
+			if (taskHistoryItem) {
+				taskHistoryItem.mode = newMode
+				await this.updateTaskHistory(taskHistoryItem)
+			}
 		}
 
 		await this.updateGlobalState("mode", newMode)