@aws/lsp-codewhisperer 0.0.28 → 0.0.30

package/out/language-server/agenticChat/qAgenticChatServer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"qAgenticChatServer.js","sourceRoot":"","sources":["../../../src/language-server/agenticChat/qAgenticChatServer.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAGH,mEAA+D;AAC/D,uFAAmF;AACnF,uDAA4E;AAC5E,0DAAsD;AACtD,gEAAmE;AACnE,oGAAgG;AAChG,4DAAmF;AACnF,oCAAkC;AAE3B,MAAM,kBAAkB;AAC3B,kBAAkB;AAClB,GAAW,EAAE,CAAC,QAAQ,CAAC,EAAE;IACrB,MAAM,EAAE,IAAI,EAAE,mBAAmB,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAA;IAEhF,IAAI,qBAAiD,CAAA;IACrD,IAAI,cAAqC,CAAA;IACzC,IAAI,4BAA0D,CAAA;IAC9D,IAAI,gBAAkC,CAAC;IAEvC,GAAG,CAAC,cAAc,CAAC,CAAC,MAAwB,EAAE,EAAE;QAC5C,OAAO;YACH,YAAY,EAAE,EAAE;YAChB,qBAAqB,EAAE;gBACnB,WAAW,EAAE;oBACT,YAAY,EAAE;wBACV,yBAAyB,EAAE;4BACvB;gCACI,QAAQ,EAAE,CAAC,gCAAiB,EAAE,iCAAkB,CAAC;6BACpD;yBACJ;qBACJ;iBACJ;aACJ;SACJ,CAAA;IACL,CAAC,CAAC,CAAA;IAEF,MAAM,0BAA0B,GAAG,KAAK,IAAI,EAAE;QAC1C,MAAM,qBAAqB,CAAC,4BAA4B,EAAE,CAAA;QAC1D,MAAM,cAAc,CAAC,mBAAmB,EAAE,CAAA;IAC9C,CAAC,CAAA;IAED,GAAG,CAAC,aAAa,CAAC,KAAK,IAAI,EAAE;QACzB,2FAA2F;QAC3F,qBAAqB,GAAG,uDAA0B,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAA;QACxE,4BAA4B,GAAG,2DAA4B;aACtD,WAAW,EAAE;aACb,yBAAyB,CAAC,qBAAqB,CAAC,CAAA;QAErD,gBAAgB,GAAG,IAAI,mCAAgB,CAC/B,qBAAqB,EACrB,mBAAmB,EACnB,SAAS,EACT,OAAO,CACV,CAAA;QAEL,MAAM,YAAY,GAAG,IAAA,eAAO,EAAC,GAAG,CAAC,yBAAyB,EAAE,EAAC,IAAI,0CAAiC,CAC1F,qEAAqE,CAAC,CAAC,CAAA;QAE/E,gBAAgB,CAAC,iBAAiB,CAAC,IAAA,sCAAqB,EAAC,YAAY,EAAE,OAAO,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAA;QAEjG,cAAc,GAAG,IAAI,6CAAqB,CAAC,4BAA4B,EAAE,QAAQ,EAAE,gBAAgB,CAAC,CAAA;QAEpG,MAAM,0BAA0B,EAAE,CAAA;IACtC,CAAC,CAAC,CAAA;IACF,GAAG,CAAC,sBAAsB,CAAC,0BAA0B,CAAC,CAAA;IAEtD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE;QACnB,OAAO,CAAC,GAAG,CAAC,eAAe,MAAM,CAAC,KAAK,EAAE,CAAC,CAAA;QAE1C,OAAO,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAA;IAC1C,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE;QACtB,OAAO,CAAC,GAAG,CAAC,oBAAoB,MAAM,CAAC,KAAK,EAAE,CAAC,CAAA;QAE/C,OAAO,cAAc,CAAC,WAAW,CAAC,MAAM,CAAC,CAAA;IAC7C,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE;QACtB,OAAO,CAAC,GAAG,CAAC,iBAAiB,MAAM,CAAC,KAAK,EAAE,CAAC,CAAA;QAE5C,OAAO,cAAc,CAAC,WAAW,CAAC,MAAM,CAAC,CAAA;IAC7C,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,SAAS,CAAC,CAAC,GAAG,MAAM,EAAE,EAAE;QACzB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAA;QACxC,OAAO,cAAc,CAAC,SAAS,CAAC,GAAG,MAAM,CAAC,CAAA;IAC9C,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,YAAY,CAAC,CAAC,GAAG,MAAM,EAAE,EAAE;QAC5B,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAA;QACnC,OAAO,cAAc,CAAC,YAAY,CAAC,GAAG,MAAM,CAAC,CAAA;IACjD,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,aAAa,CAAC,CAAC,GAAG,MAAM,EAAE,EAAE;QAC7B,OAAO,cAAc,CAAC,aAAa,CAAC,GAAG,MAAM,CAAC,CAAA;IAClD,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE;QACzB,OAAO,cAAc,CAAC,cAAc,CAAC,MAAM,CAAC,CAAA;IAChD,CAAC,CAAC,CAAA;IAEF,IAAI,CAAC,4BAA4B,CAAC,MAAM,CAAC,EAAE;QACvC,OAAO,cAAc,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAA;IAC9D,CAAC,CAAC,CAAA;IAEF,OAAO,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAA;IAEjD,OAAO,GAAG,EAAE;QACR,cAAc,EAAE,OAAO,EAAE,CAAA;IAC7B,CAAC,CAAA;AACL,CAAC,CAAA;AAtGQ,QAAA,kBAAkB,sBAsG1B"}

package/out/language-server/amazonQServiceManager/AmazonQTokenServiceManager.d.ts

@@ -0,0 +1,90 @@
+import { Logging, Lsp, CredentialsProvider, SDKInitializator, Runtime, Workspace, SsoConnectionType } from '@aws/language-server-runtimes/server-interface';
+import { CodeWhispererServiceToken } from '../codeWhispererService';
+import { BaseAmazonQServiceManager } from './BaseAmazonQServiceManager';
+import { CodeWhispererStreaming } from '@amzn/codewhisperer-streaming';
+export interface Features {
+    lsp: Lsp;
+    logging: Logging;
+    runtime: Runtime;
+    credentialsProvider: CredentialsProvider;
+    sdkInitializator: SDKInitializator;
+    workspace: Workspace;
+}
+/**
+ * AmazonQTokenServiceManager manages state and provides centralized access to
+ * instance of CodeWhispererServiceToken SDK client to any consuming code.
+ * It ensures that CodeWhispererServiceToken is configured to always access correct regionalized Amazon Q Developer API endpoint.
+ * Regional endppoint is selected based on:
+ * 1) current SSO auth connection type (BuilderId or IDC).
+ * 2) selected Amazon Q Developer profile (only for IDC connection type).
+ *
+ * @states
+ * - PENDING_CONNECTION: Initial state when no bearer token is set
+ * - PENDING_Q_PROFILE: When using Identity Center and waiting for profile selection
+ * - PENDING_Q_PROFILE_UPDATE: During profile update operation
+ * - INITIALIZED: Service is ready to handle requests
+ *
+ * @connectionTypes
+ * - none: No active connection
+ * - builderId: Connected via Builder ID
+ * - identityCenter: Connected via Identity Center
+ *
+ * AmazonQTokenServiceManager is a singleton class, which must be instantiated with Language Server runtimes [Features](https://github.com/aws/language-server-runtimes/blob/21d5d1dc7c73499475b7c88c98d2ce760e5d26c8/runtimes/server-interface/server.ts#L31-L42)
+ * To get access to current CodeWhispererServiceToken client object, call `getCodewhispererService()` mathod:
+ *
+ * @example
+ * const AmazonQServiceManager = AmazonQTokenServiceManager.getInstance(features);
+ * const codewhispererService = AmazonQServiceManager.getCodewhispererService();
+ */
+export declare class AmazonQTokenServiceManager implements BaseAmazonQServiceManager {
+    private static instance;
+    private features;
+    private logging;
+    private cachedCodewhispererService?;
+    private enableDeveloperProfileSupport?;
+    private configurationCache;
+    private activeIdcProfile?;
+    private connectionType?;
+    private profileChangeTokenSource;
+    private region?;
+    private endpoint?;
+    /**
+     * Internal state of Service connection, based on status of bearer token and Amazon Q Developer profile selection.
+     * Supported states:
+     * PENDING_CONNECTION - Waiting for Bearer Token and StartURL to be passed
+     * PENDING_Q_PROFILE - (only for identityCenter connection) waiting for setting Developer Profile
+     * PENDING_Q_PROFILE_UPDATE (only for identityCenter connection) waiting for Developer Profile to complete
+     * INITIALIZED - Service is initialized
+     */
+    private state;
+    private constructor();
+    static getInstance(features: Features): AmazonQTokenServiceManager;
+    private initialize;
+    private setupAuthListener;
+    private setupConfigurationListeners;
+    /**
+     * Validate if Bearer Token Connection type has changed mid-session.
+     * When connection type change is detected: reinitialize CodeWhispererService class with current connection type.
+     */
+    private handleSsoConnectionChange;
+    handleDidChangeConfiguration(): Promise<void>;
+    private cancelActiveProfileChangeToken;
+    private handleTokenCancellationRequest;
+    private handleProfileChange;
+    getCodewhispererService(): CodeWhispererServiceToken;
+    getStreamingClient(): CodeWhispererStreaming;
+    private resetCodewhispererService;
+    private createCodewhispererServiceInstances;
+    private getCustomUserAgent;
+    private serviceFactory;
+    private streamingClientFactory;
+    private log;
+    private logServiceState;
+    static resetInstance(): void;
+    getState(): "PENDING_CONNECTION" | "PENDING_Q_PROFILE" | "PENDING_Q_PROFILE_UPDATE" | "INITIALIZED";
+    getConnectionType(): SsoConnectionType | undefined;
+    getActiveProfileArn(): string | undefined;
+    setServiceFactory(factory: (region: string, endpoint: string) => CodeWhispererServiceToken): void;
+    getServiceFactory(): (region: string, endpoint: string) => CodeWhispererServiceToken;
+    getEnableDeveloperProfileSupport(): boolean;
+}

