Allow sentinel ext to use shutdown api (#16904)

Author: DonJayamanne

src/platform/common/constants.ts

@@ -25,6 +25,7 @@ export const JVSC_EXTENSION_ID = 'ms-toolsai.jupyter';
 export const DATA_WRANGLER_EXTENSION_ID = 'ms-toolsai.datawrangler';
 export const PROPOSED_API_ALLOWED_PUBLISHERS = ['donjayamanne'];
 export const POWER_TOYS_EXTENSION_ID = 'ms-toolsai.vscode-jupyter-powertoys';
+export const SENTINEL_EXTENSION_ID = 'ms-security.ms-sentinel';
 export const JUPYTER_HUB_EXTENSION_ID = 'ms-toolsai.jupyter-hub';
 export const AppinsightsKey = '0c6ae279ed8443289764825290e4f9e2-1a736e7c-1324-4338-be46-fc2a58ae4d14-7255';
 

src/standalone/api/kernels/kernel.ts

@@ -20,7 +20,8 @@ import {
     JVSC_EXTENSION_ID,
     POWER_TOYS_EXTENSION_ID,
     PROPOSED_API_ALLOWED_PUBLISHERS,
-    PYTHON_LANGUAGE
+    PYTHON_LANGUAGE,
+    SENTINEL_EXTENSION_ID
 } from '../../../platform/common/constants';
 import { ChatMime, generatePythonCodeToInvokeCallback } from '../../../kernels/chat/generator';
 import {
@@ -132,15 +133,15 @@ class WrappedKernelPerExtension extends DisposableBase implements Kernel {
                 return that.onDidReceiveDisplayUpdate.bind(this);
             },
             executeCode: (code: string, token: CancellationToken) => this.executeCode(code, token),
-            get shutdown() {
+            shutdown() {
                 if (
-                    ![JVSC_EXTENSION_ID, POWER_TOYS_EXTENSION_ID].includes(extensionId) &&
+                    ![JVSC_EXTENSION_ID, POWER_TOYS_EXTENSION_ID, SENTINEL_EXTENSION_ID].includes(extensionId) &&
                     !PROPOSED_API_ALLOWED_PUBLISHERS.includes(extensionId.split('.')[0])
                 ) {
                     throw new Error(`Proposed API is not supported for extension ${extensionId}`);
                 }
 
-                return () => that.shutdown();
+                return that.shutdown();
             }
         });
     }