@vscode/chat-lib 0.1.2 → 0.1.3-0

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/repository.js

@@ -0,0 +1,239 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ComputationStatus = void 0;
+exports.tryGetGitHubNWO = tryGetGitHubNWO;
+exports.extractRepoInfoInBackground = extractRepoInfoInBackground;
+exports.extractRepoInfo = extractRepoInfo;
+const gitService_1 = require("../../../../../../platform/git/common/gitService");
+const fileSystem_1 = require("../fileSystem");
+const cache_1 = require("../helpers/cache");
+const uri_1 = require("../util/uri");
+function tryGetGitHubNWO(repoInfo) {
+    if (repoInfo === undefined) {
+        return undefined;
+    }
+    if (repoInfo === ComputationStatus.PENDING) {
+        return undefined;
+    }
+    if (repoInfo.repoId?.type === 'github') {
+        return (repoInfo.repoId.org + '/' + repoInfo.repoId.repo).toLowerCase();
+    }
+    return undefined;
+}
+/**
+ * Sends off a computation to extract information about which git repo the file belongs to in the background.
+ * @param fileUri URI of a file under the repo
+ * @returns
+ *  - If the computation is still running (in particular for the first time), returns ComputationStatus.PENDING.
+ *  - If a file from this path has been looked at before, and no repository has been identified, returns undefined.
+ *  - If a file from this path has been looked at before, and a repository has been identified, returns the repo info.
+ */
+function extractRepoInfoInBackground(accessor, uri) {
+    const baseFolder = (0, uri_1.dirname)(uri);
+    return backgroundRepoInfo(accessor, baseFolder);
+}
+// Note that we assume that the same filesystem path always returns the same repository information.
+// If this changes on disk, or if two different contexts with different FileSystem implementations
+// are passed for the same path, such as for a test, then the cached value may be incorrect
+const backgroundRepoInfo = computeInBackgroundAndMemoize(extractRepoInfo, 10000);
+/**
+ * If the file is part of a git repository, return the information about the repository.
+ * @param uri URI of a folder or file in the repository
+ * @param fs The file system to be used
+ * @returns A RepoInfo object, or undefined if the file is not part of a git repository.
+ * If it does appear to be part of a git repository, but its information is not parsable,
+ * it returns a RepoInfo object with hostname, user and repo set to "".
+ */
+async function extractRepoInfo(accessor, uri) {
+    const fs = accessor.get(fileSystem_1.ICompletionsFileSystemService);
+    const fsUri = (0, uri_1.getFsUri)(uri);
+    if (!fsUri) {
+        return undefined;
+    }
+    const baseUri = await getRepoBaseUri(fs, fsUri);
+    if (!baseUri) {
+        return undefined;
+    }
+    const configUri = (0, uri_1.joinPath)(baseUri, '.git', 'config');
+    let gitConfig;
+    try {
+        gitConfig = await fs.readFileString(configUri);
+    }
+    catch (_) {
+        // typically an ENOENT or EPERM, wrapped in varying ways depending on which FileSystem implementation is used
+        return undefined;
+    }
+    const url = getRepoUrlFromConfigText(gitConfig) ?? '';
+    const parsedResult = parseRepoUrl(url);
+    const baseFolder = { uri: baseUri };
+    if (parsedResult === undefined) {
+        return { baseFolder, url, hostname: '', pathname: '', repoId: undefined };
+    }
+    else {
+        return { baseFolder, url, hostname: parsedResult.host, pathname: parsedResult.path, repoId: parsedResult.repoId };
+    }
+}
+function parseRepoUrl(url) {
+    const res = (0, gitService_1.parseRemoteUrl)(url);
+    if (!res) {
+        return undefined;
+    }
+    const repoId = (0, gitService_1.getGithubRepoIdFromFetchUrl)(url) ?? (0, gitService_1.getAdoRepoIdFromFetchUrl)(url);
+    return { ...res, repoId };
+}
+/**
+ * Returns the base folder of the git repository containing the file, or undefined if none is found.
+ * Will search recursively for a .git folder containing a config file.
+ */
+async function getRepoBaseUri(fileSystemService, uri) {
+    // to make sure the while loop terminates, we make sure the path variable decreases in length
+    let previousUri = uri + '_add_to_make_longer';
+    while (uri !== 'file:///' && uri.length < previousUri.length) {
+        const configUri = (0, uri_1.joinPath)(uri, '.git', 'config');
+        let result = false;
+        try {
+            await fileSystemService.stat(configUri);
+            result = true;
+        }
+        catch (reason) {
+            result = false;
+        }
+        if (result) {
+            return uri;
+        }
+        else {
+            previousUri = uri;
+            uri = (0, uri_1.dirname)(uri);
+        }
+    }
+    return undefined;
+}
+/**
+ * Parses a git config file, returning
+ * 1. remote.origin.url if it exists,
+ * 2. any remote.[name].url if it exists but not 1.,
+ * 3. undefined if neither exist.
+ * Will throw if the file does not exist.
+ *
+ * The config format is expected to follow https://git-scm.com/docs/git-config#_configuration_file
+ * e.g. it could include lines like
+    [remote "origin"]
+        url = git@github.com:microsoft/vscode-copilot-chat.git
+        fetch = +refs/heads/*:refs/remotes/origin/*
+ *
+ * Known limitations:
+ * - This will not respect include and includeIf directions
+ *
+ * @param gitConfig the contents of the git config file
+ * @returns the url, or undefined if none found
+ */
+function getRepoUrlFromConfigText(gitConfig) {
+    // We're looking for [remote "origin"] and [remote "name"] sections
+    // section headers must be one line,
+    // except for whitespace, they're [section "subsection"]
+    // where subsection can contain " by escaping \" and
+    // can escape \ by \\ (so that e.g. it can be the last character before the ")
+    const remoteSectionRegex = /^\s*\[\s*remote\s+"((\\\\|\\"|[^\\"])+)"/;
+    // deprecated syntax: [section.subsection]
+    const deprecatedRemoteSectionRegex = /^\s*\[remote.([^"\s]+)/;
+    // extract the name of the remote -- assume it doesn't contain whitespace, and remember # and ; start comments
+    const setUrlRegex = /^\s*url\s*=\s*([^\s#;]+)/;
+    // use the following to check whether the current section ended
+    const newSectionRegex = /^\s*\[/;
+    let remoteUrl = undefined;
+    let remoteSection = undefined;
+    let isWithinMultilineUrl = false;
+    for (const line of gitConfig.split('\n')) {
+        if (isWithinMultilineUrl && remoteUrl !== undefined) {
+            remoteUrl += line;
+            if (line.endsWith('\\')) {
+                remoteUrl = remoteUrl.substring(0, remoteUrl.length - 1);
+            }
+            else {
+                isWithinMultilineUrl = false;
+                if (remoteSection === 'origin') {
+                    // we're already finished
+                    return remoteUrl;
+                }
+            }
+        }
+        else {
+            // check whether a new section starts
+            const remoteSectionMatch = line.match(remoteSectionRegex) ?? line.match(deprecatedRemoteSectionRegex);
+            if (remoteSectionMatch) {
+                remoteSection = remoteSectionMatch[1];
+            }
+            else if (line.match(newSectionRegex)) {
+                remoteSection = undefined;
+            }
+            else if (remoteUrl && remoteSection !== 'origin') {
+                // if we already have any remote url, only "origin" is more interesting
+                continue;
+            }
+            else {
+                const urlMatch = line.match(setUrlRegex);
+                if (urlMatch) {
+                    remoteUrl = urlMatch[1];
+                    if (remoteUrl.endsWith('\\')) {
+                        remoteUrl = remoteUrl.substring(0, remoteUrl.length - 1);
+                        isWithinMultilineUrl = true;
+                    }
+                    else if (remoteSection === 'origin') {
+                        // we're already finished
+                        return remoteUrl;
+                    }
+                }
+            }
+        }
+    }
+    return remoteUrl;
+}
+/**
+ * Helper functionality for doing the computation in the background
+ */
+var ComputationStatus;
+(function (ComputationStatus) {
+    ComputationStatus[ComputationStatus["PENDING"] = 0] = "PENDING";
+})(ComputationStatus || (exports.ComputationStatus = ComputationStatus = {}));
+class CompletedComputation {
+    constructor(result) {
+        this.result = result;
+    }
+}
+/**
+ * Function wrapper that memoizes a given function to be computed in the background.
+ * Until the first computation is complete, the wrapper returns ComputationStatus.PENDING.
+ * The context is not taken into account for computing the cache key so the function may
+ * behave incorrectly if called with different contexts.
+ * @param fct A function returning a promise whose arguments are amenable to JSON.stringify.
+ * @param cacheSize Number of elements to cache.
+ * @returns The memoized function, which returns ComputationStatus.PENDING until the first computation is complete.
+ */
+function computeInBackgroundAndMemoize(fct, cacheSize) {
+    const resultsCache = new cache_1.LRUCacheMap(cacheSize);
+    const inComputation = new Set();
+    return (accessor, ...args) => {
+        const key = JSON.stringify(args);
+        const memorizedComputation = resultsCache.get(key);
+        if (memorizedComputation) {
+            return memorizedComputation.result;
+        }
+        if (inComputation.has(key)) {
+            // already being computed from a different call
+            return ComputationStatus.PENDING;
+        }
+        const computation = fct(accessor, ...args);
+        inComputation.add(key);
+        void computation.then(computedResult => {
+            // remove from inComputation
+            resultsCache.set(key, new CompletedComputation(computedResult));
+            inComputation.delete(key);
+        });
+        return ComputationStatus.PENDING;
+    };
+}
+//# sourceMappingURL=repository.js.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/repository.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"repository.js","sourceRoot":"","sources":["../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/repository.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAiChG,0CAWC;AAUD,kEAGC;AAkBD,0CA0BC;AAnGD,iFAAkK;AAElK,8CAA8E;AAC9E,4CAA+C;AAC/C,qCAA0D;AA2B1D,SAAgB,eAAe,CAAC,QAAuB;IACtD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC5B,OAAO,SAAS,CAAC;IAClB,CAAC;IACD,IAAI,QAAQ,KAAK,iBAAiB,CAAC,OAAO,EAAE,CAAC;QAC5C,OAAO,SAAS,CAAC;IAClB,CAAC;IACD,IAAI,QAAQ,CAAC,MAAM,EAAE,IAAI,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,GAAG,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,CAAC;IACzE,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAED;;;;;;;GAOG;AACH,SAAgB,2BAA2B,CAAC,QAA0B,EAAE,GAAmB;IAC1F,MAAM,UAAU,GAAG,IAAA,aAAO,EAAC,GAAG,CAAC,CAAC;IAChC,OAAO,kBAAkB,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;AACjD,CAAC;AAED,oGAAoG;AACpG,kGAAkG;AAClG,2FAA2F;AAC3F,MAAM,kBAAkB,GAAG,6BAA6B,CACvD,eAAe,EACf,KAAK,CACL,CAAC;AAEF;;;;;;;GAOG;AACI,KAAK,UAAU,eAAe,CAAC,QAA0B,EAAE,GAAmB;IACpF,MAAM,EAAE,GAAG,QAAQ,CAAC,GAAG,CAAC,0CAA6B,CAAC,CAAC;IAEvD,MAAM,KAAK,GAAG,IAAA,cAAQ,EAAC,GAAG,CAAC,CAAC;IAC5B,IAAI,CAAC,KAAK,EAAE,CAAC;QAAC,OAAO,SAAS,CAAC;IAAC,CAAC;IAEjC,MAAM,OAAO,GAAG,MAAM,cAAc,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IAChD,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,OAAO,SAAS,CAAC;IAClB,CAAC;IACD,MAAM,SAAS,GAAG,IAAA,cAAQ,EAAC,OAAO,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;IACtD,IAAI,SAAS,CAAC;IACd,IAAI,CAAC;QACJ,SAAS,GAAG,MAAM,EAAE,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;IAChD,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACZ,6GAA6G;QAC7G,OAAO,SAAS,CAAC;IAClB,CAAC;IACD,MAAM,GAAG,GAAG,wBAAwB,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;IACtD,MAAM,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;IACvC,MAAM,UAAU,GAAG,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC;IACpC,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;QAChC,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;IAC3E,CAAC;SAAM,CAAC;QACP,OAAO,EAAE,UAAU,EAAE,GAAG,EAAE,QAAQ,EAAE,YAAY,CAAC,IAAI,EAAE,QAAQ,EAAE,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,CAAC,MAAM,EAAE,CAAC;IACnH,CAAC;AACF,CAAC;AAED,SAAS,YAAY,CACpB,GAAW;IAEX,MAAM,GAAG,GAAG,IAAA,2BAAc,EAAC,GAAG,CAAC,CAAC;IAChC,IAAI,CAAC,GAAG,EAAE,CAAC;QACV,OAAO,SAAS,CAAC;IAClB,CAAC;IACD,MAAM,MAAM,GAAG,IAAA,wCAA2B,EAAC,GAAG,CAAC,IAAI,IAAA,qCAAwB,EAAC,GAAG,CAAC,CAAC;IACjF,OAAO,EAAE,GAAG,GAAG,EAAE,MAAM,EAAE,CAAC;AAC3B,CAAC;AAED;;;GAGG;AACH,KAAK,UAAU,cAAc,CAAC,iBAAgD,EAAE,GAAW;IAC1F,6FAA6F;IAC7F,IAAI,WAAW,GAAG,GAAG,GAAG,qBAAqB,CAAC;IAC9C,OAAO,GAAG,KAAK,UAAU,IAAI,GAAG,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;QAC9D,MAAM,SAAS,GAAG,IAAA,cAAQ,EAAC,GAAG,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAClD,IAAI,MAAM,GAAG,KAAK,CAAC;QAEnB,IAAI,CAAC;YACJ,MAAM,iBAAiB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACxC,MAAM,GAAG,IAAI,CAAC;QACf,CAAC;QAAC,OAAO,MAAM,EAAE,CAAC;YACjB,MAAM,GAAG,KAAK,CAAC;QAChB,CAAC;QAED,IAAI,MAAM,EAAE,CAAC;YACZ,OAAO,GAAG,CAAC;QACZ,CAAC;aAAM,CAAC;YACP,WAAW,GAAG,GAAG,CAAC;YAClB,GAAG,GAAG,IAAA,aAAO,EAAC,GAAG,CAAC,CAAC;QACpB,CAAC;IACF,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAED;;;;;;;;;;;;;;;;;;GAkBG;AACH,SAAS,wBAAwB,CAAC,SAAiB;IAClD,mEAAmE;IAEnE,oCAAoC;IACpC,wDAAwD;IACxD,oDAAoD;IACpD,8EAA8E;IAC9E,MAAM,kBAAkB,GAAG,0CAA0C,CAAC;IACtE,0CAA0C;IAC1C,MAAM,4BAA4B,GAAG,wBAAwB,CAAC;IAC9D,8GAA8G;IAC9G,MAAM,WAAW,GAAG,0BAA0B,CAAC;IAC/C,+DAA+D;IAC/D,MAAM,eAAe,GAAG,QAAQ,CAAC;IAEjC,IAAI,SAAS,GAAuB,SAAS,CAAC;IAC9C,IAAI,aAAa,GAAG,SAAS,CAAC;IAC9B,IAAI,oBAAoB,GAAG,KAAK,CAAC;IACjC,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QAC1C,IAAI,oBAAoB,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YACrD,SAAS,IAAI,IAAI,CAAC;YAClB,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBACzB,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAC1D,CAAC;iBAAM,CAAC;gBACP,oBAAoB,GAAG,KAAK,CAAC;gBAC7B,IAAI,aAAa,KAAK,QAAQ,EAAE,CAAC;oBAChC,yBAAyB;oBACzB,OAAO,SAAS,CAAC;gBAClB,CAAC;YACF,CAAC;QACF,CAAC;aAAM,CAAC;YACP,qCAAqC;YACrC,MAAM,kBAAkB,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,4BAA4B,CAAC,CAAC;YACtG,IAAI,kBAAkB,EAAE,CAAC;gBACxB,aAAa,GAAG,kBAAkB,CAAC,CAAC,CAAC,CAAC;YACvC,CAAC;iBAAM,IAAI,IAAI,CAAC,KAAK,CAAC,eAAe,CAAC,EAAE,CAAC;gBACxC,aAAa,GAAG,SAAS,CAAC;YAC3B,CAAC;iBAAM,IAAI,SAAS,IAAI,aAAa,KAAK,QAAQ,EAAE,CAAC;gBACpD,uEAAuE;gBACvE,SAAS;YACV,CAAC;iBAAM,CAAC;gBACP,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;gBACzC,IAAI,QAAQ,EAAE,CAAC;oBACd,SAAS,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;oBACxB,IAAI,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;wBAC9B,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;wBACzD,oBAAoB,GAAG,IAAI,CAAC;oBAC7B,CAAC;yBAAM,IAAI,aAAa,KAAK,QAAQ,EAAE,CAAC;wBACvC,yBAAyB;wBACzB,OAAO,SAAS,CAAC;oBAClB,CAAC;gBACF,CAAC;YACF,CAAC;QACF,CAAC;IACF,CAAC;IACD,OAAO,SAAS,CAAC;AAClB,CAAC;AAED;;GAEG;AAEH,IAAY,iBAEX;AAFD,WAAY,iBAAiB;IAC5B,+DAAO,CAAA;AACR,CAAC,EAFW,iBAAiB,iCAAjB,iBAAiB,QAE5B;AAED,MAAM,oBAAoB;IAEzB,YAAY,MAAS;QACpB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;CACD;AAED;;;;;;;;GAQG;AACH,SAAS,6BAA6B,CACrC,GAA2D,EAC3D,SAAiB;IAEjB,MAAM,YAAY,GAAG,IAAI,mBAAW,CAAkC,SAAS,CAAC,CAAC;IACjF,MAAM,aAAa,GAAgB,IAAI,GAAG,EAAE,CAAC;IAC7C,OAAO,CAAC,QAA0B,EAAE,GAAG,IAAO,EAAE,EAAE;QACjD,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,oBAAoB,GAAG,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACnD,IAAI,oBAAoB,EAAE,CAAC;YAC1B,OAAO,oBAAoB,CAAC,MAAM,CAAC;QACpC,CAAC;QACD,IAAI,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC5B,+CAA+C;YAC/C,OAAO,iBAAiB,CAAC,OAAO,CAAC;QAClC,CAAC;QACD,MAAM,WAAW,GAAG,GAAG,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,CAAC;QAC3C,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACvB,KAAK,WAAW,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE;YACtC,4BAA4B;YAC5B,YAAY,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,oBAAoB,CAAC,cAAc,CAAC,CAAC,CAAC;YAChE,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;QACH,OAAO,iBAAiB,CAAC,OAAO,CAAC;IAClC,CAAC,CAAC;AACH,CAAC"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/similarFiles/compositeRelatedFilesProvider.d.ts

@@ -0,0 +1,37 @@
+import { IIgnoreService } from '../../../../../../../platform/ignore/common/ignoreService';
+import { IInstantiationService } from '../../../../../../../util/vs/platform/instantiation/common/instantiation';
+import { CancellationToken as ICancellationToken } from '../../../../types/src';
+import { ICompletionsFeaturesService } from '../../experiments/featuresService';
+import { ICompletionsFileSystemService } from '../../fileSystem';
+import { ICompletionsLogTargetService } from '../../logger';
+import { TelemetryWithExp } from '../../telemetry';
+import { RelatedFilesDocumentInfo, RelatedFilesProvider, RelatedFilesResponse } from './relatedFiles';
+export type ProviderCallback = (uri: string, context: {
+    flags: Record<string, unknown>;
+}, cancellationToken: ICancellationToken) => Promise<RelatedFilesResponse | undefined>;
+type Provider = {
+    languageId: string;
+    extensionId: string;
+    callback: ProviderCallback;
+};
+export declare class CompositeRelatedFilesProvider extends RelatedFilesProvider {
+    private featuresService;
+    protected providers: Map<string, Map<string, Provider>>;
+    protected telemetrySent: boolean;
+    private reportedUnknownProviders;
+    constructor(instantiationService: IInstantiationService, ignoreService: IIgnoreService, featuresService: ICompletionsFeaturesService, logTarget: ICompletionsLogTargetService, fileSystemService: ICompletionsFileSystemService);
+    getRelatedFilesResponse(docInfo: RelatedFilesDocumentInfo, telemetryData: TelemetryWithExp, cancellationToken: ICancellationToken | undefined): Promise<RelatedFilesResponse | undefined>;
+    convert(uri: string, providers: Map<string, Provider>, startTime: number, telemetryData: TelemetryWithExp, token: ICancellationToken | undefined): Promise<RelatedFilesResponse | undefined>;
+    registerRelatedFilesProvider(extensionId: string, languageId: string, provider: ProviderCallback): void;
+    unregisterRelatedFilesProvider(extensionId: string, languageId: string, callback: ProviderCallback): void;
+    /**
+     * Providers should manage their own telemetry.
+     * These four methods are for backward compatibility with the C++ provider.
+     */
+    isActive(languageId: string, telemetryData: TelemetryWithExp): boolean;
+    relatedFilesTelemetry(telemetryData: TelemetryWithExp): void;
+    relatedFileNonresponseTelemetry(language: string, telemetryData: TelemetryWithExp): void;
+    performanceTelemetry(duration: number, telemetryData: TelemetryWithExp): void;
+}
+export {};
+//# sourceMappingURL=compositeRelatedFilesProvider.d.ts.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/similarFiles/compositeRelatedFilesProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"compositeRelatedFilesProvider.d.ts","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/similarFiles/compositeRelatedFilesProvider.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,cAAc,EAAE,MAAM,2DAA2D,CAAC;AAC3F,OAAO,EAAE,qBAAqB,EAAE,MAAM,0EAA0E,CAAC;AACjH,OAAO,EAAE,iBAAiB,IAAI,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAEhF,OAAO,EAAE,2BAA2B,EAAE,MAAM,mCAAmC,CAAC;AAChF,OAAO,EAAE,6BAA6B,EAAE,MAAM,kBAAkB,CAAC;AACjE,OAAO,EAAE,4BAA4B,EAAE,MAAM,cAAc,CAAC;AAC5D,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAEnD,OAAO,EAEN,wBAAwB,EACxB,oBAAoB,EACpB,oBAAoB,EAEpB,MAAM,gBAAgB,CAAC;AAexB,MAAM,MAAM,gBAAgB,GAAG,CAC9B,GAAG,EAAE,MAAM,EACX,OAAO,EAAE;IAAE,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,EAC3C,iBAAiB,EAAE,kBAAkB,KACjC,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC,CAAC;AAE/C,KAAK,QAAQ,GAAG;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,gBAAgB,CAAC;CAC3B,CAAC;AAEF,qBAAa,6BAA8B,SAAQ,oBAAoB;IAOxC,OAAO,CAAC,eAAe;IANrD,SAAS,CAAC,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAa;IACpE,SAAS,CAAC,aAAa,UAAS;IAChC,OAAO,CAAC,wBAAwB,CAAqB;gBAE7B,oBAAoB,EAAE,qBAAqB,EAClD,aAAa,EAAE,cAAc,EACR,eAAe,EAAE,2BAA2B,EACnD,SAAS,EAAE,4BAA4B,EACtC,iBAAiB,EAAE,6BAA6B;IAIjE,uBAAuB,CACrC,OAAO,EAAE,wBAAwB,EACjC,aAAa,EAAE,gBAAgB,EAC/B,iBAAiB,EAAE,kBAAkB,GAAG,SAAS;IA+B5C,OAAO,CACZ,GAAG,EAAE,MAAM,EACX,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC,EAChC,SAAS,EAAE,MAAM,EACjB,aAAa,EAAE,gBAAgB,EAC/B,KAAK,EAAE,kBAAkB,GAAG,SAAS,GACnC,OAAO,CAAC,oBAAoB,GAAG,SAAS,CAAC;IA2B5C,4BAA4B,CAAC,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,gBAAgB;IAQhG,8BAA8B,CAAC,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,gBAAgB;IASlG;;;OAGG;IACH,QAAQ,CAAC,UAAU,EAAE,MAAM,EAAE,aAAa,EAAE,gBAAgB,GAAG,OAAO;IAqBtE,qBAAqB,CAAC,aAAa,EAAE,gBAAgB;IACrD,+BAA+B,CAAC,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,gBAAgB;IACjF,oBAAoB,CAAC,QAAQ,EAAE,MAAM,EAAE,aAAa,EAAE,gBAAgB;CACtE"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/similarFiles/compositeRelatedFilesProvider.js

@@ -0,0 +1,149 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CompositeRelatedFilesProvider = void 0;
+const ignoreService_1 = require("../../../../../../../platform/ignore/common/ignoreService");
+const instantiation_1 = require("../../../../../../../util/vs/platform/instantiation/common/instantiation");
+const config_1 = require("../../config");
+const featuresService_1 = require("../../experiments/featuresService");
+const fileSystem_1 = require("../../fileSystem");
+const logger_1 = require("../../logger");
+const neighborFiles_1 = require("./neighborFiles");
+const relatedFiles_1 = require("./relatedFiles");
+const cppLanguageIds = ['cpp', 'c', 'cuda-cpp'];
+const typescriptLanguageIds = ['typescript', 'javascript', 'typescriptreact', 'javascriptreact'];
+const csharpLanguageIds = ['csharp'];
+const neighborFileTypeMap = new Map([
+    ...cppLanguageIds.map(id => [id, neighborFiles_1.NeighboringFileType.RelatedCpp]),
+    ...typescriptLanguageIds.map(id => [id, neighborFiles_1.NeighboringFileType.RelatedTypeScript]),
+    ...csharpLanguageIds.map(id => [id, neighborFiles_1.NeighboringFileType.RelatedCSharpRoslyn]),
+]);
+function getNeighboringFileType(languageId) {
+    return neighborFileTypeMap.get(languageId) ?? neighborFiles_1.NeighboringFileType.RelatedOther;
+}
+let CompositeRelatedFilesProvider = class CompositeRelatedFilesProvider extends relatedFiles_1.RelatedFilesProvider {
+    constructor(instantiationService, ignoreService, featuresService, logTarget, fileSystemService) {
+        super(instantiationService, ignoreService, logTarget, fileSystemService);
+        this.featuresService = featuresService;
+        this.providers = new Map();
+        this.telemetrySent = false;
+        this.reportedUnknownProviders = new Set();
+    }
+    async getRelatedFilesResponse(docInfo, telemetryData, cancellationToken) {
+        const startTime = Date.now();
+        const languageId = docInfo.clientLanguageId.toLowerCase();
+        const fileType = getNeighboringFileType(languageId);
+        if (fileType === neighborFiles_1.NeighboringFileType.RelatedOther && !this.reportedUnknownProviders.has(languageId)) {
+            this.reportedUnknownProviders.add(languageId);
+            relatedFiles_1.relatedFilesLogger.warn(this.logTarget, `unknown language ${languageId}`);
+        }
+        this.relatedFilesTelemetry(telemetryData);
+        relatedFiles_1.relatedFilesLogger.debug(this.logTarget, `Fetching related files for ${docInfo.uri}`);
+        if (!this.isActive(languageId, telemetryData)) {
+            relatedFiles_1.relatedFilesLogger.debug(this.logTarget, 'language-server related-files experiment is not active.');
+            return relatedFiles_1.EmptyRelatedFilesResponse;
+        }
+        const languageProviders = this.providers.get(languageId);
+        if (!languageProviders) {
+            return relatedFiles_1.EmptyRelatedFilesResponse;
+        }
+        try {
+            return this.convert(docInfo.uri, languageProviders, startTime, telemetryData, cancellationToken);
+        }
+        catch (error) {
+            // When the command returns an empty std::optional, we get an Error exception with message:
+            // "Received message which is neither a response nor a notification message: {"jsonrpc": "2.0","id": 22}"
+            this.relatedFileNonresponseTelemetry(languageId, telemetryData);
+            // Return undefined to inform the caller that the command failed.
+            return undefined;
+        }
+    }
+    async convert(uri, providers, startTime, telemetryData, token) {
+        if (!token) {
+            token = {
+                isCancellationRequested: false,
+                onCancellationRequested: () => ({ dispose() { } }),
+            };
+        }
+        const combined = { entries: [], traits: [] };
+        let allProvidersReturnedUndefined = providers.size > 0;
+        for (const provider of providers.values()) {
+            const response = await provider.callback(uri, { flags: {} }, token);
+            if (response) {
+                allProvidersReturnedUndefined = false;
+                combined.entries.push(...response.entries);
+                if (response.traits) {
+                    combined.traits.push(...response.traits);
+                }
+                for (const entry of response.entries) {
+                    for (const uri of entry.uris) {
+                        relatedFiles_1.relatedFilesLogger.debug(this.logTarget, uri.toString());
+                    }
+                }
+            }
+        }
+        this.performanceTelemetry(Date.now() - startTime, telemetryData);
+        return allProvidersReturnedUndefined ? undefined : combined;
+    }
+    registerRelatedFilesProvider(extensionId, languageId, provider) {
+        const languageProvider = this.providers.get(languageId);
+        if (languageProvider) {
+            languageProvider.set(extensionId, { extensionId, languageId, callback: provider });
+        }
+        else {
+            this.providers.set(languageId, new Map([[extensionId, { extensionId, languageId, callback: provider }]]));
+        }
+    }
+    unregisterRelatedFilesProvider(extensionId, languageId, callback) {
+        const languageProvider = this.providers.get(languageId);
+        if (languageProvider) {
+            const currentProvider = languageProvider.get(extensionId);
+            if (currentProvider && currentProvider.callback === callback) {
+                languageProvider.delete(extensionId);
+            }
+        }
+    }
+    /**
+     * Providers should manage their own telemetry.
+     * These four methods are for backward compatibility with the C++ provider.
+     */
+    isActive(languageId, telemetryData) {
+        if (csharpLanguageIds.includes(languageId)) {
+            return (this.featuresService.relatedFilesVSCodeCSharp(telemetryData) ||
+                this.instantiationService.invokeFunction((config_1.getConfig), config_1.ConfigKey.RelatedFilesVSCodeCSharp));
+        }
+        else if (typescriptLanguageIds.includes(languageId)) {
+            return (this.featuresService.relatedFilesVSCodeTypeScript(telemetryData) ||
+                this.instantiationService.invokeFunction((config_1.getConfig), config_1.ConfigKey.RelatedFilesVSCodeTypeScript));
+        }
+        else if (cppLanguageIds.includes(languageId)) {
+            return (this.featuresService.cppHeadersEnableSwitch(telemetryData));
+        }
+        return (this.featuresService.relatedFilesVSCode(telemetryData) ||
+            this.instantiationService.invokeFunction((config_1.getConfig), config_1.ConfigKey.RelatedFilesVSCode));
+    }
+    relatedFilesTelemetry(telemetryData) { }
+    relatedFileNonresponseTelemetry(language, telemetryData) { }
+    performanceTelemetry(duration, telemetryData) { }
+};
+exports.CompositeRelatedFilesProvider = CompositeRelatedFilesProvider;
+exports.CompositeRelatedFilesProvider = CompositeRelatedFilesProvider = __decorate([
+    __param(0, instantiation_1.IInstantiationService),
+    __param(1, ignoreService_1.IIgnoreService),
+    __param(2, featuresService_1.ICompletionsFeaturesService),
+    __param(3, logger_1.ICompletionsLogTargetService),
+    __param(4, fileSystem_1.ICompletionsFileSystemService)
+], CompositeRelatedFilesProvider);
+//# sourceMappingURL=compositeRelatedFilesProvider.js.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/similarFiles/compositeRelatedFilesProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"compositeRelatedFilesProvider.js","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/similarFiles/compositeRelatedFilesProvider.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;;;;;;;;;;AAEhG,6FAA2F;AAC3F,4GAAiH;AAEjH,yCAAoD;AACpD,uEAAgF;AAChF,iDAAiE;AACjE,yCAA4D;AAE5D,mDAAsD;AACtD,iDAMwB;AAExB,MAAM,cAAc,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;AAChD,MAAM,qBAAqB,GAAG,CAAC,YAAY,EAAE,YAAY,EAAE,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;AACjG,MAAM,iBAAiB,GAAG,CAAC,QAAQ,CAAC,CAAC;AACrC,MAAM,mBAAmB,GAAG,IAAI,GAAG,CAA8B;IAChE,GAAG,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,mCAAmB,CAAC,UAAU,CAAU,CAAC;IAC1E,GAAG,qBAAqB,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,mCAAmB,CAAC,iBAAiB,CAAU,CAAC;IACxF,GAAG,iBAAiB,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,mCAAmB,CAAC,mBAAmB,CAAU,CAAC;CACtF,CAAC,CAAC;AAEH,SAAS,sBAAsB,CAAC,UAAkB;IACjD,OAAO,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,mCAAmB,CAAC,YAAY,CAAC;AAChF,CAAC;AAcM,IAAM,6BAA6B,GAAnC,MAAM,6BAA8B,SAAQ,mCAAoB;IAItE,YACwB,oBAA2C,EAClD,aAA6B,EAChB,eAAoD,EACnD,SAAuC,EACtC,iBAAgD;QAE/E,KAAK,CAAC,oBAAoB,EAAE,aAAa,EAAE,SAAS,EAAE,iBAAiB,CAAC,CAAC;QAJpC,oBAAe,GAAf,eAAe,CAA6B;QANxE,cAAS,GAAuC,IAAI,GAAG,EAAE,CAAC;QAC1D,kBAAa,GAAG,KAAK,CAAC;QACxB,6BAAwB,GAAG,IAAI,GAAG,EAAU,CAAC;IASrD,CAAC;IACQ,KAAK,CAAC,uBAAuB,CACrC,OAAiC,EACjC,aAA+B,EAC/B,iBAAiD;QAEjD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,UAAU,GAAG,OAAO,CAAC,gBAAgB,CAAC,WAAW,EAAE,CAAC;QAC1D,MAAM,QAAQ,GAAG,sBAAsB,CAAC,UAAU,CAAC,CAAC;QACpD,IAAI,QAAQ,KAAK,mCAAmB,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YACrG,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YAC9C,iCAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,oBAAoB,UAAU,EAAE,CAAC,CAAC;QAC3E,CAAC;QACD,IAAI,CAAC,qBAAqB,CAAC,aAAa,CAAC,CAAC;QAE1C,iCAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,8BAA8B,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;QACtF,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,aAAa,CAAC,EAAE,CAAC;YAC/C,iCAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,yDAAyD,CAAC,CAAC;YACpG,OAAO,wCAAyB,CAAC;QAClC,CAAC;QAED,MAAM,iBAAiB,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACzD,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACxB,OAAO,wCAAyB,CAAC;QAClC,CAAC;QACD,IAAI,CAAC;YACJ,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,iBAAiB,EAAE,SAAS,EAAE,aAAa,EAAE,iBAAiB,CAAC,CAAC;QAClG,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,2FAA2F;YAC3F,yGAAyG;YACzG,IAAI,CAAC,+BAA+B,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;YAChE,iEAAiE;YACjE,OAAO,SAAS,CAAC;QAClB,CAAC;IACF,CAAC;IACD,KAAK,CAAC,OAAO,CACZ,GAAW,EACX,SAAgC,EAChC,SAAiB,EACjB,aAA+B,EAC/B,KAAqC;QAErC,IAAI,CAAC,KAAK,EAAE,CAAC;YACZ,KAAK,GAAG;gBACP,uBAAuB,EAAE,KAAK;gBAC9B,uBAAuB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,OAAO,KAAK,CAAC,EAAE,CAAC;aAClD,CAAC;QACH,CAAC;QACD,MAAM,QAAQ,GAAyB,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QACnE,IAAI,6BAA6B,GAAY,SAAS,CAAC,IAAI,GAAG,CAAC,CAAC;QAChE,KAAK,MAAM,QAAQ,IAAI,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC;YAC3C,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,CAAC;YACpE,IAAI,QAAQ,EAAE,CAAC;gBACd,6BAA6B,GAAG,KAAK,CAAC;gBACtC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;gBAC3C,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;oBACrB,QAAQ,CAAC,MAAO,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC;gBAC3C,CAAC;gBACD,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;oBACtC,KAAK,MAAM,GAAG,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;wBAC9B,iCAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC;oBAC1D,CAAC;gBACF,CAAC;YACF,CAAC;QACF,CAAC;QACD,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,EAAE,aAAa,CAAC,CAAC;QACjE,OAAO,6BAA6B,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;IAC7D,CAAC;IACD,4BAA4B,CAAC,WAAmB,EAAE,UAAkB,EAAE,QAA0B;QAC/F,MAAM,gBAAgB,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACxD,IAAI,gBAAgB,EAAE,CAAC;YACtB,gBAAgB,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpF,CAAC;aAAM,CAAC;YACP,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC,WAAW,EAAE,EAAE,WAAW,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3G,CAAC;IACF,CAAC;IACD,8BAA8B,CAAC,WAAmB,EAAE,UAAkB,EAAE,QAA0B;QACjG,MAAM,gBAAgB,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACxD,IAAI,gBAAgB,EAAE,CAAC;YACtB,MAAM,eAAe,GAAG,gBAAgB,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;YAC1D,IAAI,eAAe,IAAI,eAAe,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;gBAC9D,gBAAgB,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YACtC,CAAC;QACF,CAAC;IACF,CAAC;IACD;;;OAGG;IACH,QAAQ,CAAC,UAAkB,EAAE,aAA+B;QAC3D,IAAI,iBAAiB,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAC5C,OAAO,CACN,IAAI,CAAC,eAAe,CAAC,wBAAwB,CAAC,aAAa,CAAC;gBAC5D,IAAI,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAA,kBAAkB,CAAA,EAAE,kBAAS,CAAC,wBAAwB,CAAC,CAChG,CAAC;QACH,CAAC;aAAM,IAAI,qBAAqB,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YACvD,OAAO,CACN,IAAI,CAAC,eAAe,CAAC,4BAA4B,CAAC,aAAa,CAAC;gBAChE,IAAI,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAA,kBAAkB,CAAA,EAAE,kBAAS,CAAC,4BAA4B,CAAC,CACpG,CAAC;QACH,CAAC;aAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAChD,OAAO,CACN,IAAI,CAAC,eAAe,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAC1D,CAAC;QACH,CAAC;QACD,OAAO,CACN,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,aAAa,CAAC;YACtD,IAAI,CAAC,oBAAoB,CAAC,cAAc,CAAC,CAAA,kBAAkB,CAAA,EAAE,kBAAS,CAAC,kBAAkB,CAAC,CAC1F,CAAC;IACH,CAAC;IACD,qBAAqB,CAAC,aAA+B,IAAI,CAAC;IAC1D,+BAA+B,CAAC,QAAgB,EAAE,aAA+B,IAAI,CAAC;IACtF,oBAAoB,CAAC,QAAgB,EAAE,aAA+B,IAAI,CAAC;CAC3E,CAAA;AA7HY,sEAA6B;wCAA7B,6BAA6B;IAKvC,WAAA,qCAAqB,CAAA;IACrB,WAAA,8BAAc,CAAA;IACd,WAAA,6CAA2B,CAAA;IAC3B,WAAA,qCAA4B,CAAA;IAC5B,WAAA,0CAA6B,CAAA;GATnB,6BAA6B,CA6HzC"}

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/similarFiles/neighborFiles.d.ts

@@ -0,0 +1,66 @@
+import { ServicesAccessor } from '../../../../../../../util/vs/platform/instantiation/common/instantiation';
+import { SimilarFileInfo } from '../../../../prompt/src/prompt';
+import { CancellationToken as ICancellationToken } from '../../../../types/src';
+import { TelemetryWithExp } from '../../telemetry';
+import { RelatedFileTrait } from './relatedFiles';
+export declare enum NeighboringFileType {
+    None = "none",// Do not add neighbor files.
+    OpenTabs = "opentabs",// Add open files.
+    CursorMostRecent = "cursormostrecent",// Add the most recent cursor focused files.
+    CursorMostCount = "cursormostcount",// Add the most cursor focused files.
+    WorkspaceSharingSameFolder = "workspacesharingsamefolder",// Add the workspace files sharing the same folder with the target file.
+    WorkspaceSmallestPathDist = "workspacesmallestpathdist",// Add the workspace files according to their path distance toward the target file
+    OpenTabsAndCocommitted = "opentabsandcocommitted",// Add open files and the co-committed files.
+    RelatedCSharp = "related/csharp",// The Semantic Code Context says this file is related.
+    RelatedCSharpRoslyn = "related/csharproslyn",// The C# language service says this file is related.
+    RelatedCpp = "related/cpp",// The C++ language service says this file is related.
+    RelatedTypeScript = "related/typescript",// The Typescript language service says this file is related.
+    RelatedCppSemanticCodeContext = "related/cppsemanticcodecontext",// The Semantic Code Context says this file is related.
+    RelatedOther = "related/other"
+}
+/**
+ * We found out that considering
+ * all **open** neighbor files (independent of the language) was not helpful. However, some
+ * specific languages (e.g. frontend frameworks) benefit from this approach. Leaving this
+ * function here for future reference, in case we want to experiment this approach again for
+ * specific languages that always use cross-language files.
+ *
+ * @param languageId Language ID of the current file
+ * @param neighborLanguageId Language ID of the neighbor file
+ * @returns Boolean value indicating whether the neighbor file should be considered
+ *          (currently matching the current file's language with neighbors')
+ */
+export declare function considerNeighborFile(languageId: string, neighborLanguageId: string): boolean;
+export type NeighborsCollection = Map<string, SimilarFileInfo>;
+export interface INeighborSource {
+    getNeighborFiles(uri: string, languageId: string, maxNumNeighborFiles: number): Promise<{
+        docs: NeighborsCollection;
+        neighborSource: Map<NeighboringFileType, string[]>;
+    }>;
+}
+export declare class NeighborSource {
+    static MAX_NEIGHBOR_AGGREGATE_LENGTH: number;
+    static MAX_NEIGHBOR_FILES: number;
+    static EXCLUDED_NEIGHBORS: string[];
+    static defaultEmptyResult(): {
+        docs: Map<string, SimilarFileInfo>;
+        neighborSource: Map<NeighboringFileType, string[]>;
+        traits: RelatedFileTrait[];
+    };
+    private static instance;
+    /** Reset the singleton instance for unit test only */
+    static reset(): void;
+    static getNeighborFilesAndTraits(accessor: ServicesAccessor, uri: string, fileType: string, telemetryData: TelemetryWithExp, cancellationToken?: ICancellationToken, data?: unknown, forceRelatedFilesComputation?: boolean): Promise<{
+        docs: NeighborsCollection;
+        neighborSource: Map<NeighboringFileType, string[]>;
+        traits: RelatedFileTrait[];
+    }>;
+    static basename(uri: string): string;
+    /**
+     * Get the fileUri relative to the provided basePath
+     * or its basename if basePath is not its ancestor.
+     */
+    static getRelativePath(fileUri: string, baseUri: string): string | undefined;
+}
+export declare function isIncludeNeighborFilesActive(accessor: ServicesAccessor, languageId: string, telemetryData: TelemetryWithExp): boolean;
+//# sourceMappingURL=neighborFiles.d.ts.map

package/dist/src/_internal/extension/completions-core/vscode-node/lib/src/prompt/similarFiles/neighborFiles.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"neighborFiles.d.ts","sourceRoot":"","sources":["../../../../../../../../../../src/_internal/extension/completions-core/vscode-node/lib/src/prompt/similarFiles/neighborFiles.ts"],"names":[],"mappings":"AAKA,OAAO,EAAyB,gBAAgB,EAAE,MAAM,0EAA0E,CAAC;AACnI,OAAO,EAAuB,eAAe,EAAE,MAAM,+BAA+B,CAAC;AACrF,OAAO,EAAE,iBAAiB,IAAI,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAGhF,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAGnD,OAAO,EAAgD,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAGhG,oBAAY,mBAAmB;IAC9B,IAAI,SAAS,CAAE,6BAA6B;IAC5C,QAAQ,aAAa,CAAE,kBAAkB;IACzC,gBAAgB,qBAAqB,CAAE,4CAA4C;IACnF,eAAe,oBAAoB,CAAE,qCAAqC;IAC1E,0BAA0B,+BAA+B,CAAE,wEAAwE;IACnI,yBAAyB,8BAA8B,CAAE,kFAAkF;IAC3I,sBAAsB,2BAA2B,CAAE,6CAA6C;IAChG,aAAa,mBAAmB,CAAE,uDAAuD;IACzF,mBAAmB,yBAAyB,CAAE,qDAAqD;IACnG,UAAU,gBAAgB,CAAE,sDAAsD;IAClF,iBAAiB,uBAAuB,CAAE,6DAA6D;IACvG,6BAA6B,mCAAmC,CAAE,uDAAuD;IACzH,YAAY,kBAAkB;CAC9B;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,oBAAoB,CAAC,UAAU,EAAE,MAAM,EAAE,kBAAkB,EAAE,MAAM,GAAG,OAAO,CAE5F;AAED,MAAM,MAAM,mBAAmB,GAAG,GAAG,CAAC,MAAM,EAAE,eAAe,CAAC,CAAC;AAE/D,MAAM,WAAW,eAAe;IAC/B,gBAAgB,CACf,GAAG,EAAE,MAAM,EACX,UAAU,EAAE,MAAM,EAClB,mBAAmB,EAAE,MAAM,GACzB,OAAO,CAAC;QAAE,IAAI,EAAE,mBAAmB,CAAC;QAAC,cAAc,EAAE,GAAG,CAAC,mBAAmB,EAAE,MAAM,EAAE,CAAC,CAAA;KAAE,CAAC,CAAC;CAC9F;AAED,qBAAa,cAAc;IAE1B,MAAM,CAAC,6BAA6B,SAAU;IAC9C,MAAM,CAAC,kBAAkB,SAAM;IAE/B,MAAM,CAAC,kBAAkB,WAA6C;IAEtE,MAAM,CAAC,kBAAkB;;;gBAIT,gBAAgB,EAAE;;IAIlC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAA8B;IAErD,sDAAsD;IACtD,MAAM,CAAC,KAAK,IAAI,IAAI;WAIP,yBAAyB,CACrC,QAAQ,EAAE,gBAAgB,EAC1B,GAAG,EAAE,MAAM,EACX,QAAQ,EAAE,MAAM,EAChB,aAAa,EAAE,gBAAgB,EAC/B,iBAAiB,CAAC,EAAE,kBAAkB,EACtC,IAAI,CAAC,EAAE,OAAO,EACd,4BAA4B,CAAC,EAAE,OAAO,GACpC,OAAO,CAAC;QACV,IAAI,EAAE,mBAAmB,CAAC;QAC1B,cAAc,EAAE,GAAG,CAAC,mBAAmB,EAAE,MAAM,EAAE,CAAC,CAAC;QACnD,MAAM,EAAE,gBAAgB,EAAE,CAAC;KAC3B,CAAC;IA4EF,MAAM,CAAC,QAAQ,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM;IAIpC;;;OAGG;IACH,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS;CAU5E;AAED,wBAAgB,4BAA4B,CAAC,QAAQ,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,EAAE,aAAa,EAAE,gBAAgB,GAAG,OAAO,CAGrI"}