Native Azure Pulumi provider with Microsoft Azure SDK

// Copyright 2016-2021, Pulumi Corporation.  All rights reserved.

import * as authorization from "@pulumi/azure-native/authorization";
import * as containerregistry from "@pulumi/azure-native/containerregistry";
import * as resources from "@pulumi/azure-native/resources";
import * as pulumi from "@pulumi/pulumi";

import { AuthorizationManagementClient } from "@azure/arm-authorization";
import { TokenCredentials } from "@azure/ms-rest-js";

async function getAuthorizationManagementClient(): Promise<AuthorizationManagementClient> {
    const config = await authorization.getClientConfig();
    const token = await authorization.getClientToken();
    const credentials = new TokenCredentials(token.token);
    // Note: reuse the credentials and/or the client in case your scenario needs
    // multiple calls to Azure SDKs.
    return new AuthorizationManagementClient(credentials, config.subscriptionId);
}

async function getRoleIdByName(roleName: string, scope?: string): Promise<string> {
    const client = await getAuthorizationManagementClient();
    const roles = await client.roleDefinitions.list(
        scope || "",
        {
            filter: `roleName eq '${roleName}'`,
        },
    );
    if (roles.length === 0) {
        throw new Error(`role "${roleName}" not found at scope "${scope}"`);
    }
    if (roles.length > 1) {
        throw new Error(`too many roles "${roleName}" found at scope "${scope}". Found: ${roles.length}`);
    }
    const role = roles[0];
    return role.id!;
}

const resourceGroup = new resources.ResourceGroup("registryrg");

const registry = new containerregistry.Registry("registry", {
    resourceGroupName: resourceGroup.name,
    sku: {
        name: "Basic",
    },
    adminUserEnabled: true,
});

const currentServicePrincipalId = pulumi.output(authorization.getClientConfig()).objectId;

const grantPull = new authorization.RoleAssignment("access-from-cluster", {
    principalId: currentServicePrincipalId,
    principalType: authorization.PrincipalType.ServicePrincipal, // adjust the type if you are running as a user
    roleDefinitionId: getRoleIdByName("AcrPull"),
    scope: registry.id,
});