Add MFA code request handler

Author: liramon1

packages/amazonq/src/lsp/client.ts

@@ -30,6 +30,9 @@ import {
     ResponseError,
     LSPErrorCodes,
     updateConfigurationRequestType,
+    GetMfaCodeParams,
+    GetMfaCodeResult,
+    getMfaCodeRequestType,
 } from '@aws/language-server-runtimes/protocol'
 import {
     AuthUtil,
@@ -342,6 +345,15 @@ async function postStartLanguageServer(
         }
     )
 
+    // Handler for when Flare needs to assume a role with MFA code
+    client.onRequest<GetMfaCodeParams, GetMfaCodeResult>(
+        getMfaCodeRequestType.method,
+        async (params: GetMfaCodeParams): Promise<GetMfaCodeResult> => {
+            const mfaCode = await vscode.window.showInputBox({ title: 'Enter MFA Code' })
+            return { code: mfaCode ?? '' }
+        }
+    )
+
     const sendProfileToLsp = async () => {
         try {
             const result = await client.sendRequest(updateConfigurationRequestType.method, {