package/out/language-server/amazonQServiceManager/AmazonQTokenServiceManager.js

@@ -0,0 +1,466 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AmazonQTokenServiceManager = void 0;
+const server_interface_1 = require("@aws/language-server-runtimes/server-interface");
+const constants_1 = require("../../constants");
+const codeWhispererService_1 = require("../codeWhispererService");
+const errors_1 = require("./errors");
+const qConfigurationServer_1 = require("../configuration/qConfigurationServer");
+const lsp_core_1 = require("@aws/lsp-core");
+const codewhisperer_streaming_1 = require("@amzn/codewhisperer-streaming");
+const util_retry_1 = require("@aws-sdk/util-retry");
+const qDeveloperProfiles_1 = require("./qDeveloperProfiles");
+const telemetryUtils_1 = require("../utilities/telemetryUtils");
+const utils_1 = require("../utils");
+/**
+ * AmazonQTokenServiceManager manages state and provides centralized access to
+ * instance of CodeWhispererServiceToken SDK client to any consuming code.
+ * It ensures that CodeWhispererServiceToken is configured to always access correct regionalized Amazon Q Developer API endpoint.
+ * Regional endppoint is selected based on:
+ * 1) current SSO auth connection type (BuilderId or IDC).
+ * 2) selected Amazon Q Developer profile (only for IDC connection type).
+ *
+ * @states
+ * - PENDING_CONNECTION: Initial state when no bearer token is set
+ * - PENDING_Q_PROFILE: When using Identity Center and waiting for profile selection
+ * - PENDING_Q_PROFILE_UPDATE: During profile update operation
+ * - INITIALIZED: Service is ready to handle requests
+ *
+ * @connectionTypes
+ * - none: No active connection
+ * - builderId: Connected via Builder ID
+ * - identityCenter: Connected via Identity Center
+ *
+ * AmazonQTokenServiceManager is a singleton class, which must be instantiated with Language Server runtimes [Features](https://github.com/aws/language-server-runtimes/blob/21d5d1dc7c73499475b7c88c98d2ce760e5d26c8/runtimes/server-interface/server.ts#L31-L42)
+ * To get access to current CodeWhispererServiceToken client object, call `getCodewhispererService()` mathod:
+ *
+ * @example
+ * const AmazonQServiceManager = AmazonQTokenServiceManager.getInstance(features);
+ * const codewhispererService = AmazonQServiceManager.getCodewhispererService();
+ */
+class AmazonQTokenServiceManager {
+    static instance = null;
+    features;
+    logging;
+    cachedCodewhispererService;
+    enableDeveloperProfileSupport;
+    configurationCache = new Map();
+    activeIdcProfile;
+    connectionType;
+    profileChangeTokenSource;
+    region;
+    endpoint;
+    /**
+     * Internal state of Service connection, based on status of bearer token and Amazon Q Developer profile selection.
+     * Supported states:
+     * PENDING_CONNECTION - Waiting for Bearer Token and StartURL to be passed
+     * PENDING_Q_PROFILE - (only for identityCenter connection) waiting for setting Developer Profile
+     * PENDING_Q_PROFILE_UPDATE (only for identityCenter connection) waiting for Developer Profile to complete
+     * INITIALIZED - Service is initialized
+     */
+    state = 'PENDING_CONNECTION';
+    constructor() { }
+    static getInstance(features) {
+        if (!AmazonQTokenServiceManager.instance) {
+            AmazonQTokenServiceManager.instance = new AmazonQTokenServiceManager();
+            AmazonQTokenServiceManager.instance.initialize(features);
+        }
+        return AmazonQTokenServiceManager.instance;
+    }
+    initialize(features) {
+        if (!features || !features.logging || !features.lsp) {
+            throw new errors_1.AmazonQServiceInitializationError('Service features not initialized. Please ensure proper initialization.');
+        }
+        this.features = features;
+        this.logging = features.logging;
+        if (!this.features.lsp.getClientInitializeParams()) {
+            this.log('AmazonQTokenServiceManager initialized before LSP connection was initialized.');
+            throw new errors_1.AmazonQServiceInitializationError('AmazonQTokenServiceManager initialized before LSP connection was initialized.');
+        }
+        // Bind methods that are passed by reference to some handlers to maintain proper scope.
+        this.serviceFactory = this.serviceFactory.bind(this);
+        this.handleDidChangeConfiguration = this.handleDidChangeConfiguration.bind(this);
+        this.log('Reading enableDeveloperProfileSupport setting from AWSInitializationOptions');
+        if (this.features.lsp.getClientInitializeParams()?.initializationOptions?.aws) {
+            const awsOptions = this.features.lsp.getClientInitializeParams()?.initializationOptions?.aws || {};
+            this.enableDeveloperProfileSupport = (0, qDeveloperProfiles_1.signalsAWSQDeveloperProfilesEnabled)(awsOptions);
+            this.log(`Enabled Q Developer Profile support: ${this.enableDeveloperProfileSupport}`);
+        }
+        this.connectionType = 'none';
+        this.state = 'PENDING_CONNECTION';
+        this.setupAuthListener();
+        this.setupConfigurationListeners();
+        this.log('Manager instance is initialize');
+    }
+    setupAuthListener() {
+        this.features.credentialsProvider.onCredentialsDeleted((type) => {
+            this.log(`Received credentials delete event for type: ${type}`);
+            if (type === 'iam') {
+                return;
+            }
+            this.cancelActiveProfileChangeToken();
+            this.resetCodewhispererService();
+            this.connectionType = 'none';
+            this.state = 'PENDING_CONNECTION';
+        });
+    }
+    setupConfigurationListeners() {
+        // TODO: refactor how handleDidChangeConfiguration can be hooked to lsp Feature.
+        // Currently can't do this, as it overrides handler set by Server, using ServiceManager.
+        // this.features.lsp.onInitialized(this.handleDidChangeConfiguration)
+        // this.features.lsp.didChangeConfiguration(this.handleDidChangeConfiguration)
+        this.features.lsp.workspace.onUpdateConfiguration(async (params, _token) => {
+            try {
+                if (params.section === qConfigurationServer_1.Q_CONFIGURATION_SECTION && params.settings.profileArn !== undefined) {
+                    const profileArn = params.settings.profileArn;
+                    if (!(0, utils_1.isStringOrNull)(profileArn)) {
+                        throw new Error('Expected params.settings.profileArn to be of either type string or null');
+                    }
+                    this.log(`Profile update is requested for profile ${profileArn}`);
+                    this.cancelActiveProfileChangeToken();
+                    this.profileChangeTokenSource = new server_interface_1.CancellationTokenSource();
+                    await this.handleProfileChange(profileArn, this.profileChangeTokenSource.token);
+                }
+            }
+            catch (error) {
+                this.log('Error updating profiles: ' + error);
+                if (error instanceof errors_1.AmazonQServiceProfileUpdateCancelled) {
+                    throw new server_interface_1.ResponseError(server_interface_1.LSPErrorCodes.ServerCancelled, error.message, {
+                        awsErrorCode: error.code,
+                    });
+                }
+                if (error instanceof errors_1.AmazonQError) {
+                    throw new server_interface_1.ResponseError(server_interface_1.LSPErrorCodes.RequestFailed, error.message, {
+                        awsErrorCode: error.code,
+                    });
+                }
+                throw new server_interface_1.ResponseError(server_interface_1.LSPErrorCodes.RequestFailed, 'Failed to update configuration');
+            }
+            finally {
+                if (this.profileChangeTokenSource) {
+                    this.profileChangeTokenSource.dispose();
+                    this.profileChangeTokenSource = undefined;
+                }
+            }
+        });
+    }
+    /**
+     * Validate if Bearer Token Connection type has changed mid-session.
+     * When connection type change is detected: reinitialize CodeWhispererService class with current connection type.
+     */
+    handleSsoConnectionChange() {
+        const newConnectionType = this.features.credentialsProvider.getConnectionType();
+        this.logServiceState('Validate State of SSO Connection');
+        if (newConnectionType === 'none' || !this.features.credentialsProvider.hasCredentials('bearer')) {
+            // Connection was reset, wait for SSO connection token from client
+            this.log('No active SSO connection is detected, resetting the client');
+            this.resetCodewhispererService();
+            this.connectionType = 'none';
+            this.state = 'PENDING_CONNECTION';
+            return;
+        }
+        // Connection type hasn't change.
+        if (newConnectionType === this.connectionType) {
+            this.log('Connection type did not change.');
+            return;
+        }
+        // Connection type changed to 'builderId'
+        if (newConnectionType === 'builderId') {
+            this.log('Detected New connection type: builderId');
+            this.resetCodewhispererService();
+            // For the builderId connection type regional endpoint discovery chain is:
+            // region set by client -> runtime region -> default region
+            const clientParams = this.features.lsp.getClientInitializeParams();
+            this.createCodewhispererServiceInstances('builderId', clientParams?.initializationOptions?.aws?.region);
+            this.state = 'INITIALIZED';
+            this.log('Initialized Amazon Q service with builderId connection');
+            return;
+        }
+        // Connection type changed to 'identityCenter'
+        if (newConnectionType === 'identityCenter') {
+            this.log('Detected New connection type: identityCenter');
+            this.resetCodewhispererService();
+            if (this.enableDeveloperProfileSupport) {
+                this.connectionType = 'identityCenter';
+                this.state = 'PENDING_Q_PROFILE';
+                this.logServiceState('Pending profile selection for IDC connection');
+                return;
+            }
+            this.createCodewhispererServiceInstances('identityCenter');
+            this.state = 'INITIALIZED';
+            this.log('Initialized Amazon Q service with identityCenter connection');
+            return;
+        }
+        this.logServiceState('Unknown Connection state');
+    }
+    async handleDidChangeConfiguration() {
+        try {
+            const qConfig = await this.features.lsp.workspace.getConfiguration(qConfigurationServer_1.Q_CONFIGURATION_SECTION);
+            if (qConfig) {
+                // aws.q.customizationArn - selected customization
+                const customizationArn = lsp_core_1.textUtils.undefinedIfEmpty(qConfig.customization);
+                this.log(`Read configuration customizationArn=${customizationArn}`);
+                this.configurationCache.set('customizationArn', customizationArn);
+                // aws.q.optOutTelemetry - telemetry optout option
+                const optOutTelemetryPreference = qConfig['optOutTelemetry'] === true ? 'OPTOUT' : 'OPTIN';
+                this.log(`Read configuration optOutTelemetryPreference=${optOutTelemetryPreference}`);
+                this.configurationCache.set('optOutTelemetryPreference', optOutTelemetryPreference);
+                if (this.cachedCodewhispererService) {
+                    this.log(`Using customization=${customizationArn}`);
+                    this.cachedCodewhispererService.customizationArn = customizationArn;
+                }
+            }
+            const codeWhispererConfig = await this.features.lsp.workspace.getConfiguration('aws.codeWhisperer');
+            if (codeWhispererConfig) {
+                // aws.codeWhisperer.includeSuggestionsWithCodeReferences - return suggestions with code references
+                const includeSuggestionsWithCodeReferences = codeWhispererConfig['includeSuggestionsWithCodeReferences'] === true;
+                this.log(`Read сonfiguration includeSuggestionsWithCodeReferences=${includeSuggestionsWithCodeReferences}`);
+                this.configurationCache.set('includeSuggestionsWithCodeReferences', includeSuggestionsWithCodeReferences);
+                // aws.codeWhisperershareCodeWhispererContentWithAWS - share content with AWS
+                const shareCodeWhispererContentWithAWS = codeWhispererConfig['shareCodeWhispererContentWithAWS'] === true;
+                this.log(`Read configuration shareCodeWhispererContentWithAWS=${shareCodeWhispererContentWithAWS}`);
+                this.configurationCache.set('shareCodeWhispererContentWithAWS', shareCodeWhispererContentWithAWS);
+                if (this.cachedCodewhispererService) {
+                    this.log('Update shareCodeWhispererContentWithAWS setting on cachedCodewhispererService to ' +
+                        shareCodeWhispererContentWithAWS);
+                    this.cachedCodewhispererService.shareCodeWhispererContentWithAWS = shareCodeWhispererContentWithAWS;
+                }
+            }
+        }
+        catch (error) {
+            this.log(`Error in GetConfiguration: ${error}`);
+        }
+    }
+    cancelActiveProfileChangeToken() {
+        this.profileChangeTokenSource?.cancel();
+        this.profileChangeTokenSource?.dispose();
+        this.profileChangeTokenSource = undefined;
+    }
+    handleTokenCancellationRequest(token) {
+        if (token.isCancellationRequested) {
+            this.logServiceState('Handling CancellationToken cancellation request');
+            throw new errors_1.AmazonQServiceProfileUpdateCancelled('Requested profile update got cancelled');
+        }
+    }
+    async handleProfileChange(newProfileArn, token) {
+        if (!this.enableDeveloperProfileSupport) {
+            this.log('Developer Profiles Support is not enabled');
+            return;
+        }
+        if (typeof newProfileArn === 'string' && newProfileArn.length === 0) {
+            throw new Error('Received invalid Profile ARN (empty string)');
+        }
+        this.logServiceState('UpdateProfile is requested');
+        // Test if connection type changed
+        this.handleSsoConnectionChange();
+        if (this.connectionType === 'none') {
+            throw new errors_1.AmazonQServicePendingSigninError();
+        }
+        if (this.connectionType !== 'identityCenter') {
+            this.logServiceState('Q Profile can not be set');
+            throw new errors_1.AmazonQServiceNoProfileSupportError(`Connection type ${this.connectionType} does not support Developer Profiles feature.`);
+        }
+        if ((this.state === 'INITIALIZED' && this.activeIdcProfile) || this.state === 'PENDING_Q_PROFILE') {
+            // Change status to pending to prevent API calls until profile is updated.
+            // Because `listAvailableProfiles` below can take few seconds to complete,
+            // there is possibility that client could send requests while profile is changing.
+            this.state = 'PENDING_Q_PROFILE_UPDATE';
+        }
+        // Client sent an explicit null, indicating they want to reset the assigned profile (if any)
+        if (newProfileArn === null) {
+            this.logServiceState('Received null profile, resetting to PENDING_Q_PROFILE state');
+            this.resetCodewhispererService();
+            this.state = 'PENDING_Q_PROFILE';
+            return;
+        }
+        const profiles = await (0, qDeveloperProfiles_1.getListAllAvailableProfilesHandler)(this.serviceFactory)({
+            connectionType: 'identityCenter',
+            logging: this.logging,
+        });
+        this.handleTokenCancellationRequest(token);
+        const newProfile = profiles.find(el => el.arn === newProfileArn);
+        if (!newProfile || !newProfile.identityDetails?.region) {
+            this.log(`Amazon Q Profile ${newProfileArn} is not valid`);
+            this.resetCodewhispererService();
+            this.state = 'PENDING_Q_PROFILE';
+            throw new errors_1.AmazonQServiceInvalidProfileError('Requested Amazon Q Profile does not exist');
+        }
+        this.handleTokenCancellationRequest(token);
+        if (!this.activeIdcProfile) {
+            this.activeIdcProfile = newProfile;
+            this.createCodewhispererServiceInstances('identityCenter', newProfile.identityDetails.region);
+            this.state = 'INITIALIZED';
+            this.log(`Initialized identityCenter connection to region ${newProfile.identityDetails.region} for profile ${newProfile.arn}`);
+            return;
+        }
+        // Profile didn't change
+        if (this.activeIdcProfile && this.activeIdcProfile.arn === newProfile.arn) {
+            // Update cached profile fields, keep existing client
+            this.log(`Profile selection did not change, active profile is ${this.activeIdcProfile.arn}`);
+            this.activeIdcProfile = newProfile;
+            this.state = 'INITIALIZED';
+            return;
+        }
+        this.handleTokenCancellationRequest(token);
+        // At this point new valid profile is selected.
+        const oldRegion = this.activeIdcProfile.identityDetails?.region;
+        const newRegion = newProfile.identityDetails.region;
+        if (oldRegion === newRegion) {
+            this.log(`New profile is in the same region as old one, keeping exising service.`);
+            this.log(`New active profile is ${this.activeIdcProfile.arn}, region ${oldRegion}`);
+            this.activeIdcProfile = newProfile;
+            this.state = 'INITIALIZED';
+            if (this.cachedCodewhispererService) {
+                this.cachedCodewhispererService.profileArn = newProfile.arn;
+            }
+            return;
+        }
+        this.log(`Switching service client region from ${oldRegion} to ${newRegion}`);
+        this.handleTokenCancellationRequest(token);
+        // Selected new profile is in different region. Re-initialize service
+        this.resetCodewhispererService();
+        this.activeIdcProfile = newProfile;
+        this.createCodewhispererServiceInstances('identityCenter', newProfile.identityDetails.region);
+        this.state = 'INITIALIZED';
+        return;
+    }
+    getCodewhispererService() {
+        // Prevent initiating requests while profile change is in progress.
+        if (this.state === 'PENDING_Q_PROFILE_UPDATE') {
+            throw new errors_1.AmazonQServicePendingProfileUpdateError();
+        }
+        this.handleSsoConnectionChange();
+        if (this.state === 'INITIALIZED' && this.cachedCodewhispererService) {
+            return this.cachedCodewhispererService;
+        }
+        if (this.state === 'PENDING_CONNECTION') {
+            throw new errors_1.AmazonQServicePendingSigninError();
+        }
+        if (this.state === 'PENDING_Q_PROFILE') {
+            throw new errors_1.AmazonQServicePendingProfileError();
+        }
+        throw new errors_1.AmazonQServiceNotInitializedError();
+    }
+    getStreamingClient() {
+        this.log('Getting instance of CodeWhispererStreaming client');
+        // Trigger checks in token service
+        const tokenService = this.getCodewhispererService();
+        if (!tokenService || !this.region || !this.endpoint) {
+            throw new errors_1.AmazonQServiceNotInitializedError();
+        }
+        return this.streamingClientFactory(this.region, this.endpoint);
+    }
+    resetCodewhispererService() {
+        this.cachedCodewhispererService?.abortInflightRequests();
+        this.cachedCodewhispererService = undefined;
+        this.activeIdcProfile = undefined;
+    }
+    createCodewhispererServiceInstances(connectionType, region) {
+        this.logServiceState('Initializing CodewhispererService');
+        let awsQRegion;
+        let awsQEndpoint;
+        if (region) {
+            this.log(`Selecting region (found: ${region}) provided by ${connectionType === 'builderId' ? 'client' : 'profile'}`);
+            awsQRegion = region;
+            // @ts-ignore
+            awsQEndpoint = constants_1.AWS_Q_ENDPOINTS[awsQRegion];
+        }
+        else {
+            const runtimeRegion = this.features.runtime.getConfiguration(constants_1.AWS_Q_REGION_ENV_VAR);
+            if (runtimeRegion) {
+                this.log(`Selecting region (found: ${runtimeRegion}) provided by runtime`);
+                awsQRegion = runtimeRegion;
+                // prettier-ignore
+                awsQEndpoint = // @ts-ignore
+                    this.features.runtime.getConfiguration(constants_1.AWS_Q_ENDPOINT_URL_ENV_VAR) ?? constants_1.AWS_Q_ENDPOINTS[awsQRegion];
+            }
+            else {
+                this.log('Region not provided by caller or runtime, falling back to default region and endpoint');
+                awsQRegion = constants_1.DEFAULT_AWS_Q_REGION;
+                awsQEndpoint = constants_1.DEFAULT_AWS_Q_ENDPOINT_URL;
+            }
+        }
+        if (!awsQEndpoint) {
+            this.log(`Unable to determine endpoint (found: ${awsQEndpoint}) for region: ${awsQRegion}, falling back to default region and endpoint`);
+            awsQRegion = constants_1.DEFAULT_AWS_Q_REGION;
+            awsQEndpoint = constants_1.DEFAULT_AWS_Q_ENDPOINT_URL;
+        }
+        // Cache active region and endpoint selection
+        this.connectionType = connectionType;
+        this.region = awsQRegion;
+        this.endpoint = awsQEndpoint;
+        this.cachedCodewhispererService = this.serviceFactory(awsQRegion, awsQEndpoint);
+        this.log(`CodeWhispererToken service for connection type ${connectionType} was initialized, region=${awsQRegion}`);
+        this.logServiceState('CodewhispererService Initialization finished');
+    }
+    getCustomUserAgent() {
+        const initializeParams = this.features.lsp.getClientInitializeParams() || {};
+        return (0, telemetryUtils_1.getUserAgent)(initializeParams, this.features.runtime.serverInfo);
+    }
+    serviceFactory(region, endpoint) {
+        const service = new codeWhispererService_1.CodeWhispererServiceToken(this.features.credentialsProvider, this.features.workspace, region, endpoint, this.features.sdkInitializator);
+        const customUserAgent = this.getCustomUserAgent();
+        service.updateClientConfig({
+            customUserAgent: customUserAgent,
+        });
+        service.customizationArn = lsp_core_1.textUtils.undefinedIfEmpty(this.configurationCache.get('customizationArn'));
+        service.profileArn = this.activeIdcProfile?.arn;
+        service.shareCodeWhispererContentWithAWS =
+            this.configurationCache.get('shareCodeWhispererContentWithAWS') === true;
+        this.log('Configured CodeWhispererServiceToken instance settings:');
+        this.log(`customUserAgent=${customUserAgent}, customizationArn=${service.customizationArn}, shareCodeWhispererContentWithAWS=${service.shareCodeWhispererContentWithAWS}`);
+        return service;
+    }
+    streamingClientFactory(region, endpoint) {
+        const token = (0, utils_1.getBearerTokenFromProvider)(this.features.credentialsProvider);
+        // TODO: Follow-up with creating CodeWhispererStreaming client which supports inplace access to CredentialsProvider instead of caching static value.
+        // Without this, we need more complex mechanism for managing token change state when caching streaming client.
+        const streamingClient = this.features.sdkInitializator(codewhisperer_streaming_1.CodeWhispererStreaming, {
+            region,
+            endpoint,
+            token: { token: token },
+            retryStrategy: new util_retry_1.ConfiguredRetryStrategy(0, (attempt) => 500 + attempt ** 10),
+            customUserAgent: this.getCustomUserAgent(),
+        });
+        this.logging.debug(`Created streaming client instance region=${region}, endpoint=${endpoint}`);
+        return streamingClient;
+    }
+    log(message) {
+        const prefix = 'Amazon Q Token Service Manager';
+        this.logging?.log(`${prefix}: ${message}`);
+    }
+    logServiceState(context) {
+        this.logging?.debug(JSON.stringify({
+            context,
+            state: {
+                serviceStatus: this.state,
+                connectionType: this.connectionType,
+                activeIdcProfile: this.activeIdcProfile,
+            },
+        }));
+    }
+    // For Unit Tests
+    static resetInstance() {
+        AmazonQTokenServiceManager.instance = null;
+    }
+    getState() {
+        return this.state;
+    }
+    getConnectionType() {
+        return this.connectionType;
+    }
+    getActiveProfileArn() {
+        return this.activeIdcProfile?.arn;
+    }
+    setServiceFactory(factory) {
+        this.serviceFactory = factory.bind(this);
+    }
+    getServiceFactory() {
+        return this.serviceFactory;
+    }
+    getEnableDeveloperProfileSupport() {
+        return this.enableDeveloperProfileSupport === undefined ? false : this.enableDeveloperProfileSupport;
+    }
+}
+exports.AmazonQTokenServiceManager = AmazonQTokenServiceManager;
+//# sourceMappingURL=AmazonQTokenServiceManager.js.map

