@aws/lsp-codewhisperer 0.0.83 → 0.0.85

package/node_modules/@smithy/node-http-handler/dist-types/node-http2-connection-pool.d.ts

@@ -0,0 +1,13 @@
+/// <reference types="node" />
+import { ConnectionPool } from "@smithy/types";
+import { ClientHttp2Session } from "http2";
+export declare class NodeHttp2ConnectionPool implements ConnectionPool<ClientHttp2Session> {
+    private sessions;
+    constructor(sessions?: ClientHttp2Session[]);
+    poll(): ClientHttp2Session | void;
+    offerLast(session: ClientHttp2Session): void;
+    contains(session: ClientHttp2Session): boolean;
+    remove(session: ClientHttp2Session): void;
+    [Symbol.iterator](): IterableIterator<ClientHttp2Session>;
+    destroy(connection: ClientHttp2Session): void;
+}

package/node_modules/@smithy/node-http-handler/dist-types/node-http2-handler.d.ts

@@ -0,0 +1,57 @@
+import { HttpHandler, HttpRequest, HttpResponse } from "@smithy/protocol-http";
+import { HttpHandlerOptions, Provider } from "@smithy/types";
+/**
+ * Represents the http2 options that can be passed to a node http2 client.
+ */
+export interface NodeHttp2HandlerOptions {
+    /**
+     * The maximum time in milliseconds that a stream may remain idle before it
+     * is closed.
+     */
+    requestTimeout?: number;
+    /**
+     * The maximum time in milliseconds that a session or socket may remain idle
+     * before it is closed.
+     * https://nodejs.org/docs/latest-v12.x/api/http2.html#http2_http2session_and_sockets
+     */
+    sessionTimeout?: number;
+    /**
+     * Disables processing concurrent streams on a ClientHttp2Session instance. When set
+     * to true, a new session instance is created for each request to a URL.
+     * **Default:** false.
+     * https://nodejs.org/api/http2.html#http2_class_clienthttp2session
+     */
+    disableConcurrentStreams?: boolean;
+    /**
+     * Maximum number of concurrent Http2Stream instances per ClientHttp2Session. Each session
+     * may have up to 2^31-1 Http2Stream instances over its lifetime.
+     * This value must be greater than or equal to 0.
+     * https://nodejs.org/api/http2.html#class-http2stream
+     */
+    maxConcurrentStreams?: number;
+}
+export declare class NodeHttp2Handler implements HttpHandler<NodeHttp2HandlerOptions> {
+    private config?;
+    private configProvider;
+    readonly metadata: {
+        handlerProtocol: string;
+    };
+    private readonly connectionManager;
+    /**
+     * @returns the input if it is an HttpHandler of any class,
+     * or instantiates a new instance of this handler.
+     */
+    static create(instanceOrOptions?: HttpHandler<any> | NodeHttp2HandlerOptions | Provider<NodeHttp2HandlerOptions | void>): HttpHandler<any> | NodeHttp2Handler;
+    constructor(options?: NodeHttp2HandlerOptions | Provider<NodeHttp2HandlerOptions | void>);
+    destroy(): void;
+    handle(request: HttpRequest, { abortSignal }?: HttpHandlerOptions): Promise<{
+        response: HttpResponse;
+    }>;
+    updateHttpClientConfig(key: keyof NodeHttp2HandlerOptions, value: NodeHttp2HandlerOptions[typeof key]): void;
+    httpHandlerConfigs(): NodeHttp2HandlerOptions;
+    /**
+     * Destroys a session.
+     * @param session The session to destroy.
+     */
+    private destroySession;
+}

package/node_modules/@smithy/node-http-handler/dist-types/readable.mock.d.ts

@@ -0,0 +1,14 @@
+/// <reference types="node" />
+/// <reference types="node" />
+import { Readable, ReadableOptions } from "stream";
+export interface ReadFromBuffersOptions extends ReadableOptions {
+    buffers: Buffer[];
+    errorAfter?: number;
+}
+export declare class ReadFromBuffers extends Readable {
+    private buffersToRead;
+    private numBuffersRead;
+    private errorAfter;
+    constructor(options: ReadFromBuffersOptions);
+    _read(): boolean | undefined;
+}

package/node_modules/@smithy/node-http-handler/dist-types/set-connection-timeout.d.ts

@@ -0,0 +1,2 @@
+import { ClientRequest } from "http";
+export declare const setConnectionTimeout: (request: ClientRequest, reject: (err: Error) => void, timeoutInMs?: number) => void;

package/node_modules/@smithy/node-http-handler/dist-types/set-socket-keep-alive.d.ts

