feat(Refactor user message bubble): When user uses refactoring comman…

…d we can show the prompt which user has entered in the user message bubbles
CommandDash · Feb 27, 2024 · a896f22 · a896f22
1 parent 91f8e7a
commit a896f22
Show file tree

Hide file tree

Showing 5 changed files with 55 additions and 41 deletions.
diff --git a/user_config.json b/user_config.json
@@ -0,0 +1 @@
+{"version":"0.2.1"}
diff --git a/vscode/media/onboarding/onboarding.js b/vscode/media/onboarding/onboarding.js
diff --git a/vscode/src/providers/chat_view_provider.ts b/vscode/src/providers/chat_view_provider.ts
@@ -158,6 +158,8 @@ export class FlutterGPTViewProvider implements vscode.WebviewViewProvider {
         const chipsData = data.chipsData;
         data.message = data.message.replace(`/${actionType}`, '').trim();
         if (actionType === 'refactor') {
+            this._publicConversationHistory.push({ role: 'user', parts: data.message });
+            this._view?.webview.postMessage({ type: 'displayMessages', value: this._publicConversationHistory });
             this._view?.webview.postMessage({ type: 'showLoadingIndicator' });
             const result = await RefactorActionManager.handleRequest(chipsData, data, this.aiRepo!, this.context, this.analyzer!, this);
             this._view?.webview.postMessage({ type: 'hideLoadingIndicator' });

diff --git a/vscode/src/tools/refactor/refactor_from_instructions.ts b/vscode/src/tools/refactor/refactor_from_instructions.ts
@@ -46,6 +46,7 @@ export async function refactorCode(generationRepository: GenerationRepository, g
 
             const workspaceFolders = vscode.workspace.workspaceFolders;
             let relativePath = editor.document.fileName;
+            const referenceContent = editor.document.getText(editor.selection);
             if (workspaceFolders && workspaceFolders.length > 0) {
                 const workspaceRoot = workspaceFolders[0].uri.fsPath;
                 relativePath = path.relative(workspaceRoot, editor.document.fileName);
@@ -62,7 +63,7 @@ export async function refactorCode(generationRepository: GenerationRepository, g
             });
             flutterGPTViewProvider.postMessageToWebview({
                 type: 'addToReference', value: JSON.stringify({
-                    relativePath: relativePath.trim(), referenceContent: '', referenceData: {
+                    relativePath: relativePath.trim(), referenceContent: `\`\n${relativePath.trim()}\n\`\n\`\`\`\n${referenceContent.toString()}\n\`\`\`\n`, referenceData: {
                         'selection': {
                             'start': {
                                 'line': replaceRange.start.line,

diff --git a/vscode/src/tools/reference/add_reference.ts b/vscode/src/tools/reference/add_reference.ts
@@ -26,6 +26,7 @@ export async function addToReference(globalState: vscode.Memento, flutterGPTView
     relativePath = path.relative(workspaceRoot, editor.document.fileName);
   }
 
+  debugger;
   flutterGPTViewProvider.postMessageToWebview({
     type: 'addToReference', value: JSON.stringify({
       relativePath: relativePath.trim(), referenceContent: `\`\n${relativePath.trim()}\n\`\n\`\`\`\n${referenceContent.toString()}\n\`\`\`\n`, referenceData: {