Create Vault commands

Author: devsriMS

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/Commands.RecoveryServices.csproj

@@ -136,6 +136,7 @@
     <Compile Include="PSRecoveryServicesClient\PSRecoveryServicesVMClient.cs" />
     <Compile Include="RecoveryServicesCmdletBase.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />
+    <Compile Include="Service\CreateAzureSiteRecoveryVault.cs" />
     <Compile Include="Service\GetAzureSiteRecoveryStorage.cs" />
     <Compile Include="Service\GetAzureSiteRecoveryStorageMapping.cs" />
     <Compile Include="Service\GetAzureSiteRecoveryVaults.cs" />
@@ -147,7 +148,7 @@
     <Compile Include="Service\GetAzureSiteRecoveryNetwork.cs" />
     <Compile Include="Service\CreateAzureSiteRecoveryRecoveryPlan.cs" />
     <Compile Include="Service\GetAzureSiteRecoveryRecoveryPlanFile.cs" />
-    <Compile Include="Service\GetVaultCredentialsFile.cs" />
+    <Compile Include="Service\GetAzureSiteRecoveryVaultCredential.cs" />
     <Compile Include="Service\RemoveAzureSiteRecoveryRecoveryPlan.cs" />
     <Compile Include="Service\GetAzureSiteRecoveryProtectionEntity.cs" />
     <Compile Include="Service\RestartAzureSiteRecoveryJob.cs" />
@@ -176,9 +177,7 @@
     </ProjectReference>
   </ItemGroup>
   <ItemGroup>
-    <Content Include="lib\Microsoft.Azure.RecoveryServices.dll">
-      <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
-    </Content>
+    <Content Include="lib\Microsoft.Azure.RecoveryServices.dll" />
     <Content Include="lib\Security.Cryptography.dll">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
     </Content>

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/PSRecoveryServicesClient/PSRecoveryServicesCloudServiceClient.cs

@@ -12,7 +12,10 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
+using System;
 using System.Collections.Generic;
+using System.Net;
+using System.Threading.Tasks;
 using Microsoft.WindowsAzure;
 using Microsoft.WindowsAzure.Management.RecoveryServices.Models;
 
@@ -69,5 +72,42 @@ public CloudService GetCloudServiceForVault(string vaultName, string region)
 
             return cloudServiceToReturn;
         }
+
+        /// <summary>
+        /// Method to Either find or create the cloud service.
+        /// </summary>
+        /// <param name="cloudServiceName">name of the cloud service to be created</param>
+        /// <param name="cloudServiceInput">cloud service input to create the service.</param>
+        public void FindOrCreateCloudService(string cloudServiceName, CloudServiceCreateArgs cloudServiceInput)
+        {
+            bool cloudServicePresent = this.DoesCloudServiceExits(cloudServiceName);
+
+            if (!cloudServicePresent)
+            {
+                this.GetRecoveryServicesClient.CloudServices.Create(cloudServiceName, cloudServiceInput);
+            }
+        }
+
+        /// <summary>
+        /// Checks whether a cloud service is present or not.
+        /// </summary>
+        /// <param name="cloudServiceName">name of the cloud service to be created</param>
+        /// <returns>returns true in case the cloud service exits and false otherwise.</returns>
+        private bool DoesCloudServiceExits(string cloudServiceName)
+        {
+            IEnumerable<CloudService> cloudServiceList = this.GetCloudServices();
+            bool cloudServicePresent = false;
+
+            foreach (var cloudService in cloudServiceList)
+            {
+                if (cloudServiceName.Equals(cloudService.Name, StringComparison.InvariantCultureIgnoreCase))
+                {
+                    cloudServicePresent = true;
+                    break;
+                }
+            }
+
+            return cloudServicePresent;
+        }
     }
 }

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/PSRecoveryServicesClient/PSRecoveryServicesVaultClient.cs

@@ -12,9 +12,8 @@
 // limitations under the License.
 // ----------------------------------------------------------------------------------
 