@@ -0,0 +1,6 @@
+import { ClientRequest } from "http";
+export interface SocketKeepAliveOptions {
+    keepAlive: boolean;
+    keepAliveMsecs?: number;
+}
+export declare const setSocketKeepAlive: (request: ClientRequest, { keepAlive, keepAliveMsecs }: SocketKeepAliveOptions) => void;

package/node_modules/@smithy/node-http-handler/dist-types/set-socket-timeout.d.ts

@@ -0,0 +1,2 @@
+import { ClientRequest } from "http";
+export declare const setSocketTimeout: (request: ClientRequest, reject: (err: Error) => void, timeoutInMs?: number) => void;

package/node_modules/@smithy/node-http-handler/dist-types/stream-collector/collector.d.ts

@@ -0,0 +1,7 @@
+/// <reference types="node" />
+/// <reference types="node" />
+import { Writable } from "stream";
+export declare class Collector extends Writable {
+    readonly bufferedBytes: Buffer[];
+    _write(chunk: Buffer, encoding: string, callback: (err?: Error) => void): void;
+}

package/node_modules/@smithy/node-http-handler/dist-types/stream-collector/index.d.ts

@@ -0,0 +1,2 @@
+import { StreamCollector } from "@smithy/types";
+export declare const streamCollector: StreamCollector;

package/node_modules/@smithy/node-http-handler/dist-types/stream-collector/readable.mock.d.ts

@@ -0,0 +1,14 @@
+/// <reference types="node" />
+/// <reference types="node" />
+import { Readable, ReadableOptions } from "stream";
+export interface ReadFromBuffersOptions extends ReadableOptions {
+    buffers: Buffer[];
+    errorAfter?: number;
+}
+export declare class ReadFromBuffers extends Readable {
+    private buffersToRead;
+    private numBuffersRead;
+    private errorAfter;
+    constructor(options: ReadFromBuffersOptions);
+    _read(size: number): boolean | undefined;
+}

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/node-http-handler.d.ts

@@ -0,0 +1,36 @@
+/// <reference types="node" />
+import { HttpHandler, HttpRequest, HttpResponse } from "@smithy/protocol-http";
+import { NodeHttpHandlerOptions } from "@smithy/types";
+import { HttpHandlerOptions, Provider } from "@smithy/types";
+import { Agent as hAgent } from "http";
+import { Agent as hsAgent } from "https";
+export { NodeHttpHandlerOptions };
+export declare const DEFAULT_REQUEST_TIMEOUT = 0;
+export declare class NodeHttpHandler implements HttpHandler<NodeHttpHandlerOptions> {
+    private config?;
+    private configProvider;
+    private socketWarningTimestamp;
+    readonly metadata: {
+        handlerProtocol: string;
+    };
+    /**
+     * @returns the input if it is an HttpHandler of any class,
+     * or instantiates a new instance of this handler.
+     */
+    static create(instanceOrOptions?: HttpHandler<any> | NodeHttpHandlerOptions | Provider<NodeHttpHandlerOptions | void>): NodeHttpHandler | HttpHandler<any>;
+    /**
+     * @internal
+     *
+     * @param agent - http(s) agent in use by the NodeHttpHandler instance.
+     * @returns timestamp of last emitted warning.
+     */
+    static checkSocketUsage(agent: hAgent | hsAgent, socketWarningTimestamp: number): number;
+    constructor(options?: NodeHttpHandlerOptions | Provider<NodeHttpHandlerOptions | void>);
+    private resolveDefaultConfig;
+    destroy(): void;
+    handle(request: HttpRequest, { abortSignal }?: HttpHandlerOptions): Promise<{
+        response: HttpResponse;
+    }>;
+    updateHttpClientConfig(key: keyof NodeHttpHandlerOptions, value: NodeHttpHandlerOptions[typeof key]): void;
+    httpHandlerConfigs(): NodeHttpHandlerOptions;
+}

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/node-http2-connection-manager.d.ts

@@ -0,0 +1,22 @@
+/// <reference types="node" />
+import { RequestContext } from "@smithy/types";
+import { ConnectConfiguration } from "@smithy/types";
+import { ConnectionManager, ConnectionManagerConfiguration } from "@smithy/types";
+import { ClientHttp2Session } from "http2";
+export declare class NodeHttp2ConnectionManager implements ConnectionManager<ClientHttp2Session> {
+    constructor(config: ConnectionManagerConfiguration);
+    private config;
+    private readonly sessionCache;
+    lease(requestContext: RequestContext, connectionConfiguration: ConnectConfiguration): ClientHttp2Session;
+    /**
+     * Delete a session from the connection pool.
+     * @param authority The authority of the session to delete.
+     * @param session The session to delete.
+     */
+    deleteSession(authority: string, session: ClientHttp2Session): void;
+    release(requestContext: RequestContext, session: ClientHttp2Session): void;
+    destroy(): void;
+    setMaxConcurrentStreams(maxConcurrentStreams: number): void;
+    setDisableConcurrentStreams(disableConcurrentStreams: boolean): void;
+    private getUrlString;
+}

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/node-http2-connection-pool.d.ts

