Program.cs

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License. See License.txt in the project root for license information. 

using Microsoft.Azure.Management.Compute.Fluent;
using Microsoft.Azure.Management.Compute.Fluent.Models;
using Microsoft.Azure.Management.Fluent;
using Microsoft.Azure.Management.ResourceManager.Fluent;
using Microsoft.Azure.Management.ResourceManager.Fluent.Authentication;
using Microsoft.Azure.Management.ResourceManager.Fluent.Core;
using Microsoft.Azure.Management.Samples.Common;
using System;
using System.Collections.Generic;

namespace ManageVirtualMachineExtension
{
    public class Program
    {
        // Linux configurations
        //
        readonly static string FirstLinuxUserName = Utilities.CreateUsername();
        readonly static string FirstLinuxUserPassword = Utilities.CreatePassword();
        readonly static string FirstLinuxUserNewPassword = Utilities.CreatePassword();

        readonly static string SecondLinuxUserName = "seconduser";
        readonly static string SecondLinuxUserPassword = Utilities.CreatePassword();
        readonly static string SecondLinuxUserExpiration = "2020-12-31";

        readonly static string ThirdLinuxUserName = "thirduser";
        readonly static string ThirdLinuxUserPassword = Utilities.CreatePassword();
        readonly static string ThirdLinuxUserExpiration = "2020-12-31";

        readonly static string LinuxCustomScriptExtensionName = "CustomScriptForLinux";
        readonly static string LinuxCustomScriptExtensionPublisherName = "Microsoft.OSTCExtensions";
        readonly static string LinuxCustomScriptExtensionTypeName = "CustomScriptForLinux";
        readonly static string LinuxCustomScriptExtensionVersionName = "1.4";

        readonly static string MySqlScriptLinuxInstallCommand = "bash install_mysql_server_5.6.sh Abc.123x(";
        readonly static List<string> MySQLLinuxInstallScriptFileUris = new List<string>()
        {
            "https://raw.githubusercontent.com/Azure/azure-quickstart-templates/4397e808d07df60ff3cdfd1ae40999f0130eb1b3/mysql-standalone-server-ubuntu/scripts/install_mysql_server_5.6.sh"
        };

        readonly static string windowsCustomScriptExtensionName = "CustomScriptExtension";
        readonly static string windowsCustomScriptExtensionPublisherName = "Microsoft.Compute";
        readonly static string windowsCustomScriptExtensionTypeName = "CustomScriptExtension";
        readonly static string windowsCustomScriptExtensionVersionName = "1.7";

        readonly static string mySqlScriptWindowsInstallCommand = "powershell.exe -ExecutionPolicy Unrestricted -File installMySQL.ps1";
        readonly static List<string> mySQLWindowsInstallScriptFileUris = new List<string>()
        {
            "https://raw.githubusercontent.com/Azure/azure-libraries-for-net/master/Samples/Asset/installMySQL.ps1"
        };

        readonly static string linuxVmAccessExtensionName = "VMAccessForLinux";
        readonly static string linuxVmAccessExtensionPublisherName = "Microsoft.OSTCExtensions";
        readonly static string linuxVmAccessExtensionTypeName = "VMAccessForLinux";
        readonly static string linuxVmAccessExtensionVersionName = "1.4";

        // Windows configurations
        //
        readonly static string firstWindowsUserName = Utilities.CreateUsername();
        readonly static string firstWindowsUserPassword = Utilities.CreatePassword();
        readonly static string firstWindowsUserNewPassword = Utilities.CreatePassword();

        readonly static string secondWindowsUserName = "seconduser";
        readonly static string secondWindowsUserPassword = Utilities.CreatePassword();

        readonly static string thirdWindowsUserName = "thirduser";
        readonly static string thirdWindowsUserPassword = Utilities.CreatePassword();

        readonly static string windowsVmAccessExtensionName = "VMAccessAgent";
        readonly static string windowsVmAccessExtensionPublisherName = "Microsoft.Compute";
        readonly static string windowsVmAccessExtensionTypeName = "VMAccessAgent";
        readonly static string windowsVmAccessExtensionVersionName = "2.3";