package/out/language-server/amazonQServiceManager/AmazonQTokenServiceManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"AmazonQTokenServiceManager.js","sourceRoot":"","sources":["../../../src/language-server/amazonQServiceManager/AmazonQTokenServiceManager.ts"],"names":[],"mappings":";;;AAAA,qFAeuD;AACvD,+CAMwB;AACxB,kEAAmE;AACnE,qCAUiB;AAEjB,gFAA+E;AAC/E,4CAAyC;AACzC,2EAAsE;AACtE,oDAA6D;AAC7D,6DAI6B;AAC7B,gEAA0D;AAC1D,oCAAqE;AAWrE;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,MAAa,0BAA0B;IAC3B,MAAM,CAAC,QAAQ,GAAsC,IAAI,CAAA;IACzD,QAAQ,CAAW;IACnB,OAAO,CAAU;IACjB,0BAA0B,CAA4B;IACtD,6BAA6B,CAAU;IACvC,kBAAkB,GAAG,IAAI,GAAG,EAAE,CAAA;IAC9B,gBAAgB,CAA0B;IAC1C,cAAc,CAAoB;IAClC,wBAAwB,CAAqC;IAC7D,MAAM,CAAS;IACf,QAAQ,CAAS;IACzB;;;;;;;OAOG;IACK,KAAK,GACT,oBAAoB,CAAA;IAExB,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW,CAAC,QAAkB;QACxC,IAAI,CAAC,0BAA0B,CAAC,QAAQ,EAAE,CAAC;YACvC,0BAA0B,CAAC,QAAQ,GAAG,IAAI,0BAA0B,EAAE,CAAA;YACtE,0BAA0B,CAAC,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAA;QAC5D,CAAC;QACD,OAAO,0BAA0B,CAAC,QAAQ,CAAA;IAC9C,CAAC;IAEO,UAAU,CAAC,QAAkB;QACjC,IAAI,CAAC,QAAQ,IAAI,CAAC,QAAQ,CAAC,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;YAClD,MAAM,IAAI,0CAAiC,CACvC,wEAAwE,CAC3E,CAAA;QACL,CAAC;QAED,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAA;QACxB,IAAI,CAAC,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAA;QAE/B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,yBAAyB,EAAE,EAAE,CAAC;YACjD,IAAI,CAAC,GAAG,CAAC,+EAA+E,CAAC,CAAA;YACzF,MAAM,IAAI,0CAAiC,CACvC,+EAA+E,CAClF,CAAA;QACL,CAAC;QAED,uFAAuF;QACvF,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACpD,IAAI,CAAC,4BAA4B,GAAG,IAAI,CAAC,4BAA4B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAEhF,IAAI,CAAC,GAAG,CAAC,6EAA6E,CAAC,CAAA;QACvF,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,yBAAyB,EAAE,EAAE,qBAAqB,EAAE,GAAG,EAAE,CAAC;YAC5E,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,yBAAyB,EAAE,EAAE,qBAAqB,EAAE,GAAG,IAAI,EAAE,CAAA;YAClG,IAAI,CAAC,6BAA6B,GAAG,IAAA,wDAAmC,EAAC,UAAU,CAAC,CAAA;YAEpF,IAAI,CAAC,GAAG,CAAC,wCAAwC,IAAI,CAAC,6BAA6B,EAAE,CAAC,CAAA;QAC1F,CAAC;QAED,IAAI,CAAC,cAAc,GAAG,MAAM,CAAA;QAC5B,IAAI,CAAC,KAAK,GAAG,oBAAoB,CAAA;QAEjC,IAAI,CAAC,iBAAiB,EAAE,CAAA;QACxB,IAAI,CAAC,2BAA2B,EAAE,CAAA;QAElC,IAAI,CAAC,GAAG,CAAC,gCAAgC,CAAC,CAAA;IAC9C,CAAC;IAEO,iBAAiB;QACrB,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,oBAAoB,CAAC,CAAC,IAAqB,EAAE,EAAE;YAC7E,IAAI,CAAC,GAAG,CAAC,+CAA+C,IAAI,EAAE,CAAC,CAAA;YAC/D,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;gBACjB,OAAM;YACV,CAAC;YACD,IAAI,CAAC,8BAA8B,EAAE,CAAA;YAErC,IAAI,CAAC,yBAAyB,EAAE,CAAA;YAChC,IAAI,CAAC,cAAc,GAAG,MAAM,CAAA;YAC5B,IAAI,CAAC,KAAK,GAAG,oBAAoB,CAAA;QACrC,CAAC,CAAC,CAAA;IACN,CAAC;IAEO,2BAA2B;QAC/B,gFAAgF;QAChF,wFAAwF;QACxF,qEAAqE;QACrE,8EAA8E;QAE9E,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,qBAAqB,CAC7C,KAAK,EAAE,MAAiC,EAAE,MAAyB,EAAE,EAAE;YACnE,IAAI,CAAC;gBACD,IAAI,MAAM,CAAC,OAAO,KAAK,8CAAuB,IAAI,MAAM,CAAC,QAAQ,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;oBACzF,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAA;oBAE7C,IAAI,CAAC,IAAA,sBAAc,EAAC,UAAU,CAAC,EAAE,CAAC;wBAC9B,MAAM,IAAI,KAAK,CAAC,yEAAyE,CAAC,CAAA;oBAC9F,CAAC;oBAED,IAAI,CAAC,GAAG,CAAC,2CAA2C,UAAU,EAAE,CAAC,CAAA;oBACjE,IAAI,CAAC,8BAA8B,EAAE,CAAA;oBACrC,IAAI,CAAC,wBAAwB,GAAG,IAAI,0CAAuB,EAAE,CAAA;oBAE7D,MAAM,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAA;gBACnF,CAAC;YACL,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,GAAG,CAAC,2BAA2B,GAAG,KAAK,CAAC,CAAA;gBAC7C,IAAI,KAAK,YAAY,6CAAoC,EAAE,CAAC;oBACxD,MAAM,IAAI,gCAAa,CAAC,gCAAa,CAAC,eAAe,EAAE,KAAK,CAAC,OAAO,EAAE;wBAClE,YAAY,EAAE,KAAK,CAAC,IAAI;qBAC3B,CAAC,CAAA;gBACN,CAAC;gBACD,IAAI,KAAK,YAAY,qBAAY,EAAE,CAAC;oBAChC,MAAM,IAAI,gCAAa,CAAC,gCAAa,CAAC,aAAa,EAAE,KAAK,CAAC,OAAO,EAAE;wBAChE,YAAY,EAAE,KAAK,CAAC,IAAI;qBAC3B,CAAC,CAAA;gBACN,CAAC;gBAED,MAAM,IAAI,gCAAa,CAAC,gCAAa,CAAC,aAAa,EAAE,gCAAgC,CAAC,CAAA;YAC1F,CAAC;oBAAS,CAAC;gBACP,IAAI,IAAI,CAAC,wBAAwB,EAAE,CAAC;oBAChC,IAAI,CAAC,wBAAwB,CAAC,OAAO,EAAE,CAAA;oBACvC,IAAI,CAAC,wBAAwB,GAAG,SAAS,CAAA;gBAC7C,CAAC;YACL,CAAC;QACL,CAAC,CACJ,CAAA;IACL,CAAC;IAED;;;OAGG;IACK,yBAAyB;QAC7B,MAAM,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,iBAAiB,EAAE,CAAA;QAE/E,IAAI,CAAC,eAAe,CAAC,kCAAkC,CAAC,CAAA;QAExD,IAAI,iBAAiB,KAAK,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC9F,kEAAkE;YAClE,IAAI,CAAC,GAAG,CAAC,4DAA4D,CAAC,CAAA;YACtE,IAAI,CAAC,yBAAyB,EAAE,CAAA;YAChC,IAAI,CAAC,cAAc,GAAG,MAAM,CAAA;YAC5B,IAAI,CAAC,KAAK,GAAG,oBAAoB,CAAA;YAEjC,OAAM;QACV,CAAC;QAED,iCAAiC;QAEjC,IAAI,iBAAiB,KAAK,IAAI,CAAC,cAAc,EAAE,CAAC;YAC5C,IAAI,CAAC,GAAG,CAAC,iCAAiC,CAAC,CAAA;YAE3C,OAAM;QACV,CAAC;QAED,yCAAyC;QAEzC,IAAI,iBAAiB,KAAK,WAAW,EAAE,CAAC;YACpC,IAAI,CAAC,GAAG,CAAC,yCAAyC,CAAC,CAAA;YACnD,IAAI,CAAC,yBAAyB,EAAE,CAAA;YAEhC,0EAA0E;YAC1E,2DAA2D;YAC3D,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,yBAAyB,EAAE,CAAA;YAElE,IAAI,CAAC,mCAAmC,CAAC,WAAW,EAAE,YAAY,EAAE,qBAAqB,EAAE,GAAG,EAAE,MAAM,CAAC,CAAA;YACvG,IAAI,CAAC,KAAK,GAAG,aAAa,CAAA;YAC1B,IAAI,CAAC,GAAG,CAAC,wDAAwD,CAAC,CAAA;YAElE,OAAM;QACV,CAAC;QAED,8CAA8C;QAE9C,IAAI,iBAAiB,KAAK,gBAAgB,EAAE,CAAC;YACzC,IAAI,CAAC,GAAG,CAAC,8CAA8C,CAAC,CAAA;YAExD,IAAI,CAAC,yBAAyB,EAAE,CAAA;YAEhC,IAAI,IAAI,CAAC,6BAA6B,EAAE,CAAC;gBACrC,IAAI,CAAC,cAAc,GAAG,gBAAgB,CAAA;gBACtC,IAAI,CAAC,KAAK,GAAG,mBAAmB,CAAA;gBAChC,IAAI,CAAC,eAAe,CAAC,8CAA8C,CAAC,CAAA;gBAEpE,OAAM;YACV,CAAC;YAED,IAAI,CAAC,mCAAmC,CAAC,gBAAgB,CAAC,CAAA;YAC1D,IAAI,CAAC,KAAK,GAAG,aAAa,CAAA;YAC1B,IAAI,CAAC,GAAG,CAAC,6DAA6D,CAAC,CAAA;YAEvE,OAAM;QACV,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,0BAA0B,CAAC,CAAA;IACpD,CAAC;IAEM,KAAK,CAAC,4BAA4B;QACrC,IAAI,CAAC;YACD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,8CAAuB,CAAC,CAAA;YAC3F,IAAI,OAAO,EAAE,CAAC;gBACV,kDAAkD;gBAClD,MAAM,gBAAgB,GAAG,oBAAS,CAAC,gBAAgB,CAAC,OAAO,CAAC,aAAa,CAAC,CAAA;gBAC1E,IAAI,CAAC,GAAG,CAAC,uCAAuC,gBAAgB,EAAE,CAAC,CAAA;gBACnE,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,kBAAkB,EAAE,gBAAgB,CAAC,CAAA;gBAEjE,kDAAkD;gBAClD,MAAM,yBAAyB,GAAG,OAAO,CAAC,iBAAiB,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAA;gBAC1F,IAAI,CAAC,GAAG,CAAC,gDAAgD,yBAAyB,EAAE,CAAC,CAAA;gBACrF,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,2BAA2B,EAAE,yBAAyB,CAAC,CAAA;gBAEnF,IAAI,IAAI,CAAC,0BAA0B,EAAE,CAAC;oBAClC,IAAI,CAAC,GAAG,CAAC,uBAAuB,gBAAgB,EAAE,CAAC,CAAA;oBACnD,IAAI,CAAC,0BAA0B,CAAC,gBAAgB,GAAG,gBAAgB,CAAA;gBACvE,CAAC;YACL,CAAC;YAED,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,CAAA;YACnG,IAAI,mBAAmB,EAAE,CAAC;gBACtB,mGAAmG;gBACnG,MAAM,oCAAoC,GACtC,mBAAmB,CAAC,sCAAsC,CAAC,KAAK,IAAI,CAAA;gBACxE,IAAI,CAAC,GAAG,CACJ,2DAA2D,oCAAoC,EAAE,CACpG,CAAA;gBACD,IAAI,CAAC,kBAAkB,CAAC,GAAG,CACvB,sCAAsC,EACtC,oCAAoC,CACvC,CAAA;gBAED,6EAA6E;gBAC7E,MAAM,gCAAgC,GAClC,mBAAmB,CAAC,kCAAkC,CAAC,KAAK,IAAI,CAAA;gBACpE,IAAI,CAAC,GAAG,CAAC,uDAAuD,gCAAgC,EAAE,CAAC,CAAA;gBACnG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,kCAAkC,EAAE,gCAAgC,CAAC,CAAA;gBAEjG,IAAI,IAAI,CAAC,0BAA0B,EAAE,CAAC;oBAClC,IAAI,CAAC,GAAG,CACJ,mFAAmF;wBAC/E,gCAAgC,CACvC,CAAA;oBACD,IAAI,CAAC,0BAA0B,CAAC,gCAAgC,GAAG,gCAAgC,CAAA;gBACvG,CAAC;YACL,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,8BAA8B,KAAK,EAAE,CAAC,CAAA;QACnD,CAAC;IACL,CAAC;IAEO,8BAA8B;QAClC,IAAI,CAAC,wBAAwB,EAAE,MAAM,EAAE,CAAA;QACvC,IAAI,CAAC,wBAAwB,EAAE,OAAO,EAAE,CAAA;QACxC,IAAI,CAAC,wBAAwB,GAAG,SAAS,CAAA;IAC7C,CAAC;IAEO,8BAA8B,CAAC,KAAwB;QAC3D,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;YAChC,IAAI,CAAC,eAAe,CAAC,iDAAiD,CAAC,CAAA;YACvE,MAAM,IAAI,6CAAoC,CAAC,wCAAwC,CAAC,CAAA;QAC5F,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,aAA4B,EAAE,KAAwB;QACpF,IAAI,CAAC,IAAI,CAAC,6BAA6B,EAAE,CAAC;YACtC,IAAI,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAA;YACrD,OAAM;QACV,CAAC;QAED,IAAI,OAAO,aAAa,KAAK,QAAQ,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAClE,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAA;QAClE,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,4BAA4B,CAAC,CAAA;QAElD,kCAAkC;QAClC,IAAI,CAAC,yBAAyB,EAAE,CAAA;QAEhC,IAAI,IAAI,CAAC,cAAc,KAAK,MAAM,EAAE,CAAC;YACjC,MAAM,IAAI,yCAAgC,EAAE,CAAA;QAChD,CAAC;QAED,IAAI,IAAI,CAAC,cAAc,KAAK,gBAAgB,EAAE,CAAC;YAC3C,IAAI,CAAC,eAAe,CAAC,0BAA0B,CAAC,CAAA;YAChD,MAAM,IAAI,4CAAmC,CACzC,mBAAmB,IAAI,CAAC,cAAc,+CAA+C,CACxF,CAAA;QACL,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,KAAK,KAAK,aAAa,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,IAAI,CAAC,KAAK,KAAK,mBAAmB,EAAE,CAAC;YAChG,0EAA0E;YAC1E,0EAA0E;YAC1E,kFAAkF;YAClF,IAAI,CAAC,KAAK,GAAG,0BAA0B,CAAA;QAC3C,CAAC;QAED,4FAA4F;QAC5F,IAAI,aAAa,KAAK,IAAI,EAAE,CAAC;YACzB,IAAI,CAAC,eAAe,CAAC,6DAA6D,CAAC,CAAA;YACnF,IAAI,CAAC,yBAAyB,EAAE,CAAA;YAChC,IAAI,CAAC,KAAK,GAAG,mBAAmB,CAAA;YAEhC,OAAM;QACV,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,IAAA,uDAAkC,EAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC3E,cAAc,EAAE,gBAAgB;YAChC,OAAO,EAAE,IAAI,CAAC,OAAO;SACxB,CAAC,CAAA;QAEF,IAAI,CAAC,8BAA8B,CAAC,KAAK,CAAC,CAAA;QAE1C,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,GAAG,KAAK,aAAa,CAAC,CAAA;QAEhE,IAAI,CAAC,UAAU,IAAI,CAAC,UAAU,CAAC,eAAe,EAAE,MAAM,EAAE,CAAC;YACrD,IAAI,CAAC,GAAG,CAAC,oBAAoB,aAAa,eAAe,CAAC,CAAA;YAC1D,IAAI,CAAC,yBAAyB,EAAE,CAAA;YAChC,IAAI,CAAC,KAAK,GAAG,mBAAmB,CAAA;YAEhC,MAAM,IAAI,0CAAiC,CAAC,2CAA2C,CAAC,CAAA;QAC5F,CAAC;QAED,IAAI,CAAC,8BAA8B,CAAC,KAAK,CAAC,CAAA;QAE1C,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACzB,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAA;YAClC,IAAI,CAAC,mCAAmC,CAAC,gBAAgB,EAAE,UAAU,CAAC,eAAe,CAAC,MAAM,CAAC,CAAA;YAC7F,IAAI,CAAC,KAAK,GAAG,aAAa,CAAA;YAC1B,IAAI,CAAC,GAAG,CACJ,mDAAmD,UAAU,CAAC,eAAe,CAAC,MAAM,gBAAgB,UAAU,CAAC,GAAG,EAAE,CACvH,CAAA;YAED,OAAM;QACV,CAAC;QAED,wBAAwB;QACxB,IAAI,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,KAAK,UAAU,CAAC,GAAG,EAAE,CAAC;YACxE,qDAAqD;YACrD,IAAI,CAAC,GAAG,CAAC,uDAAuD,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,CAAC,CAAA;YAC5F,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAA;YAClC,IAAI,CAAC,KAAK,GAAG,aAAa,CAAA;YAE1B,OAAM;QACV,CAAC;QAED,IAAI,CAAC,8BAA8B,CAAC,KAAK,CAAC,CAAA;QAE1C,+CAA+C;QAE/C,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE,MAAM,CAAA;QAC/D,MAAM,SAAS,GAAG,UAAU,CAAC,eAAe,CAAC,MAAM,CAAA;QACnD,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC1B,IAAI,CAAC,GAAG,CAAC,wEAAwE,CAAC,CAAA;YAClF,IAAI,CAAC,GAAG,CAAC,yBAAyB,IAAI,CAAC,gBAAgB,CAAC,GAAG,YAAY,SAAS,EAAE,CAAC,CAAA;YACnF,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAA;YAClC,IAAI,CAAC,KAAK,GAAG,aAAa,CAAA;YAE1B,IAAI,IAAI,CAAC,0BAA0B,EAAE,CAAC;gBAClC,IAAI,CAAC,0BAA0B,CAAC,UAAU,GAAG,UAAU,CAAC,GAAG,CAAA;YAC/D,CAAC;YAED,OAAM;QACV,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,wCAAwC,SAAS,OAAO,SAAS,EAAE,CAAC,CAAA;QAE7E,IAAI,CAAC,8BAA8B,CAAC,KAAK,CAAC,CAAA;QAE1C,qEAAqE;QACrE,IAAI,CAAC,yBAAyB,EAAE,CAAA;QAEhC,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAA;QAElC,IAAI,CAAC,mCAAmC,CAAC,gBAAgB,EAAE,UAAU,CAAC,eAAe,CAAC,MAAM,CAAC,CAAA;QAC7F,IAAI,CAAC,KAAK,GAAG,aAAa,CAAA;QAE1B,OAAM;IACV,CAAC;IAEM,uBAAuB;QAC1B,mEAAmE;QACnE,IAAI,IAAI,CAAC,KAAK,KAAK,0BAA0B,EAAE,CAAC;YAC5C,MAAM,IAAI,gDAAuC,EAAE,CAAA;QACvD,CAAC;QAED,IAAI,CAAC,yBAAyB,EAAE,CAAA;QAEhC,IAAI,IAAI,CAAC,KAAK,KAAK,aAAa,IAAI,IAAI,CAAC,0BAA0B,EAAE,CAAC;YAClE,OAAO,IAAI,CAAC,0BAA0B,CAAA;QAC1C,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,KAAK,oBAAoB,EAAE,CAAC;YACtC,MAAM,IAAI,yCAAgC,EAAE,CAAA;QAChD,CAAC;QAED,IAAI,IAAI,CAAC,KAAK,KAAK,mBAAmB,EAAE,CAAC;YACrC,MAAM,IAAI,0CAAiC,EAAE,CAAA;QACjD,CAAC;QAED,MAAM,IAAI,0CAAiC,EAAE,CAAA;IACjD,CAAC;IAEM,kBAAkB;QACrB,IAAI,CAAC,GAAG,CAAC,mDAAmD,CAAC,CAAA;QAE7D,kCAAkC;QAClC,MAAM,YAAY,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAA;QAEnD,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClD,MAAM,IAAI,0CAAiC,EAAE,CAAA;QACjD,CAAC;QAED,OAAO,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAA;IAClE,CAAC;IAEO,yBAAyB;QAC7B,IAAI,CAAC,0BAA0B,EAAE,qBAAqB,EAAE,CAAA;QACxD,IAAI,CAAC,0BAA0B,GAAG,SAAS,CAAA;QAC3C,IAAI,CAAC,gBAAgB,GAAG,SAAS,CAAA;IACrC,CAAC;IAEO,mCAAmC,CAAC,cAA8C,EAAE,MAAe;QACvG,IAAI,CAAC,eAAe,CAAC,mCAAmC,CAAC,CAAA;QACzD,IAAI,UAAkB,CAAA;QACtB,IAAI,YAAgC,CAAA;QAEpC,IAAI,MAAM,EAAE,CAAC;YACT,IAAI,CAAC,GAAG,CACJ,4BAA4B,MAAM,iBAAiB,cAAc,KAAK,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,EAAE,CAC7G,CAAA;YACD,UAAU,GAAG,MAAM,CAAA;YACnB,aAAa;YACb,YAAY,GAAG,2BAAe,CAAC,UAAU,CAAC,CAAA;QAC9C,CAAC;aAAM,CAAC;YACJ,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,gCAAoB,CAAC,CAAA;YAElF,IAAI,aAAa,EAAE,CAAC;gBAChB,IAAI,CAAC,GAAG,CAAC,4BAA4B,aAAa,uBAAuB,CAAC,CAAA;gBAC1E,UAAU,GAAG,aAAa,CAAA;gBAC1B,kBAAkB;gBAClB,YAAY,GAAG,aAAa;oBACxB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC,sCAA0B,CAAC,IAAI,2BAAe,CAAC,UAAU,CAAC,CAAA;YACzG,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,GAAG,CAAC,uFAAuF,CAAC,CAAA;gBACjG,UAAU,GAAG,gCAAoB,CAAA;gBACjC,YAAY,GAAG,sCAA0B,CAAA;YAC7C,CAAC;QACL,CAAC;QAED,IAAI,CAAC,YAAY,EAAE,CAAC;YAChB,IAAI,CAAC,GAAG,CACJ,wCAAwC,YAAY,iBAAiB,UAAU,+CAA+C,CACjI,CAAA;YACD,UAAU,GAAG,gCAAoB,CAAA;YACjC,YAAY,GAAG,sCAA0B,CAAA;QAC7C,CAAC;QAED,6CAA6C;QAC7C,IAAI,CAAC,cAAc,GAAG,cAAc,CAAA;QACpC,IAAI,CAAC,MAAM,GAAG,UAAU,CAAA;QACxB,IAAI,CAAC,QAAQ,GAAG,YAAY,CAAA;QAE5B,IAAI,CAAC,0BAA0B,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,YAAY,CAAC,CAAA;QAE/E,IAAI,CAAC,GAAG,CACJ,kDAAkD,cAAc,4BAA4B,UAAU,EAAE,CAC3G,CAAA;QACD,IAAI,CAAC,eAAe,CAAC,8CAA8C,CAAC,CAAA;IACxE,CAAC;IAEO,kBAAkB;QACtB,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,yBAAyB,EAAE,IAAI,EAAE,CAAA;QAE5E,OAAO,IAAA,6BAAY,EAAC,gBAAoC,EAAE,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,UAAU,CAAC,CAAA;IAC/F,CAAC;IAEO,cAAc,CAAC,MAAc,EAAE,QAAgB;QACnD,MAAM,OAAO,GAAG,IAAI,gDAAyB,CACzC,IAAI,CAAC,QAAQ,CAAC,mBAAmB,EACjC,IAAI,CAAC,QAAQ,CAAC,SAAS,EACvB,MAAM,EACN,QAAQ,EACR,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CACjC,CAAA;QAED,MAAM,eAAe,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAA;QACjD,OAAO,CAAC,kBAAkB,CAAC;YACvB,eAAe,EAAE,eAAe;SACnC,CAAC,CAAA;QACF,OAAO,CAAC,gBAAgB,GAAG,oBAAS,CAAC,gBAAgB,CAAC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,CAAA;QACtG,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,gBAAgB,EAAE,GAAG,CAAA;QAC/C,OAAO,CAAC,gCAAgC;YACpC,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,kCAAkC,CAAC,KAAK,IAAI,CAAA;QAE5E,IAAI,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAA;QACnE,IAAI,CAAC,GAAG,CACJ,mBAAmB,eAAe,sBAAsB,OAAO,CAAC,gBAAgB,sCAAsC,OAAO,CAAC,gCAAgC,EAAE,CACnK,CAAA;QAED,OAAO,OAAO,CAAA;IAClB,CAAC;IAEO,sBAAsB,CAAC,MAAc,EAAE,QAAgB;QAC3D,MAAM,KAAK,GAAG,IAAA,kCAA0B,EAAC,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAA;QAE3E,oJAAoJ;QACpJ,8GAA8G;QAC9G,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC,gDAAsB,EAAE;YAC3E,MAAM;YACN,QAAQ;YACR,KAAK,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;YACvB,aAAa,EAAE,IAAI,oCAAuB,CAAC,CAAC,EAAE,CAAC,OAAe,EAAE,EAAE,CAAC,GAAG,GAAG,OAAO,IAAI,EAAE,CAAC;YACvF,eAAe,EAAE,IAAI,CAAC,kBAAkB,EAAE;SAC7C,CAAC,CAAA;QACF,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,4CAA4C,MAAM,cAAc,QAAQ,EAAE,CAAC,CAAA;QAE9F,OAAO,eAAe,CAAA;IAC1B,CAAC;IAEO,GAAG,CAAC,OAAe;QACvB,MAAM,MAAM,GAAG,gCAAgC,CAAA;QAC/C,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,GAAG,MAAM,KAAK,OAAO,EAAE,CAAC,CAAA;IAC9C,CAAC;IAEO,eAAe,CAAC,OAAe;QACnC,IAAI,CAAC,OAAO,EAAE,KAAK,CACf,IAAI,CAAC,SAAS,CAAC;YACX,OAAO;YACP,KAAK,EAAE;gBACH,aAAa,EAAE,IAAI,CAAC,KAAK;gBACzB,cAAc,EAAE,IAAI,CAAC,cAAc;gBACnC,gBAAgB,EAAE,IAAI,CAAC,gBAAgB;aAC1C;SACJ,CAAC,CACL,CAAA;IACL,CAAC;IAED,iBAAiB;IACV,MAAM,CAAC,aAAa;QACvB,0BAA0B,CAAC,QAAQ,GAAG,IAAI,CAAA;IAC9C,CAAC;IAEM,QAAQ;QACX,OAAO,IAAI,CAAC,KAAK,CAAA;IACrB,CAAC;IAEM,iBAAiB;QACpB,OAAO,IAAI,CAAC,cAAc,CAAA;IAC9B,CAAC;IAEM,mBAAmB;QACtB,OAAO,IAAI,CAAC,gBAAgB,EAAE,GAAG,CAAA;IACrC,CAAC;IAEM,iBAAiB,CAAC,OAAwE;QAC7F,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC5C,CAAC;IAEM,iBAAiB;QACpB,OAAO,IAAI,CAAC,cAAc,CAAA;IAC9B,CAAC;IAEM,gCAAgC;QACnC,OAAO,IAAI,CAAC,6BAA6B,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,6BAA6B,CAAA;IACxG,CAAC;;AAtjBL,gEAujBC"}

