@aws/lsp-codewhisperer 0.0.31 → 0.0.33

package/out/shared/constants.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AWS_Q_ENDPOINT_URL_ENV_VAR = exports.AWS_Q_REGION_ENV_VAR = exports.AWS_Q_ENDPOINTS = exports.DEFAULT_AWS_Q_REGION = exports.DEFAULT_AWS_Q_ENDPOINT_URL = exports.BUILDER_ID_START_URL = exports.GENERIC_UNAUTHORIZED_ERROR = exports.INVALID_TOKEN = exports.MISSING_BEARER_TOKEN_ERROR = void 0;
+exports.driveLetterRegex = exports.crashMonitoringDirName = exports.CODE_WHISPERER_CONFIGURATION_SECTION = exports.Q_CONFIGURATION_SECTION = exports.AWS_Q_ENDPOINT_URL_ENV_VAR = exports.AWS_Q_REGION_ENV_VAR = exports.AWS_Q_ENDPOINTS = exports.DEFAULT_AWS_Q_REGION = exports.DEFAULT_AWS_Q_ENDPOINT_URL = exports.BUILDER_ID_START_URL = exports.GENERIC_UNAUTHORIZED_ERROR = exports.INVALID_TOKEN = exports.MISSING_BEARER_TOKEN_ERROR = void 0;
 exports.MISSING_BEARER_TOKEN_ERROR = 'credentialsProvider does not have bearer token credentials';
 exports.INVALID_TOKEN = 'The bearer token included in the request is invalid.';
 exports.GENERIC_UNAUTHORIZED_ERROR = 'User is not authorized to make this call';
@@ -9,8 +9,19 @@ exports.DEFAULT_AWS_Q_ENDPOINT_URL = 'https://codewhisperer.us-east-1.amazonaws.
 exports.DEFAULT_AWS_Q_REGION = 'us-east-1';
 exports.AWS_Q_ENDPOINTS = new Map([
     [exports.DEFAULT_AWS_Q_REGION, exports.DEFAULT_AWS_Q_ENDPOINT_URL],
+    ['us-east-1', 'https://codewhisperer.us-east-1.amazonaws.com/'],
     ['eu-central-1', 'https://q.eu-central-1.amazonaws.com/'],
 ]);
 exports.AWS_Q_REGION_ENV_VAR = 'AWS_Q_REGION';
 exports.AWS_Q_ENDPOINT_URL_ENV_VAR = 'AWS_Q_ENDPOINT_URL';
+exports.Q_CONFIGURATION_SECTION = 'aws.q';
+exports.CODE_WHISPERER_CONFIGURATION_SECTION = 'aws.codeWhisperer';
+/**
+ * Names of directories relevant to the crash reporting functionality.
+ *
+ * Moved here to resolve circular dependency issues.
+ */
+exports.crashMonitoringDirName = 'crashMonitoring';
+/** Matches Windows drive letter ("C:"). */
+exports.driveLetterRegex = /^[a-zA-Z]\:/;
 //# sourceMappingURL=constants.js.map

package/out/shared/constants.js.map

