Include saved contexts and file ranges in duplicated workspace state

Author: robertpiosik

packages/vscode/src/commands/duplicate-workspace-command.ts

@@ -5,8 +5,11 @@ import {
   API_EDIT_FORMAT_STATE_KEY,
   CHAT_EDIT_FORMAT_STATE_KEY,
   DUPLICATE_WORKSPACE_CONTEXT_STATE_KEY,
+  SAVED_CONTEXTS_STATE_KEY,
+  RANGES_STATE_KEY,
   type DuplicateWorkspaceContext
 } from '../constants/state-keys'
+import { SavedContext } from '@/types/context'
 
 export function duplicate_workspace_command(
   workspace_provider: WorkspaceProvider,
@@ -39,12 +42,19 @@ export function duplicate_workspace_command(
         }
       }
 
+      const ranges =
+        context.workspaceState.get<Record<string, string>>(RANGES_STATE_KEY)
+      const saved_contexts = context.workspaceState.get<SavedContext[]>(
+        SAVED_CONTEXTS_STATE_KEY
+      )
       const context_to_save: DuplicateWorkspaceContext = {
         checked_files,
         checked_websites,
         workspace_root_folders,
         timestamp: Date.now(),
-        open_editors
+        open_editors,
+        ranges,
+        saved_contexts
       }
 
       const api_edit_format = context.workspaceState.get(

packages/vscode/src/constants/state-keys.ts

@@ -1,3 +1,5 @@
+import { SavedContext } from '@/types/context'
+
 // Note: Use kebab-case for new keys
 export const LAST_APPLIED_CHANGES_STATE_KEY = 'last-applied-changes-state'
 export const LAST_APPLIED_CHANGES_EDITOR_STATE_STATE_KEY =
@@ -79,6 +81,8 @@ export type DuplicateWorkspaceContext = {
   timestamp: number
   workspace_root_folders: string[]
   open_editors?: { path: string; view_column?: number }[]
+  ranges?: Record<string, string>
+  saved_contexts?: SavedContext[]
 }
 
 export type HistoryEntry = {

packages/vscode/src/context/context-initialization.ts

@@ -13,6 +13,8 @@ import {
   CONTEXT_CHECKED_PATHS_STATE_KEY,
   CONTEXT_CHECKED_URLS_STATE_KEY,
   DUPLICATE_WORKSPACE_CONTEXT_STATE_KEY,
+  SAVED_CONTEXTS_STATE_KEY,
+  RANGES_STATE_KEY,
   type DuplicateWorkspaceContext
 } from '../constants/state-keys'
 import { ContextProvider } from './providers/context-provider'
@@ -61,6 +63,18 @@ const restore_duplicated_workspace_context = async (
           CONTEXT_CHECKED_URLS_STATE_KEY,
           duplicated_context.checked_websites
         )
+        if (duplicated_context.ranges) {
+          await context.workspaceState.update(
+            RANGES_STATE_KEY,
+            duplicated_context.ranges
+          )
+        }
+        if (duplicated_context.saved_contexts) {
+          await context.workspaceState.update(
+            SAVED_CONTEXTS_STATE_KEY,
+            duplicated_context.saved_contexts
+          )
+        }
         if (duplicated_context.open_editors) {
           for (const editor_info of duplicated_context.open_editors) {
             try {