private static void InitVMIdentity(VirtualMachineInner vmInner, ResourceIdentityType identityType) { if (!identityType.Equals(ResourceIdentityType.UserAssigned) && !identityType.Equals(ResourceIdentityType.SystemAssigned)) { throw new ArgumentException("Invalid argument: " + identityType); } if (vmInner.Identity == null) { vmInner.Identity = new VirtualMachineIdentity(); } ResourceIdentityType?parsedIdentityType = ResourceIdentityTypeEnumExtension.ParseResourceIdentityType(vmInner.Identity.Type); if (parsedIdentityType == null || parsedIdentityType.Equals(ResourceIdentityType.None) || parsedIdentityType.Equals(identityType)) { vmInner.Identity.Type = ResourceIdentityTypeEnumExtension.ToSerializedValue(identityType); } else { vmInner.Identity.Type = ResourceIdentityTypeEnumExtension.ToSerializedValue(ResourceIdentityType.SystemAssignedUserAssigned); } if (vmInner.Identity.IdentityIds == null) { if (identityType.Equals(ResourceIdentityType.UserAssigned) || identityType.Equals(ResourceIdentityType.SystemAssignedUserAssigned)) { vmInner.Identity.IdentityIds = new List <string>(); } } }
/// <summary> /// Gets the MSI identity type. /// </summary> /// <param name="inner">the virtual machine inner</param> /// <returns>the MSI identity type</returns> internal static ResourceIdentityType?ManagedServiceIdentityType(VirtualMachineInner inner) { if (inner.Identity != null) { return(ResourceIdentityTypeEnumExtension.ParseResourceIdentityType(inner.Identity.Type)); } return(null); }
/// <summary> /// Gets the MSI identity type. /// </summary> /// <param name="inner">the virtual machine scale set inner</param> /// <returns>the MSI identity type</returns> internal static ResourceIdentityType?ManagedServiceIdentityType(VirtualMachineScaleSetInner inner) { if (inner.Identity != null) { //ResourceIdentityTypeEnumExtension.ToSerializedValue(ResourceIdentityType.SystemAssigned); return(ResourceIdentityTypeEnumExtension.ParseResourceIdentityType(inner.Identity.Type)); } return(null); }
/// <summary> /// Set user assigned identity ids to the given virtual machine inner model /// </summary> /// <param name="virtualMachineInner">the virtual machine inner model</param> /// <param name="vmTaskGroup">the virtual machine task group</param> internal void HandleUserAssignedIdentities(VirtualMachineInner virtualMachineInner, CreatorTaskGroup <IHasId> vmTaskGroup) { try { if (virtualMachineInner.Identity == null || virtualMachineInner.Identity.Type == null) { return; } ResourceIdentityType?parsedIdentityType = ResourceIdentityTypeEnumExtension.ParseResourceIdentityType(virtualMachineInner.Identity.Type); if (parsedIdentityType.Equals(ResourceIdentityType.None) || parsedIdentityType.Equals(ResourceIdentityType.SystemAssigned)) { return; } foreach (var key in this.userAssignedIdentityCreatableKeys) { var identity = (IIdentity)vmTaskGroup.CreatedResource(key); if (!this.userAssignedIdentityIdsToAssociate.Contains(identity.Id)) { this.userAssignedIdentityIdsToAssociate.Add(identity.Id); } } if (virtualMachineInner.Identity.IdentityIds == null) { virtualMachineInner.Identity.IdentityIds = new List <string>(); } foreach (var identityId in this.userAssignedIdentityIdsToAssociate) { if (!virtualMachineInner.Identity.IdentityIds.Contains(identityId)) { virtualMachineInner.Identity.IdentityIds.Add(identityId); } } foreach (var identityId in this.userAssignedIdentityIdsToRemove) { if (virtualMachineInner.Identity.IdentityIds.Contains(identityId)) { virtualMachineInner.Identity.IdentityIds.Remove(identityId); } } if (virtualMachineInner.Identity.IdentityIds.Any()) { this.installExtensionIfNotInstalled = true; } } finally { this.userAssignedIdentityCreatableKeys.Clear(); this.userAssignedIdentityIdsToAssociate.Clear(); this.userAssignedIdentityIdsToRemove.Clear(); } }
/// <summary> /// Checks system assigned identity is enabled for the virtual machine. /// </summary> /// <param name="virtualMachine">the virtual machine</param> /// <returns>true if system assigned MSI is enabled, false otherwise</returns> private static bool IsSystemAssignedMSIEnabled(IVirtualMachine virtualMachine) { VirtualMachineInner VMInner = virtualMachine.Inner; if (VMInner.Identity == null) { return(false); } ResourceIdentityType?parsedIdentityType = ResourceIdentityTypeEnumExtension.ParseResourceIdentityType(VMInner.Identity.Type); if (parsedIdentityType == null || parsedIdentityType.Equals(ResourceIdentityType.None)) { return(false); } else { return(parsedIdentityType.Equals(ResourceIdentityType.SystemAssigned) || parsedIdentityType.Equals(ResourceIdentityType.SystemAssignedUserAssigned)); } }