package/out/language-server/amazonQServiceManager/BaseAmazonQServiceManager.d.ts

@@ -0,0 +1,5 @@
+import { CodeWhispererServiceBase } from '../codeWhispererService';
+export interface BaseAmazonQServiceManager {
+    handleDidChangeConfiguration: () => Promise<void>;
+    getCodewhispererService: () => CodeWhispererServiceBase;
+}

package/out/language-server/amazonQServiceManager/BaseAmazonQServiceManager.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=BaseAmazonQServiceManager.js.map

package/out/language-server/amazonQServiceManager/BaseAmazonQServiceManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"BaseAmazonQServiceManager.js","sourceRoot":"","sources":["../../../src/language-server/amazonQServiceManager/BaseAmazonQServiceManager.ts"],"names":[],"mappings":""}

package/out/language-server/amazonQServiceManager/errors.d.ts

@@ -0,0 +1,28 @@
+export declare class AmazonQError extends Error {
+    code: string;
+    constructor(message: string, code: string);
+}
+export declare class AmazonQServiceInitializationError extends AmazonQError {
+    constructor(message?: string);
+}
+export declare class AmazonQServiceNotInitializedError extends AmazonQError {
+    constructor(message?: string);
+}
+export declare class AmazonQServicePendingSigninError extends AmazonQError {
+    constructor(message?: string);
+}
+export declare class AmazonQServicePendingProfileError extends AmazonQError {
+    constructor(message?: string);
+}
+export declare class AmazonQServicePendingProfileUpdateError extends AmazonQError {
+    constructor(message?: string);
+}
+export declare class AmazonQServiceProfileUpdateCancelled extends AmazonQError {
+    constructor(message?: string);
+}
+export declare class AmazonQServiceInvalidProfileError extends AmazonQError {
+    constructor(message?: string);
+}
+export declare class AmazonQServiceNoProfileSupportError extends AmazonQError {
+    constructor(message?: string);
+}