-using System.Threading.Tasks;
 using Microsoft.WindowsAzure;
-using Microsoft.WindowsAzure.Management.SiteRecovery.Models;
+using Microsoft.WindowsAzure.Management.RecoveryServices.Models;
 
 namespace Microsoft.Azure.Commands.RecoveryServices
 {
@@ -24,13 +23,15 @@ namespace Microsoft.Azure.Commands.RecoveryServices
     public partial class PSRecoveryServicesClient
     {
         /// <summary>
-        /// Updates the vault certificate
+        /// Method to create Azure Site Recovery Vault
         /// </summary>
-        /// <param name="args">the certificate update arguments</param>
-        /// <returns>Upload Certificate Response</returns>
-        public async Task<UploadCertificateResponse> UpdateVaultCertificate(CertificateArgs args)
+        /// <param name="cloudServiceName">name of the cloud service</param>
+        /// <param name="vaultName">name of the vault</param>
+        /// <param name="vaultCreateInput">vault creation input object</param>
+        /// <returns>creation response object.</returns>
+        public VaultCreateResponse CreateVault(string cloudServiceName, string vaultName, VaultCreateArgs vaultCreateInput)
         {
-            return await this.GetSiteRecoveryClient().Vaults.UploadCertificateAsync(args, this.GetRequestHeaders(false));
+            return this.GetRecoveryServicesClient.Vaults.Create(cloudServiceName, vaultName, vaultCreateInput);
         }
     }
 }

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/PSRecoveryServicesClient/PSRecoveryServicesVaultExtendedInfoClient.cs

@@ -43,5 +43,15 @@ public OperationResponse CreateExtendedInfo(ResourceExtendedInformationArgs exte
         {
             return this.GetSiteRecoveryClient().VaultExtendedInfo.CreateExtendedInfo(extendedInfoArgs, this.GetRequestHeaders(false));
         }
+
+        /// <summary>
+        /// Updates the vault certificate
+        /// </summary>
+        /// <param name="args">the certificate update arguments</param>
+        /// <returns>Upload Certificate Response</returns>
+        public async Task<UploadCertificateResponse> UpdateVaultCertificate(CertificateArgs args)
+        {
+            return await this.GetSiteRecoveryClient().VaultExtendedInfo.UploadCertificateAsync(args, this.GetRequestHeaders(false));
+        }
     }
 }

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/Service/CreateAzureSiteRecoveryVault.cs

