feat: improve multi-folder workspace support

- Fix Task to store initial workspace folder and use it consistently throughout task lifetime
- Fix file mentions to use relative paths in multi-folder workspaces
- Fix .roo folder detection to check if .roo is a workspace folder itself
- Add support for workspace folder selection based on active editor
- Add tests for multi-folder workspace functionality

Fixes #6897 and #6720

Author: roomote

src/core/config/CustomModesManager.ts

@@ -96,8 +96,18 @@ export class CustomModesManager {
 			return undefined
 		}
 
-		const workspaceRoot = getWorkspacePath()
-		const roomodesPath = path.join(workspaceRoot, ROOMODES_FILENAME)
+		// Check if .roo is a workspace folder itself
+		const rooWorkspaceFolder = workspaceFolders.find((folder) => path.basename(folder.uri.fsPath) === ".roo")
+
+		let roomodesPath: string
+		if (rooWorkspaceFolder) {
+			// .roo is a workspace folder itself, look for .roomodes directly in it
+			roomodesPath = path.join(rooWorkspaceFolder.uri.fsPath, ROOMODES_FILENAME.replace(".roo/", ""))
+		} else {
+			// Use the current workspace path (which considers the active file's workspace)
+			const workspaceRoot = getWorkspacePath()
+			roomodesPath = path.join(workspaceRoot, ROOMODES_FILENAME)
+		}
 		const exists = await fileExistsAtPath(roomodesPath)
 		return exists ? roomodesPath : undefined
 	}

src/core/mentions/index.ts

@@ -274,20 +274,41 @@ async function getFileOrFolderContent(
 	maxReadFileLine?: number,
 ): Promise<string> {
 	const unescapedPath = unescapeSpaces(mentionPath)
-	const absPath = path.resolve(cwd, unescapedPath)
+
+	// Check if this is an absolute path from a different workspace folder
+	let absPath: string
+	let displayPath: string = mentionPath
+
+	if (path.isAbsolute(unescapedPath)) {
+		absPath = unescapedPath
+
+		// In multi-folder workspaces, convert absolute paths to relative paths
+		// based on the workspace folder they belong to
+		if (vscode.workspace.workspaceFolders && vscode.workspace.workspaceFolders.length > 1) {
+			const workspaceFolder = vscode.workspace.getWorkspaceFolder(vscode.Uri.file(absPath))
+			if (workspaceFolder) {
+				// Use relative path from the workspace folder
+				const relativePath = path.relative(workspaceFolder.uri.fsPath, absPath)
+				displayPath = path.join(workspaceFolder.name, relativePath).toPosix()
+			}
+		}
+	} else {
+		absPath = path.resolve(cwd, unescapedPath)
+		displayPath = mentionPath
+	}
 
 	try {
 		const stats = await fs.stat(absPath)
 
 		if (stats.isFile()) {
 			if (rooIgnoreController && !rooIgnoreController.validateAccess(absPath)) {
-				return `(File ${mentionPath} is ignored by .rooignore)`
+				return `(File ${displayPath} is ignored by .rooignore)`
 			}
 			try {
 				const content = await extractTextFromFile(absPath, maxReadFileLine)
 				return content
 			} catch (error) {
-				return `(Failed to read contents of ${mentionPath}): ${error.message}`
+				return `(Failed to read contents of ${displayPath}): ${error.message}`
 			}
 		} else if (stats.isDirectory()) {
 			const entries = await fs.readdir(absPath, { withFileTypes: true })
@@ -315,8 +336,24 @@ async function getFileOrFolderContent(
 				if (entry.isFile()) {
 					folderContent += `${linePrefix}${displayName}\n`
 					if (!isIgnored) {
-						const filePath = path.join(mentionPath, entry.name)
 						const absoluteFilePath = path.resolve(absPath, entry.name)
+
+						// Determine the display path for the file
+						let fileDisplayPath: string
+						if (vscode.workspace.workspaceFolders && vscode.workspace.workspaceFolders.length > 1) {
+							const workspaceFolder = vscode.workspace.getWorkspaceFolder(
+								vscode.Uri.file(absoluteFilePath),
+							)
+							if (workspaceFolder) {
+								const relativePath = path.relative(workspaceFolder.uri.fsPath, absoluteFilePath)
+								fileDisplayPath = path.join(workspaceFolder.name, relativePath).toPosix()
+							} else {
+								fileDisplayPath = path.join(displayPath, entry.name).toPosix()
+							}
+						} else {
+							fileDisplayPath = path.join(displayPath, entry.name).toPosix()
+						}
+
 						fileContentPromises.push(
 							(async () => {
 								try {
@@ -325,7 +362,7 @@ async function getFileOrFolderContent(
 										return undefined
 									}
 									const content = await extractTextFromFile(absoluteFilePath, maxReadFileLine)
-									return `<file_content path="${filePath.toPosix()}">\n${content}\n</file_content>`
+									return `<file_content path="${fileDisplayPath}">\n${content}\n</file_content>`
 								} catch (error) {
 									return undefined
 								}
@@ -341,10 +378,10 @@ async function getFileOrFolderContent(
 			const fileContents = (await Promise.all(fileContentPromises)).filter((content) => content)
 			return `${folderContent}\n${fileContents.join("\n\n")}`.trim()
 		} else {
-			return `(Failed to read contents of ${mentionPath})`
+			return `(Failed to read contents of ${displayPath})`
 		}
 	} catch (error) {
-		throw new Error(`Failed to access path "${mentionPath}": ${error.message}`)
+		throw new Error(`Failed to access path "${displayPath}": ${error.message}`)
 	}
 }
 

src/core/task/Task.ts

@@ -279,10 +279,23 @@ export class Task extends EventEmitter<TaskEvents> implements TaskLike {
 
 		this.taskId = historyItem ? historyItem.id : crypto.randomUUID()
 
-		// Normal use-case is usually retry similar history task with new workspace.
-		this.workspacePath = parentTask
-			? parentTask.workspacePath
-			: getWorkspacePath(path.join(os.homedir(), "Desktop"))
+		// Store the initial workspace path when the task is created
+		// In multi-folder workspaces, we want to maintain the same workspace context throughout the task
+		if (parentTask) {
+			// Inherit workspace from parent task
+			this.workspacePath = parentTask.workspacePath
+		} else {
+			// Determine workspace based on current context
+			// If there's an active editor, use its workspace folder
+			// Otherwise, use the first workspace folder or fallback to Desktop
+			const currentFileUri = vscode.window.activeTextEditor?.document.uri
+			if (currentFileUri) {
+				const workspaceFolder = vscode.workspace.getWorkspaceFolder(currentFileUri)
+				this.workspacePath = workspaceFolder?.uri.fsPath || getWorkspacePath(path.join(os.homedir(), "Desktop"))
+			} else {
+				this.workspacePath = getWorkspacePath(path.join(os.homedir(), "Desktop"))
+			}
+		}
 
 		this.instanceId = crypto.randomUUID().slice(0, 8)
 		this.taskNumber = -1