Implement IAM setup function

Author: liramon1

packages/core/src/auth/auth2.ts

@@ -28,6 +28,8 @@ import {
     AuthorizationFlowKind,
     CancellationToken,
     CancellationTokenSource,
+    iamCredentialsUpdateRequestType,
+    iamCredentialsDeleteNotificationType,
     bearerCredentialsDeleteNotificationType,
     bearerCredentialsUpdateRequestType,
     SsoTokenChangedKind,
@@ -45,6 +47,10 @@ import { getCacheDir, getCacheFileWatcher, getFlareCacheFileName } from './sso/c
 import { VSCODE_EXTENSION_ID } from '../shared/extensions'
 
 export const notificationTypes = {
+    updateIamCredential: new RequestType<UpdateCredentialsParams, ResponseMessage, Error>(
+        iamCredentialsUpdateRequestType.method
+    ),
+    deleteIamCredential: new NotificationType(iamCredentialsDeleteNotificationType.method),
     updateBearerToken: new RequestType<UpdateCredentialsParams, ResponseMessage, Error>(
         bearerCredentialsUpdateRequestType.method
     ),

packages/core/src/login/webview/vue/amazonq/backend_amazonq.ts

@@ -15,8 +15,9 @@ import { debounce } from 'lodash'
 import { AuthError, AuthFlowState, userCancelled } from '../types'
 import { ToolkitError } from '../../../../shared/errors'
 import { withTelemetryContext } from '../../../../shared/telemetry/util'
+import { Commands } from '../../../../shared/vscode/commands2'
 import { builderIdStartUrl } from '../../../../auth/sso/constants'
-import { RegionProfile } from '../../../../codewhisperer/models/model'
+import { RegionProfile, vsCodeState } from '../../../../codewhisperer/models/model'
 import { randomUUID } from '../../../../shared/crypto'
 import globals from '../../../../shared/extensionGlobals'
 import { telemetry } from '../../../../shared/telemetry/telemetry'
@@ -196,12 +197,40 @@ export class AmazonQLoginWebview extends CommonAuthWebview {
         return []
     }
 
-    override startIamCredentialSetup(
+    async startIamCredentialSetup(
         profileName: string,
         accessKey: string,
         secretKey: string
     ): Promise<AuthError | undefined> {
-        throw new Error('Method not implemented.')
+        getLogger().debug(`called startIamCredentialSetup()`)
+        // Defining separate auth function to emit telemetry before returning from setup
+        const runAuth = async (): Promise<AuthError | undefined> => {
+            try {
+                await AuthUtil.instance.login(accessKey, secretKey)
+                // Add auth telemetry
+                this.storeMetricMetadata(await AuthUtil.instance.getTelemetryMetadata())
+                // Show sign-in message
+                void vscode.window.showInformationMessage('AmazonQ: Successfully connected to AWS IAM Credentials')
+            } catch (e) {
+                getLogger().error('Failed submitting credentials %O', e)
+                return { id: this.id, text: e as string }
+            }
+            // Enable code suggestions
+            vsCodeState.isFreeTierLimitReached = false
+            await Commands.tryExecute('aws.amazonq.enableCodeSuggestions')
+        }
+
+        const result = await runAuth()
+
+        // Emit telemetry
+        this.storeMetricMetadata({
+            credentialSourceId: 'sharedCredentials',
+            authEnabledFeatures: 'codewhisperer',
+            ...this.getResultForMetrics(result),
+        })
+        this.emitAuthMetric()
+
+        return result
     }
 
     /** If users are unauthenticated in Q/CW, we should always display the auth screen. */