debuggerDataViewer experiment (#14530)

joyceerhl · web-flow · commit ad0f632e248b · 2020-10-26T15:11:43.000-07:00
diff --git a/src/client/common/experiments/debuggerDataViewerExperimentEnabler.ts b/src/client/common/experiments/debuggerDataViewerExperimentEnabler.ts
@@ -0,0 +1,26 @@
+import { inject, injectable } from 'inversify';
+import { IExtensionSingleActivationService } from '../../activation/types';
+import { ICommandManager } from '../../common/application/types';
+import { ContextKey } from '../../common/contextKey';
+import { traceError } from '../../common/logger';
+import { IExperimentService } from '../../common/types';
+
+@injectable()
+export class DebuggerDataViewerExperimentEnabler implements IExtensionSingleActivationService {
+    constructor(
+        @inject(ICommandManager) private readonly commandManager: ICommandManager,
+        @inject(IExperimentService) private readonly experimentService: IExperimentService
+    ) {}
+    public async activate() {
+        this.activateInternal().catch(traceError.bind('Failed to activate debuggerDataViewerExperimentEnabler'));
+    }
+    private async activateInternal() {
+        // This context key controls the visibility of the 'View Variable in Data Viewer'
+        // context menu item from the variable window context menu during a debugging session
+        const isDataViewerExperimentEnabled = new ContextKey(
+            'python.isDebuggerDataViewerExperimentEnabled',
+            this.commandManager
+        );
+        await isDataViewerExperimentEnabled.set(await this.experimentService.inExperiment('debuggerDataViewer'));
+    }
+}
diff --git a/src/client/common/serviceRegistry.ts b/src/client/common/serviceRegistry.ts
@@ -38,6 +38,7 @@ import { AsyncDisposableRegistry } from './asyncDisposableRegistry';
 import { ConfigurationService } from './configuration/service';
 import { CryptoUtils } from './crypto';
 import { EditorUtils } from './editor';
+import { DebuggerDataViewerExperimentEnabler } from './experiments/debuggerDataViewerExperimentEnabler';
 import { ExperimentsManager } from './experiments/manager';
 import { ExperimentService } from './experiments/service';
 import { FeatureDeprecationManager } from './featureDeprecationManager';
@@ -200,6 +201,10 @@ export function registerTypes(serviceManager: IServiceManager) {
         IExtensionSingleActivationService,
         ReloadVSCodeCommandHandler
     );
+    serviceManager.addSingleton<IExtensionSingleActivationService>(
+        IExtensionSingleActivationService,
+        DebuggerDataViewerExperimentEnabler
+    );
     serviceManager.addSingleton<IExtensionChannelService>(IExtensionChannelService, ExtensionChannelService);
     serviceManager.addSingleton<IExtensionChannelRule>(
         IExtensionChannelRule,