@@ -0,0 +1,13 @@
+/// <reference types="node" />
+import { ConnectionPool } from "@smithy/types";
+import { ClientHttp2Session } from "http2";
+export declare class NodeHttp2ConnectionPool implements ConnectionPool<ClientHttp2Session> {
+    private sessions;
+    constructor(sessions?: ClientHttp2Session[]);
+    poll(): ClientHttp2Session | void;
+    offerLast(session: ClientHttp2Session): void;
+    contains(session: ClientHttp2Session): boolean;
+    remove(session: ClientHttp2Session): void;
+    [Symbol.iterator](): IterableIterator<ClientHttp2Session>;
+    destroy(connection: ClientHttp2Session): void;
+}

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/node-http2-handler.d.ts

@@ -0,0 +1,57 @@
+import { HttpHandler, HttpRequest, HttpResponse } from "@smithy/protocol-http";
+import { HttpHandlerOptions, Provider } from "@smithy/types";
+/**
+ * Represents the http2 options that can be passed to a node http2 client.
+ */
+export interface NodeHttp2HandlerOptions {
+    /**
+     * The maximum time in milliseconds that a stream may remain idle before it
+     * is closed.
+     */
+    requestTimeout?: number;
+    /**
+     * The maximum time in milliseconds that a session or socket may remain idle
+     * before it is closed.
+     * https://nodejs.org/docs/latest-v12.x/api/http2.html#http2_http2session_and_sockets
+     */
+    sessionTimeout?: number;
+    /**
+     * Disables processing concurrent streams on a ClientHttp2Session instance. When set
+     * to true, a new session instance is created for each request to a URL.
+     * **Default:** false.
+     * https://nodejs.org/api/http2.html#http2_class_clienthttp2session
+     */
+    disableConcurrentStreams?: boolean;
+    /**
+     * Maximum number of concurrent Http2Stream instances per ClientHttp2Session. Each session
+     * may have up to 2^31-1 Http2Stream instances over its lifetime.
+     * This value must be greater than or equal to 0.
+     * https://nodejs.org/api/http2.html#class-http2stream
+     */
+    maxConcurrentStreams?: number;
+}
+export declare class NodeHttp2Handler implements HttpHandler<NodeHttp2HandlerOptions> {
+    private config?;
+    private configProvider;
+    readonly metadata: {
+        handlerProtocol: string;
+    };
+    private readonly connectionManager;
+    /**
+     * @returns the input if it is an HttpHandler of any class,
+     * or instantiates a new instance of this handler.
+     */
+    static create(instanceOrOptions?: HttpHandler<any> | NodeHttp2HandlerOptions | Provider<NodeHttp2HandlerOptions | void>): HttpHandler<any> | NodeHttp2Handler;
+    constructor(options?: NodeHttp2HandlerOptions | Provider<NodeHttp2HandlerOptions | void>);
+    destroy(): void;
+    handle(request: HttpRequest, { abortSignal }?: HttpHandlerOptions): Promise<{
+        response: HttpResponse;
+    }>;
+    updateHttpClientConfig(key: keyof NodeHttp2HandlerOptions, value: NodeHttp2HandlerOptions[typeof key]): void;
+    httpHandlerConfigs(): NodeHttp2HandlerOptions;
+    /**
+     * Destroys a session.
+     * @param session The session to destroy.
+     */
+    private destroySession;
+}

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/readable.mock.d.ts

@@ -0,0 +1,13 @@
+/// <reference types="node" />
+import { Readable, ReadableOptions } from "stream";
+export interface ReadFromBuffersOptions extends ReadableOptions {
+    buffers: Buffer[];
+    errorAfter?: number;
+}
+export declare class ReadFromBuffers extends Readable {
+    private buffersToRead;
+    private numBuffersRead;
+    private errorAfter;
+    constructor(options: ReadFromBuffersOptions);
+    _read(): boolean | undefined;
+}

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/set-connection-timeout.d.ts

@@ -0,0 +1,2 @@
+import { ClientRequest } from "http";
+export declare const setConnectionTimeout: (request: ClientRequest, reject: (err: Error) => void, timeoutInMs?: number) => void;

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/set-socket-keep-alive.d.ts

