public static void SetDefaultStorage()
        {
            if (!string.IsNullOrEmpty(GetDefaultStorage(CredentialHelper.DefaultStorageName, CredentialHelper.Location)))
            {
                defaultAzureSubscription = vmPowershellCmdlets.SetAzureSubscription(defaultAzureSubscription.SubscriptionName, CredentialHelper.DefaultStorageName);

                storageAccountKey = vmPowershellCmdlets.GetAzureStorageAccountKey(defaultAzureSubscription.CurrentStorageAccountName);
                Assert.AreEqual(defaultAzureSubscription.CurrentStorageAccountName, storageAccountKey.StorageAccountName);
                blobUrlRoot = (vmPowershellCmdlets.GetAzureStorageAccount(defaultAzureSubscription.CurrentStorageAccountName)[0].Endpoints.ToArray())[0];
            }
            else
            {
                Console.WriteLine("Unable to get the default storege account");
            }
        }
        public void GetStorageKeysProcess()
        {
            Func <string, StorageService>     func = null;
            StorageServiceKeyOperationContext storageServiceKeyOperationContext = null;

            using (OperationContextScope operationContextScope = new OperationContextScope((IContextChannel)base.Channel))
            {
                try
                {
                    GetAzureStorageKeyCommand getAzureStorageKeyCommand = this;
                    if (func == null)
                    {
                        func = (string s) => base.Channel.GetStorageKeys(s, this.StorageAccountName);
                    }
                    StorageService storageService = ((CmdletBase <IServiceManagement>)getAzureStorageKeyCommand).RetryCall <StorageService>(func);
                    if (storageService != null)
                    {
                        StorageServiceKeyOperationContext storageAccountName = new StorageServiceKeyOperationContext();
                        storageAccountName.StorageAccountName = this.StorageAccountName;
                        storageAccountName.Primary            = storageService.StorageServiceKeys.Primary;
                        storageAccountName.Secondary          = storageService.StorageServiceKeys.Secondary;
                        storageServiceKeyOperationContext     = storageAccountName;
                    }
                    Operation operation = base.WaitForOperation(base.CommandRuntime.ToString());
                    storageServiceKeyOperationContext.set_OperationDescription(base.CommandRuntime.ToString());
                    storageServiceKeyOperationContext.set_OperationId(operation.OperationTrackingId);
                    storageServiceKeyOperationContext.set_OperationStatus(operation.Status);
                    base.WriteObject(storageServiceKeyOperationContext, true);
                }
                catch (CommunicationException communicationException1)
                {
                    CommunicationException communicationException = communicationException1;
                    if (communicationException as EndpointNotFoundException == null || base.IsVerbose())
                    {
                        this.WriteErrorDetails(communicationException);
                    }
                    else
                    {
                        base.WriteObject(null);
                        return;
                    }
                }
            }
        }
Exemple #3
0
        protected static void DownloadVhds()
        {
            storageAccountKey = vmPowershellCmdlets.GetAzureStorageAccountKey(defaultAzureSubscription.CurrentStorageAccountName);

            foreach (var vhdFile in VhdFiles)
            {
                string vhdBlobLocation = string.Format("{0}{1}/{2}", blobUrlRoot, VhdFilesContainerName, vhdFile);

                var vhdLocalPath = new FileInfo(Directory.GetCurrentDirectory() + "\\" + vhdFile);

                if (!File.Exists(vhdLocalPath.FullName))
                {
                    // Set the source blob
                    BlobHandle blobHandle = Utilities.GetBlobHandle(vhdBlobLocation, storageAccountKey.Primary);

                    SaveVhd(blobHandle, vhdLocalPath, storageAccountKey.Primary);
                }
            }
        }
Exemple #4
0
        public void NewStorageKeyProcess()
        {
            Func <string, StorageService>     func = null;
            StorageServiceKeyOperationContext storageServiceKeyOperationContext = null;
            RegenerateKeys regenerateKey = new RegenerateKeys();

            regenerateKey.KeyType = this.KeyType;
            RegenerateKeys regenerateKey1 = regenerateKey;

            using (OperationContextScope operationContextScope = new OperationContextScope((IContextChannel)base.Channel))
            {
                try
                {
                    CmdletExtensions.WriteVerboseOutputForObject(this, regenerateKey1);
                    NewAzureStorageKeyCommand newAzureStorageKeyCommand = this;
                    if (func == null)
                    {
                        func = (string s) => base.Channel.RegenerateStorageServiceKeys(s, this.StorageAccountName, regenerateKey1);
                    }
                    StorageService storageService = ((CmdletBase <IServiceManagement>)newAzureStorageKeyCommand).RetryCall <StorageService>(func);
                    if (storageService != null)
                    {
                        StorageServiceKeyOperationContext storageAccountName = new StorageServiceKeyOperationContext();
                        storageAccountName.StorageAccountName = this.StorageAccountName;
                        storageAccountName.Primary            = storageService.StorageServiceKeys.Primary;
                        storageAccountName.Secondary          = storageService.StorageServiceKeys.Secondary;
                        storageServiceKeyOperationContext     = storageAccountName;
                    }
                    Operation operation = base.WaitForOperation(base.CommandRuntime.ToString());
                    storageServiceKeyOperationContext.set_OperationDescription(base.CommandRuntime.ToString());
                    storageServiceKeyOperationContext.set_OperationId(operation.OperationTrackingId);
                    storageServiceKeyOperationContext.set_OperationStatus(operation.Status);
                    base.WriteObject(storageServiceKeyOperationContext, true);
                }
                catch (CommunicationException communicationException1)
                {
                    CommunicationException communicationException = communicationException1;
                    this.WriteErrorDetails(communicationException);
                }
            }
        }
        public static void SetDefaultStorage()
        {
            if (!string.IsNullOrEmpty(GetDefaultStorage(CredentialHelper.DefaultStorageName, CredentialHelper.Location)))
            {
                defaultAzureSubscription = vmPowershellCmdlets.SetAzureSubscription(defaultAzureSubscription.SubscriptionName, CredentialHelper.DefaultStorageName);

                storageAccountKey = vmPowershellCmdlets.GetAzureStorageAccountKey(defaultAzureSubscription.CurrentStorageAccountName);
                Assert.AreEqual(defaultAzureSubscription.CurrentStorageAccountName, storageAccountKey.StorageAccountName);
                blobUrlRoot = (vmPowershellCmdlets.GetAzureStorageAccount(defaultAzureSubscription.CurrentStorageAccountName)[0].Endpoints.ToArray())[0];
            }
            else
            {
                Console.WriteLine("Unable to get the default storege account");
            }
        }
        protected static void DownloadVhds()
        {
            storageAccountKey = vmPowershellCmdlets.GetAzureStorageAccountKey(defaultAzureSubscription.CurrentStorageAccountName);

            foreach (var vhdFile in VhdFiles)
            {
                string vhdBlobLocation = string.Format("{0}{1}/{2}", blobUrlRoot, VhdFilesContainerName, vhdFile);

                var vhdLocalPath = new FileInfo(Directory.GetCurrentDirectory() + "\\" + vhdFile);

                if (!File.Exists(vhdLocalPath.FullName))
                {
                    // Set the source blob
                    BlobHandle blobHandle = Utilities.GetBlobHandle(vhdBlobLocation, storageAccountKey.Primary);

                    SaveVhd(blobHandle, vhdLocalPath, storageAccountKey.Primary);
                }
            }
        }