@@ -0,0 +1,95 @@
+// ----------------------------------------------------------------------------------
+//
+// Copyright Microsoft Corporation
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ----------------------------------------------------------------------------------
+
+using System;
+using System.Management.Automation;
+using Microsoft.Azure.Commands.RecoveryServices.SiteRecovery;
+using Microsoft.Azure.Portal.RecoveryServices.Models.Common;
+using Microsoft.WindowsAzure.Management.RecoveryServices.Models;
+using Microsoft.WindowsAzure.Management.SiteRecovery.Models;
+
+namespace Microsoft.Azure.Commands.RecoveryServices
+{
+    /// <summary>
+    /// Used to initiate a vault create operation.
+    /// </summary>
+    [Cmdlet(VerbsCommon.New, "AzureSiteRecoveryVault")]
+    [OutputType(typeof(string))]
+    public class CreateAzureSiteRecoveryVault : RecoveryServicesCmdletBase
+    {
+        #region Parameters
+
+        /// <summary>
+        /// Gets or sets the vault name
+        /// </summary>
+        [Parameter(ParameterSetName = ASRParameterSets.ByParam, HelpMessage = "Vault Name for which the cred file to be generated")]
+        [ValidateNotNullOrEmpty]
+        public string Name { get; set; }
+
+        /// <summary>
+        /// Gets or sets the location of the vault
+        /// </summary>
+        [Parameter(ParameterSetName = ASRParameterSets.ByParam, HelpMessage = "Geo Location Name")]
+        [ValidateNotNullOrEmpty]
+        public string Location { get; set; }
+
+        #endregion
+
+        /// <summary>
+        /// ProcessRecord of the command.
+        /// </summary>
+        public override void ExecuteCmdlet()
+        {
+            try
+            {
+                string cloudServiceName = Utilities.GenerateCloudServiceName(this.Location);
+                byte[] bytes = System.Text.Encoding.UTF8.GetBytes(cloudServiceName);
+                string base64Label = Convert.ToBase64String(bytes);
+
+                CloudServiceCreateArgs cloudServiceCreateArgs = new CloudServiceCreateArgs()
+                {
+                    GeoRegion = this.Location,
+                    Label = base64Label,
+                    Description = base64Label
+                };
+
+                RecoveryServicesClient.FindOrCreateCloudService(cloudServiceName, cloudServiceCreateArgs);
+
+                VaultCreateArgs vaultCreateArgs = new VaultCreateArgs()
+                {
+                    Name = this.Name,
+                    Plan = string.Empty,
+                    ResourceProviderNamespace = "WAHyperVRecoveryManager", // TODO:devsri - do not hard code, find a good place to keep it.
+                    Type = Constants.ASRVaulType,
+                    ETag = Guid.NewGuid().ToString(),
+                    SchemaVersion = Constants.RpSchemaVersion
+                };
+
+                Utilities.UpdateVaultSettings(new ASRVaultCreds()
+                {
+                    CloudServiceName = cloudServiceName,
+                    ResourceName = this.Name
+                });
+
+                VaultCreateResponse response = RecoveryServicesClient.CreateVault(cloudServiceName, this.Name, vaultCreateArgs);
+
+                this.WriteObject(response.RequestId, true);
+            }
+            catch (Exception exception)
+            {
+                this.HandleException(exception);
+            }
+        }
+    }
+}

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/Service/GetVaultCredentialsFile.cs renamed to src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/Service/GetAzureSiteRecoveryVaultCredential.cs

@@ -170,6 +170,21 @@ public class Constants
         /// A valid value for the string field Microsoft.WindowsAzure.CloudServiceManagement.resource.OperationStatus.InProgress
         /// </summary>
         public const string RdfeOperationStatusResultInProgress = "InProgress";
+
+        /// <summary>
+        /// Cloud service name prefix
+        /// </summary>
+        public const string CloudServiceNameExtensionPrefix = "CS-";
+
+        /// <summary>
+        /// Cloud service name suffix
+        /// </summary>
+        public const string CloudServiceNameExtensionSuffix = "-RecoveryServices";
+
+        /// <summary>
+        /// Schema Version of RP
+        /// </summary>
+        public const string RpSchemaVersion = "1.1";
     }
 
     /// <summary>

src/ServiceManagement/RecoveryServices/Commands.RecoveryServices/lib/Microsoft.Azure.RecoveryServices.dll

@@ -13,9 +13,11 @@
 // ----------------------------------------------------------------------------------
 
 using System;
+using System.Globalization;
 using System.IO;
 using System.Runtime.Serialization;
 using System.Security.Cryptography;
+using System.Text;
 using Microsoft.Azure.Portal.RecoveryServices.Models.Common;
 
 namespace Microsoft.Azure.Commands.RecoveryServices
@@ -138,5 +140,20 @@ public static string GenerateRandomKey(int size)
             crypto.GetBytes(key);
             return Convert.ToBase64String(key);
         }
+
+        /// <summary>
+        /// Method to generate the cloud service name 
+        /// </summary>
+        /// <param name="region">region name</param>
+        /// <returns>cloud service name as string</returns>
+        public static string GenerateCloudServiceName(string region)
+        {
+            return string.Format(
+                CultureInfo.InvariantCulture,
+                "{0}{1}{2}",
+                Constants.CloudServiceNameExtensionPrefix,
+                region.Replace(' ', '-'),
+                Constants.CloudServiceNameExtensionSuffix);       
+        }
     }
 }