feat: Implement file creation confirmation and handling in chat responses

Author: TheUnknown550

vscode-extension/src/extension.ts

@@ -228,6 +228,28 @@ async function handleChatMessage(message: any) {
         const isCode = response.completion_type === 'code';
         console.log('Sending message to webview:', isCode ? 'codeResponse' : 'chatResponse', response.completion);
 
+        // Detect file creation intent (e.g., response starts with 'Filename: ...')
+        const fileMatch = response.completion.match(/^Filename:\s*(\S+)\s*\n([\s\S]*)/);
+        if (fileMatch) {
+            const filename = fileMatch[1];
+            const fileContent = fileMatch[2].trim();
+            // Ask user to confirm file creation
+            const confirm = await vscode.window.showInformationMessage(
+                `AI wants to create file: ${filename}. Proceed?`,
+                'Create', 'Cancel'
+            );
+            if (confirm === 'Create') {
+                await createFileInWorkspace(filename, fileContent);
+                vscode.window.showInformationMessage(`File '${filename}' created!`);
+            }
+            currentPanel?.webview.postMessage({
+                type: 'chatResponse',
+                content: `File creation: ${filename}\n\n${fileContent}`,
+                loading: false
+            });
+            return;
+        }
+
         if (isCode) {
             currentPanel?.webview.postMessage({
                 type: 'codeResponse',
@@ -243,6 +265,21 @@ async function handleChatMessage(message: any) {
                 loading: false
             });
         }
+// Create a new file in the workspace
+async function createFileInWorkspace(filename: string, content: string) {
+    const folders = vscode.workspace.workspaceFolders;
+    if (!folders || folders.length === 0) {
+        vscode.window.showErrorMessage('No workspace folder open.');
+        return;
+    }
+    const folderUri = folders[0].uri;
+    const fileUri = vscode.Uri.joinPath(folderUri, filename);
+    try {
+        await vscode.workspace.fs.writeFile(fileUri, Buffer.from(content, 'utf8'));
+    } catch (e) {
+        vscode.window.showErrorMessage(`Failed to create file: ${e}`);
+    }
+}
 
     } catch (err: any) {
         console.error('Exception in handleChatMessage:', err);

vscode-extension/src/webview.js

@@ -14,44 +14,43 @@ alert('Webview loaded!');
         messageDiv.className = `message ${type}`;
 
         if (isCode) {
+            // Detect file creation pattern
+            const fileMatch = content.match(/^Filename:\s*(\S+)\s*\n([\s\S]*)/);
             const pre = document.createElement('pre');
             const code = document.createElement('code');
             code.textContent = content;
             pre.appendChild(code);
             messageDiv.appendChild(pre);
-            
-            // Add action buttons for code
-            const actionsDiv = document.createElement('div');
-            actionsDiv.className = 'code-actions';
-            
-            const insertBtn = document.createElement('button');
-            insertBtn.textContent = 'Insert at Cursor';
-            insertBtn.onclick = () => {
-                vscode.postMessage({
-                    type: 'insertCode',
-                    code: content
-                });
-            };
-            
-            const replaceBtn = document.createElement('button');
-            replaceBtn.textContent = 'Replace Selection';
-            replaceBtn.onclick = () => {
-                vscode.postMessage({
-                    type: 'replaceCode',
-                    code: content
-                });
-            };
-            
-            const copyBtn = document.createElement('button');
-            copyBtn.textContent = 'Copy';
-            copyBtn.onclick = () => {
-                navigator.clipboard.writeText(content);
-            };
-            
-            actionsDiv.appendChild(insertBtn);
-            actionsDiv.appendChild(replaceBtn);
-            actionsDiv.appendChild(copyBtn);
-            messageDiv.appendChild(actionsDiv);
+            if (!fileMatch) {
+                // Add action buttons for code (not for file creation)
+                const actionsDiv = document.createElement('div');
+                actionsDiv.className = 'code-actions';
+                const insertBtn = document.createElement('button');
+                insertBtn.textContent = 'Insert at Cursor';
+                insertBtn.onclick = () => {
+                    vscode.postMessage({
+                        type: 'insertCode',
+                        code: content
+                    });
+                };
+                const replaceBtn = document.createElement('button');
+                replaceBtn.textContent = 'Replace Selection';
+                replaceBtn.onclick = () => {
+                    vscode.postMessage({
+                        type: 'replaceCode',
+                        code: content
+                    });
+                };
+                const copyBtn = document.createElement('button');
+                copyBtn.textContent = 'Copy';
+                copyBtn.onclick = () => {
+                    navigator.clipboard.writeText(content);
+                };
+                actionsDiv.appendChild(insertBtn);
+                actionsDiv.appendChild(replaceBtn);
+                actionsDiv.appendChild(copyBtn);
+                messageDiv.appendChild(actionsDiv);
+            }
         } else {
             // Format text with markdown-like support
             const formattedContent = formatText(content);