Merge pull request #1579 from kumsmrit/kumsmrit/featureDev-bugfix

fix: Handle case when user cancels and close dialog without new folder selection

Author: jpinkney-aws

packages/toolkit/src/amazonqFeatureDev/controllers/chat/controller.ts

@@ -478,9 +478,6 @@ export class FeatureDevController {
         } finally {
             // Finish processing the event
             this.messenger.sendAsyncEventProgress(message.tabID, false, undefined)
-
-            // Lock the chat input until they explicitly click one of the follow ups
-            this.messenger.sendChatInputEnabled(message.tabID, false)
         }
     }
 
@@ -522,6 +519,21 @@ export class FeatureDevController {
             canSelectFiles: false,
         }).prompt()
 
+        if (!(uri instanceof vscode.Uri)) {
+            this.messenger.sendAnswer({
+                tabID: message.tabID,
+                type: 'system-prompt',
+                followUps: [
+                    {
+                        pillText: 'Select files for context',
+                        type: 'ModifyDefaultSourceFolder',
+                        status: 'info',
+                    },
+                ],
+            })
+            return
+        }
+
         if (uri instanceof vscode.Uri && !vscode.workspace.getWorkspaceFolder(uri)) {
             this.messenger.sendAnswer({
                 tabID: message.tabID,