first commit

wangzhibo

2026-05-09 53418655c061ac26f66058253f39200aced1ab1d

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
/*! @azure/msal-node v5.1.3 2026-04-16 */
'use strict';
import { AppService } from './ManagedIdentitySources/AppService.mjs';
import { AzureArc } from './ManagedIdentitySources/AzureArc.mjs';
import { CloudShell } from './ManagedIdentitySources/CloudShell.mjs';
import { Imds } from './ManagedIdentitySources/Imds.mjs';
import { ServiceFabric } from './ManagedIdentitySources/ServiceFabric.mjs';
import { createManagedIdentityError } from '../error/ManagedIdentityError.mjs';
import { ManagedIdentitySourceNames } from '../utils/Constants.mjs';
import { MachineLearning } from './ManagedIdentitySources/MachineLearning.mjs';
import { unableToCreateSource } from '../error/ManagedIdentityErrorCodes.mjs';
 
/* 
 * Copyright (c) Microsoft Corporation. All rights reserved. 
 * Licensed under the MIT License. 
 */ 
/* 
 * Class to initialize a managed identity and identify the service. 
 * Original source of code: https://github.com/Azure/azure-sdk-for-net/blob/main/sdk/identity/Azure.Identity/src/ManagedIdentityClient.cs 
 */ 
class ManagedIdentityClient { 
    constructor(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries) { 
        this.logger = logger; 
        this.nodeStorage = nodeStorage; 
        this.networkClient = networkClient; 
        this.cryptoProvider = cryptoProvider; 
        this.disableInternalRetries = disableInternalRetries; 
    } 
    async sendManagedIdentityTokenRequest(managedIdentityRequest, managedIdentityId, fakeAuthority, refreshAccessToken) { 
        if (!ManagedIdentityClient.identitySource) { 
            ManagedIdentityClient.identitySource = 
                this.selectManagedIdentitySource(this.logger, this.nodeStorage, this.networkClient, this.cryptoProvider, this.disableInternalRetries, managedIdentityId); 
        } 
        return ManagedIdentityClient.identitySource.acquireTokenWithManagedIdentity(managedIdentityRequest, managedIdentityId, fakeAuthority, refreshAccessToken); 
    } 
    allEnvironmentVariablesAreDefined(environmentVariables) { 
        return Object.values(environmentVariables).every((environmentVariable) => { 
            return environmentVariable !== undefined; 
        }); 
    } 
    /** 
     * Determine the Managed Identity Source based on available environment variables. This API is consumed by ManagedIdentityApplication's getManagedIdentitySource. 
     * @returns ManagedIdentitySourceNames - The Managed Identity source's name 
     */ 
    getManagedIdentitySource() { 
        ManagedIdentityClient.sourceName = 
            this.allEnvironmentVariablesAreDefined(ServiceFabric.getEnvironmentVariables()) 
                ? ManagedIdentitySourceNames.SERVICE_FABRIC 
                : this.allEnvironmentVariablesAreDefined(AppService.getEnvironmentVariables()) 
                    ? ManagedIdentitySourceNames.APP_SERVICE 
                    : this.allEnvironmentVariablesAreDefined(MachineLearning.getEnvironmentVariables()) 
                        ? ManagedIdentitySourceNames.MACHINE_LEARNING 
                        : this.allEnvironmentVariablesAreDefined(CloudShell.getEnvironmentVariables()) 
                            ? ManagedIdentitySourceNames.CLOUD_SHELL 
                            : this.allEnvironmentVariablesAreDefined(AzureArc.getEnvironmentVariables()) 
                                ? ManagedIdentitySourceNames.AZURE_ARC 
                                : ManagedIdentitySourceNames.DEFAULT_TO_IMDS; 
        return ManagedIdentityClient.sourceName; 
    } 
    /** 
     * Tries to create a managed identity source for all sources 
     * @returns the managed identity Source 
     */ 
    selectManagedIdentitySource(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries, managedIdentityId) { 
        const source = ServiceFabric.tryCreate(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries, managedIdentityId) || 
            AppService.tryCreate(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries) || 
            MachineLearning.tryCreate(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries) || 
            CloudShell.tryCreate(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries, managedIdentityId) || 
            AzureArc.tryCreate(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries, managedIdentityId) || 
            Imds.tryCreate(logger, nodeStorage, networkClient, cryptoProvider, disableInternalRetries); 
        if (!source) { 
            throw createManagedIdentityError(unableToCreateSource); 
        } 
        return source; 
    } 
}
 
export { ManagedIdentityClient };
//# sourceMappingURL=ManagedIdentityClient.mjs.map