azuredatastudio/src/vs/workbench/services/telemetry/browser/telemetryService.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the Source EULA. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import { ITelemetryService, ITelemetryInfo, ITelemetryData } from 'vs/platform/telemetry/common/telemetry';
import { NullTelemetryService, combinedAppender, LogAppender, ITelemetryAppender, validateTelemetryData } from 'vs/platform/telemetry/common/telemetryUtils';
import { IConfigurationService } from 'vs/platform/configuration/common/configuration';
import { Disposable } from 'vs/base/common/lifecycle';
import { IWorkbenchEnvironmentService } from 'vs/workbench/services/environment/common/environmentService';
import { ILogService } from 'vs/platform/log/common/log';
import { TelemetryService as BaseTelemetryService, ITelemetryServiceConfig } from 'vs/platform/telemetry/common/telemetryService';
import { registerSingleton } from 'vs/platform/instantiation/common/extensions';
import { ClassifiedEvent, StrictPropertyCheck, GDPRClassification } from 'vs/platform/telemetry/common/gdprTypings';
import { IStorageService } from 'vs/platform/storage/common/storage';
import { resolveWorkbenchCommonProperties } from 'vs/platform/telemetry/browser/workbenchCommonProperties';
import { IProductService } from 'vs/platform/product/common/product';

interface IConfig {
	instrumentationKey?: string;
	endpointUrl?: string;
	emitLineDelimitedJson?: boolean;
	accountId?: string;
	sessionRenewalMs?: number;
	sessionExpirationMs?: number;
	maxBatchSizeInBytes?: number;
	maxBatchInterval?: number;
	enableDebug?: boolean;
	disableExceptionTracking?: boolean;
	disableTelemetry?: boolean;
	verboseLogging?: boolean;
	diagnosticLogInterval?: number;
	samplingPercentage?: number;
	autoTrackPageVisitTime?: boolean;
	disableAjaxTracking?: boolean;
	overridePageViewDuration?: boolean;
	maxAjaxCallsPerView?: number;
	disableDataLossAnalysis?: boolean;
	disableCorrelationHeaders?: boolean;
	correlationHeaderExcludedDomains?: string[];
	disableFlushOnBeforeUnload?: boolean;
	enableSessionStorageBuffer?: boolean;
	isCookieUseDisabled?: boolean;
	cookieDomain?: string;
	isRetryDisabled?: boolean;
	url?: string;
	isStorageUseDisabled?: boolean;
	isBeaconApiDisabled?: boolean;
	sdkExtension?: string;
	isBrowserLinkTrackingEnabled?: boolean;
	appId?: string;
	enableCorsCorrelation?: boolean;
}

declare class Microsoft {
	public static ApplicationInsights: {
		Initialization: {
			new(init: { config: IConfig }): AppInsights;
		}
	};
}

declare interface IAppInsightsClient {
	config: IConfig;

	/** Log a user action or other occurrence. */
	trackEvent: (name: string, properties?: { [key: string]: string }, measurements?: { [key: string]: number }) => void;

	/** Immediately send all queued telemetry. Synchronous. */
	flush(): void;
}

interface AppInsights {
	loadAppInsights: () => IAppInsightsClient;
}

export class WebTelemetryAppender implements ITelemetryAppender {
	private _aiClient?: IAppInsightsClient;

	constructor(aiKey: string, private _logService: ILogService) {
		const initConfig = {
			config: {
				instrumentationKey: aiKey,
				endpointUrl: 'https://vortex.data.microsoft.com/collect/v1',
				emitLineDelimitedJson: true,
				autoTrackPageVisitTime: false,
				disableExceptionTracking: true,
				disableAjaxTracking: true
			}
		};

		const appInsights = new Microsoft.ApplicationInsights.Initialization(initConfig);
		this._aiClient = appInsights.loadAppInsights();
	}

	log(eventName: string, data: any): void {
		if (!this._aiClient) {
			return;
		}

		data = validateTelemetryData(data);
		this._logService.trace(`telemetry/${eventName}`, data);

		this._aiClient.trackEvent('monacoworkbench/' + eventName, data.properties, data.measurements);
	}

	flush(): Promise<any> | undefined {
		if (this._aiClient) {
			return new Promise(resolve => {
				this._aiClient!.flush();
				this._aiClient = undefined;
				resolve(undefined);
			});
		}

		return undefined;
	}
}

export class TelemetryService extends Disposable implements ITelemetryService {

	_serviceBrand: any;

	private impl: ITelemetryService;

	constructor(
		@IWorkbenchEnvironmentService environmentService: IWorkbenchEnvironmentService,
		@ILogService logService: ILogService,
		@IConfigurationService configurationService: IConfigurationService,
		@IStorageService storageService: IStorageService,
		@IProductService productService: IProductService
	) {
		super();

		const aiKey = productService.aiConfig && productService.aiConfig.asimovKey;
		if (!environmentService.isExtensionDevelopment && !environmentService.args['disable-telemetry'] && !!productService.enableTelemetry && !!aiKey) {
			const config: ITelemetryServiceConfig = {
				appender: combinedAppender(new WebTelemetryAppender(aiKey, logService), new LogAppender(logService)),
				commonProperties: resolveWorkbenchCommonProperties(storageService, productService.commit, productService.version, environmentService.configuration.machineId, environmentService.configuration.remoteAuthority),
				piiPaths: [environmentService.appRoot]
			};

			this.impl = this._register(new BaseTelemetryService(config, configurationService));
		} else {
			this.impl = NullTelemetryService;
		}
	}

	setEnabled(value: boolean): void {
		return this.impl.setEnabled(value);
	}

	get isOptedIn(): boolean {
		return this.impl.isOptedIn;
	}

	publicLog(eventName: string, data?: ITelemetryData, anonymizeFilePaths?: boolean): Promise<void> {
		return this.impl.publicLog(eventName, data, anonymizeFilePaths);
	}

	publicLog2<E extends ClassifiedEvent<T> = never, T extends GDPRClassification<T> = never>(eventName: string, data?: StrictPropertyCheck<T, E>, anonymizeFilePaths?: boolean) {
		return this.publicLog(eventName, data as ITelemetryData, anonymizeFilePaths);
	}

	getTelemetryInfo(): Promise<ITelemetryInfo> {
		return this.impl.getTelemetryInfo();
	}
}

registerSingleton(ITelemetryService, TelemetryService);