Use toolkit MFA prompt in Q

Author: liramon1

packages/amazonq/src/lsp/client.ts

@@ -58,7 +58,7 @@ import { processUtils } from 'aws-core-vscode/shared'
 import { activate as activateChat } from './chat/activation'
 import { activate as activeInlineChat } from '../inlineChat/activation'
 import { AmazonQResourcePaths } from './lspInstaller'
-import { auth2 } from 'aws-core-vscode/auth'
+import { auth2, getMfaTokenFromUser } from 'aws-core-vscode/auth'
 import { ConfigSection, isValidConfigSection, pushConfigUpdate, toAmazonQLSPLogLevel } from './config'
 import { telemetry } from 'aws-core-vscode/telemetry'
 import { SessionManager } from '../app/inline/sessionManager'
@@ -348,7 +348,7 @@ async function postStartLanguageServer(
     client.onRequest(
         auth2.notificationTypes.getMfaCode.method,
         async (params: GetMfaCodeParams): Promise<GetMfaCodeResult> => {
-            const mfaCode = await vscode.window.showInputBox({ title: 'Enter MFA Code' })
+            const mfaCode = await getMfaTokenFromUser(params.mfaSerial, params.profileName)
             return { code: mfaCode ?? '' }
         }
     )

packages/core/src/auth/index.ts

@@ -22,6 +22,7 @@ export {
 } from './connection'
 export { Auth } from './auth'
 export { CredentialsStore } from './credentials/store'
+export { getMfaTokenFromUser } from './credentials/utils'
 export { LoginManager } from './deprecated/loginManager'
 export * as constants from './sso/constants'
 export * as cache from './sso/cache'