@@ -1 +1 @@
-{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/shared/constants.ts"],"names":[],"mappings":";;;AAAa,QAAA,0BAA0B,GAAG,4DAA4D,CAAA;AACzF,QAAA,aAAa,GAAG,sDAAsD,CAAA;AACtE,QAAA,0BAA0B,GAAG,0CAA0C,CAAA;AACvE,QAAA,oBAAoB,GAAG,gCAAgC,CAAA;AACvD,QAAA,0BAA0B,GAAG,gDAAgD,CAAA;AAC7E,QAAA,oBAAoB,GAAG,WAAW,CAAA;AAElC,QAAA,eAAe,GAAG,IAAI,GAAG,CAAC;IACnC,CAAC,4BAAoB,EAAE,kCAA0B,CAAC;IAClD,CAAC,cAAc,EAAE,uCAAuC,CAAC;CAC5D,CAAC,CAAA;AAEW,QAAA,oBAAoB,GAAG,cAAc,CAAA;AACrC,QAAA,0BAA0B,GAAG,oBAAoB,CAAA"}
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/shared/constants.ts"],"names":[],"mappings":";;;AAAa,QAAA,0BAA0B,GAAG,4DAA4D,CAAA;AACzF,QAAA,aAAa,GAAG,sDAAsD,CAAA;AACtE,QAAA,0BAA0B,GAAG,0CAA0C,CAAA;AACvE,QAAA,oBAAoB,GAAG,gCAAgC,CAAA;AACvD,QAAA,0BAA0B,GAAG,gDAAgD,CAAA;AAC7E,QAAA,oBAAoB,GAAG,WAAW,CAAA;AAElC,QAAA,eAAe,GAAG,IAAI,GAAG,CAAC;IACnC,CAAC,4BAAoB,EAAE,kCAA0B,CAAC;IAClD,CAAC,WAAW,EAAE,gDAAgD,CAAC;IAC/D,CAAC,cAAc,EAAE,uCAAuC,CAAC;CAC5D,CAAC,CAAA;AAEW,QAAA,oBAAoB,GAAG,cAAc,CAAA;AACrC,QAAA,0BAA0B,GAAG,oBAAoB,CAAA;AAEjD,QAAA,uBAAuB,GAAG,OAAO,CAAA;AACjC,QAAA,oCAAoC,GAAG,mBAAmB,CAAA;AAEvE;;;;GAIG;AACU,QAAA,sBAAsB,GAAG,iBAAiB,CAAA;AAEvD,2CAA2C;AAC9B,QAAA,gBAAgB,GAAG,aAAa,CAAA"}

package/out/shared/localProjectContextController.d.ts

@@ -0,0 +1,56 @@
+import { Logging, WorkspaceFolder } from '@aws/language-server-runtimes/server-interface';
+import type { AdditionalContextPrompt, Chunk, ContextCommandItem, InlineProjectContext, QueryInlineProjectContextRequestV2, QueryRequest, UpdateMode } from 'local-indexing';
+export interface SizeConstraints {
+    maxFileSize: number;
+    remainingIndexSize: number;
+}
+export interface LocalProjectContextInitializationOptions {
+    vectorLib?: any;
+    ignoreFilePatterns?: string[];
+    respectUserGitIgnores?: boolean;
+    fileExtensions?: string[];
+    includeSymlinks?: boolean;
+    maxFileSizeMB?: number;
+    maxIndexSizeMB?: number;
+    indexCacheDirPath?: string;
+    enableGpuAcceleration?: boolean;
+    indexWorkerThreads?: number;
+    enableIndexing?: boolean;
+}
+export declare class LocalProjectContextController {
+    private static instance;
+    private workspaceFolders;
+    private _vecLib?;
+    private _contextCommandSymbolsUpdated;
+    private readonly clientName;
+    private readonly log;
+    private _isIndexingEnabled;
+    private ignoreFilePatterns?;
+    private includeSymlinks?;
+    private maxFileSizeMB?;
+    private maxIndexSizeMB?;
+    private respectUserGitIgnores?;
+    private indexCacheDirPath;
+    private readonly fileExtensions;
+    private readonly DEFAULT_MAX_INDEX_SIZE_MB;
+    private readonly DEFAULT_MAX_FILE_SIZE_MB;
+    private readonly MB_TO_BYTES;
+    constructor(clientName: string, workspaceFolders: WorkspaceFolder[], logging: Logging);
+    get isEnabled(): boolean;
+    static getInstance(): Promise<LocalProjectContextController>;
+    init({ vectorLib, ignoreFilePatterns, respectUserGitIgnores, includeSymlinks, maxFileSizeMB, maxIndexSizeMB, indexCacheDirPath, enableGpuAcceleration, indexWorkerThreads, enableIndexing, }?: LocalProjectContextInitializationOptions): Promise<void>;
+    private getVectorLibraryPath;
+    dispose(): Promise<void>;
+    updateIndex(filePaths: string[], operation: UpdateMode): Promise<void>;
+    buildIndex(): Promise<void>;
+    updateWorkspaceFolders(added: WorkspaceFolder[], removed: WorkspaceFolder[]): Promise<void>;
+    queryInlineProjectContext(request: QueryInlineProjectContextRequestV2): Promise<InlineProjectContext[]>;
+    queryVectorIndex(request: QueryRequest): Promise<Chunk[]>;
+    getContextCommandItems(): Promise<ContextCommandItem[]>;
+    shouldUpdateContextCommand(filePaths: string[], isAdd: boolean): Promise<boolean>;
+    shouldUpdateContextCommandSymbolsOnce(): Promise<boolean>;
+    getContextCommandPrompt(contextCommandItems: ContextCommandItem[]): Promise<AdditionalContextPrompt[]>;
+    private fileMeetsFileSizeConstraints;
+    private processWorkspaceFolders;
+    private areWorkspaceFoldersEqual;
+}

package/out/shared/localProjectContextController.js

@@ -0,0 +1,371 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LocalProjectContextController = void 0;
+const path_1 = require("path");
+const languageDetection_1 = require("./languageDetection");
+const os_1 = require("os");
+const vscode_uri_1 = require("vscode-uri");
+const timeoutUtils_1 = require("@aws/lsp-core/out/util/timeoutUtils");
+const fs = require("fs");
+const path = require("path");
+const ignore = require("ignore");
+const fdir_1 = require("fdir");
+const url_1 = require("url");
+const LIBRARY_DIR = (() => {
+    if (require.main?.filename) {
+        return path.join((0, path_1.dirname)(require.main.filename), 'indexing');
+    }
+    return path.join(__dirname, 'indexing');
+})();
+class LocalProjectContextController {
+    static instance;
+    workspaceFolders;
+    _vecLib;
+    _contextCommandSymbolsUpdated = false;
+    clientName;
+    log;
+    _isIndexingEnabled = false;
+    ignoreFilePatterns;
+    includeSymlinks;
+    maxFileSizeMB;
+    maxIndexSizeMB;
+    respectUserGitIgnores;
+    indexCacheDirPath = path.join((0, os_1.homedir)(), '.aws', 'amazonq', 'cache');
+    fileExtensions = Object.keys(languageDetection_1.languageByExtension);
+    DEFAULT_MAX_INDEX_SIZE_MB = 2048;
+    DEFAULT_MAX_FILE_SIZE_MB = 10;
+    MB_TO_BYTES = 1024 * 1024;
+    constructor(clientName, workspaceFolders, logging) {
+        this.workspaceFolders = workspaceFolders;
+        this.clientName = clientName;
+        this.log = logging;
+    }
+    get isEnabled() {
+        return this._vecLib !== undefined && this._vecLib !== null;
+    }
+    static async getInstance() {
+        try {
+            await (0, timeoutUtils_1.waitUntil)(async () => this.instance, {
+                interval: 100,
+                timeout: 600,
+                truthy: true,
+            });
+            if (!this.instance) {
+                throw new Error('LocalProjectContextController initialization timeout after 60 seconds');
+            }
+            return this.instance;
+        }
+        catch (error) {
+            // throw new Error(`Failed to get LocalProjectContextController instance: ${error}`)
+            return {
+                isEnabled: true,
+                getContextCommandItems: () => [],
+                shouldUpdateContextCommandSymbolsOnce: () => false,
+            };
+        }
+    }
+    async init({ vectorLib, ignoreFilePatterns = [], respectUserGitIgnores = true, includeSymlinks = false, maxFileSizeMB = this.DEFAULT_MAX_FILE_SIZE_MB, maxIndexSizeMB = this.DEFAULT_MAX_INDEX_SIZE_MB, indexCacheDirPath = path.join((0, os_1.homedir)(), '.aws', 'amazonq', 'cache'), enableGpuAcceleration = false, indexWorkerThreads = 0, enableIndexing = false, } = {}) {
+        try {
+            // update states according to configuration
+            this.includeSymlinks = includeSymlinks;
+            this.maxFileSizeMB = maxFileSizeMB;
+            this.maxIndexSizeMB = maxIndexSizeMB;
+            this.respectUserGitIgnores = respectUserGitIgnores;
+            this.ignoreFilePatterns = ignoreFilePatterns;
+            if (indexCacheDirPath?.length > 0 && path.parse(indexCacheDirPath)) {
+                this.indexCacheDirPath = indexCacheDirPath;
+            }
+            if (enableGpuAcceleration) {
+                process.env.Q_ENABLE_GPU = 'true';
+            }
+            else {
+                delete process.env.Q_ENABLE_GPU;
+            }
+            if (indexWorkerThreads && indexWorkerThreads > 0 && indexWorkerThreads < 100) {
+                process.env.Q_WORKER_THREADS = indexWorkerThreads.toString();
+            }
+            else {
+                delete process.env.Q_WORKER_THREADS;
+            }
+            this.log.info(`Vector library initializing with GPU acceleration: ${enableGpuAcceleration}, ` +
+                `index worker thread count: ${indexWorkerThreads}`);
+            // build index if vecLib was initialized but indexing was not enabled before
+            if (this._vecLib) {
+                if (enableIndexing && !this._isIndexingEnabled) {
+                    void this.buildIndex();
+                }
+                this._isIndexingEnabled = enableIndexing;
+                return;
+            }
+            // initialize vecLib and index if needed
+            const libraryPath = this.getVectorLibraryPath();
+            const vecLib = vectorLib ?? (await eval(`import("${libraryPath}")`));
+            if (vecLib) {
+                this._vecLib = await vecLib.start(LIBRARY_DIR, this.clientName, this.indexCacheDirPath);
+                if (enableIndexing) {
+                    void this.buildIndex();
+                }
+                LocalProjectContextController.instance = this;
+                this._isIndexingEnabled = enableIndexing;
+            }
+            else {
+                this.log.warn(`Vector library could not be imported from: ${libraryPath}`);
+            }
+        }
+        catch (error) {
+            this.log.error('Vector library failed to initialize:' + error);
+        }
+    }
+    getVectorLibraryPath() {
+        const libraryPath = path.join(LIBRARY_DIR, 'dist', 'extension.js');
+        if (process.platform === 'win32') {
+            // On Windows, the path must be loaded using a URL.
+            // Using the file path directly results in ERR_UNSUPPORTED_ESM_URL_SCHEME
+            // More details: https://github.com/nodejs/node/issues/31710
+            return (0, url_1.pathToFileURL)(libraryPath).toString();
+        }
+        return libraryPath;
+    }
+    async dispose() {
+        if (this._vecLib) {
+            await this._vecLib?.clear?.();
+            this._vecLib = undefined;
+        }
+    }
+    async updateIndex(filePaths, operation) {
+        if (!this._vecLib) {
+            return;
+        }
+        try {
+            await this._vecLib?.updateIndexV2(filePaths, operation);
+        }
+        catch (error) {
+            this.log.error(`Error updating index: ${error}`);
+        }
+    }
+    // public for test
+    async buildIndex() {
+        try {
+            if (this._vecLib) {
+                if (!this.workspaceFolders.length) {
+                    this.log.info('skip building index because no workspace folder found');
+                    return;
+                }
+                const sourceFiles = await this.processWorkspaceFolders(this.workspaceFolders, this.ignoreFilePatterns, this.respectUserGitIgnores, this.includeSymlinks, this.fileExtensions, this.maxFileSizeMB, this.maxIndexSizeMB);
+                const projectRoot = this.workspaceFolders.sort()[0].uri;
+                await this._vecLib?.buildIndex(sourceFiles, projectRoot, 'all');
+                this.log.info('Context index built successfully');
+            }
+        }
+        catch (error) {
+            this.log.error(`Error building index: ${error}`);
+        }
+    }
+    async updateWorkspaceFolders(added, removed) {
+        try {
+            const afterRemovals = this.workspaceFolders.filter(existing => !removed.some(removal => this.areWorkspaceFoldersEqual(existing, removal)));
+            const merged = [...afterRemovals];
+            for (const addition of added) {
+                if (!merged.some(existing => this.areWorkspaceFoldersEqual(existing, addition))) {
+                    merged.push(addition);
+                }
+            }
+            if (this._vecLib && this._isIndexingEnabled) {
+                void this.buildIndex();
+            }
+        }
+        catch (error) {
+            this.log.error(`Error in updateWorkspaceFolders: ${error}`);
+        }
+    }
+    async queryInlineProjectContext(request) {
+        if (!this._vecLib) {
+            return [];
+        }
+        try {
+            const resp = await this._vecLib?.queryInlineProjectContext(request.query, request.filePath, request.target);
+            return resp ?? [];
+        }
+        catch (error) {
+            this.log.error(`Error in queryInlineProjectContext: ${error}`);
+            return [];
+        }
+    }
+    async queryVectorIndex(request) {
+        if (!this._vecLib) {
+            return [];
+        }
+        try {
+            const resp = await this._vecLib?.queryVectorIndex(request.query);
+            return resp ?? [];
+        }
+        catch (error) {
+            this.log.error(`Error in queryVectorIndex: ${error}`);
+            return [];
+        }
+    }
+    async getContextCommandItems() {
+        if (!this._vecLib) {
+            return [];
+        }
+        try {
+            const foldersPath = this.workspaceFolders.map(folder => vscode_uri_1.URI.parse(folder.uri).fsPath);
+            const resp = await this._vecLib?.getContextCommandItems(foldersPath);
+            this.log.log(`received ${resp.length} context command items`);
+            return resp ?? [];
+        }
+        catch (error) {
+            this.log.error(`Error in getContextCommandItems: ${error}`);
+            return [];
+        }
+    }
+    async shouldUpdateContextCommand(filePaths, isAdd) {
+        try {
+            const indexSeqNum = await this._vecLib?.getIndexSequenceNumber();
+            await this.updateIndex(filePaths, isAdd ? 'add' : 'remove');
+            await (0, timeoutUtils_1.waitUntil)(async () => {
+                const newIndexSeqNum = await this._vecLib?.getIndexSequenceNumber();
+                if (newIndexSeqNum && indexSeqNum && newIndexSeqNum > indexSeqNum) {
+                    return true;
+                }
+                return false;
+            }, { interval: 500, timeout: 5_000, truthy: true });
+            return true;
+        }
+        catch (error) {
+            this.log.error(`Error in shouldUpdateContextCommand(: ${error}`);
+            return false;
+        }
+    }
+    async shouldUpdateContextCommandSymbolsOnce() {
+        if (this._contextCommandSymbolsUpdated) {
+            return false;
+        }
+        this._contextCommandSymbolsUpdated = true;
+        try {
+            const indexSeqNum = await this._vecLib?.getIndexSequenceNumber();
+            await this.updateIndex([], 'context_command_symbol_update');
+            await (0, timeoutUtils_1.waitUntil)(async () => {
+                const newIndexSeqNum = await this._vecLib?.getIndexSequenceNumber();
+                if (newIndexSeqNum && indexSeqNum && newIndexSeqNum > indexSeqNum) {
+                    return true;
+                }
+                return false;
+            }, { interval: 1000, timeout: 60_000, truthy: true });
+            return true;
+        }
+        catch (error) {
+            this.log.error(`Error in shouldUpdateContextCommandSymbolsOnce: ${error}`);
+            return false;
+        }
+    }
+    async getContextCommandPrompt(contextCommandItems) {
+        if (!this._vecLib) {
+            return [];
+        }
+        try {
+            const resp = await this._vecLib?.getContextCommandPrompt(contextCommandItems);
+            return resp ?? [];
+        }
+        catch (error) {
+            this.log.error(`Error in getContextCommandPrompt: ${error}`);
+            return [];
+        }
+    }
+    fileMeetsFileSizeConstraints(filePath, sizeConstraints) {
+        let fileSize;
+        try {
+            fileSize = fs.statSync(filePath).size;
+        }
+        catch (error) {
+            this.log.error(`Error reading file size for ${filePath}: ${error}`);
+            return false;
+        }
+        if (fileSize > sizeConstraints.maxFileSize || fileSize > sizeConstraints.remainingIndexSize) {
+            return false;
+        }
+        sizeConstraints.remainingIndexSize -= fileSize;
+        return true;
+    }
+    async processWorkspaceFolders(workspaceFolders, ignoreFilePatterns, respectUserGitIgnores, includeSymLinks, fileExtensions, maxFileSizeMB, maxIndexSizeMB) {
+        if (!workspaceFolders?.length) {
+            this.log.info(`Skipping indexing: no workspace folders available`);
+            return [];
+        }
+        this.log.info(`Indexing ${workspaceFolders.length} workspace folders...`);
+        const filter = ignore().add(ignoreFilePatterns ?? []);
+        maxFileSizeMB = Math.min(maxFileSizeMB ?? Infinity, this.DEFAULT_MAX_FILE_SIZE_MB);
+        maxIndexSizeMB = Math.min(maxIndexSizeMB ?? Infinity, this.DEFAULT_MAX_INDEX_SIZE_MB);
+        const sizeConstraints = {
+            maxFileSize: maxFileSizeMB * this.MB_TO_BYTES,
+            remainingIndexSize: maxIndexSizeMB * this.MB_TO_BYTES,
+        };
+        const controller = new AbortController();
+        const workspaceSourceFiles = await Promise.all(workspaceFolders.map(async (folder) => {
+            const absolutePath = path.resolve(vscode_uri_1.URI.parse(folder.uri).fsPath);
+            const localGitIgnoreFiles = [];
+            const crawler = new fdir_1.fdir()
+                .withSymlinks({ resolvePaths: !includeSymLinks })
+                .withAbortSignal(controller.signal)
+                .exclude((dirName, dirPath) => {
+                const relativePath = path.relative(absolutePath, dirPath);
+                return relativePath.startsWith('..') || filter.ignores(relativePath);
+            })
+                .glob(...(fileExtensions?.map(ext => `**/*${ext}`) ?? []), '**/.gitignore')
+                .filter((filePath, isDirectory) => {
+                const relativePath = path.relative(absolutePath, filePath);
+                if (isDirectory || relativePath.startsWith('..') || filter.ignores(relativePath)) {
+                    return false;
+                }
+                if (!respectUserGitIgnores && sizeConstraints.remainingIndexSize <= 0) {
+                    controller.abort();
+                    return false;
+                }
+                if (path.basename(filePath) === '.gitignore') {
+                    localGitIgnoreFiles.push(filePath);
+                    return false;
+                }
+                return respectUserGitIgnores || this.fileMeetsFileSizeConstraints(filePath, sizeConstraints);
+            });
+            return crawler
+                .crawl(absolutePath)
+                .withPromise()
+                .then(async (sourceFiles) => {
+                if (!respectUserGitIgnores) {
+                    return sourceFiles;
+                }
+                const userGitIgnoreFilterByFile = new Map(await Promise.all(localGitIgnoreFiles.map(async (filePath) => {
+                    const filter = ignore();
+                    try {
+                        filter.add((await fs.promises.readFile(filePath)).toString());
+                    }
+                    catch (error) {
+                        this.log.error(`Error reading .gitignore file ${filePath}: ${error}`);
+                    }
+                    return [filePath, filter];
+                })));
+                return sourceFiles.reduce((filteredSourceFiles, filePath) => {
+                    if (sizeConstraints.remainingIndexSize <= 0) {
+                        return filteredSourceFiles;
+                    }
+                    const isIgnored = [...userGitIgnoreFilterByFile].some(([gitIgnorePath, filter]) => {
+                        const gitIgnoreDir = path.dirname(path.resolve(gitIgnorePath));
+                        const relativePath = path.relative(gitIgnoreDir, filePath);
+                        return !relativePath.startsWith('..') && filter.ignores(relativePath);
+                    });
+                    if (!isIgnored && this.fileMeetsFileSizeConstraints(filePath, sizeConstraints)) {
+                        filteredSourceFiles.push(filePath);
+                    }
+                    return filteredSourceFiles;
+                }, []);
+            });
+        })).then((nestedFilePaths) => nestedFilePaths.flat());
+        this.log.info(`Indexing complete: found ${workspaceSourceFiles.length} files.`);
+        return workspaceSourceFiles;
+    }
+    areWorkspaceFoldersEqual(a, b) {
+        return a.uri === b.uri && a.name === b.name;
+    }
+}
+exports.LocalProjectContextController = LocalProjectContextController;
+//# sourceMappingURL=localProjectContextController.js.map

package/out/shared/localProjectContextController.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"localProjectContextController.js","sourceRoot":"","sources":["../../src/shared/localProjectContextController.ts"],"names":[],"mappings":";;;AACA,+BAA8B;AAC9B,2DAAyD;AACzD,2BAA4B;AAW5B,2CAAgC;AAChC,sEAA+D;AAE/D,yBAAwB;AACxB,6BAA4B;AAE5B,iCAAgC;AAChC,+BAA2B;AAC3B,6BAAmC;AAEnC,MAAM,WAAW,GAAG,CAAC,GAAG,EAAE;IACtB,IAAI,OAAO,CAAC,IAAI,EAAE,QAAQ,EAAE,CAAC;QACzB,OAAO,IAAI,CAAC,IAAI,CAAC,IAAA,cAAO,EAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,UAAU,CAAC,CAAA;IAChE,CAAC;IACD,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,UAAU,CAAC,CAAA;AAC3C,CAAC,CAAC,EAAE,CAAA;AAqBJ,MAAa,6BAA6B;IAC9B,MAAM,CAAC,QAAQ,CAA2C;IAE1D,gBAAgB,CAAmB;IACnC,OAAO,CAAe;IACtB,6BAA6B,GAAG,KAAK,CAAA;IAC5B,UAAU,CAAQ;IAClB,GAAG,CAAS;IACrB,kBAAkB,GAAY,KAAK,CAAA;IAEnC,kBAAkB,CAAW;IAC7B,eAAe,CAAU;IACzB,aAAa,CAAS;IACtB,cAAc,CAAS;IACvB,qBAAqB,CAAU;IAC/B,iBAAiB,GAAW,IAAI,CAAC,IAAI,CAAC,IAAA,YAAO,GAAE,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,CAAA;IAEnE,cAAc,GAAa,MAAM,CAAC,IAAI,CAAC,uCAAmB,CAAC,CAAA;IAC3D,yBAAyB,GAAG,IAAI,CAAA;IAChC,wBAAwB,GAAG,EAAE,CAAA;IAC7B,WAAW,GAAG,IAAI,GAAG,IAAI,CAAA;IAE1C,YAAY,UAAkB,EAAE,gBAAmC,EAAE,OAAgB;QACjF,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAA;QACxC,IAAI,CAAC,UAAU,GAAG,UAAU,CAAA;QAC5B,IAAI,CAAC,GAAG,GAAG,OAAO,CAAA;IACtB,CAAC;IAED,IAAI,SAAS;QACT,OAAO,IAAI,CAAC,OAAO,KAAK,SAAS,IAAI,IAAI,CAAC,OAAO,KAAK,IAAI,CAAA;IAC9D,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,WAAW;QAC3B,IAAI,CAAC;YACD,MAAM,IAAA,wBAAS,EAAC,KAAK,IAAI,EAAE,CAAC,IAAI,CAAC,QAAQ,EAAE;gBACvC,QAAQ,EAAE,GAAG;gBACb,OAAO,EAAE,GAAG;gBACZ,MAAM,EAAE,IAAI;aACf,CAAC,CAAA;YAEF,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,uEAAuE,CAAC,CAAA;YAC5F,CAAC;YAED,OAAO,IAAI,CAAC,QAAQ,CAAA;QACxB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,oFAAoF;YACpF,OAAO;gBACH,SAAS,EAAE,IAAI;gBACf,sBAAsB,EAAE,GAAG,EAAE,CAAC,EAAE;gBAChC,qCAAqC,EAAE,GAAG,EAAE,CAAC,KAAK;aACT,CAAA;QACjD,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,IAAI,CAAC,EACd,SAAS,EACT,kBAAkB,GAAG,EAAE,EACvB,qBAAqB,GAAG,IAAI,EAC5B,eAAe,GAAG,KAAK,EACvB,aAAa,GAAG,IAAI,CAAC,wBAAwB,EAC7C,cAAc,GAAG,IAAI,CAAC,yBAAyB,EAC/C,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,IAAA,YAAO,GAAE,EAAE,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,EACpE,qBAAqB,GAAG,KAAK,EAC7B,kBAAkB,GAAG,CAAC,EACtB,cAAc,GAAG,KAAK,MACoB,EAAE;QAC5C,IAAI,CAAC;YACD,2CAA2C;YAC3C,IAAI,CAAC,eAAe,GAAG,eAAe,CAAA;YACtC,IAAI,CAAC,aAAa,GAAG,aAAa,CAAA;YAClC,IAAI,CAAC,cAAc,GAAG,cAAc,CAAA;YACpC,IAAI,CAAC,qBAAqB,GAAG,qBAAqB,CAAA;YAClD,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAA;YAC5C,IAAI,iBAAiB,EAAE,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACjE,IAAI,CAAC,iBAAiB,GAAG,iBAAiB,CAAA;YAC9C,CAAC;YAED,IAAI,qBAAqB,EAAE,CAAC;gBACxB,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,MAAM,CAAA;YACrC,CAAC;iBAAM,CAAC;gBACJ,OAAO,OAAO,CAAC,GAAG,CAAC,YAAY,CAAA;YACnC,CAAC;YACD,IAAI,kBAAkB,IAAI,kBAAkB,GAAG,CAAC,IAAI,kBAAkB,GAAG,GAAG,EAAE,CAAC;gBAC3E,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,kBAAkB,CAAC,QAAQ,EAAE,CAAA;YAChE,CAAC;iBAAM,CAAC;gBACJ,OAAO,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAA;YACvC,CAAC;YACD,IAAI,CAAC,GAAG,CAAC,IAAI,CACT,sDAAsD,qBAAqB,IAAI;gBAC3E,8BAA8B,kBAAkB,EAAE,CACzD,CAAA;YAED,4EAA4E;YAC5E,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,IAAI,cAAc,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC;oBAC7C,KAAK,IAAI,CAAC,UAAU,EAAE,CAAA;gBAC1B,CAAC;gBACD,IAAI,CAAC,kBAAkB,GAAG,cAAc,CAAA;gBACxC,OAAM;YACV,CAAC;YAED,wCAAwC;YACxC,MAAM,WAAW,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAA;YAC/C,MAAM,MAAM,GAAG,SAAS,IAAI,CAAC,MAAM,IAAI,CAAC,WAAW,WAAW,IAAI,CAAC,CAAC,CAAA;YACpE,IAAI,MAAM,EAAE,CAAC;gBACT,IAAI,CAAC,OAAO,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAA;gBACvF,IAAI,cAAc,EAAE,CAAC;oBACjB,KAAK,IAAI,CAAC,UAAU,EAAE,CAAA;gBAC1B,CAAC;gBACD,6BAA6B,CAAC,QAAQ,GAAG,IAAI,CAAA;gBAC7C,IAAI,CAAC,kBAAkB,GAAG,cAAc,CAAA;YAC5C,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,8CAA8C,WAAW,EAAE,CAAC,CAAA;YAC9E,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,sCAAsC,GAAG,KAAK,CAAC,CAAA;QAClE,CAAC;IACL,CAAC;IAEO,oBAAoB;QACxB,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,EAAE,cAAc,CAAC,CAAA;QAElE,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;YAC/B,mDAAmD;YACnD,yEAAyE;YACzE,4DAA4D;YAC5D,OAAO,IAAA,mBAAa,EAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,CAAA;QAChD,CAAC;QAED,OAAO,WAAW,CAAA;IACtB,CAAC;IAEM,KAAK,CAAC,OAAO;QAChB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAA;YAC7B,IAAI,CAAC,OAAO,GAAG,SAAS,CAAA;QAC5B,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,WAAW,CAAC,SAAmB,EAAE,SAAqB;QAC/D,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAChB,OAAM;QACV,CAAC;QAED,IAAI,CAAC;YACD,MAAM,IAAI,CAAC,OAAO,EAAE,aAAa,CAAC,SAAS,EAAE,SAAS,CAAC,CAAA;QAC3D,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,yBAAyB,KAAK,EAAE,CAAC,CAAA;QACpD,CAAC;IACL,CAAC;IAED,kBAAkB;IAClB,KAAK,CAAC,UAAU;QACZ,IAAI,CAAC;YACD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC;oBAChC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,uDAAuD,CAAC,CAAA;oBACtE,OAAM;gBACV,CAAC;gBACD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAClD,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,kBAAkB,EACvB,IAAI,CAAC,qBAAqB,EAC1B,IAAI,CAAC,eAAe,EACpB,IAAI,CAAC,cAAc,EACnB,IAAI,CAAC,aAAa,EAClB,IAAI,CAAC,cAAc,CACtB,CAAA;gBAED,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,CAAA;gBACvD,MAAM,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,WAAW,EAAE,WAAW,EAAE,KAAK,CAAC,CAAA;gBAC/D,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,kCAAkC,CAAC,CAAA;YACrD,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,yBAAyB,KAAK,EAAE,CAAC,CAAA;QACpD,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,sBAAsB,CAAC,KAAwB,EAAE,OAA0B;QACpF,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAC9C,QAAQ,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CACzF,CAAA;YAED,MAAM,MAAM,GAAG,CAAC,GAAG,aAAa,CAAC,CAAA;YACjC,KAAK,MAAM,QAAQ,IAAI,KAAK,EAAE,CAAC;gBAC3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,EAAE,CAAC;oBAC9E,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;gBACzB,CAAC;YACL,CAAC;YACD,IAAI,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC1C,KAAK,IAAI,CAAC,UAAU,EAAE,CAAA;YAC1B,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,oCAAoC,KAAK,EAAE,CAAC,CAAA;QAC/D,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAClC,OAA2C;QAE3C,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAChB,OAAO,EAAE,CAAA;QACb,CAAC;QAED,IAAI,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,yBAAyB,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;YAC3G,OAAO,IAAI,IAAI,EAAE,CAAA;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,uCAAuC,KAAK,EAAE,CAAC,CAAA;YAC9D,OAAO,EAAE,CAAA;QACb,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,OAAqB;QAC/C,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAChB,OAAO,EAAE,CAAA;QACb,CAAC;QAED,IAAI,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;YAChE,OAAO,IAAI,IAAI,EAAE,CAAA;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,8BAA8B,KAAK,EAAE,CAAC,CAAA;YACrD,OAAO,EAAE,CAAA;QACb,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,sBAAsB;QAC/B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAChB,OAAO,EAAE,CAAA;QACb,CAAC;QAED,IAAI,CAAC;YACD,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,gBAAG,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAA;YACrF,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,sBAAsB,CAAC,WAAW,CAAC,CAAA;YACpE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,MAAM,wBAAwB,CAAC,CAAA;YAC7D,OAAO,IAAI,IAAI,EAAE,CAAA;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,oCAAoC,KAAK,EAAE,CAAC,CAAA;YAC3D,OAAO,EAAE,CAAA;QACb,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,0BAA0B,CAAC,SAAmB,EAAE,KAAc;QACvE,IAAI,CAAC;YACD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,sBAAsB,EAAE,CAAA;YAChE,MAAM,IAAI,CAAC,WAAW,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAA;YAC3D,MAAM,IAAA,wBAAS,EACX,KAAK,IAAI,EAAE;gBACP,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,sBAAsB,EAAE,CAAA;gBACnE,IAAI,cAAc,IAAI,WAAW,IAAI,cAAc,GAAG,WAAW,EAAE,CAAC;oBAChE,OAAO,IAAI,CAAA;gBACf,CAAC;gBACD,OAAO,KAAK,CAAA;YAChB,CAAC,EACD,EAAE,QAAQ,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,CAClD,CAAA;YACD,OAAO,IAAI,CAAA;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,yCAAyC,KAAK,EAAE,CAAC,CAAA;YAChE,OAAO,KAAK,CAAA;QAChB,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,qCAAqC;QAC9C,IAAI,IAAI,CAAC,6BAA6B,EAAE,CAAC;YACrC,OAAO,KAAK,CAAA;QAChB,CAAC;QACD,IAAI,CAAC,6BAA6B,GAAG,IAAI,CAAA;QACzC,IAAI,CAAC;YACD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,sBAAsB,EAAE,CAAA;YAChE,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,+BAA+B,CAAC,CAAA;YAC3D,MAAM,IAAA,wBAAS,EACX,KAAK,IAAI,EAAE;gBACP,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,sBAAsB,EAAE,CAAA;gBACnE,IAAI,cAAc,IAAI,WAAW,IAAI,cAAc,GAAG,WAAW,EAAE,CAAC;oBAChE,OAAO,IAAI,CAAA;gBACf,CAAC;gBACD,OAAO,KAAK,CAAA;YAChB,CAAC,EACD,EAAE,QAAQ,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,CACpD,CAAA;YACD,OAAO,IAAI,CAAA;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,mDAAmD,KAAK,EAAE,CAAC,CAAA;YAC1E,OAAO,KAAK,CAAA;QAChB,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,uBAAuB,CAChC,mBAAyC;QAEzC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAChB,OAAO,EAAE,CAAA;QACb,CAAC;QAED,IAAI,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE,uBAAuB,CAAC,mBAAmB,CAAC,CAAA;YAC7E,OAAO,IAAI,IAAI,EAAE,CAAA;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,qCAAqC,KAAK,EAAE,CAAC,CAAA;YAC5D,OAAO,EAAE,CAAA;QACb,CAAC;IACL,CAAC;IAEO,4BAA4B,CAAC,QAAgB,EAAE,eAAgC;QACnF,IAAI,QAAQ,CAAA;QAEZ,IAAI,CAAC;YACD,QAAQ,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAA;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,+BAA+B,QAAQ,KAAK,KAAK,EAAE,CAAC,CAAA;YACnE,OAAO,KAAK,CAAA;QAChB,CAAC;QAED,IAAI,QAAQ,GAAG,eAAe,CAAC,WAAW,IAAI,QAAQ,GAAG,eAAe,CAAC,kBAAkB,EAAE,CAAC;YAC1F,OAAO,KAAK,CAAA;QAChB,CAAC;QAED,eAAe,CAAC,kBAAkB,IAAI,QAAQ,CAAA;QAC9C,OAAO,IAAI,CAAA;IACf,CAAC;IAEO,KAAK,CAAC,uBAAuB,CACjC,gBAA2C,EAC3C,kBAA6B,EAC7B,qBAA+B,EAC/B,eAAyB,EACzB,cAAyB,EACzB,aAAsB,EACtB,cAAuB;QAEvB,IAAI,CAAC,gBAAgB,EAAE,MAAM,EAAE,CAAC;YAC5B,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,mDAAmD,CAAC,CAAA;YAClE,OAAO,EAAE,CAAA;QACb,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,gBAAgB,CAAC,MAAM,uBAAuB,CAAC,CAAA;QAEzE,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,GAAG,CAAC,kBAAkB,IAAI,EAAE,CAAC,CAAA;QAErD,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,aAAa,IAAI,QAAQ,EAAE,IAAI,CAAC,wBAAwB,CAAC,CAAA;QAClF,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,cAAc,IAAI,QAAQ,EAAE,IAAI,CAAC,yBAAyB,CAAC,CAAA;QAErF,MAAM,eAAe,GAAoB;YACrC,WAAW,EAAE,aAAa,GAAG,IAAI,CAAC,WAAW;YAC7C,kBAAkB,EAAE,cAAc,GAAG,IAAI,CAAC,WAAW;SACxD,CAAA;QAED,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAA;QAExC,MAAM,oBAAoB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC1C,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAE,MAAuB,EAAE,EAAE;YACnD,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,gBAAG,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAA;YAC/D,MAAM,mBAAmB,GAAa,EAAE,CAAA;YAExC,MAAM,OAAO,GAAG,IAAI,WAAI,EAAE;iBACrB,YAAY,CAAC,EAAE,YAAY,EAAE,CAAC,eAAe,EAAE,CAAC;iBAChD,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC;iBAClC,OAAO,CAAC,CAAC,OAAe,EAAE,OAAe,EAAE,EAAE;gBAC1C,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAA;gBACzD,OAAO,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAA;YACxE,CAAC,CAAC;iBACD,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,OAAO,GAAG,EAAE,CAAC,IAAI,EAAE,CAAC,EAAE,eAAe,CAAC;iBAC1E,MAAM,CAAC,CAAC,QAAgB,EAAE,WAAoB,EAAE,EAAE;gBAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAA;gBAE1D,IAAI,WAAW,IAAI,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;oBAC/E,OAAO,KAAK,CAAA;gBAChB,CAAC;gBAED,IAAI,CAAC,qBAAqB,IAAI,eAAe,CAAC,kBAAkB,IAAI,CAAC,EAAE,CAAC;oBACpE,UAAU,CAAC,KAAK,EAAE,CAAA;oBAClB,OAAO,KAAK,CAAA;gBAChB,CAAC;gBAED,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,YAAY,EAAE,CAAC;oBAC3C,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;oBAClC,OAAO,KAAK,CAAA;gBAChB,CAAC;gBAED,OAAO,qBAAqB,IAAI,IAAI,CAAC,4BAA4B,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAA;YAChG,CAAC,CAAC,CAAA;YAEN,OAAO,OAAO;iBACT,KAAK,CAAC,YAAY,CAAC;iBACnB,WAAW,EAAE;iBACb,IAAI,CAAC,KAAK,EAAE,WAAqB,EAAE,EAAE;gBAClC,IAAI,CAAC,qBAAqB,EAAE,CAAC;oBACzB,OAAO,WAAW,CAAA;gBACtB,CAAC;gBAED,MAAM,yBAAyB,GAAG,IAAI,GAAG,CACrC,MAAM,OAAO,CAAC,GAAG,CACb,mBAAmB,CAAC,GAAG,CAAC,KAAK,EAAC,QAAQ,EAAC,EAAE;oBACrC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAA;oBACvB,IAAI,CAAC;wBACD,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAA;oBACjE,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACb,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,iCAAiC,QAAQ,KAAK,KAAK,EAAE,CAAC,CAAA;oBACzE,CAAC;oBACD,OAAO,CAAC,QAAQ,EAAE,MAAM,CAAU,CAAA;gBACtC,CAAC,CAAC,CACL,CACJ,CAAA;gBAED,OAAO,WAAW,CAAC,MAAM,CAAC,CAAC,mBAAmB,EAAE,QAAQ,EAAE,EAAE;oBACxD,IAAI,eAAe,CAAC,kBAAkB,IAAI,CAAC,EAAE,CAAC;wBAC1C,OAAO,mBAAmB,CAAA;oBAC9B,CAAC;oBAED,MAAM,SAAS,GAAG,CAAC,GAAG,yBAAyB,CAAC,CAAC,IAAI,CACjD,CAAC,CAAC,aAAa,EAAE,MAAM,CAAgB,EAAE,EAAE;wBACvC,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,CAAA;wBAC9D,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,QAAQ,CAAC,CAAA;wBAE1D,OAAO,CAAC,YAAY,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAA;oBACzE,CAAC,CACJ,CAAA;oBAED,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,4BAA4B,CAAC,QAAQ,EAAE,eAAe,CAAC,EAAE,CAAC;wBAC7E,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;oBACtC,CAAC;oBAED,OAAO,mBAAmB,CAAA;gBAC9B,CAAC,EAAE,EAAc,CAAC,CAAA;YACtB,CAAC,CAAC,CAAA;QACV,CAAC,CAAC,CACL,CAAC,IAAI,CAAC,CAAC,eAA2B,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAA;QAE/D,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,4BAA4B,oBAAoB,CAAC,MAAM,SAAS,CAAC,CAAA;QAC/E,OAAO,oBAAoB,CAAA;IAC/B,CAAC;IAEO,wBAAwB,CAAC,CAAkB,EAAE,CAAkB;QACnE,OAAO,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,IAAI,CAAA;IAC/C,CAAC;CACJ;AAvbD,sEAubC"}