@@ -0,0 +1,6 @@
+import { ClientRequest } from "http";
+export interface SocketKeepAliveOptions {
+    keepAlive: boolean;
+    keepAliveMsecs?: number;
+}
+export declare const setSocketKeepAlive: (request: ClientRequest, { keepAlive, keepAliveMsecs }: SocketKeepAliveOptions) => void;

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/set-socket-timeout.d.ts

@@ -0,0 +1,2 @@
+import { ClientRequest } from "http";
+export declare const setSocketTimeout: (request: ClientRequest, reject: (err: Error) => void, timeoutInMs?: number) => void;

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/stream-collector/collector.d.ts

@@ -0,0 +1,6 @@
+/// <reference types="node" />
+import { Writable } from "stream";
+export declare class Collector extends Writable {
+    readonly bufferedBytes: Buffer[];
+    _write(chunk: Buffer, encoding: string, callback: (err?: Error) => void): void;
+}

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/stream-collector/index.d.ts

@@ -0,0 +1,2 @@
+import { StreamCollector } from "@smithy/types";
+export declare const streamCollector: StreamCollector;

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/stream-collector/readable.mock.d.ts

@@ -0,0 +1,13 @@
+/// <reference types="node" />
+import { Readable, ReadableOptions } from "stream";
+export interface ReadFromBuffersOptions extends ReadableOptions {
+    buffers: Buffer[];
+    errorAfter?: number;
+}
+export declare class ReadFromBuffers extends Readable {
+    private buffersToRead;
+    private numBuffersRead;
+    private errorAfter;
+    constructor(options: ReadFromBuffersOptions);
+    _read(size: number): boolean | undefined;
+}

package/node_modules/@smithy/node-http-handler/dist-types/ts3.4/write-request-body.d.ts

@@ -0,0 +1,12 @@
+/// <reference types="node" />
+import { HttpRequest } from "@smithy/types";
+import { ClientRequest } from "http";
+import { ClientHttp2Stream } from "http2";
+/**
+ * This resolves when writeBody has been called.
+ *
+ * @param httpRequest - opened Node.js request.
+ * @param request - container with the request body.
+ * @param maxContinueTimeoutMs - maximum time to wait for the continue event. Minimum of 1000ms.
+ */
+export declare function writeRequestBody(httpRequest: ClientRequest | ClientHttp2Stream, request: HttpRequest, maxContinueTimeoutMs?: number): Promise<void>;

package/node_modules/@smithy/node-http-handler/dist-types/write-request-body.d.ts

@@ -0,0 +1,13 @@
+/// <reference types="node" />
+/// <reference types="node" />
+import { HttpRequest } from "@smithy/types";
+import { ClientRequest } from "http";
+import { ClientHttp2Stream } from "http2";
+/**
+ * This resolves when writeBody has been called.
+ *
+ * @param httpRequest - opened Node.js request.
+ * @param request - container with the request body.
+ * @param maxContinueTimeoutMs - maximum time to wait for the continue event. Minimum of 1000ms.
+ */
+export declare function writeRequestBody(httpRequest: ClientRequest | ClientHttp2Stream, request: HttpRequest, maxContinueTimeoutMs?: number): Promise<void>;

package/node_modules/@smithy/node-http-handler/package.json

@@ -0,0 +1,67 @@
+{
+  "name": "@smithy/node-http-handler",
+  "version": "2.5.0",
+  "description": "Provides a way to make requests",
+  "scripts": {
+    "build": "concurrently 'yarn:build:cjs' 'yarn:build:es' 'yarn:build:types && yarn build:types:downlevel'",
+    "build:cjs": "node ../../scripts/inline node-http-handler",
+    "build:es": "yarn g:tsc -p tsconfig.es.json",
+    "build:types": "yarn g:tsc -p tsconfig.types.json",
+    "build:types:downlevel": "downlevel-dts dist-types dist-types/ts3.4",
+    "stage-release": "rimraf ./.release && yarn pack && mkdir ./.release && tar zxvf ./package.tgz --directory ./.release && rm ./package.tgz",
+    "clean": "rimraf ./dist-* && rimraf *.tsbuildinfo || exit 0",
+    "lint": "eslint -c ../../.eslintrc.js \"src/**/*.ts\"",
+    "format": "prettier --config ../../prettier.config.js --ignore-path ../.prettierignore --write \"**/*.{ts,md,json}\"",
+    "extract:docs": "api-extractor run --local",
+    "test": "yarn g:jest --coverage --forceExit"
+  },
+  "author": {
+    "name": "AWS SDK for JavaScript Team",
+    "email": "",
+    "url": "https://aws.amazon.com/javascript/"
+  },
+  "license": "Apache-2.0",
+  "main": "./dist-cjs/index.js",
+  "module": "./dist-es/index.js",
+  "types": "./dist-types/index.d.ts",
+  "dependencies": {
+    "@smithy/abort-controller": "^2.2.0",
+    "@smithy/protocol-http": "^3.3.0",
+    "@smithy/querystring-builder": "^2.2.0",
+    "@smithy/types": "^2.12.0",
+    "tslib": "^2.6.2"
+  },
+  "devDependencies": {
+    "@tsconfig/recommended": "1.0.1",
+    "@types/node": "^14.14.31",
+    "concurrently": "7.0.0",
+    "downlevel-dts": "0.10.1",
+    "rimraf": "3.0.2",
+    "typedoc": "0.23.23"
+  },
+  "engines": {
+    "node": ">=14.0.0"
+  },
+  "typesVersions": {
+    "<4.0": {
+      "dist-types/*": [
+        "dist-types/ts3.4/*"
+      ]
+    }
+  },
+  "files": [
+    "dist-*/**"
+  ],
+  "homepage": "https://github.com/awslabs/smithy-typescript/tree/main/packages/node-http-handler",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/awslabs/smithy-typescript.git",
+    "directory": "packages/node-http-handler"
+  },
+  "typedoc": {
+    "entryPoint": "src/index.ts"
+  },
+  "publishConfig": {
+    "directory": ".release/package"
+  }
+}

