refactor insert code pill text logic

Author: kelvin-klchu

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

@@ -289,7 +289,11 @@ export class FeatureDevController {
                     tabID: message.tabID,
                     followUps: [
                         {
-                            pillText: i18n('AWS.amazonq.featureDev.pillText.acceptAllChanges'),
+                            pillText:
+                                session?.getInsertCodePillText([
+                                    ...(session?.state.filePaths ?? []),
+                                    ...(session?.state.deletedFiles ?? []),
+                                ]) ?? i18n('AWS.amazonq.featureDev.pillText.acceptAllChanges'),
                             type: FollowUpTypes.InsertCode,
                             icon: 'ok' as MynahIcons,
                             status: 'success',

packages/core/src/amazonqFeatureDev/session/session.ts

@@ -157,16 +157,7 @@ export class Session {
     public async updateFilesPaths(params: UpdateFilesPathsParams) {
         const { tabID, filePaths, deletedFiles, messageId, disableFileActions = false } = params
         this.messenger.updateFileComponent(tabID, filePaths, deletedFiles, messageId, disableFileActions)
-        const allFiles = [...filePaths, ...deletedFiles]
-        if (allFiles.some((file) => file.rejected || file.changeApplied)) {
-            if (allFiles.every((file) => file.rejected || file.changeApplied)) {
-                await this.updateChatAnswer(tabID, i18n('AWS.amazonq.featureDev.pillText.continue'))
-            } else {
-                await this.updateChatAnswer(tabID, i18n('AWS.amazonq.featureDev.pillText.acceptRemainingChanges'))
-            }
-        } else {
-            await this.updateChatAnswer(tabID, i18n('AWS.amazonq.featureDev.pillText.acceptAllChanges'))
-        }
+        await this.updateChatAnswer(tabID, this.getInsertCodePillText([...filePaths, ...deletedFiles]))
     }
 
     public async updateChatAnswer(tabID: string, insertCodePillText: string) {
@@ -272,6 +263,16 @@ export class Session {
         this._acceptCodeTelemetrySent = sent
     }
 
+    public getInsertCodePillText(files: Array<NewFileInfo | DeletedFileInfo>) {
+        if (files.every((file) => file.rejected || file.changeApplied)) {
+            return i18n('AWS.amazonq.featureDev.pillText.continue')
+        }
+        if (files.some((file) => file.rejected || file.changeApplied)) {
+            return i18n('AWS.amazonq.featureDev.pillText.acceptRemainingChanges')
+        }
+        return i18n('AWS.amazonq.featureDev.pillText.acceptAllChanges')
+    }
+
     get state() {
         if (!this._state) {
             throw new Error("State should be initialized before it's read")