scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-09-07 08:26:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor(migration): Isolate raw context access to migration logic

Browse source
This commit is contained in:
Daniel Riccio 2025-07-02 12:38:13 -05:00
parent 9b68fe09b5
commit f706071c34
No known key found for this signature in database
GPG key ID: FFD5FD825F8E8209
1 changed files with 14 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
src/utils/migrateSettings.ts
View file

@ -8,6 +8,7 @@ import type { ContextProxy } from "../core/config/ContextProxy"
import type { HistoryItem } from "@roo-code/types"
const deprecatedCustomModesJSONFilename = "custom_modes.json"
const TASK_HISTORY_MIGRATION_KEY = "taskHistoryMigratedToWorkspace"
/**
* Migrates old settings files to new file names
@ -61,7 +62,7 @@ export async function migrateSettings(
// Migrate task history from global state to workspace state using ContextProxy
const workspaceFolder = vscode.workspace.workspaceFolders?.[0]
await migrateTaskHistoryWithContextProxy(contextProxy, workspaceFolder)
await migrateTaskHistoryWithContextProxy(context, contextProxy, workspaceFolder)
} catch (error) {
outputChannel.appendLine(`Error in file migrations: ${error}`)
}
@ -124,22 +125,26 @@ async function migrateCustomModesToYaml(settingsDir: string, outputChannel: vsco
/**
* Migrates task history from global state to workspace state using ContextProxy
* This is used for the new architecture with ContextProxy
* @param context The original vscode.ExtensionContext for raw state access
* @param contextProxy The context proxy instance
* @param workspaceFolder The current workspace folder
*/
export async function migrateTaskHistoryWithContextProxy(
context: vscode.ExtensionContext,
contextProxy: ContextProxy,
workspaceFolder: vscode.WorkspaceFolder | undefined,
): Promise<void> {
const alreadyMigrated = context.globalState.get<boolean>(TASK_HISTORY_MIGRATION_KEY)
if (alreadyMigrated) {
return
}
if (!workspaceFolder) {
return
}
try {
// Access the raw context to get the global state directly
const context = (contextProxy as any).context as vscode.ExtensionContext
// Get the raw global state
// Get the raw global state directly from context
const rawGlobalState = context.globalState.get<any>("globalSettings", {})
const taskHistory = rawGlobalState.taskHistory as HistoryItem[] | undefined
@ -178,8 +183,11 @@ export async function migrateTaskHistoryWithContextProxy(
delete rawGlobalState.taskHistory
}
// Update the raw global state directly
// Update the raw global state directly using context
await context.globalState.update("globalSettings", rawGlobalState)
// Set the migration flag to prevent future migrations
await context.globalState.update(TASK_HISTORY_MIGRATION_KEY, true)
} catch (error) {
console.error("Failed to migrate task history to workspace:", error)
}