mirror of
https://github.com/microsoft/vscode.git
synced 2026-02-21 10:17:25 +00:00
86efdcd2c1
Mostly plumbing... this enables:
```
vscode.authentication.getSession('microsoft', scopes, { issuer: "https://login.microsoftonline.com/common/v2.0" });
```
And the respective API for an auth providers to handle it being passed in.
This props up work in MCP land which needs a way to map an issuer to an auth provider... but I certainly see utility outside of that space.
Fixes https://github.com/microsoft/vscode/issues/248775#issuecomment-2876711396
101 lines
3.9 KiB
TypeScript
101 lines
3.9 KiB
TypeScript
/*---------------------------------------------------------------------------------------------
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
|
* Licensed under the MIT License. See License.txt in the project root for license information.
|
|
*--------------------------------------------------------------------------------------------*/
|
|
|
|
import { Environment, EnvironmentParameters } from '@azure/ms-rest-azure-env';
|
|
import Logger from './logger';
|
|
import { MsalAuthProvider } from './node/authProvider';
|
|
import { UriEventHandler } from './UriEventHandler';
|
|
import { authentication, commands, ExtensionContext, l10n, window, workspace, Disposable, Uri } from 'vscode';
|
|
import { MicrosoftAuthenticationTelemetryReporter, MicrosoftSovereignCloudAuthenticationTelemetryReporter } from './common/telemetryReporter';
|
|
|
|
async function initMicrosoftSovereignCloudAuthProvider(
|
|
context: ExtensionContext,
|
|
uriHandler: UriEventHandler
|
|
): Promise<Disposable | undefined> {
|
|
const environment = workspace.getConfiguration('microsoft-sovereign-cloud').get<string | undefined>('environment');
|
|
let authProviderName: string | undefined;
|
|
if (!environment) {
|
|
return undefined;
|
|
}
|
|
|
|
if (environment === 'custom') {
|
|
const customEnv = workspace.getConfiguration('microsoft-sovereign-cloud').get<EnvironmentParameters>('customEnvironment');
|
|
if (!customEnv) {
|
|
const res = await window.showErrorMessage(l10n.t('You must also specify a custom environment in order to use the custom environment auth provider.'), l10n.t('Open settings'));
|
|
if (res) {
|
|
await commands.executeCommand('workbench.action.openSettingsJson', 'microsoft-sovereign-cloud.customEnvironment');
|
|
}
|
|
return undefined;
|
|
}
|
|
try {
|
|
Environment.add(customEnv);
|
|
} catch (e) {
|
|
const res = await window.showErrorMessage(l10n.t('Error validating custom environment setting: {0}', e.message), l10n.t('Open settings'));
|
|
if (res) {
|
|
await commands.executeCommand('workbench.action.openSettings', 'microsoft-sovereign-cloud.customEnvironment');
|
|
}
|
|
return undefined;
|
|
}
|
|
authProviderName = customEnv.name;
|
|
} else {
|
|
authProviderName = environment;
|
|
}
|
|
|
|
const env = Environment.get(authProviderName);
|
|
if (!env) {
|
|
await window.showErrorMessage(l10n.t('The environment `{0}` is not a valid environment.', authProviderName), l10n.t('Open settings'));
|
|
return undefined;
|
|
}
|
|
|
|
const authProvider = await MsalAuthProvider.create(
|
|
context,
|
|
new MicrosoftSovereignCloudAuthenticationTelemetryReporter(context.extension.packageJSON.aiKey),
|
|
window.createOutputChannel(l10n.t('Microsoft Sovereign Cloud Authentication'), { log: true }),
|
|
uriHandler,
|
|
env
|
|
);
|
|
const disposable = authentication.registerAuthenticationProvider(
|
|
'microsoft-sovereign-cloud',
|
|
authProviderName,
|
|
authProvider,
|
|
{ supportsMultipleAccounts: true }
|
|
);
|
|
context.subscriptions.push(disposable);
|
|
return disposable;
|
|
}
|
|
|
|
export async function activate(context: ExtensionContext, mainTelemetryReporter: MicrosoftAuthenticationTelemetryReporter) {
|
|
const uriHandler = new UriEventHandler();
|
|
context.subscriptions.push(uriHandler);
|
|
const authProvider = await MsalAuthProvider.create(
|
|
context,
|
|
mainTelemetryReporter,
|
|
Logger,
|
|
uriHandler
|
|
);
|
|
context.subscriptions.push(authentication.registerAuthenticationProvider(
|
|
'microsoft',
|
|
'Microsoft',
|
|
authProvider,
|
|
{
|
|
supportsMultipleAccounts: true,
|
|
supportedIssuers: [
|
|
Uri.parse('https://login.microsoftonline.com/*/v2.0')
|
|
]
|
|
}
|
|
));
|
|
|
|
let microsoftSovereignCloudAuthProviderDisposable = await initMicrosoftSovereignCloudAuthProvider(context, uriHandler);
|
|
|
|
context.subscriptions.push(workspace.onDidChangeConfiguration(async e => {
|
|
if (e.affectsConfiguration('microsoft-sovereign-cloud')) {
|
|
microsoftSovereignCloudAuthProviderDisposable?.dispose();
|
|
microsoftSovereignCloudAuthProviderDisposable = await initMicrosoftSovereignCloudAuthProvider(context, uriHandler);
|
|
}
|
|
}));
|
|
}
|
|
|
|
export function deactivate() { }
|