Transition to workspace-level edits and remove automatic document opening across chat response handlers

Author: robertpiosik

packages/vscode/src/commands/apply-chat-response-command/handlers/conflict-markers-handler.ts

@@ -200,8 +200,6 @@ export const handle_conflict_markers = async (
           }
 
           await fs.promises.writeFile(safe_path, content_to_write, 'utf8')
-          const document = await vscode.workspace.openTextDocument(safe_path)
-          await vscode.window.showTextDocument(document, { preview: false })
 
           original_states.push({
             file_path: file.file_path,
@@ -217,23 +215,23 @@ export const handle_conflict_markers = async (
           const document = await vscode.workspace.openTextDocument(safe_path)
           const original_content = document.getText()
 
-          const editor = await vscode.window.showTextDocument(document)
           let final_content = file.content
           if (
             original_content.endsWith('\n') &&
             !final_content.endsWith('\n')
           ) {
             final_content += '\n'
           }
-          await editor.edit((edit) => {
-            edit.replace(
-              new vscode.Range(
-                document.positionAt(0),
-                document.positionAt(document.getText().length)
-              ),
-              final_content
-            )
-          })
+          const edit = new vscode.WorkspaceEdit()
+          edit.replace(
+            document.uri,
+            new vscode.Range(
+              document.positionAt(0),
+              document.positionAt(document.getText().length)
+            ),
+            final_content
+          )
+          await vscode.workspace.applyEdit(edit)
           await document.save()
 
           original_states.push({

packages/vscode/src/commands/apply-chat-response-command/handlers/diff-handler.ts

@@ -307,9 +307,6 @@ const handle_new_file_patch = async (
 
     await fs.promises.writeFile(safe_path, new_content, 'utf8')
 
-    const document = await vscode.workspace.openTextDocument(safe_path)
-    await vscode.window.showTextDocument(document, { preview: false })
-
     return { success: true, original_states }
   } catch (error: any) {
     Logger.error({

packages/vscode/src/commands/apply-chat-response-command/handlers/fast-replace-handler.ts

@@ -162,9 +162,6 @@ export const handle_fast_replace = async (
             Buffer.from(new_content, 'utf8')
           )
 
-          const document = await vscode.workspace.openTextDocument(safe_path)
-          await vscode.window.showTextDocument(document, { preview: false })
-
           original_states.push({
             file_path: file.file_path,
             content: rename_source_content,
@@ -201,23 +198,24 @@ export const handle_fast_replace = async (
             content: original_content,
             workspace_name: file.workspace_name
           })
-          const editor = await vscode.window.showTextDocument(document)
           let final_content = file.content
           if (
             original_content.endsWith('\n') &&
             !final_content.endsWith('\n')
           ) {
             final_content += '\n'
           }
-          await editor.edit((edit) => {
-            edit.replace(
-              new vscode.Range(
-                document.positionAt(0),
-                document.positionAt(document.getText().length)
-              ),
-              final_content
-            )
-          })
+
+          const edit = new vscode.WorkspaceEdit()
+          edit.replace(
+            document.uri,
+            new vscode.Range(
+              document.positionAt(0),
+              document.positionAt(document.getText().length)
+            ),
+            final_content
+          )
+          await vscode.workspace.applyEdit(edit)
 
           await document.save()
           Logger.info({
@@ -263,8 +261,6 @@ export const handle_fast_replace = async (
               message: 'New file created',
               data: safe_path
             })
-            const document = await vscode.workspace.openTextDocument(safe_path)
-            await vscode.window.showTextDocument(document, { preview: false })
           } catch (error) {
             Logger.error({
               function_name: 'handle_fast_replace',

packages/vscode/src/commands/apply-chat-response-command/handlers/intelligent-update-handler.ts

@@ -532,16 +532,16 @@ export const handle_intelligent_update = async (params: {
           continue
         }
         try {
-          const editor = await vscode.window.showTextDocument(document)
-          await editor.edit((edit) => {
-            edit.replace(
-              new vscode.Range(
-                document.positionAt(0),
-                document.positionAt(document.getText().length)
-              ),
-              change.content
-            )
-          })
+          const edit = new vscode.WorkspaceEdit()
+          edit.replace(
+            document.uri,
+            new vscode.Range(
+              document.positionAt(0),
+              document.positionAt(document.getText().length)
+            ),
+            change.content
+          )
+          await vscode.workspace.applyEdit(edit)
           await document.save()
         } catch (error) {
           Logger.error({

packages/vscode/src/commands/apply-chat-response-command/handlers/restore-preview-handler.ts

@@ -136,23 +136,23 @@ export const handle_restore_preview = async (
           try {
             const file_uri = vscode.Uri.file(safe_path)
             const document = await vscode.workspace.openTextDocument(file_uri)
-            const editor = await vscode.window.showTextDocument(document)
             let final_content = file.content
             if (
               original_content.endsWith('\n') &&
               !final_content.endsWith('\n')
             ) {
               final_content += '\n'
             }
-            await editor.edit((edit) => {
-              edit.replace(
-                new vscode.Range(
-                  document.positionAt(0),
-                  document.positionAt(document.getText().length)
-                ),
-                final_content
-              )
-            })
+            const edit = new vscode.WorkspaceEdit()
+            edit.replace(
+              document.uri,
+              new vscode.Range(
+                document.positionAt(0),
+                document.positionAt(document.getText().length)
+              ),
+              final_content
+            )
+            await vscode.workspace.applyEdit(edit)
 
             await document.save()
             Logger.info({
@@ -202,8 +202,6 @@ export const handle_restore_preview = async (
               message: 'New file created',
               data: safe_path
             })
-            const document = await vscode.workspace.openTextDocument(safe_path)
-            await vscode.window.showTextDocument(document, { preview: false })
           } catch (error) {
             Logger.error({
               function_name: 'handle_restore_preview',

packages/vscode/src/commands/apply-chat-response-command/utils/file-operations.ts

@@ -152,8 +152,6 @@ export const create_file_if_needed = async (params: {
       message: 'File created',
       data: safe_path
     })
-    const document = await vscode.workspace.openTextDocument(safe_path)
-    await vscode.window.showTextDocument(document, { preview: false })
     return true
   } catch (error) {
     Logger.error({
@@ -228,10 +226,6 @@ const relocate_file = async (params: {
       workspace_root: params.workspace_root
     })
 
-    // Open the file at new location
-    const document = await vscode.workspace.openTextDocument(new_safe_path)
-    await vscode.window.showTextDocument(document)
-
     Logger.info({
       function_name: 'relocate_file',
       message: 'File successfully relocated',