From 6b5d043df45bed181d3490271f57ebdadac85e7f Mon Sep 17 00:00:00 2001
From: Leandro Wajswajn Pereyra <leandro.wajswajn@ar.ey.com>
Date: Tue, 30 Apr 2019 22:57:50 -0300
Subject: [PATCH] Added scopeto Update-Module. Added localized error message

---
 src/PowerShellGet/PSGet.Resource.psd1         |  1 +
 .../public/psgetfunctions/Update-Module.ps1   | 22 ++++++++++++++++++-
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/src/PowerShellGet/PSGet.Resource.psd1 b/src/PowerShellGet/PSGet.Resource.psd1
index b6704d17..8d472b70 100644
--- a/src/PowerShellGet/PSGet.Resource.psd1
+++ b/src/PowerShellGet/PSGet.Resource.psd1
@@ -22,6 +22,7 @@ ConvertFrom-StringData @'
         RequiredVersionAllowedOnlyWithSingleModuleName=The RequiredVersion parameter is allowed only when a single module name is specified as the value of the Name parameter, without any wildcard characters.
         MinimumVersionIsGreaterThanMaximumVersion=The specified MinimumVersion '{0}' is greater than the specified MaximumVersion '{1}'.
         AllowPrereleaseRequiredToUsePrereleaseStringInVersion=The '-AllowPrerelease' parameter must be specified when using the Prerelease string in MinimumVersion, MaximumVersion, or RequiredVersion.
+        UpdateModuleAdminPrivilegeRequiredForAllUsersScope=Administrator rights are required to update modules in '{0}'. Log on to the computer with an account that has Administrator rights, and then try again, or update '{1}' by adding "-Scope CurrentUser" to your command. You can also try running the Windows PowerShell session with elevated rights (Run as Administrator).
         InstallModuleAdminPrivilegeRequiredForAllUsersScope=Administrator rights are required to install modules in '{0}'. Log on to the computer with an account that has Administrator rights, and then try again, or install '{1}' by adding "-Scope CurrentUser" to your command. You can also try running the Windows PowerShell session with elevated rights (Run as Administrator).
         InstallScriptAdminPrivilegeRequiredForAllUsersScope=Administrator rights are required to install scripts in '{0}'. Log on to the computer with an account that has Administrator rights, and then try again, or install '{1}' by adding "-Scope CurrentUser" to your command. You can also try running the Windows PowerShell session with elevated rights (Run as Administrator).
         AdministratorRightsNeededOrSpecifyCurrentUserScope=Administrator rights are required to install or update. Log on to the computer with an account that has Administrator rights, and then try again, or install by adding "-Scope CurrentUser" to your command. You can also try running the Windows PowerShell session with elevated rights (Run as Administrator).
diff --git a/src/PowerShellGet/public/psgetfunctions/Update-Module.ps1 b/src/PowerShellGet/public/psgetfunctions/Update-Module.ps1
index 8a24deb4..92adff7d 100644
--- a/src/PowerShellGet/public/psgetfunctions/Update-Module.ps1
+++ b/src/PowerShellGet/public/psgetfunctions/Update-Module.ps1
@@ -26,6 +26,11 @@ function Update-Module {
         [PSCredential]
         $Credential,
 
+        [Parameter()]
+        [ValidateSet("CurrentUser", "AllUsers")]
+        [string]
+        $Scope,
+
         [Parameter(ValueFromPipelineByPropertyName = $true)]
         [ValidateNotNullOrEmpty()]
         [Uri]
@@ -51,6 +56,17 @@ function Update-Module {
     Begin {
         Install-NuGetClientBinaries -CallerPSCmdlet $PSCmdlet -Proxy $Proxy -ProxyCredential $ProxyCredential
 
+        if ($Scope -eq "AllUsers" -and -not (Test-RunningAsElevated)) {
+          # Throw an error when Update-Module is used as a non-admin user and '-Scope AllUsers'
+          $message = $LocalizedData.UpdateModuleAdminPrivilegeRequiredForAllUsersScope -f @($script:programFilesModulesPath, $script:MyDocumentsModulesPath)
+
+          ThrowError -ExceptionName "System.ArgumentException" `
+              -ExceptionMessage $message `
+              -ErrorId "UpdateModuleAdminPrivilegeRequiredForAllUsersScope" `
+              -CallerPSCmdlet $PSCmdlet `
+              -ErrorCategory InvalidArgument
+        }
+
         # Module names already tried in the current pipeline
         $moduleNamesInPipeline = @()
     }
@@ -135,7 +151,11 @@ function Update-Module {
             $PSBoundParameters["PackageManagementProvider"] = $providerName
             $PSBoundParameters["InstallUpdate"] = $true
 
-            $PSBoundParameters["Scope"] = Get-InstallationScope -PreviousInstallLocation $psgetItemInfo.InstalledLocation -CurrentUserPath $script:MyDocumentsModulesPath
+            if (-not $Scope) {
+              $Scope = Get-InstallationScope -PreviousInstallLocation $psgetItemInfo.InstalledLocation -CurrentUserPath $script:MyDocumentsModulesPath
+            }
+
+            $PSBoundParameters["Scope"] = $Scope
 
             $sid = PackageManagement\Install-Package @PSBoundParameters
         }