Preserve and restore open editors and their view columns when duplicating a workspace

Author: robertpiosik

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

@@ -24,11 +24,27 @@ export function duplicate_workspace_command(
         vscode.workspace.workspaceFolders?.map((folder) => folder.uri.fsPath) ??
         []
 
+      const open_editors: { path: string; view_column?: number }[] = []
+      for (const tab_group of vscode.window.tabGroups.all) {
+        for (const tab of tab_group.tabs) {
+          if (
+            tab.input instanceof vscode.TabInputText &&
+            tab.input.uri.scheme == 'file'
+          ) {
+            open_editors.push({
+              path: tab.input.uri.fsPath,
+              view_column: tab_group.viewColumn
+            })
+          }
+        }
+      }
+
       const context_to_save: DuplicateWorkspaceContext = {
         checked_files,
         checked_websites,
         workspace_root_folders,
-        timestamp: Date.now()
+        timestamp: Date.now(),
+        open_editors
       }
 
       const api_edit_format = context.workspaceState.get(

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

@@ -77,6 +77,7 @@ export type DuplicateWorkspaceContext = {
   checked_websites: string[]
   timestamp: number
   workspace_root_folders: string[]
+  open_editors?: { path: string; view_column?: number }[]
 }
 
 export type HistoryEntry = {

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

@@ -61,6 +61,23 @@ const restore_duplicated_workspace_context = async (
           CONTEXT_CHECKED_URLS_STATE_KEY,
           duplicated_context.checked_websites
         )
+        if (duplicated_context.open_editors) {
+          for (const editor_info of duplicated_context.open_editors) {
+            try {
+              const uri = vscode.Uri.file(editor_info.path)
+              const document = await vscode.workspace.openTextDocument(uri)
+              await vscode.window.showTextDocument(document, {
+                viewColumn: editor_info.view_column,
+                preview: false
+              })
+            } catch (error) {
+              console.error(
+                `Failed to restore open editor for ${editor_info.path}:`,
+                error
+              )
+            }
+          }
+        }
       }
     }