Support activating the extension with limited support in untrusted workspaces

Author: JoeRobich

l10n/bundle.l10n.json

@@ -1,4 +1,6 @@
 {
+  "Reload Window": "Reload Window",
+  "Workspace trust has changed. Would you like to reload the window to activate the C# extension?": "Workspace trust has changed. Would you like to reload the window to activate the C# extension?",
   "How to setup Remote Debugging": "How to setup Remote Debugging",
   "The C# extension for Visual Studio Code is incompatible on {0} {1} with the VS Code Remote Extensions. To see avaliable workarounds, click on '{2}'.": "The C# extension for Visual Studio Code is incompatible on {0} {1} with the VS Code Remote Extensions. To see avaliable workarounds, click on '{2}'.",
   "The C# extension for Visual Studio Code is incompatible on {0} {1}.": "The C# extension for Visual Studio Code is incompatible on {0} {1}.",
@@ -146,7 +148,6 @@
   "OmniSharp requires a complete install of Mono (including MSBuild) to provide language services when `omnisharp.useModernNet` is disabled in Settings. Please install the latest Mono and restart.": "OmniSharp requires a complete install of Mono (including MSBuild) to provide language services when `omnisharp.useModernNet` is disabled in Settings. Please install the latest Mono and restart.",
   "Download Mono": "Download Mono",
   "Open settings": "Open settings",
-  "Reload Window": "Reload Window",
   "dotnet.server.useOmnisharp option has changed. Please reload the window to apply the change": "dotnet.server.useOmnisharp option has changed. Please reload the window to apply the change",
   "C# configuration has changed. Would you like to relaunch the Language Server with your changes?": "C# configuration has changed. Would you like to relaunch the Language Server with your changes?",
   "Restart Language Server": "Restart Language Server",

package.json

@@ -33,7 +33,7 @@
   "capabilities": {
     "virtualWorkspaces": false,
     "untrustedWorkspaces": {
-      "supported": false
+      "supported": "limited"
     }
   },
   "extensionKind": [

src/activateOmniSharp.ts

@@ -58,6 +58,7 @@ export function activateOmniSharp(
     const coreClrDebugPromise = getCoreClrDebugPromise(omnisharpLangServicePromise);
 
     const exports: OmnisharpExtensionExports = {
+        isLimitedActivation: false,
         initializationFinished: async () => {
             const langService = await omnisharpLangServicePromise;
             await langService!.server.waitForInitialize();

src/activateRoslyn.ts

@@ -87,6 +87,7 @@ export function activateRoslyn(
 
     const languageServerExport = new RoslynLanguageServerExport(roslynLanguageServerStartedPromise);
     const exports: CSharpExtensionExports = {
+        isLimitedActivation: false,
         initializationFinished: async () => {
             await coreClrDebugPromise;
             await razorLanguageServerStartedPromise;

src/csharpExtensionExports.ts

@@ -11,7 +11,12 @@ import { GlobalBrokeredServiceContainer } from '@microsoft/servicehub-framework'
 import { PartialResultParams, ProtocolRequestType, RequestType } from 'vscode-languageclient/node';
 import { LanguageServerEvents } from './lsptoolshost/server/languageServerEvents';
 
+export interface LimitedExtensionExports {
+    isLimitedActivation: true;
+}
+
 export interface OmnisharpExtensionExports {
+    isLimitedActivation: false;
     initializationFinished: () => Promise<void>;
     getAdvisor: () => Promise<Advisor>;
     getTestManager: () => Promise<TestManager>;
@@ -20,6 +25,7 @@ export interface OmnisharpExtensionExports {
 }
 
 export interface CSharpExtensionExports {
+    isLimitedActivation: false;
     initializationFinished: () => Promise<void>;
     logDirectory: string;
     profferBrokeredServices: (container: GlobalBrokeredServiceContainer) => void;

src/main.ts

@@ -21,18 +21,19 @@ import IInstallDependencies from './packageManager/IInstallDependencies';
 import { installRuntimeDependencies } from './installRuntimeDependencies';
 import { isValidDownload } from './packageManager/isValidDownload';
 import { MigrateOptions } from './shared/migrateOptions';
-import { CSharpExtensionExports, OmnisharpExtensionExports } from './csharpExtensionExports';
+import { CSharpExtensionExports, LimitedExtensionExports, OmnisharpExtensionExports } from './csharpExtensionExports';
 import { getCSharpDevKit } from './utils/getCSharpDevKit';
 import { commonOptions, omnisharpOptions } from './shared/options';
 import { TelemetryEventNames } from './shared/telemetryEventNames';
 import { checkDotNetRuntimeExtensionVersion } from './checkDotNetRuntimeExtensionVersion';
 import { checkIsSupportedPlatform } from './checkSupportedPlatform';
 import { activateOmniSharp } from './activateOmniSharp';
 import { activateRoslyn } from './activateRoslyn';
+import { CommandOption, showInformationMessage } from './shared/observers/utils/showMessage';
 
 export async function activate(
     context: vscode.ExtensionContext
-): Promise<CSharpExtensionExports | OmnisharpExtensionExports | null> {
+): Promise<CSharpExtensionExports | OmnisharpExtensionExports | LimitedExtensionExports | null> {
     // Start measuring the activation time
     const startActivation = process.hrtime();
 
@@ -110,8 +111,25 @@ export async function activate(
         return coreClrDebugPromise;
     };
 
-    let exports: CSharpExtensionExports | OmnisharpExtensionExports;
-    if (!useOmnisharpServer) {
+    let activationEvent = TelemetryEventNames.CSharpActivated;
+    let exports: CSharpExtensionExports | OmnisharpExtensionExports | LimitedExtensionExports;
+    if (vscode.workspace.isTrusted !== true) {
+        activationEvent = TelemetryEventNames.CSharpLimitedActivation;
+        exports = { isLimitedActivation: true };
+        csharpChannel.trace('C# Extension activated in limited mode due to workspace trust not being granted.');
+        context.subscriptions.push(
+            vscode.workspace.onDidGrantWorkspaceTrust(() => {
+                const reloadTitle: CommandOption = {
+                    title: vscode.l10n.t('Reload Window'),
+                    command: 'workbench.action.reloadWindow',
+                };
+                const message = vscode.l10n.t(
+                    'Workspace trust has changed. Would you like to reload the window to activate the C# extension?'
+                );
+                showInformationMessage(vscode, message, reloadTitle);
+            })
+        );
+    } else if (!useOmnisharpServer) {
         exports = activateRoslyn(
             context,
             platformInfo,
@@ -142,7 +160,7 @@ export async function activate(
         serverKind: useOmnisharpServer ? 'OmniSharp' : 'Roslyn',
         timeTaken: timeTakenStr,
     };
-    reporter.sendTelemetryEvent(TelemetryEventNames.CSharpActivated, activationProperties);
+    reporter.sendTelemetryEvent(activationEvent, activationProperties);
 
     return exports;
 }

src/shared/telemetryEventNames.ts

@@ -9,6 +9,7 @@
 export enum TelemetryEventNames {
     // Common extension events
     CSharpActivated = 'CSharpActivated',
+    CSharpLimitedActivation = 'CSharpLimitedActivation',
 
     // Events related to the roslyn language server.