mirror of
https://github.com/tvytlx/ai-agent-deep-dive.git
synced 2026-04-04 16:14:50 +08:00
104 lines
5.9 KiB
JavaScript
104 lines
5.9 KiB
JavaScript
/*! @azure/msal-node v3.8.1 2025-10-29 */
|
|
'use strict';
|
|
import { ManagedIdentityRequestParameters } from '../../config/ManagedIdentityRequestParameters.mjs';
|
|
import { BaseManagedIdentitySource } from './BaseManagedIdentitySource.mjs';
|
|
import { ManagedIdentityEnvironmentVariableNames, ManagedIdentitySourceNames, ManagedIdentityIdType, ManagedIdentityHeaders, ManagedIdentityQueryParameters, HttpMethod } from '../../utils/Constants.mjs';
|
|
import { createManagedIdentityError } from '../../error/ManagedIdentityError.mjs';
|
|
import { unableToCreateCloudShell } from '../../error/ManagedIdentityErrorCodes.mjs';
|
|
|
|
/*
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
|
* Licensed under the MIT License.
|
|
*/
|
|
/**
|
|
* Azure Cloud Shell managed identity source implementation.
|
|
*
|
|
* This class handles authentication for applications running in Azure Cloud Shell environment.
|
|
* Cloud Shell provides a browser-accessible shell for managing Azure resources and includes
|
|
* a pre-configured managed identity for authentication.
|
|
*
|
|
* Original source of code: https://github.com/Azure/azure-sdk-for-net/blob/main/sdk/identity/Azure.Identity/src/CloudShellManagedIdentitySource.cs
|
|
*/
|
|
class CloudShell extends BaseManagedIdentitySource {
|
|
/**
|
|
* Creates a new CloudShell managed identity source instance.
|
|
*
|
|
* @param logger - Logger instance for diagnostic logging
|
|
* @param nodeStorage - Node.js storage implementation for caching
|
|
* @param networkClient - HTTP client for making requests to the managed identity endpoint
|
|
* @param cryptoProvider - Cryptographic operations provider
|
|
* @param disableInternalRetries - Whether to disable automatic retry logic for failed requests
|
|
* @param msiEndpoint - The MSI endpoint URL obtained from environment variables
|
|
*/
|
|
constructor(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries, msiEndpoint) {
|
|
super(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries);
|
|
this.msiEndpoint = msiEndpoint;
|
|
}
|
|
/**
|
|
* Retrieves the required environment variables for Cloud Shell managed identity.
|
|
*
|
|
* Cloud Shell requires the MSI_ENDPOINT environment variable to be set, which
|
|
* contains the URL of the managed identity service endpoint.
|
|
*
|
|
* @returns An array containing the MSI_ENDPOINT environment variable value (or undefined if not set)
|
|
*/
|
|
static getEnvironmentVariables() {
|
|
const msiEndpoint = process.env[ManagedIdentityEnvironmentVariableNames.MSI_ENDPOINT];
|
|
return [msiEndpoint];
|
|
}
|
|
/**
|
|
* Attempts to create a CloudShell managed identity source instance.
|
|
*
|
|
* This method validates that the required environment variables are present and
|
|
* creates a CloudShell instance if the environment is properly configured.
|
|
* Cloud Shell only supports system-assigned managed identities.
|
|
*
|
|
* @param logger - Logger instance for diagnostic logging
|
|
* @param nodeStorage - Node.js storage implementation for caching
|
|
* @param networkClient - HTTP client for making requests
|
|
* @param cryptoProvider - Cryptographic operations provider
|
|
* @param disableInternalRetries - Whether to disable automatic retry logic
|
|
* @param managedIdentityId - The managed identity configuration (must be system-assigned)
|
|
*
|
|
* @returns A CloudShell instance if the environment is valid, null otherwise
|
|
*
|
|
* @throws {ManagedIdentityError} When a user-assigned managed identity is requested,
|
|
* as Cloud Shell only supports system-assigned identities
|
|
*/
|
|
static tryCreate(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries, managedIdentityId) {
|
|
const [msiEndpoint] = CloudShell.getEnvironmentVariables();
|
|
// if the msi endpoint environment variable is undefined, this MSI provider is unavailable.
|
|
if (!msiEndpoint) {
|
|
logger.info(`[Managed Identity] ${ManagedIdentitySourceNames.CLOUD_SHELL} managed identity is unavailable because the '${ManagedIdentityEnvironmentVariableNames.MSI_ENDPOINT} environment variable is not defined.`);
|
|
return null;
|
|
}
|
|
const validatedMsiEndpoint = CloudShell.getValidatedEnvVariableUrlString(ManagedIdentityEnvironmentVariableNames.MSI_ENDPOINT, msiEndpoint, ManagedIdentitySourceNames.CLOUD_SHELL, logger);
|
|
logger.info(`[Managed Identity] Environment variable validation passed for ${ManagedIdentitySourceNames.CLOUD_SHELL} managed identity. Endpoint URI: ${validatedMsiEndpoint}. Creating ${ManagedIdentitySourceNames.CLOUD_SHELL} managed identity.`);
|
|
if (managedIdentityId.idType !== ManagedIdentityIdType.SYSTEM_ASSIGNED) {
|
|
throw createManagedIdentityError(unableToCreateCloudShell);
|
|
}
|
|
return new CloudShell(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries, msiEndpoint);
|
|
}
|
|
/**
|
|
* Creates an HTTP request to acquire an access token from the Cloud Shell managed identity endpoint.
|
|
*
|
|
* This method constructs a POST request to the MSI endpoint with the required headers and
|
|
* body parameters for Cloud Shell authentication. The request includes the target resource
|
|
* for which the access token is being requested.
|
|
*
|
|
* @param resource - The target resource/scope for which to request an access token (e.g., "https://graph.microsoft.com/.default")
|
|
*
|
|
* @returns A configured ManagedIdentityRequestParameters object ready for network execution
|
|
*/
|
|
createRequest(resource) {
|
|
const request = new ManagedIdentityRequestParameters(HttpMethod.POST, this.msiEndpoint);
|
|
request.headers[ManagedIdentityHeaders.METADATA_HEADER_NAME] = "true";
|
|
request.bodyParameters[ManagedIdentityQueryParameters.RESOURCE] =
|
|
resource;
|
|
return request;
|
|
}
|
|
}
|
|
|
|
export { CloudShell };
|
|
//# sourceMappingURL=CloudShell.mjs.map
|