package/out/client/sigv4/codewhisperer.d.ts

@@ -1,5 +1,5 @@
-import { ServiceConfigurationOptions } from 'aws-sdk/lib/service';
-import CodeWhispererClient = require('./codewhisperersigv4client');
+import { CodeWhispererClient, CodeWhispererClientConfig } from '@amzn/codewhisperer';
 import { SDKInitializator, Logging } from '@aws/language-server-runtimes/server-interface';
-export type CodeWhispererSigv4ClientConfigurationOptions = ServiceConfigurationOptions;
-export declare function createCodeWhispererSigv4Client(options: ServiceConfigurationOptions, sdkInitializator: SDKInitializator, logging: Logging): CodeWhispererClient;
+export type CodeWhispererSigv4ClientConfigurationOptions = CodeWhispererClientConfig;
+export declare function createCodeWhispererSigv4Client(options: CodeWhispererClientConfig, sdkInitializator: SDKInitializator, logging: Logging, shareCodeWhispererContentWithAWS?: boolean): CodeWhispererClient;
+export type CodeWhispererSigv4Client = CodeWhispererClient;

package/out/client/sigv4/codewhisperer.js

@@ -1,14 +1,45 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createCodeWhispererSigv4Client = createCodeWhispererSigv4Client;
-const aws_sdk_1 = require("aws-sdk");
-const apiConfig = require('./service.json');
-function createCodeWhispererSigv4Client(options, sdkInitializator, logging) {
-    return createService(options, sdkInitializator, logging);
-}
-function createService(options, sdkInitializator, logging) {
-    logging.log(`Passing client for class Service to sdkInitializator (v2) for additional setup (e.g. proxy)`);
-    const client = sdkInitializator.v2(aws_sdk_1.Service, { apiConfig, ...options });
+const codewhisperer_1 = require("@amzn/codewhisperer");
+function createCodeWhispererSigv4Client(options, sdkInitializator, logging, shareCodeWhispererContentWithAWS = false) {
+    logging.log(`Passing client for class CodeWhispererClient to sdkInitializator (v3) for additional setup (e.g. proxy)`);
+    const client = sdkInitializator(codewhisperer_1.CodeWhispererClient, {
+        ...options,
+    });
+    // Add middleware to set opt-out header
+    client.middlewareStack.add(next => async (args) => {
+        if (args.request &&
+            typeof args.request === 'object' &&
+            args.request !== null &&
+            'headers' in args.request) {
+            ;
+            args.request.headers['x-amzn-codewhisperer-optout'] = `${!shareCodeWhispererContentWithAWS}`;
+        }
+        return next(args);
+    }, {
+        step: 'build',
+        name: 'addOptOutHeader',
+        priority: 'high',
+    });
+    // Add middleware to capture HTTP headers
+    client.middlewareStack.add(next => async (args) => {
+        const result = await next(args);
+        // Store headers on the response metadata
+        if (result.response) {
+            const httpResponse = result.response;
+            if (httpResponse.headers && result.output?.$metadata) {
+                // Extend metadata to include headers
+                ;
+                result.output.$metadata.httpHeaders = httpResponse.headers;
+            }
+        }
+        return result;
+    }, {
+        step: 'deserialize',
+        name: 'captureHeaders',
+        priority: 'high',
+    });
     return client;
 }
 //# sourceMappingURL=codewhisperer.js.map

package/out/client/sigv4/codewhisperer.js.map

@@ -1 +1 @@
-{"version":3,"file":"codewhisperer.js","sourceRoot":"","sources":["../../../src/client/sigv4/codewhisperer.ts"],"names":[],"mappings":";;AAQA,wEAMC;AAdD,qCAAiC;AAEjC,MAAM,SAAS,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAAA;AAM3C,SAAgB,8BAA8B,CAC1C,OAAoC,EACpC,gBAAkC,EAClC,OAAgB;IAEhB,OAAO,aAAa,CAAC,OAAO,EAAE,gBAAgB,EAAE,OAAO,CAAwB,CAAA;AACnF,CAAC;AAED,SAAS,aAAa,CAClB,OAAoC,EACpC,gBAAkC,EAClC,OAAgB;IAEhB,OAAO,CAAC,GAAG,CAAC,6FAA6F,CAAC,CAAA;IAC1G,MAAM,MAAM,GAAG,gBAAgB,CAAC,EAAE,CAAC,iBAAO,EAAE,EAAE,SAAS,EAAE,GAAG,OAAO,EAAS,CAAC,CAAA;IAC7E,OAAO,MAAM,CAAA;AACjB,CAAC"}
+{"version":3,"file":"codewhisperer.js","sourceRoot":"","sources":["../../../src/client/sigv4/codewhisperer.ts"],"names":[],"mappings":";;AAMA,wEA0DC;AAhED,uDAAoF;AAMpF,SAAgB,8BAA8B,CAC1C,OAAkC,EAClC,gBAAkC,EAClC,OAAgB,EAChB,mCAA4C,KAAK;IAEjD,OAAO,CAAC,GAAG,CACP,yGAAyG,CAC5G,CAAA;IAED,MAAM,MAAM,GAAG,gBAAgB,CAAC,mCAAmB,EAAE;QACjD,GAAG,OAAO;KACb,CAAC,CAAA;IAEF,uCAAuC;IACvC,MAAM,CAAC,eAAe,CAAC,GAAG,CACtB,IAAI,CAAC,EAAE,CAAC,KAAK,EAAC,IAAI,EAAC,EAAE;QACjB,IACI,IAAI,CAAC,OAAO;YACZ,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ;YAChC,IAAI,CAAC,OAAO,KAAK,IAAI;YACrB,SAAS,IAAI,IAAI,CAAC,OAAO,EAC3B,CAAC;YACC,CAAC;YAAC,IAAI,CAAC,OAAe,CAAC,OAAO,CAAC,6BAA6B,CAAC,GAAG,GAAG,CAAC,gCAAgC,EAAE,CAAA;QAC1G,CAAC;QACD,OAAO,IAAI,CAAC,IAAI,CAAC,CAAA;IACrB,CAAC,EACD;QACI,IAAI,EAAE,OAAO;QACb,IAAI,EAAE,iBAAiB;QACvB,QAAQ,EAAE,MAAM;KACnB,CACJ,CAAA;IAED,yCAAyC;IACzC,MAAM,CAAC,eAAe,CAAC,GAAG,CACtB,IAAI,CAAC,EAAE,CAAC,KAAK,EAAC,IAAI,EAAC,EAAE;QACjB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,CAAA;QAE/B,yCAAyC;QACzC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,YAAY,GAAG,MAAM,CAAC,QAAwB,CAAA;YACpD,IAAI,YAAY,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC;gBACnD,qCAAqC;gBACrC,CAAC;gBAAC,MAAM,CAAC,MAAM,CAAC,SAAiB,CAAC,WAAW,GAAG,YAAY,CAAC,OAAO,CAAA;YACxE,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAA;IACjB,CAAC,EACD;QACI,IAAI,EAAE,aAAa;QACnB,IAAI,EAAE,gBAAgB;QACtB,QAAQ,EAAE,MAAM;KACnB,CACJ,CAAA;IAED,OAAO,MAAM,CAAA;AACjB,CAAC"}

package/out/client/token/codewhisperer.d.ts

@@ -1,15 +1,6 @@
-import { AWSError } from 'aws-sdk';
-import { ServiceConfigurationOptions } from 'aws-sdk/lib/service';
-import CodeWhispererClient = require('./codewhispererbearertokenclient');
-import { SDKInitializator, Logging } from '@aws/language-server-runtimes/server-interface';
-export interface RequestExtras {
-    readonly service: AWS.Service;
-    readonly operation: string;
-    readonly params?: any;
+import { CodeWhispererRuntimeClient, CodeWhispererRuntimeClientConfig } from '@amzn/codewhisperer-runtime';
+import { SDKInitializator, Logging, CredentialsProvider } from '@aws/language-server-runtimes/server-interface';
+export interface CodeWhispererTokenClientConfigurationOptions extends CodeWhispererRuntimeClientConfig {
 }
-type RequestListener = (request: AWS.Request<any, AWSError> & RequestExtras) => void;
-export interface CodeWhispererTokenClientConfigurationOptions extends ServiceConfigurationOptions {
-    onRequestSetup?: RequestListener | RequestListener[];
-}
-export declare function createCodeWhispererTokenClient(options: CodeWhispererTokenClientConfigurationOptions, sdkInitializator: SDKInitializator, logging: Logging): CodeWhispererClient;
-export {};
+export declare function createCodeWhispererTokenClient(options: CodeWhispererTokenClientConfigurationOptions, sdkInitializator: SDKInitializator, logging: Logging, credentialsProvider: CredentialsProvider, shareCodeWhispererContentWithAWS: boolean): CodeWhispererRuntimeClient;
+export type CodeWhispererTokenClient = CodeWhispererRuntimeClient;

package/out/client/token/codewhisperer.js

@@ -1,23 +1,39 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.createCodeWhispererTokenClient = createCodeWhispererTokenClient;
-const aws_sdk_1 = require("aws-sdk");
-const apiConfig = require('./bearer-token-service.json');
-function createCodeWhispererTokenClient(options, sdkInitializator, logging) {
-    return createService(options, sdkInitializator, logging);
-}
-function createService(options, sdkInitializator, logging) {
-    const onRequest = options?.onRequestSetup ?? [];
-    const listeners = Array.isArray(onRequest) ? onRequest : [onRequest];
-    const opt = { ...options };
-    delete opt.onRequestSetup;
-    logging.log(`Passing client for class Service to sdkInitializator (v2) for additional setup (e.g. proxy)`);
-    const client = sdkInitializator.v2(aws_sdk_1.Service, { apiConfig, ...options });
-    const originalClient = client.setupRequestListeners.bind(client);
-    client.setupRequestListeners = (request) => {
-        originalClient(request);
-        listeners.forEach(l => l(request));
-    };
+const codewhisperer_runtime_1 = require("@amzn/codewhisperer-runtime");
+function createCodeWhispererTokenClient(options, sdkInitializator, logging, credentialsProvider, shareCodeWhispererContentWithAWS) {
+    logging.log(`Passing client for class CodeWhispererRuntimeClient to sdkInitializator (v3) for additional setup (e.g. proxy)`);
+    const client = sdkInitializator(codewhisperer_runtime_1.CodeWhispererRuntimeClient, {
+        ...options,
+    });
+    // Add middleware for custom headers
+    client.middlewareStack.add(next => async (args) => {
+        const request = args.request;
+        request.headers['x-amzn-codewhisperer-optout'] = `${!shareCodeWhispererContentWithAWS}`;
+        if (credentialsProvider.getConnectionType() === 'external_idp') {
+            request.headers['TokenType'] = 'EXTERNAL_IDP';
+        }
+        return next(args);
+    }, { step: 'build', priority: 'high' });
+    // Add middleware to capture HTTP headers
+    client.middlewareStack.add(next => async (args) => {
+        const result = await next(args);
+        // Store headers on the response metadata
+        if (result.response) {
+            const httpResponse = result.response;
+            if (httpResponse.headers && result.output?.$metadata) {
+                // Extend metadata to include headers
+                ;
+                result.output.$metadata.httpHeaders = httpResponse.headers;
+            }
+        }
+        return result;
+    }, {
+        step: 'deserialize',
+        name: 'captureHeaders',
+        priority: 'high',
+    });
     return client;
 }
 //# sourceMappingURL=codewhisperer.js.map

package/out/client/token/codewhisperer.js.map

@@ -1 +1 @@
-{"version":3,"file":"codewhisperer.js","sourceRoot":"","sources":["../../../src/client/token/codewhisperer.ts"],"names":[],"mappings":";;AAqBA,wEAMC;AA3BD,qCAAoD;AAEpD,MAAM,SAAS,GAAG,OAAO,CAAC,6BAA6B,CAAC,CAAA;AAmBxD,SAAgB,8BAA8B,CAC1C,OAAqD,EACrD,gBAAkC,EAClC,OAAgB;IAEhB,OAAO,aAAa,CAAC,OAAO,EAAE,gBAAgB,EAAE,OAAO,CAAwB,CAAA;AACnF,CAAC;AAED,SAAS,aAAa,CAClB,OAAqD,EACrD,gBAAkC,EAClC,OAAgB;IAEhB,MAAM,SAAS,GAAG,OAAO,EAAE,cAAc,IAAI,EAAE,CAAA;IAC/C,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAA;IACpE,MAAM,GAAG,GAAG,EAAE,GAAG,OAAO,EAAE,CAAA;IAC1B,OAAO,GAAG,CAAC,cAAc,CAAA;IACzB,OAAO,CAAC,GAAG,CAAC,6FAA6F,CAAC,CAAA;IAC1G,MAAM,MAAM,GAAG,gBAAgB,CAAC,EAAE,CAAC,iBAAO,EAAE,EAAE,SAAS,EAAE,GAAG,OAAO,EAAS,CAAC,CAAA;IAC7E,MAAM,cAAc,GAAG,MAAM,CAAC,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IAEhE,MAAM,CAAC,qBAAqB,GAAG,CAAC,OAA+B,EAAE,EAAE;QAC/D,cAAc,CAAC,OAAO,CAAC,CAAA;QACvB,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAqD,CAAC,CAAC,CAAA;IACpF,CAAC,CAAA;IAED,OAAO,MAAM,CAAA;AACjB,CAAC"}
+{"version":3,"file":"codewhisperer.js","sourceRoot":"","sources":["../../../src/client/token/codewhisperer.ts"],"names":[],"mappings":";;AAQA,wEAsDC;AA9DD,uEAA0G;AAQ1G,SAAgB,8BAA8B,CAC1C,OAAqD,EACrD,gBAAkC,EAClC,OAAgB,EAChB,mBAAwC,EACxC,gCAAyC;IAEzC,OAAO,CAAC,GAAG,CACP,gHAAgH,CACnH,CAAA;IAED,MAAM,MAAM,GAAG,gBAAgB,CAAC,kDAA0B,EAAE;QACxD,GAAG,OAAO;KACb,CAAC,CAAA;IAEF,oCAAoC;IACpC,MAAM,CAAC,eAAe,CAAC,GAAG,CACtB,IAAI,CAAC,EAAE,CAAC,KAAK,EAAC,IAAI,EAAC,EAAE;QACjB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAsB,CAAA;QAC3C,OAAO,CAAC,OAAO,CAAC,6BAA6B,CAAC,GAAG,GAAG,CAAC,gCAAgC,EAAE,CAAA;QAEvF,IAAI,mBAAmB,CAAC,iBAAiB,EAAE,KAAK,cAAc,EAAE,CAAC;YAC7D,OAAO,CAAC,OAAO,CAAC,WAAW,CAAC,GAAG,cAAc,CAAA;QACjD,CAAC;QAED,OAAO,IAAI,CAAC,IAAI,CAAC,CAAA;IACrB,CAAC,EACD,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CACtC,CAAA;IAED,yCAAyC;IACzC,MAAM,CAAC,eAAe,CAAC,GAAG,CACtB,IAAI,CAAC,EAAE,CAAC,KAAK,EAAC,IAAI,EAAC,EAAE;QACjB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,CAAA;QAE/B,yCAAyC;QACzC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,YAAY,GAAG,MAAM,CAAC,QAAwB,CAAA;YACpD,IAAI,YAAY,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC;gBACnD,qCAAqC;gBACrC,CAAC;gBAAC,MAAM,CAAC,MAAM,CAAC,SAAiB,CAAC,WAAW,GAAG,YAAY,CAAC,OAAO,CAAA;YACxE,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAA;IACjB,CAAC,EACD;QACI,IAAI,EAAE,aAAa;QACnB,IAAI,EAAE,gBAAgB;QACtB,QAAQ,EAAE,MAAM;KACnB,CACJ,CAAA;IAED,OAAO,MAAM,CAAA;AACjB,CAAC"}

package/out/language-server/agenticChat/agenticChatController.d.ts

@@ -15,7 +15,7 @@ import { ChatSessionService } from '../chat/chatSessionService';
 import { AgenticChatResultStream } from './agenticChatResultStream';
 import { AdditionalContentEntryAddition } from './context/agenticChatTriggerContext';
 import { PaidTierMode } from '../paidTier/paidTier';
-import { UserContext } from '../../client/token/codewhispererbearertokenclient';
+import { UserContext } from '@amzn/codewhisperer-runtime';
 type ChatHandlers = Omit<LspHandlers<Chat>, 'openTab' | 'sendChatUpdate' | 'sendContextCommands' | 'onListConversations' | 'onConversationClick' | 'onListMcpServers' | 'onMcpServerClick' | 'onTabBarAction' | 'getSerializedChat' | 'chatOptionsUpdate' | 'onListRules' | 'sendPinnedContext' | 'onActiveEditorChanged' | 'onPinnedContextAdd' | 'onPinnedContextRemove' | 'onOpenFileDialog' | 'onListAvailableModels' | 'sendSubscriptionDetails' | 'onSubscriptionUpgrade'>;
 export declare class AgenticChatController implements ChatHandlers {
     #private;