Changing alias infrastructure for sql aliases

Author: markcowl

setup/azurecmdfiles.wxi

@@ -6,7 +6,7 @@
     <ProductVersion>
     </ProductVersion>
     <SchemaVersion>2.0</SchemaVersion>
-    <ProjectGuid>{127D0D51-FDEA-4E1A-8CD8-34DEB5C2F7F6}</ProjectGuid>
+    <ProjectGuid>{59D1B5DC-9175-43EC-90C6-CBA601B3565F}</ProjectGuid>
     <OutputType>Library</OutputType>
     <AppDesignerFolder>Properties</AppDesignerFolder>
     <RootNamespace>Microsoft.Azure.Commands.Automation.Test</RootNamespace>

src/ResourceManager/Automation/Commands.Automation.Test/Commands.ResourceManagement.Automation.Test.csproj

@@ -47,9 +47,6 @@ ProcessorArchitecture = 'None'
 # Modules that must be imported into the global environment prior to importing this module
 RequiredModules = @()
 
-# Module definitions that will be imported prior. This line was added to make Aliases in NestedModules work.
-ModuleToProcess = 'AzureResourceManager.psm1'
-
 # Assemblies that must be loaded prior to importing this module
 RequiredAssemblies = @()
 

src/ResourceManager/Resources/Commands.Resources/AzureResourceManager.psd1

@@ -257,9 +257,9 @@
     <Content Include="AzureResourceManager.psd1">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
     </Content>
-    <Content Include="AzureResourceManager.psm1">
+    <None Include="ResourceManagerStartup.ps1">
       <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
-    </Content>
+    </None>
     <None Include="MSSharedLibKey.snk" />
     <None Include="packages.config" />
     <None Include="PostBuild.ps1" />

src/ResourceManager/Resources/Commands.Resources/AzureResourceManager.psm1

@@ -30,7 +30,7 @@ if (Test-Path -Path $sourcePath) {
     Remove-Item -Path $sourcePath -Force;
 }
 
-$sourcePath = $sourceDir + "AzureResourceManager.psm1"
+$sourcePath = $sourceDir + "ResourceManagerStartup.ps1"
 $destDir = Split-Path -Path $sourceDir
 
 if (Test-Path -Path $sourcePath) {

src/ResourceManager/Resources/Commands.Resources/Commands.Resources.csproj

@@ -14,19 +14,42 @@
 
 using System.Collections.Generic;
 using System.Management.Automation;
+using System.Reflection;
 using Microsoft.Azure.Commands.Resources.Models;
+using System.IO;
+using Microsoft.Azure.Common.Authentication;
 
 namespace Microsoft.Azure.Commands.Resources
 {
     /// <summary>
     /// Get the available locations for certain resource types.
     /// </summary>
     [Cmdlet(VerbsCommon.Get, "AzureLocation"), OutputType(typeof(List<PSResourceProviderLocationInfo>))]
-    public class GetAzureLocationCommand : ResourcesBaseCmdlet
+    public class GetAzureLocationCommand : ResourcesBaseCmdlet, IModuleAssemblyInitializer
     {
         public override void ExecuteCmdlet()
         {
             WriteObject(ResourcesClient.GetLocations(), true);
         }
+
+        /// <summary>
+        /// Load global aliases for ARM
+        /// </summary>
+        public void OnImport()
+        {
+            try
+            {
+                System.Management.Automation.PowerShell invoker = null;
+                invoker = System.Management.Automation.PowerShell.Create(RunspaceMode.CurrentRunspace);
+                invoker.AddScript(File.ReadAllText(FileUtilities.GetContentFilePath(
+                    Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location),
+                    "ResourceManagerStartup.ps1")));
+                invoker.Invoke();
+            }
+            catch
+            {
+                // This will throw exception for tests, ignore.
+            }
+        }
     }
 }

src/ResourceManager/Resources/Commands.Resources/PostBuild.ps1

@@ -0,0 +1,24 @@
+# ----------------------------------------------------------------------------------
+#
+# 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.
+# ----------------------------------------------------------------------------------
+
+
+$aliases = @{
+    # Sql aliases
+    "Get-AzureSqlDatabaseServerAuditingPolicy"="Get-AzureSqlServerAuditingPolicy";
+    "Remove-AzureSqlDatabaseServerAuditing"="Remove-AzureSqlServerAuditing";
+    "Set-AzureSqlDatabaseServerAuditingPolicy"="Set-AzureSqlServerAuditingPolicy";
+    "Use-AzureSqlDatabaseServerAuditingPolicy"="Use-AzureSqlServerAuditingPolicy";
+}
+
+$aliases.GetEnumerator() | Select @{Name='Name'; Expression={$_.Key}}, @{Name='Value'; Expression={$_.Value}} | New-Alias -Description "AzureAlias"