        /**
         * Azure Compute sample for managing virtual machine extensions. -
         *  - Create a Linux and Windows virtual machine
         *  - Add three users (user names and passwords for windows, SSH keys for Linux)
         *  - Resets user credentials
         *  - Remove a user
         *  - Install MySQL on Linux | something significant on Windows
         *  - Remove extensions
         */
        public static void RunSample(IAzure azure)
        {
            string rgName = SdkContext.RandomResourceName("rgCOVE", 15);
            string linuxVmName = SdkContext.RandomResourceName("lVM", 10);
            string windowsVmName = SdkContext.RandomResourceName("wVM", 10);
            string pipDnsLabelLinuxVM = SdkContext.RandomResourceName("rgPip1", 25);
            string pipDnsLabelWindowsVM = SdkContext.RandomResourceName("rgPip2", 25);

            try
            {
                //=============================================================
                // Create a Linux VM with root (sudo) user
                Utilities.Log("Creating a Linux VM");

                IVirtualMachine linuxVM = azure.VirtualMachines.Define(linuxVmName)
                        .WithRegion(Region.USEast)
                        .WithNewResourceGroup(rgName)
                        .WithNewPrimaryNetwork("10.0.0.0/28")
                        .WithPrimaryPrivateIPAddressDynamic()
                        .WithNewPrimaryPublicIPAddress(pipDnsLabelLinuxVM)
                        .WithPopularLinuxImage(KnownLinuxVirtualMachineImage.UbuntuServer14_04_Lts)
                        .WithRootUsername(FirstLinuxUserName)
                        .WithRootPassword(FirstLinuxUserPassword)
                        .WithSize(VirtualMachineSizeTypes.Parse("Standard_D2a_v4"))
                        .Create();

                Utilities.Log("Created a Linux VM:" + linuxVM.Id);
                Utilities.PrintVirtualMachine(linuxVM);

                //=============================================================
                // Add a second sudo user to Linux VM using VMAccess extension

                linuxVM.Update()
                        .DefineNewExtension(linuxVmAccessExtensionName)
                            .WithPublisher(linuxVmAccessExtensionPublisherName)
                            .WithType(linuxVmAccessExtensionTypeName)
                            .WithVersion(linuxVmAccessExtensionVersionName)
                            .WithProtectedSetting("username", SecondLinuxUserName)
                            .WithProtectedSetting("password", SecondLinuxUserPassword)
                            .WithProtectedSetting("expiration", SecondLinuxUserExpiration)
                            .Attach()
                        .Apply();

                Utilities.Log("Added a second sudo user to the Linux VM");

                //=============================================================
                // Add a third sudo user to Linux VM by updating VMAccess extension

                linuxVM.Update()
                        .UpdateExtension(linuxVmAccessExtensionName)
                            .WithProtectedSetting("username", ThirdLinuxUserName)
                            .WithProtectedSetting("password", ThirdLinuxUserPassword)
                            .WithProtectedSetting("expiration", ThirdLinuxUserExpiration)
                        .Parent()
                        .Apply();

                Utilities.Log("Added a third sudo user to the Linux VM");


                //=============================================================
                // Reset ssh password of first user of Linux VM by updating VMAccess extension

                linuxVM.Update()
                        .UpdateExtension(linuxVmAccessExtensionName)
                            .WithProtectedSetting("username", FirstLinuxUserName)
                            .WithProtectedSetting("password", FirstLinuxUserNewPassword)
                            .WithProtectedSetting("reset_ssh", "true")
                        .Parent()
                        .Apply();

                Utilities.Log("Password of first user of Linux VM has been updated");

                //=============================================================
                // Removes the second sudo user from Linux VM using VMAccess extension

                linuxVM.Update()
                        .UpdateExtension(linuxVmAccessExtensionName)
                            .WithProtectedSetting("remove_user", SecondLinuxUserName)
                        .Parent()
                        .Apply();

                //=============================================================
                // Install MySQL in Linux VM using CustomScript extension

                linuxVM.Update()
                        .DefineNewExtension(LinuxCustomScriptExtensionName)
                            .WithPublisher(LinuxCustomScriptExtensionPublisherName)
                            .WithType(LinuxCustomScriptExtensionTypeName)
                            .WithVersion(LinuxCustomScriptExtensionVersionName)
                            .WithMinorVersionAutoUpgrade()
                            .WithPublicSetting("fileUris", MySQLLinuxInstallScriptFileUris)
                            .WithPublicSetting("commandToExecute", MySqlScriptLinuxInstallCommand)
                        .Attach()
                        .Apply();

                Utilities.Log("Installed MySql using custom script extension");
                Utilities.PrintVirtualMachine(linuxVM);

                //=============================================================
                // Removes the extensions from Linux VM

                linuxVM.Update()
                        .WithoutExtension(LinuxCustomScriptExtensionName)
                        .WithoutExtension(linuxVmAccessExtensionName)
                        .Apply();
                Utilities.Log("Removed the custom script and VM Access extensions from Linux VM");
                Utilities.PrintVirtualMachine(linuxVM);

                //=============================================================
                // Create a Windows VM with admin user

                Utilities.Log("Creating a Windows VM");

                IVirtualMachine windowsVM = azure.VirtualMachines.Define(windowsVmName)
                        .WithRegion(Region.USEast)
                        .WithExistingResourceGroup(rgName)
                        .WithNewPrimaryNetwork("10.0.0.0/28")
                        .WithPrimaryPrivateIPAddressDynamic()
                        .WithNewPrimaryPublicIPAddress(pipDnsLabelWindowsVM)
                        .WithPopularWindowsImage(KnownWindowsVirtualMachineImage.WindowsServer2012R2Datacenter)
                        .WithAdminUsername(firstWindowsUserName)
                        .WithAdminPassword(firstWindowsUserPassword)
                        .WithSize(VirtualMachineSizeTypes.Parse("Standard_D2a_v4"))
                        .DefineNewExtension(windowsCustomScriptExtensionName)
                            .WithPublisher(windowsCustomScriptExtensionPublisherName)
                            .WithType(windowsCustomScriptExtensionTypeName)
                            .WithVersion(windowsCustomScriptExtensionVersionName)
                            .WithMinorVersionAutoUpgrade()
                            .WithPublicSetting("fileUris", mySQLWindowsInstallScriptFileUris)
                            .WithPublicSetting("commandToExecute", mySqlScriptWindowsInstallCommand)
                        .Attach()
                        .Create();

                Utilities.Log("Created a Windows VM:" + windowsVM.Id);
                Utilities.PrintVirtualMachine(windowsVM);

                //=============================================================
                // Add a second admin user to Windows VM using VMAccess extension

                windowsVM.Update()
                        .DefineNewExtension(windowsVmAccessExtensionName)
                            .WithPublisher(windowsVmAccessExtensionPublisherName)
                            .WithType(windowsVmAccessExtensionTypeName)
                            .WithVersion(windowsVmAccessExtensionVersionName)
                            .WithProtectedSetting("username", secondWindowsUserName)
                            .WithProtectedSetting("password", secondWindowsUserPassword)
                        .Attach()
                        .Apply();

                Utilities.Log("Added a second admin user to the Windows VM");

                //=============================================================
                // Add a third admin user to Windows VM by updating VMAccess extension

                windowsVM.Update()
                        .UpdateExtension(windowsVmAccessExtensionName)
                            .WithProtectedSetting("username", thirdWindowsUserName)
                            .WithProtectedSetting("password", thirdWindowsUserPassword)
                        .Parent()
                        .Apply();

                Utilities.Log("Added a third admin user to the Windows VM");

                //=============================================================
                // Reset admin password of first user of Windows VM by updating VMAccess extension

                windowsVM.Update()
                        .UpdateExtension(windowsVmAccessExtensionName)
                            .WithProtectedSetting("username", firstWindowsUserName)
                            .WithProtectedSetting("password", firstWindowsUserNewPassword)
                        .Parent()
                        .Apply();

                Utilities.Log("Password of first user of Windows VM has been updated");

                //=============================================================
                // Removes the extensions from Linux VM

                windowsVM.Update()
                        .WithoutExtension(windowsVmAccessExtensionName)
                        .Apply();
                Utilities.Log("Removed the VM Access extensions from Windows VM");
                Utilities.PrintVirtualMachine(windowsVM);
            }
            finally
            {
                try
                {
                    Utilities.Log("Deleting Resource Group: " + rgName);
                    azure.ResourceGroups.DeleteByName(rgName);
                    Utilities.Log("Deleted Resource Group: " + rgName);
                }
                catch (Exception ex)
                {
                    Utilities.Log(ex);
                }
            }
        }

        public static void Main(string[] args)
        {

            try
            {
                //=============================================================
                // Authenticate
                AzureCredentials credentials = SdkContext.AzureCredentialsFactory.FromFile(Environment.GetEnvironmentVariable("AZURE_AUTH_LOCATION"));

                var azure = Azure
                    .Configure()
                    .WithLogLevel(HttpLoggingDelegatingHandler.Level.Basic)
                    .Authenticate(credentials)
                    .WithDefaultSubscription();

                // Print selected subscription
                Utilities.Log("Selected subscription: " + azure.SubscriptionId);

                RunSample(azure);
            }
            catch (Exception ex)
            {
                Utilities.Log(ex);
            }
        }
    }
}