Skip to content

Commit 1235112

Browse files
hyonholeehyonholee
committed
Merge branch 'dev' of https://github.com/Azure/azure-powershell into dev
2 parents 57df40c + eaa63d3 commit 1235112

File tree

28 files changed

+34715
-15944
lines changed

28 files changed

+34715
-15944
lines changed

‎setup/Setup/RemoveGalleryModules.ps1

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
# ----------------------------------------------------------------------------------
2+
#
3+
# Copyright Microsoft Corporation
4+
# Licensed under the Apache License, Version 2.0 (the "License");
5+
# you may not use this file except in compliance with the License.
6+
# You may obtain a copy of the License at
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
# Unless required by applicable law or agreed to in writing, software
9+
# distributed under the License is distributed on an "AS IS" BASIS,
10+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11+
# See the License for the specific language governing permissions and
12+
# limitations under the License.
13+
# ----------------------------------------------------------------------------------
14+
15+
function Remove-Modules
16+
{
17+
param([string]$basePath)
18+
$paths = "Azure", "AzureRM", "AzureRM.*", "Azure.Storage"
19+
$paths | ForEach-Object {
20+
$modulePath = ([System.IO.Path]::Combine($basePath, $_))
21+
try {
22+
Write-Host Removing $_
23+
if (Test-Path $modulePath)
24+
{
25+
Remove-Item -Recurse $modulePath -Force
26+
}
27+
}
28+
catch {}
29+
}
30+
}
31+
32+
if (Test-Path 'Env:\ProgramFiles(x86)')
33+
{
34+
Remove-Modules ([System.IO.Path]::Combine(${env:ProgramFiles(x86)}, "WindowsPowerShell", "Modules"))
35+
}
36+
37+
if (Test-Path Env:\ProgramFiles)
38+
{
39+
Remove-Modules ([System.IO.Path]::Combine($env:ProgramFiles, "WindowsPowerShell", "Modules"))
40+
}
41+

‎setup/Setup/SetExecutionPolicy.ps1

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@
1212
# limitations under the License.
1313
# ----------------------------------------------------------------------------------
1414

