Implement the duplicateWorkspace command to save current context state checked files/websites before duplicating the workspace and restore this state in the new window

Author: robertpiosik

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

@@ -0,0 +1,42 @@
+import * as vscode from 'vscode'
+import { WorkspaceProvider } from '../context/providers/workspace-provider'
+import { WebsitesProvider } from '../context/providers/websites-provider'
+import {
+  DUPLICATE_WORKSPACE_CONTEXT_STATE_KEY,
+  type DuplicateWorkspaceContext
+} from '../constants/state-keys'
+
+export function duplicate_workspace_command(
+  workspace_provider: WorkspaceProvider,
+  websites_provider: WebsitesProvider,
+  context: vscode.ExtensionContext
+) {
+  return vscode.commands.registerCommand(
+    'codeWebChat.duplicateWorkspace',
+    async () => {
+      const checked_files = workspace_provider.get_all_checked_paths()
+      const checked_websites = websites_provider
+        .get_checked_websites()
+        .map((w) => w.url)
+      const workspace_root_folders =
+        vscode.workspace.workspaceFolders?.map((folder) => folder.uri.fsPath) ??
+        []
+
+      const context_to_save: DuplicateWorkspaceContext = {
+        checked_files,
+        checked_websites,
+        workspace_root_folders,
+        timestamp: Date.now()
+      }
+
+      await context.globalState.update(
+        DUPLICATE_WORKSPACE_CONTEXT_STATE_KEY,
+        context_to_save
+      )
+
+      vscode.commands.executeCommand(
+        'workbench.action.duplicateWorkspaceInNewWindow'
+      )
+    }
+  )
+}

packages/vscode/src/commands/index.ts

@@ -2,6 +2,7 @@ export * from './apply-chat-response-command'
 export * from './close-all-editors-command'
 export * from './checkpoints-command'
 export * from './close-editor-command'
+export * from './duplicate-workspace-command'
 export * from './code-completion-commands'
 export * from './delete-command'
 export * from './new-file-command'

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

@@ -31,6 +31,9 @@ export const LAST_REFACTOR_INSTRUCTION_STATE_KEY = 'last-refactor-instruction'
 export const CHECKPOINTS_STATE_KEY = 'checkpoints'
 export const TEMPORARY_CHECKPOINT_STATE_KEY = 'temporary-checkpoint'
 
+export const DUPLICATE_WORKSPACE_CONTEXT_STATE_KEY =
+  'duplicate-workspace-context'
+
 export const CHAT_EDIT_FORMAT_STATE_KEY = 'chat-edit-format'
 export const API_EDIT_FORMAT_STATE_KEY = 'api-edit-format'
 export const WEB_MODE_STATE_KEY = 'web-mode'
@@ -67,6 +70,13 @@ export const get_last_selected_preset_key = (web_mode: string) =>
 export const get_last_selected_group_state_key = (web_mode: string) =>
   `last-selected-group-${web_mode}`
 
+export interface DuplicateWorkspaceContext {
+  checked_files: string[]
+  checked_websites: string[]
+  timestamp: number
+  workspace_root_folders: string[]
+}
+
 export interface HistoryEntry {
   text: string
   createdAt: number

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

@@ -9,6 +9,12 @@ import { marked } from 'marked'
 import { EventEmitter } from 'events'
 import { apply_context_command } from '../commands/apply-context-command/apply-context-command'
 import { dictionary } from '@shared/constants/dictionary'
+import {
+  CONTEXT_CHECKED_PATHS_STATE_KEY,
+  CONTEXT_CHECKED_URLS_STATE_KEY,
+  DUPLICATE_WORKSPACE_CONTEXT_STATE_KEY,
+  type DuplicateWorkspaceContext
+} from '../constants/state-keys'
 import { ContextProvider } from './providers/context-provider'
 
 export const token_count_emitter = new EventEmitter()
@@ -20,13 +26,60 @@ const round_token_count_for_badge = (count: number): number => {
   return Math.floor(count / 1000) * 1000
 }
 
-export const context_initialization = (
+const restore_duplicated_workspace_context = async (
   context: vscode.ExtensionContext
-): {
+) => {
+  const duplicated_context = context.globalState.get<DuplicateWorkspaceContext>(
+    DUPLICATE_WORKSPACE_CONTEXT_STATE_KEY
+  )
+
+  if (duplicated_context?.timestamp) {
+    const now = Date.now()
+    const age = now - duplicated_context.timestamp
+
+    if (age <= 60000) {
+      const current_workspace_folders =
+        vscode.workspace.workspaceFolders?.map((folder) => folder.uri.fsPath) ??
+        []
+
+      const sorted_current_folders = [...current_workspace_folders].sort()
+      const sorted_saved_folders = [
+        ...(duplicated_context.workspace_root_folders ?? [])
+      ].sort()
+
+      const are_workspaces_the_same =
+        sorted_current_folders.length == sorted_saved_folders.length &&
+        sorted_current_folders.every(
+          (value, index) => value == sorted_saved_folders[index]
+        )
+      if (are_workspaces_the_same) {
+        await context.workspaceState.update(
+          CONTEXT_CHECKED_PATHS_STATE_KEY,
+          duplicated_context.checked_files
+        )
+        await context.workspaceState.update(
+          CONTEXT_CHECKED_URLS_STATE_KEY,
+          duplicated_context.checked_websites
+        )
+      }
+    }
+
+    await context.globalState.update(
+      DUPLICATE_WORKSPACE_CONTEXT_STATE_KEY,
+      undefined
+    )
+  }
+}
+
+export const context_initialization = async (
+  context: vscode.ExtensionContext
+): Promise<{
   workspace_provider: WorkspaceProvider
   open_editors_provider: OpenEditorsProvider
   websites_provider: WebsitesProvider
-} => {
+}> => {
+  await restore_duplicated_workspace_context(context)
+
   let was_above_threshold = false
 
   const workspace_folders = vscode.workspace.workspaceFolders ?? []

packages/vscode/src/extension.ts

@@ -17,6 +17,7 @@ import {
   new_file_command,
   open_file_from_workspace_command,
   new_folder_command,
+  duplicate_workspace_command,
   rename_command,
   delete_command,
   save_context_command,
@@ -31,7 +32,7 @@ let websocket_server_instance: WebSocketManager | null = null
 
 export async function activate(context: vscode.ExtensionContext) {
   const { workspace_provider, open_editors_provider, websites_provider } =
-    context_initialization(context)
+    await context_initialization(context)
 
   websocket_server_instance = new WebSocketManager(context, websites_provider)
 
@@ -89,11 +90,7 @@ export async function activate(context: vscode.ExtensionContext) {
     delete_command(),
     save_context_command(workspace_provider, context),
     ...checkpoints_command(workspace_provider, context),
-    vscode.commands.registerCommand('codeWebChat.duplicateWorkspace', () => {
-      vscode.commands.executeCommand(
-        'workbench.action.duplicateWorkspaceInNewWindow'
-      )
-    }),
+    duplicate_workspace_command(workspace_provider, websites_provider, context),
     open_url_command({
       command: 'codeWebChat.openRepository',
       url: 'https://github.com/robertpiosik/CodeWebChat'