@codingame/monaco-vscode-notebook-service-override 3.2.2 → 4.0.0

package/vscode/src/vs/workbench/contrib/search/browser/notebookSearch/notebookSearchService.js

@@ -1,237 +0,0 @@
-import { __decorate, __param } from '../../../../../../../../external/tslib/tslib.es6.js';
-import { CancellationToken } from 'vscode/vscode/vs/base/common/cancellation';
-import { ResourceSet, ResourceMap } from 'vscode/vscode/vs/base/common/map';
-import { IConfigurationService } from 'vscode/vscode/vs/platform/configuration/common/configuration';
-import { ILogService } from 'vscode/vscode/vs/platform/log/common/log';
-import { IUriIdentityService } from 'vscode/vscode/vs/platform/uriIdentity/common/uriIdentity';
-import { INotebookService } from 'vscode/vscode/vs/workbench/contrib/notebook/common/notebookService';
-import { contentMatchesToTextSearchMatches, webviewMatchesToTextSearchMatches } from 'vscode/vscode/vs/workbench/contrib/search/browser/notebookSearch/searchNotebookHelpers';
-import { pathIncludedInQuery, ISearchService } from 'vscode/vscode/vs/workbench/services/search/common/search';
-import { coalesce } from 'vscode/vscode/vs/base/common/arrays';
-import { isNumber } from 'vscode/vscode/vs/base/common/types';
-import { IEditorResolverService } from 'vscode/vscode/vs/workbench/services/editor/common/editorResolverService';
-import { INotebookEditorService } from 'vscode/vscode/vs/workbench/contrib/notebook/browser/services/notebookEditorService';
-import { QueryBuilder } from 'vscode/vscode/vs/workbench/services/search/common/queryBuilder';
-import { IInstantiationService } from 'vscode/vscode/vs/platform/instantiation/common/instantiation';
-
-let NotebookSearchService = class NotebookSearchService {
-    constructor(uriIdentityService, notebookEditorService, logService, notebookService, configurationService, editorResolverService, searchService, instantiationService) {
-        this.uriIdentityService = uriIdentityService;
-        this.notebookEditorService = notebookEditorService;
-        this.logService = logService;
-        this.notebookService = notebookService;
-        this.configurationService = configurationService;
-        this.editorResolverService = editorResolverService;
-        this.searchService = searchService;
-        this.queryBuilder = instantiationService.createInstance(QueryBuilder);
-    }
-    notebookSearch(query, token, searchInstanceID, onProgress) {
-        if (query.type !== 2 ) {
-            return {
-                openFilesToScan: ( new ResourceSet()),
-                completeData: Promise.resolve({
-                    messages: [],
-                    limitHit: false,
-                    results: [],
-                }),
-                allScannedFiles: Promise.resolve(( new ResourceSet())),
-            };
-        }
-        const localNotebookWidgets = this.getLocalNotebookWidgets();
-        const localNotebookFiles = ( localNotebookWidgets.map(widget => widget.viewModel.uri));
-        const getAllResults = () => {
-            const searchStart = Date.now();
-            const localResultPromise = this.getLocalNotebookResults(query, token ?? CancellationToken.None, localNotebookWidgets, searchInstanceID);
-            const searchLocalEnd = Date.now();
-            const experimentalNotebooksEnabled = this.configurationService.getValue('search').experimental?.closedNotebookRichContentResults ?? false;
-            let closedResultsPromise = Promise.resolve(undefined);
-            if (experimentalNotebooksEnabled) {
-                closedResultsPromise = this.getClosedNotebookResults(query, ( new ResourceSet(
-                    localNotebookFiles,
-                    uri => this.uriIdentityService.extUri.getComparisonKey(uri)
-                )), token ?? CancellationToken.None);
-            }
-            const promise = Promise.all([localResultPromise, closedResultsPromise]);
-            return {
-                completeData: promise.then((resolvedPromise) => {
-                    const openNotebookResult = resolvedPromise[0];
-                    const closedNotebookResult = resolvedPromise[1];
-                    const resolved = resolvedPromise.filter((e) => !!e);
-                    const resultArray = [...( openNotebookResult.results.values()), ...(closedNotebookResult?.results.values() ?? [])];
-                    const results = coalesce(resultArray);
-                    if (onProgress) {
-                        results.forEach(onProgress);
-                    }
-                    this.logService.trace(`local notebook search time | ${searchLocalEnd - searchStart}ms`);
-                    return {
-                        messages: [],
-                        limitHit: resolved.reduce((prev, cur) => prev || cur.limitHit, false),
-                        results,
-                    };
-                }),
-                allScannedFiles: promise.then(resolvedPromise => {
-                    const openNotebookResults = resolvedPromise[0];
-                    const closedNotebookResults = resolvedPromise[1];
-                    const results = coalesce([...( openNotebookResults.results.keys()), ...(closedNotebookResults?.results.keys() ?? [])]);
-                    return ( new ResourceSet(results, uri => this.uriIdentityService.extUri.getComparisonKey(uri)));
-                })
-            };
-        };
-        const promiseResults = getAllResults();
-        return {
-            openFilesToScan: ( new ResourceSet(localNotebookFiles)),
-            completeData: promiseResults.completeData,
-            allScannedFiles: promiseResults.allScannedFiles
-        };
-    }
-    async doesFileExist(includes, folderQueries, token) {
-        const promises = ( includes.map(async (includePattern) => {
-            const query = this.queryBuilder.file(( folderQueries.map(e => e.folder)), {
-                includePattern: includePattern.startsWith('/') ? includePattern : '**/' + includePattern,
-                exists: true
-            });
-            return this.searchService.fileSearch(query, token).then((ret) => {
-                if (!ret.limitHit) {
-                    throw Error('File not found');
-                }
-            });
-        }));
-        return Promise.any(promises).then(() => true).catch(() => false);
-    }
-    async getClosedNotebookResults(textQuery, scannedFiles, token) {
-        const userAssociations = this.editorResolverService.getAllUserAssociations();
-        const allPriorityInfo = ( new Map());
-        const contributedNotebookTypes = this.notebookService.getContributedNotebookTypes();
-        userAssociations.forEach(association => {
-            if (!association.filenamePattern) {
-                return;
-            }
-            const info = {
-                isFromSettings: true,
-                filenamePatterns: [association.filenamePattern]
-            };
-            const existingEntry = allPriorityInfo.get(association.viewType);
-            if (existingEntry) {
-                allPriorityInfo.set(association.viewType, existingEntry.concat(info));
-            }
-            else {
-                allPriorityInfo.set(association.viewType, [info]);
-            }
-        });
-        const promises = [];
-        contributedNotebookTypes.forEach((notebook) => {
-            if (notebook.selectors.length > 0) {
-                promises.push((async () => {
-                    const includes = ( notebook.selectors.map((selector) => {
-                        const globPattern = selector.include || selector;
-                        return ( globPattern.toString());
-                    }));
-                    const isInWorkspace = await this.doesFileExist(includes, textQuery.folderQueries, token);
-                    if (isInWorkspace) {
-                        const canResolve = await this.notebookService.canResolve(notebook.id);
-                        if (!canResolve) {
-                            return undefined;
-                        }
-                        const serializer = (await this.notebookService.withNotebookDataProvider(notebook.id)).serializer;
-                        return await serializer.searchInNotebooks(textQuery, token, allPriorityInfo);
-                    }
-                    else {
-                        return undefined;
-                    }
-                })());
-            }
-        });
-        const start = Date.now();
-        const searchComplete = coalesce(await Promise.all(promises));
-        const results = searchComplete.flatMap(e => e.results);
-        let limitHit = ( searchComplete.some(e => e.limitHit));
-        const uniqueResults = ( new ResourceMap(uri => this.uriIdentityService.extUri.getComparisonKey(uri)));
-        let numResults = 0;
-        for (const result of results) {
-            if (textQuery.maxResults && numResults >= textQuery.maxResults) {
-                limitHit = true;
-                break;
-            }
-            if (!( scannedFiles.has(result.resource)) && !( uniqueResults.has(result.resource))) {
-                uniqueResults.set(result.resource, result.cellResults.length > 0 ? result : null);
-                numResults++;
-            }
-        }
-        const end = Date.now();
-        this.logService.trace(`query: ${textQuery.contentPattern.pattern}`);
-        this.logService.trace(`closed notebook search time | ${end - start}ms`);
-        return {
-            results: uniqueResults,
-            limitHit
-        };
-    }
-    async getLocalNotebookResults(query, token, widgets, searchID) {
-        const localResults = ( new ResourceMap(uri => this.uriIdentityService.extUri.getComparisonKey(uri)));
-        let limitHit = false;
-        for (const widget of widgets) {
-            if (!widget.hasModel()) {
-                continue;
-            }
-            const askMax = isNumber(query.maxResults) ? query.maxResults + 1 : Number.MAX_SAFE_INTEGER;
-            const uri = widget.viewModel.uri;
-            if (!pathIncludedInQuery(query, uri.fsPath)) {
-                continue;
-            }
-            let matches = await widget
-                .find(query.contentPattern.pattern, {
-                regex: query.contentPattern.isRegExp,
-                wholeWord: query.contentPattern.isWordMatch,
-                caseSensitive: query.contentPattern.isCaseSensitive,
-                includeMarkupInput: query.contentPattern.notebookInfo?.isInNotebookMarkdownInput ?? true,
-                includeMarkupPreview: query.contentPattern.notebookInfo?.isInNotebookMarkdownPreview ?? true,
-                includeCodeInput: query.contentPattern.notebookInfo?.isInNotebookCellInput ?? true,
-                includeOutput: query.contentPattern.notebookInfo?.isInNotebookCellOutput ?? true,
-            }, token, false, true, searchID);
-            if (matches.length) {
-                if (askMax && matches.length >= askMax) {
-                    limitHit = true;
-                    matches = matches.slice(0, askMax - 1);
-                }
-                const cellResults = ( matches.map(match => {
-                    const contentResults = contentMatchesToTextSearchMatches(match.contentMatches, match.cell);
-                    const webviewResults = webviewMatchesToTextSearchMatches(match.webviewMatches);
-                    return {
-                        cell: match.cell,
-                        index: match.index,
-                        contentResults: contentResults,
-                        webviewResults: webviewResults,
-                    };
-                }));
-                const fileMatch = {
-                    resource: uri, cellResults: cellResults
-                };
-                localResults.set(uri, fileMatch);
-            }
-            else {
-                localResults.set(uri, null);
-            }
-        }
-        return {
-            results: localResults,
-            limitHit
-        };
-    }
-    getLocalNotebookWidgets() {
-        const notebookWidgets = this.notebookEditorService.retrieveAllExistingWidgets();
-        return ( notebookWidgets
-            .map(widget => widget.value))
-            .filter((val) => !!val && val.hasModel());
-    }
-};
-NotebookSearchService = ( __decorate([
-    ( __param(0, IUriIdentityService)),
-    ( __param(1, INotebookEditorService)),
-    ( __param(2, ILogService)),
-    ( __param(3, INotebookService)),
-    ( __param(4, IConfigurationService)),
-    ( __param(5, IEditorResolverService)),
-    ( __param(6, ISearchService)),
-    ( __param(7, IInstantiationService))
-], NotebookSearchService));
-
-export { NotebookSearchService };