15-
1615
try
1716
{
1817
Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Force -ErrorAction "SilentlyContinue"

‎setup/azurecmd.wxs

Lines changed: 23 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
<?xml version="1.0" encoding="UTF-8"?>
2-
<Wix xmlns="http://schemas.microsoft.com/wix/2006/wi">
2+
<Wix xmlns="http://schemas.microsoft.com/wix/2006/wi" xmlns:util="http://schemas.microsoft.com/wix/UtilExtension">
33

44
<?define productName="Microsoft Azure PowerShell - November 2015" ?>
55
<?define sourceDir="$(var.SolutionDir)..\src\Package\$(var.Configuration)" ?>
@@ -8,7 +8,7 @@
88
<?define version="1.0.1" ?>
99
<?define versionedStartMenuFolder="Microsoft Azure" ?>
1010
<?define staleStartMenuFolder="Windows Azure" ?>
11-
11+
1212
<Product Id="*"
1313
Name="$(var.productName)"
1414
Language="1033"
@@ -40,27 +40,9 @@
4040
</SetProperty>
4141

4242
<SetProperty Id="POWERSHELLEXE" Value="[WindowsFolder]System32\WindowsPowerShell\v1.0\powershell.exe" Before="AppSearch"/>
43-
44-
<Property Id="POWERSHELLGETMODULESINSTALLED" >
45-
<DirectorySearch Id="PROGRAMFILESCONTAINER" Path="[BaseModulesFolder]">
46-
<DirectorySearch Id= "PROFILEMODULE" Path="AzureRM.Profile"/>
47-
</DirectorySearch>
48-
</Property>
49-
50-
<Property Id="POWERSHELLGETAZUREINSTALLED" >
51-
<DirectorySearch Id="AZURECONTAINER" Path="[BaseModulesFolder]">
52-
<DirectorySearch Id= "AZUREMODULE" Path="Azure"/>
53-
</DirectorySearch>
54-
</Property>
55-
56-
<Condition Message="Azure Modules from the PowerShell Gallery are installed on this machine. Please remove these modules before installing this MSI.">
57-
<![CDATA[Installed or NOT(POWERSHELLGETMODULESINSTALLED >< "AzureRM.Profile")]]>
58-
</Condition>
59-
60-
<Condition Message="Azure Modules from the PowerShell Gallery are installed on this machine. Please remove these modules before installing this MSI.">
61-
<![CDATA[Installed or NOT(POWERSHELLGETAZUREINSTALLED >< "Azure")]]>
62-
</Condition>
63-
43+
44+
<util:CloseApplication Id="ClosePowerShellExe" Target="powershell.exe" ElevatedCloseMessage="yes" CloseMessage="yes" PromptToContinue="yes" Description="PowerShell must be closed to continue with this installation. Please close any PowerShell windows and click 'Retry', or click 'Cancel' to stop installation. If you click 'Ignore' the installer will attempt to close any open PowerShell window."/>
45+
<util:CloseApplication Id="ClosePowerShellIseExe" Target="powershell_ise.exe" ElevatedCloseMessage="yes" CloseMessage="yes" PromptToContinue="yes" Description="PowerShell ISE must be closed to continue with this installation. Please close any PowerShell ISE windows and click 'Retry', or click 'Cancel' to stop installation. If you click 'Ignore' the installer will attempt to close any open PowerShell ISE window."/>
6446

6547
<Condition Message="This setup requires the Windows PowerShell 3.0 or compatible version to be installed.">
6648
<!-- This condition will break if PowerShell has version x3.0 and it is not compatible with 3.0 -->
@@ -88,19 +70,19 @@
8870
</Directory>
8971

9072
<Property Id="POWERSHELLPATH">
91-
<RegistrySearch Id="POWERSHELLPATH"
92-
Type="raw"
93-
Root="HKLM"
94-
Key="SOFTWARE\Microsoft\PowerShell\3\PowerShellEngine"
95-
Name="ApplicationBase" />
73+
<RegistrySearch Id="POWERSHELLPATH"
74+
Type="raw"
75+
Root="HKLM"
76+
Key="SOFTWARE\Microsoft\PowerShell\3\PowerShellEngine"
77+
Name="ApplicationBase" />
9678
</Property>
9779

9880
<DirectoryRef Id="PowerShellFolder">
9981
<Component Id="PSModulePath.System" Guid="273525B9-7AAB-421A-90C8-8E50A1840B8D">
10082
<CreateFolder />
10183
<!-- Work around bug that PowerShell does not always consider default module paths. -->
10284
<Environment Id="PSModulePath.SystemAppRoot" Action="set" Name="PSMODULEPATH" Part="last" Value="[BaseModulesFolder];[PowerShellFolder]ResourceManager\AzureResourceManager;[PowerShellFolder]ServiceManagement" System="yes" />
103-
</Component>
85+
</Component>
10486
</DirectoryRef>
10587

10688
<DirectoryRef Id="PowerShellFolder">
@@ -109,6 +91,12 @@
10991
</Component>
11092
</DirectoryRef>
11193

94+
<DirectoryRef Id="PowerShellFolder">
95+
<Component Id="AzureSdkRemoveModulesScript" Guid="7d75fe67-88e4-40da-bc3d-874cb0438d30">
96+
<File Id="AzureSdkRemoveModulesScriptFile" KeyPath="yes" Source="$(var.SolutionDir)\Setup\RemoveGalleryModules.ps1" />
97+
</Component>
98+
</DirectoryRef>
99+
112100
<DirectoryRef Id="AzureSdkMenu">
113101
<Component Id="AzureSdkShortcut" Guid="A9B58A22-F746-451D-8840-F887D8014C3C">
114102
<Shortcut Id="AzureSdkStartMenuShortcut"
@@ -127,17 +115,20 @@
127115
<ComponentRef Id="PSModulePath.System" />
128116
<ComponentRef Id="AzureSdkShortcut" />
129117
<ComponentRef Id="AzureSdkExecutionPolicyScript" />
130-
</Feature>
118+
<ComponentRef Id="AzureSdkRemoveModulesScript" />
119+
</Feature>
131120

132121
<Binary Id="CustomActions" SourceFile="$(var.caSourceDir)\Microsoft.WindowsAzure.Setup.CA.dll"/>
133122
<CustomAction Id="UpdatePSShortcut" BinaryKey="CustomActions" DllEntry="UpdatePSShortcut" Execute ="deferred" Impersonate="no"/>
134123
<CustomAction Id="SetCustomActionDataValue" Return="check" Property="UpdatePSShortcut"
135124
Value="ShortcutPath=[AzureSdkMenu]Microsoft Azure PowerShell.lnk;DefaultShortcutPath=[ProgramMenuFolder]System Tools\Windows PowerShell.lnk" />
136-
<CustomAction Id="SetExecutionPolicy" Property="POWERSHELLEXE" ExeCommand="-NoLogo -NoProfile -ExecutionPolicy Bypass -Command &quot;. \&quot;[PowerShellFolder]SetExecutionPolicy.ps1\&quot;&quot;" Execute="deferred" Impersonate="no" Return="check"/>
125+
<CustomAction Id="SetExecutionPolicy" Property="POWERSHELLEXE" ExeCommand="-NoLogo -NoProfile -WindowStyle Hidden -NonInteractive -ExecutionPolicy Bypass -Command &quot;. \&quot;[PowerShellFolder]SetExecutionPolicy.ps1\&quot;&quot;" Execute="deferred" Impersonate="no" Return="check"/>
126+
<CustomAction Id="RemoveGalleryModules" Property="POWERSHELLEXE" ExeCommand="-NoLogo -NoProfile -WindowStyle Hidden -NonInteractive -ExecutionPolicy Bypass -Command &quot;. \&quot;[PowerShellFolder]RemoveGalleryModules.ps1\&quot;&quot;" Execute="deferred" Impersonate="no" Return="check"/>
137127

138128
<InstallExecuteSequence>
139129
<Custom Action="SetCustomActionDataValue" After="CreateShortcuts">NOT Installed</Custom>
140-
<Custom Action="UpdatePSShortcut" After="SetCustomActionDataValue">NOT Installed</Custom>
130+
<Custom Action="RemoveGalleryModules" After="SetCustomActionDataValue">NOT Installed</Custom>
131+
<Custom Action="UpdatePSShortcut" After="RemoveGalleryModules">NOT Installed</Custom>
141132
<Custom Action="SetExecutionPolicy" After="UpdatePSShortcut">NOT Installed</Custom>
142133
</InstallExecuteSequence>
143134

‎setup/azurecmdfiles.wxi

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,9 @@
88
<Component Id="cmp0761C025FCF8EF027A5BC23C72777C52" Guid="*">
99
<File Id="fil80798000B169AC6D62DD8331A55CF6FF" KeyPath="yes" Source="$(var.sourceDir)\ResourceManager\AzureResourceManager\Azure.Storage\Azure.Storage.psd1" />
1010
</Component>
11+
<Component Id="cmpCC0D4240D3640D2D425852E0F2A4D31E" Guid="*">
12+
<File Id="fil12F99A3166EFE4DC83A3FA8C7EBCF53F" KeyPath="yes" Source="$(var.sourceDir)\ResourceManager\AzureResourceManager\Azure.Storage\AzureStorageStartup.ps1" />
13+
</Component>
1114
<Component Id="cmp658F4F28B5894F240AF3AE214E228C7A" Guid="*">
1215
<File Id="fil73B16246DA5417876E50086E40C8A00A" KeyPath="yes" Source="$(var.sourceDir)\ResourceManager\AzureResourceManager\Azure.Storage\Hyak.Common.dll" />
1316
</Component>
@@ -1716,6 +1719,9 @@
17161719
<Component Id="cmpEDB84317615898A496C0FF620B372C87" Guid="*">
17171720
<File Id="fil56F72CE4FFC9F006CE1B279238995FC8" KeyPath="yes" Source="$(var.sourceDir)\ResourceManager\AzureResourceManager\AzureRM.RecoveryServices\Microsoft.ApplicationInsights.dll" />
17181721
</Component>
1722+
<Component Id="cmp66ECF842EEB4EB595925E310358E9750" Guid="*">
1723+
<File Id="filF271979440640E53A189C08A7ADBAD7A" KeyPath="yes" Source="$(var.sourceDir)\ResourceManager\AzureResourceManager\AzureRM.RecoveryServices\Microsoft.Azure.Commands.RecoveryServices.ARM.dll-help.xml" />
1724+
</Component>
17191725
<Component Id="cmp5CAA968EE7C749E4480233FD90EDFE63" Guid="*">
17201726
<File Id="filC143DADE40E067E4692D2246F93BA8A4" KeyPath="yes" Source="$(var.sourceDir)\ResourceManager\AzureResourceManager\AzureRM.RecoveryServices\Microsoft.Azure.Commands.RecoveryServices.ARM.dll" />
17211727
</Component>
@@ -2803,6 +2809,9 @@
28032809
<Component Id="cmpA4896332D2E9278A5CFFC76462E3D397" Guid="*">
28042810
<File Id="filB992523DF0C46B6B6E8C09B1BB31B817" KeyPath="yes" Source="$(var.sourceDir)\ServiceManagement\Azure\Azure.Storage\Azure.Storage.psd1" />
28052811
</Component>
2812+
<Component Id="cmp4E5F2CFF9BC468A2A8873B7D3261F7AE" Guid="*">
2813+
<File Id="fil56B4216361B8FAE2391F878C1BBEE9A4" KeyPath="yes" Source="$(var.sourceDir)\ServiceManagement\Azure\Azure.Storage\AzureStorageStartup.ps1" />
2814+
</Component>
28062815
<Component Id="cmp1D11794D4DE13CD2849C407A7736B398" Guid="*">
28072816
<File Id="fil96E9CABBEC4A47800C1290A56CCB4839" KeyPath="yes" Source="$(var.sourceDir)\ServiceManagement\Azure\Azure.Storage\Hyak.Common.dll" />
28082817
</Component>
@@ -4764,6 +4773,7 @@
47644773
<Fragment>
47654774
<ComponentGroup Id="azurecmdfiles">
47664775
<ComponentRef Id="cmp0761C025FCF8EF027A5BC23C72777C52" />
4776+
<ComponentRef Id="cmpCC0D4240D3640D2D425852E0F2A4D31E" />
47674777
<ComponentRef Id="cmp658F4F28B5894F240AF3AE214E228C7A" />
47684778
<ComponentRef Id="cmpBE01F246267602912E7BB7FAC7C89B62" />
47694779
<ComponentRef Id="cmp12AAE81384B3AF67B8D558AB9941F111" />
@@ -5322,6 +5332,7 @@
53225332
<ComponentRef Id="cmpC96DAF4A31F86D19246E39394F496F83" />
53235333
<ComponentRef Id="cmpA77D5875EABE5151973A245C04E3858E" />
53245334
<ComponentRef Id="cmpEDB84317615898A496C0FF620B372C87" />
5335+
<ComponentRef Id="cmp66ECF842EEB4EB595925E310358E9750" />
53255336
<ComponentRef Id="cmp5CAA968EE7C749E4480233FD90EDFE63" />
53265337
<ComponentRef Id="cmpE7C8C6576C1069564C717CBBBD229F39" />
53275338
<ComponentRef Id="cmpC9F93119DA28BCE2B325C737951EFABA" />
@@ -5675,6 +5686,7 @@
56755686
<ComponentRef Id="cmp01E003A3FA26FB81C6B89817FE7250BE" />
56765687
<ComponentRef Id="cmp5E7A927FFF2CD1AC2D9A4D4746132F63" />
56775688
<ComponentRef Id="cmpA4896332D2E9278A5CFFC76462E3D397" />
5689+
<ComponentRef Id="cmp4E5F2CFF9BC468A2A8873B7D3261F7AE" />
56785690
<ComponentRef Id="cmp1D11794D4DE13CD2849C407A7736B398" />
56795691
<ComponentRef Id="cmpCCBFA5B04DFD985227AF6B654A18F375" />
56805692
<ComponentRef Id="cmp9F56F57F285139F1E6B4A645E24EA274" />

‎src/Common/Storage/Commands.Storage/AzureStorageStartup.ps1

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
# ----------------------------------------------------------------------------------
2+
#
3+
# Copyright Microsoft Corporation
4+
# Licensed under the Apache License, Version 2.0 (the "License");
5+
# you may not use this file except in compliance with the License.
6+
# You may obtain a copy of the License at
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
# Unless required by applicable law or agreed to in writing, software
9+
# distributed under the License is distributed on an "AS IS" BASIS,
10+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11+
# See the License for the specific language governing permissions and
12+
# limitations under the License.
13+
# ----------------------------------------------------------------------------------
14+
15+
$script:aliases = @{
16+
# Storage aliases
17+
"Get-AzureStorageContainerAcl" = "Get-AzureStorageContainer";
18+
"Start-CopyAzureStorageBlob" = "Start-AzureStorageBlobCopy";
19+
"Stop-CopyAzureStorageBlob" = "Stop-AzureStorageBlobCopy";
20+
}
21+
22+
$aliases.GetEnumerator() | Select @{Name='Name'; Expression={$_.Key}}, @{Name='Value'; Expression={$_.Value}} | New-Alias -Description "AzureAlias"

‎src/Common/Storage/Commands.Storage/Blob/Cmdlet/StartAzureStorageBlobCopy.cs

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
namespace Microsoft.WindowsAzure.Commands.Storage.Blob.Cmdlet
1616
{
1717
using System;
18+
using System.IO;
1819
using System.Collections.Generic;
1920
using System.Management.Automation;
2021
using System.Security.Permissions;
@@ -28,10 +29,12 @@ namespace Microsoft.WindowsAzure.Commands.Storage.Blob.Cmdlet
2829
using Microsoft.WindowsAzure.Storage.Blob;
2930
using Microsoft.WindowsAzure.Storage.DataMovement;
3031
using Microsoft.WindowsAzure.Storage.File;
32+
using Microsoft.Azure.Common.Authentication;
33+
using System.Reflection;
3134

3235
[Cmdlet(VerbsLifecycle.Start, StorageNouns.CopyBlob, ConfirmImpact = ConfirmImpact.High, DefaultParameterSetName = ContainerNameParameterSet),
3336
OutputType(typeof(AzureStorageBlob))]
34-
public class StartAzureStorageBlobCopy : StorageDataMovementCmdletBase
37+
public class StartAzureStorageBlobCopy : StorageDataMovementCmdletBase, IModuleAssemblyInitializer
3538
{
3639
/// <summary>
3740
/// Blob Pipeline parameter set name
@@ -620,5 +623,23 @@ private CloudBlob GetDestinationBlobWithCopyId(IStorageBlobManagement destChanne
620623
CloudBlob blob = destChannel.GetBlobReferenceFromServer(container, blobName, accessCondition, options, OperationContext);
621624
return blob;
622625
}
626+
627+
public void OnImport()
628+
{
629+
try
630+
{
631+
System.Management.Automation.PowerShell invoker = null;
632+
invoker = System.Management.Automation.PowerShell.Create(RunspaceMode.CurrentRunspace);
633+
invoker.AddScript(File.ReadAllText(FileUtilities.GetContentFilePath(
634+
Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location),
635+
"AzureStorageStartup.ps1")));
636+
invoker.Invoke();
637+
}
638+
catch
639+
{
640+
// Ignore exception.
641+
}
642+
}
643+
623644
}
624645
}

‎src/Common/Storage/Commands.Storage/Commands.Storage.csproj

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -293,6 +293,9 @@
293293
<Link>Azure.Storage.psd1</Link>
294294
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
295295
</None>
296+
<None Include="AzureStorageStartup.ps1" >
297+
<CopyToOutputDirectory>Always</CopyToOutputDirectory>
298+
</None>
296299
<None Include="Microsoft.WindowsAzure.Commands.Storage.format.ps1xml">
297300
<CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
298301
<SubType>Designer</SubType>

‎src/Common/Storage/Commands.Storage/Common/StorageCloudCmdletBase.cs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,10 +106,11 @@ protected void DoInitChannelCurrentSubscription(bool force)
106106
}
107107
}
108108

109-
public override void ExecuteCmdlet()
109+
protected override void ProcessRecord()
110110
{
111111
Validate.ValidateInternetConnection();
112112
InitChannelCurrentSubscription();
113+
base.ProcessRecord();
113114
}
114115

115116

0 commit comments

Comments
 (0)