package/out/shared/models/constants.d.ts

@@ -5,10 +5,15 @@ export declare enum UserGroup {
     RightContext = "RightContext"
 }
 export declare const supplemetalContextFetchingTimeoutMsg = "Amazon Q supplemental context fetching timeout";
+export declare const supplementalContextMaxTotalLength = 20480;
+export declare const supplementalContextTimeoutInMs = 100;
 export declare const crossFileContextConfig: {
     numberOfChunkToFetch: number;
     topK: number;
     numberOfLinesEachChunk: number;
+    maximumTotalLength: number;
+    maxLengthEachChunk: number;
+    maxContextCount: number;
 };
 export declare const utgConfig: {
     maxSegmentSize: number;

package/out/shared/models/constants.js

@@ -2,7 +2,7 @@
 // Port of implementation in AWS Toolkit for VSCode
 // https://github.com/aws/aws-toolkit-vscode/blob/ac9a78db90192fe13dcd82ac0b30e6f84c39b9c8/packages/core/src/codewhisperer/models/constants.ts
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.utgConfig = exports.crossFileContextConfig = exports.supplemetalContextFetchingTimeoutMsg = exports.UserGroup = void 0;
+exports.utgConfig = exports.crossFileContextConfig = exports.supplementalContextTimeoutInMs = exports.supplementalContextMaxTotalLength = exports.supplemetalContextFetchingTimeoutMsg = exports.UserGroup = void 0;
 var UserGroup;
 (function (UserGroup) {
     UserGroup["Classifier"] = "Classifier";
@@ -11,10 +11,15 @@ var UserGroup;
     UserGroup["RightContext"] = "RightContext";
 })(UserGroup || (exports.UserGroup = UserGroup = {}));
 exports.supplemetalContextFetchingTimeoutMsg = 'Amazon Q supplemental context fetching timeout';
+exports.supplementalContextMaxTotalLength = 20480;
+exports.supplementalContextTimeoutInMs = 100;
 exports.crossFileContextConfig = {
     numberOfChunkToFetch: 60,
     topK: 3,
     numberOfLinesEachChunk: 10,
+    maximumTotalLength: 20480,
+    maxLengthEachChunk: 10240,
+    maxContextCount: 5,
 };
 exports.utgConfig = {
     maxSegmentSize: 10200,

package/out/shared/models/constants.js.map

@@ -1 +1 @@
-{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../src/shared/models/constants.ts"],"names":[],"mappings":";AAAA,mDAAmD;AACnD,8IAA8I;;;AAE9I,IAAY,SAKX;AALD,WAAY,SAAS;IACjB,sCAAyB,CAAA;IACzB,oCAAuB,CAAA;IACvB,gCAAmB,CAAA;IACnB,0CAA6B,CAAA;AACjC,CAAC,EALW,SAAS,yBAAT,SAAS,QAKpB;AAEY,QAAA,oCAAoC,GAAG,gDAAgD,CAAA;AAEvF,QAAA,sBAAsB,GAAG;IAClC,oBAAoB,EAAE,EAAE;IACxB,IAAI,EAAE,CAAC;IACP,sBAAsB,EAAE,EAAE;CAC7B,CAAA;AAEY,QAAA,SAAS,GAAG;IACrB,cAAc,EAAE,KAAK;CACxB,CAAA"}
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../src/shared/models/constants.ts"],"names":[],"mappings":";AAAA,mDAAmD;AACnD,8IAA8I;;;AAE9I,IAAY,SAKX;AALD,WAAY,SAAS;IACjB,sCAAyB,CAAA;IACzB,oCAAuB,CAAA;IACvB,gCAAmB,CAAA;IACnB,0CAA6B,CAAA;AACjC,CAAC,EALW,SAAS,yBAAT,SAAS,QAKpB;AAEY,QAAA,oCAAoC,GAAG,gDAAgD,CAAA;AAEvF,QAAA,iCAAiC,GAAG,KAAK,CAAA;AAEzC,QAAA,8BAA8B,GAAG,GAAG,CAAA;AAEpC,QAAA,sBAAsB,GAAG;IAClC,oBAAoB,EAAE,EAAE;IACxB,IAAI,EAAE,CAAC;IACP,sBAAsB,EAAE,EAAE;IAC1B,kBAAkB,EAAE,KAAK;IACzB,kBAAkB,EAAE,KAAK;IACzB,eAAe,EAAE,CAAC;CACrB,CAAA;AAEY,QAAA,SAAS,GAAG;IACrB,cAAc,EAAE,KAAK;CACxB,CAAA"}

package/out/shared/models/model.d.ts

@@ -1,6 +1,7 @@
 export type UtgStrategy = 'ByName' | 'ByContent';
 export type CrossFileStrategy = 'OpenTabs_BM25';
-export type SupplementalContextStrategy = CrossFileStrategy | UtgStrategy | 'Empty';
+export type ProjectContextStrategy = 'codemap';
+export type SupplementalContextStrategy = CrossFileStrategy | ProjectContextStrategy | UtgStrategy | 'Empty';
 export interface CodeWhispererSupplementalContext {
     isUtg: boolean;
     isProcessTimeout: boolean;

package/out/shared/proxy-server.d.ts

@@ -2,6 +2,8 @@ export declare const CodeWhispererServerTokenProxy: import("@aws/language-server
 export declare const CodeWhispererServerIAMProxy: import("@aws/language-server-runtimes/server-interface").Server;
 export declare const CodeWhispererSecurityScanServerTokenProxy: import("@aws/language-server-runtimes/server-interface").Server;
 export declare const QNetTransformServerTokenProxy: import("@aws/language-server-runtimes/server-interface").Server;
-export declare const QChatServerProxy: import("@aws/language-server-runtimes/server-interface").Server;
-export declare const QAgenticChatServerProxy: import("@aws/language-server-runtimes/server-interface").Server;
+export declare const QChatServerTokenProxy: import("@aws/language-server-runtimes/server-interface").Server;
+export declare const QChatServerIAMProxy: import("@aws/language-server-runtimes/server-interface").Server;
+export declare const QAgenticChatServerTokenProxy: import("@aws/language-server-runtimes/server-interface").Server;
 export declare const QConfigurationServerTokenProxy: import("@aws/language-server-runtimes/server-interface").Server;
+export declare const QLocalProjectContextServerTokenProxy: import("@aws/language-server-runtimes/server-interface").Server;

package/out/shared/proxy-server.js

@@ -1,21 +1,22 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.QConfigurationServerTokenProxy = exports.QAgenticChatServerProxy = exports.QChatServerProxy = exports.QNetTransformServerTokenProxy = exports.CodeWhispererSecurityScanServerTokenProxy = exports.CodeWhispererServerIAMProxy = exports.CodeWhispererServerTokenProxy = void 0;
+exports.QLocalProjectContextServerTokenProxy = exports.QConfigurationServerTokenProxy = exports.QAgenticChatServerTokenProxy = exports.QChatServerIAMProxy = exports.QChatServerTokenProxy = exports.QNetTransformServerTokenProxy = exports.CodeWhispererSecurityScanServerTokenProxy = exports.CodeWhispererServerIAMProxy = exports.CodeWhispererServerTokenProxy = void 0;
 const qAgenticChatServer_1 = require("../language-server/agenticChat/qAgenticChatServer");
 const codeWhispererSecurityScanServer_1 = require("../language-server/securityScan/codeWhispererSecurityScanServer");
 const codeWhispererServer_1 = require("../language-server/inline-completion/codeWhispererServer");
-const codeWhispererService_1 = require("./codeWhispererService");
 const netTransformServer_1 = require("../language-server/netTransform/netTransformServer");
 const qChatServer_1 = require("../language-server/chat/qChatServer");
 const qConfigurationServer_1 = require("../language-server/configuration/qConfigurationServer");
-const factories_1 = require("../shared/amazonQServiceManager/factories");
-exports.CodeWhispererServerTokenProxy = (0, codeWhispererServer_1.CodewhispererServerFactory)(factories_1.initBaseTokenServiceManager);
-exports.CodeWhispererServerIAMProxy = (0, codeWhispererServer_1.CodewhispererServerFactory)(factories_1.initBaseIAMServiceManager);
+const AmazonQTokenServiceManager_1 = require("./amazonQServiceManager/AmazonQTokenServiceManager");
+const AmazonQIAMServiceManager_1 = require("./amazonQServiceManager/AmazonQIAMServiceManager");
+const localProjectContextServer_1 = require("../language-server/localProjectContext/localProjectContextServer");
+exports.CodeWhispererServerTokenProxy = (0, codeWhispererServer_1.CodewhispererServerFactory)(AmazonQTokenServiceManager_1.getOrThrowBaseTokenServiceManager);
+exports.CodeWhispererServerIAMProxy = (0, codeWhispererServer_1.CodewhispererServerFactory)(AmazonQIAMServiceManager_1.getOrThrowBaseIAMServiceManager);
 exports.CodeWhispererSecurityScanServerTokenProxy = (0, codeWhispererSecurityScanServer_1.SecurityScanServerToken)();
-exports.QNetTransformServerTokenProxy = (0, netTransformServer_1.QNetTransformServerToken)((credentialsProvider, workspace, awsQRegion, awsQEndpointUrl, sdkInitializator) => {
-    return new codeWhispererService_1.CodeWhispererServiceToken(credentialsProvider, workspace, awsQRegion, awsQEndpointUrl, sdkInitializator);
-});
-exports.QChatServerProxy = (0, qChatServer_1.QChatServer)();
-exports.QAgenticChatServerProxy = (0, qAgenticChatServer_1.QAgenticChatServer)();
+exports.QNetTransformServerTokenProxy = (0, netTransformServer_1.QNetTransformServerToken)();
+exports.QChatServerTokenProxy = (0, qChatServer_1.QChatServerFactory)(AmazonQTokenServiceManager_1.getOrThrowBaseTokenServiceManager);
+exports.QChatServerIAMProxy = (0, qChatServer_1.QChatServerFactory)(AmazonQIAMServiceManager_1.getOrThrowBaseIAMServiceManager);
+exports.QAgenticChatServerTokenProxy = (0, qAgenticChatServer_1.QAgenticChatServer)();
 exports.QConfigurationServerTokenProxy = (0, qConfigurationServer_1.QConfigurationServerToken)();
+exports.QLocalProjectContextServerTokenProxy = (0, localProjectContextServer_1.LocalProjectContextServer)();
 //# sourceMappingURL=proxy-server.js.map

package/out/shared/proxy-server.js.map

@@ -1 +1 @@
-{"version":3,"file":"proxy-server.js","sourceRoot":"","sources":["../../src/shared/proxy-server.ts"],"names":[],"mappings":";;;AAAA,0FAAsF;AACtF,qHAAyG;AACzG,kGAAqG;AACrG,iEAAkE;AAClE,2FAA6F;AAC7F,qEAAiE;AACjE,gGAAiG;AACjG,yEAAkH;AAErG,QAAA,6BAA6B,GAAG,IAAA,gDAA0B,EAAC,uCAA2B,CAAC,CAAA;AAEvF,QAAA,2BAA2B,GAAG,IAAA,gDAA0B,EAAC,qCAAyB,CAAC,CAAA;AAEnF,QAAA,yCAAyC,GAAG,IAAA,yDAAuB,GAAE,CAAA;AAErE,QAAA,6BAA6B,GAAG,IAAA,6CAAwB,EACjE,CAAC,mBAAmB,EAAE,SAAS,EAAE,UAAU,EAAE,eAAe,EAAE,gBAAgB,EAAE,EAAE;IAC9E,OAAO,IAAI,gDAAyB,CAChC,mBAAmB,EACnB,SAAS,EACT,UAAU,EACV,eAAe,EACf,gBAAgB,CACnB,CAAA;AACL,CAAC,CACJ,CAAA;AAEY,QAAA,gBAAgB,GAAG,IAAA,yBAAW,GAAE,CAAA;AAEhC,QAAA,uBAAuB,GAAG,IAAA,uCAAkB,GAAE,CAAA;AAE9C,QAAA,8BAA8B,GAAG,IAAA,gDAAyB,GAAE,CAAA"}
+{"version":3,"file":"proxy-server.js","sourceRoot":"","sources":["../../src/shared/proxy-server.ts"],"names":[],"mappings":";;;AAAA,0FAAsF;AACtF,qHAAyG;AACzG,kGAAqG;AACrG,2FAA6F;AAC7F,qEAAwE;AACxE,gGAAiG;AACjG,mGAAsG;AACtG,+FAAkG;AAClG,gHAA4G;AAE/F,QAAA,6BAA6B,GAAG,IAAA,gDAA0B,EAAC,8DAAiC,CAAC,CAAA;AAE7F,QAAA,2BAA2B,GAAG,IAAA,gDAA0B,EAAC,0DAA+B,CAAC,CAAA;AAEzF,QAAA,yCAAyC,GAAG,IAAA,yDAAuB,GAAE,CAAA;AAErE,QAAA,6BAA6B,GAAG,IAAA,6CAAwB,GAAE,CAAA;AAE1D,QAAA,qBAAqB,GAAG,IAAA,gCAAkB,EAAC,8DAAiC,CAAC,CAAA;AAC7E,QAAA,mBAAmB,GAAG,IAAA,gCAAkB,EAAC,0DAA+B,CAAC,CAAA;AAEzE,QAAA,4BAA4B,GAAG,IAAA,uCAAkB,GAAE,CAAA;AAEnD,QAAA,8BAA8B,GAAG,IAAA,gDAAyB,GAAE,CAAA;AAE5D,QAAA,oCAAoC,GAAG,IAAA,qDAAyB,GAAE,CAAA"}

package/out/shared/streamingClientService.d.ts

@@ -1,10 +1,26 @@
-import { CodeWhispererStreaming, SendMessageCommandInput, SendMessageCommandOutput } from '@amzn/codewhisperer-streaming';
-import { CredentialsProvider, SDKInitializator } from '@aws/language-server-runtimes/server-interface';
-export declare class StreamingClientService {
+import { CodeWhispererStreaming, GenerateAssistantResponseCommandInput as GenerateAssistantResponseCommandInputCodeWhispererStreaming, GenerateAssistantResponseCommandOutput as GenerateAssistantResponseCommandOutputCodeWhispererStreaming, SendMessageCommandInput as SendMessageCommandInputCodeWhispererStreaming, SendMessageCommandOutput as SendMessageCommandOutputCodeWhispererStreaming } from '@amzn/codewhisperer-streaming';
+import { QDeveloperStreaming, SendMessageCommandInput as SendMessageCommandInputQDeveloperStreaming, SendMessageCommandOutput as SendMessageCommandOutputQDeveloperStreaming } from '@amzn/amazon-q-developer-streaming-client';
+import { CredentialsProvider, SDKInitializator, Logging } from '@aws/language-server-runtimes/server-interface';
+export type SendMessageCommandInput = SendMessageCommandInputCodeWhispererStreaming | SendMessageCommandInputQDeveloperStreaming;
+export type SendMessageCommandOutput = SendMessageCommandOutputCodeWhispererStreaming | SendMessageCommandOutputQDeveloperStreaming;
+export declare abstract class StreamingClientServiceBase {
+    protected readonly region: string;
+    protected readonly endpoint: string;
+    inflightRequests: Set<AbortController>;
+    abstract client: CodeWhispererStreaming | QDeveloperStreaming;
+    constructor(region: string, endpoint: string);
+    abstract sendMessage(request: SendMessageCommandInput, abortController?: AbortController): Promise<SendMessageCommandOutput>;
+    abortInflightRequests(): void;
+}
+export declare class StreamingClientServiceToken extends StreamingClientServiceBase {
     client: CodeWhispererStreaming;
-    private inflightRequests;
     profileArn?: string;
-    constructor(credentialsProvider: CredentialsProvider, sdkInitializator: SDKInitializator, region: string, endpoint: string, customUserAgent: string);
-    sendMessage(request: SendMessageCommandInput, abortController?: AbortController): Promise<SendMessageCommandOutput>;
-    abortInflightRequests(): void;
+    constructor(credentialsProvider: CredentialsProvider, sdkInitializator: SDKInitializator, logging: Logging, region: string, endpoint: string, customUserAgent: string);
+    sendMessage(request: SendMessageCommandInputCodeWhispererStreaming, abortController?: AbortController): Promise<SendMessageCommandOutputCodeWhispererStreaming>;
+    generateAssistantResponse(request: GenerateAssistantResponseCommandInputCodeWhispererStreaming, abortController?: AbortController): Promise<GenerateAssistantResponseCommandOutputCodeWhispererStreaming>;
+}
+export declare class StreamingClientServiceIAM extends StreamingClientServiceBase {
+    client: QDeveloperStreaming;
+    constructor(credentialsProvider: CredentialsProvider, sdkInitializator: SDKInitializator, logging: Logging, region: string, endpoint: string);
+    sendMessage(request: SendMessageCommandInputQDeveloperStreaming, abortController?: AbortController): Promise<SendMessageCommandOutputQDeveloperStreaming>;
 }