Extract workspaceContains related logic to a separate file (#75284)

src/vs/workbench/api/common/shared/workspaceContains.ts

@@ -0,0 +1,105 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as path from 'vs/base/common/path';
+import { URI, UriComponents } from 'vs/base/common/uri';
+import { CancellationTokenSource, CancellationToken } from 'vs/base/common/cancellation';
+import * as errors from 'vs/base/common/errors';
+import { ExtensionIdentifier, IExtensionDescription } from 'vs/platform/extensions/common/extensions';
+
+const WORKSPACE_CONTAINS_TIMEOUT = 7000;
+
+export interface IExtensionActivationHost {
+	readonly folders: readonly UriComponents[];
+	readonly forceUsingSearch: boolean;
+
+	exists(path: string): Promise<boolean>;
+	checkExists(folders: readonly UriComponents[], includes: string[], token: CancellationToken): Promise<boolean>;
+}
+
+export interface IExtensionActivationResult {
+	activationEvent: string;
+}
+
+export function checkActivateWorkspaceContainsExtension(host: IExtensionActivationHost, desc: IExtensionDescription): Promise<IExtensionActivationResult | undefined> {
+	const activationEvents = desc.activationEvents;
+	if (!activationEvents) {
+		return Promise.resolve(undefined);
+	}
+
+	const fileNames: string[] = [];
+	const globPatterns: string[] = [];
+
+	for (const activationEvent of activationEvents) {
+		if (/^workspaceContains:/.test(activationEvent)) {
+			const fileNameOrGlob = activationEvent.substr('workspaceContains:'.length);
+			if (fileNameOrGlob.indexOf('*') >= 0 || fileNameOrGlob.indexOf('?') >= 0 || host.forceUsingSearch) {
+				globPatterns.push(fileNameOrGlob);
+			} else {
+				fileNames.push(fileNameOrGlob);
+			}
+		}
+	}
+
+	if (fileNames.length === 0 && globPatterns.length === 0) {
+		return Promise.resolve(undefined);
+	}
+
+	let resolveResult: (value: IExtensionActivationResult | undefined) => void;
+	const result = new Promise<IExtensionActivationResult | undefined>((resolve, reject) => { resolveResult = resolve; });
+	const activate = (activationEvent: string) => resolveResult({ activationEvent });
+
+	const fileNamePromise = Promise.all(fileNames.map((fileName) => _activateIfFileName(host, fileName, activate))).then(() => { });
+	const globPatternPromise = _activateIfGlobPatterns(host, desc.identifier, globPatterns, activate);
+
+	Promise.all([fileNamePromise, globPatternPromise]).then(() => {
+		// when all are done, resolve with undefined (relevant only if it was not activated so far)
+		resolveResult(undefined);
+	});
+
+	return result;
+}
+
+async function _activateIfFileName(host: IExtensionActivationHost, fileName: string, activate: (activationEvent: string) => void): Promise<void> {
+	// find exact path
+	for (const uri of host.folders) {
+		if (await host.exists(path.join(URI.revive(uri).fsPath, fileName))) {
+			// the file was found
+			activate(`workspaceContains:${fileName}`);
+			return;
+		}
+	}
+}
+
+async function _activateIfGlobPatterns(host: IExtensionActivationHost, extensionId: ExtensionIdentifier, globPatterns: string[], activate: (activationEvent: string) => void): Promise<void> {
+	if (globPatterns.length === 0) {
+		return Promise.resolve(undefined);
+	}
+
+	const tokenSource = new CancellationTokenSource();
+	const searchP = host.checkExists(host.folders, globPatterns, tokenSource.token);
+
+	const timer = setTimeout(async () => {
+		tokenSource.cancel();
+		activate(`workspaceContainsTimeout:${globPatterns.join(',')}`);
+	}, WORKSPACE_CONTAINS_TIMEOUT);
+
+	let exists: boolean = false;
+	try {
+		exists = await searchP;
+	} catch (err) {
+		if (!errors.isPromiseCanceledError(err)) {
+			errors.onUnexpectedError(err);
+		}
+	}
+
+	tokenSource.dispose();
+	clearTimeout(timer);
+
+	if (exists) {
+		// a file was found matching one of the glob patterns
+		activate(`workspaceContains:${globPatterns.join(',')}`);
+	}
+}