Update Set-AzureRmOSDisk cmdlet

Author: hyonholee

src/ResourceManager/Compute/Commands.Compute/Properties/Resources.Designer.cs

@@ -452,4 +452,7 @@ The file needs to be a PowerShell script (.ps1 or .psm1) or a ZIP archive (.zip)
     <value>Since the VM is created using premium storage, existing standard storage account, {0}, is used for boot diagnostics.</value>
     <comment>{0} = existing standard storage account name</comment>
   </data>
+  <data name="VMOSDiskDiskEncryptionBothKekVaultIdAndKekUrlRequired" xml:space="preserve">
+    <value>You have to specify either both of KeyEncryptionKeyVaultId and KeyEncryptionKeyUrl or none of them.</value>
+  </data>
 </root>

src/ResourceManager/Compute/Commands.Compute/Properties/Resources.resx

@@ -124,7 +124,7 @@ public class SetAzureVMOSDiskCommand : Microsoft.Azure.Commands.ResourceManager.
             Position = 7,
             ValueFromPipelineByPropertyName = true,
             HelpMessage = HelpMessages.VMOSDiskLinuxOSType)]
-        public Uri DiskEncryptionKeyUrl { get; set; }
+        public string DiskEncryptionKeyUrl { get; set; }
 
         [Parameter(
             ParameterSetName = WindowsDiskEncryptionParameterSet,
@@ -152,7 +152,7 @@ public class SetAzureVMOSDiskCommand : Microsoft.Azure.Commands.ResourceManager.
             Position = 9,
             ValueFromPipelineByPropertyName = true,
             HelpMessage = HelpMessages.VMOSDiskLinuxOSType)]
-        public Uri KeyEncryptionKeyUrl { get; set; }
+        public string KeyEncryptionKeyUrl { get; set; }
 
         [Parameter(
             ParameterSetName = WindowsDiskEncryptionParameterSet,
@@ -175,6 +175,14 @@ protected override void ProcessRecord()
                 this.VM.StorageProfile = new StorageProfile();
             }
 
+            if ((this.KeyEncryptionKeyVaultId == null && this.KeyEncryptionKeyUrl != null)
+                || (this.KeyEncryptionKeyVaultId != null && this.KeyEncryptionKeyUrl == null))
+            {
+                WriteError(new ErrorRecord(
+                        new Exception(Properties.Resources.VMOSDiskDiskEncryptionBothKekVaultIdAndKekUrlRequired),
+                        string.Empty, ErrorCategory.InvalidArgument, null));
+            }
+
             this.VM.StorageProfile.OSDisk = new OSDisk
             {
                 Caching = this.Caching,