From 49f0a16b65c4711c33fea309f88a2d2a1f9e76e5 Mon Sep 17 00:00:00 2001
From: twangboy <slee@saltstack.com>
Date: Wed, 25 May 2016 16:02:03 -0700
Subject: [PATCH] Fix parameters, escalate privelages

---
 bootstrap-salt.ps1 | 71 ++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 62 insertions(+), 9 deletions(-)

diff --git a/bootstrap-salt.ps1 b/bootstrap-salt.ps1
index af8e650..b97032f 100644
--- a/bootstrap-salt.ps1
+++ b/bootstrap-salt.ps1
@@ -23,7 +23,7 @@ Useful for testing locally from the command line with the --local switch
 .EXAMPLE
 ./bootstrap-salt.ps1 -minion minion-box -master master-box
 Specifies the minion and master ids in the minion config. 
-Defaults to the installer values of "minion" and "master".
+Defaults to the installer values of host name for the minion id and "salt" for the master.
 
 .EXAMPLE
 ./bootstrap-salt.ps1 -minion minion-box -master master-box -version 2015.5.2 -runservice false
@@ -32,12 +32,12 @@ Specifies all the optional parameters in no particular order.
 .PARAMETER version - Default version defined in this script. 
 
 .PARAMETER runservice - Boolean flag to stop the windows service and set to "manual". 
-                        Installer starts it by default. 
+                        Installer starts it by default.
 
 .PARAMETER minion - Name of the minion being installed on this host. 
-                    Installer defaults to "minion".
+                    Installer defaults to the host name.
 
-.PARAMETER master - Name or IP of the master server the minion. Installer defaults to "master".
+.PARAMETER master - Name or IP of the master server the minion. Installer defaults to "salt".
 
 .NOTES
 All of the parameters are optional. The default should be the latest version. The architecture
@@ -49,6 +49,10 @@ Original Vagrant Provisioner Project -https://github.com/saltstack/salty-vagrant
 Vagrant Project (utilizes this script) - https://github.com/mitchellh/vagrant
 SaltStack Download Location - https://repo.saltstack.com/windows/
 #>
+
+#==============================================================================
+# Commandlet Binding
+#==============================================================================
 [CmdletBinding()]
 Param(
   [Parameter(Mandatory=$false,ValueFromPipeline=$true)]
@@ -67,6 +71,55 @@ Param(
   [string]$master = "not-specified"
 )
 
+#==============================================================================
+# Script Functions
+#==============================================================================
+function Get-IsAdministrator
+{
+    $Identity = [System.Security.Principal.WindowsIdentity]::GetCurrent()
+    $Principal = New-Object System.Security.Principal.WindowsPrincipal($Identity)
+    $Principal.IsInRole([System.Security.Principal.WindowsBuiltInRole]::Administrator)
+}
+
+function Get-IsUacEnabled
+{
+    (Get-ItemProperty HKLM:\Software\Microsoft\Windows\CurrentVersion\Policies\System).EnableLua -ne 0
+}
+
+#==============================================================================
+# Check for Elevated Privileges
+#==============================================================================
+If (!(Get-IsAdministrator)) {
+    If (Get-IsUacEnabled) {
+        # We are not running "as Administrator" - so relaunch as administrator
+        # Create a new process object that starts PowerShell
+        $newProcess = new-object System.Diagnostics.ProcessStartInfo "PowerShell";
+
+        # Specify the current script path and name as a parameter`
+        $parameters = ""
+        If($minion -ne "not-specified") {$parameters = "-minion $minion"}
+        If($master -ne "not-specified") {$parameters = "$parameters -master $master"}
+        If($runservice -eq $false) {$parameters = "$parameters -runservice false"}
+        If($version -ne '') {$parameters = "$parameters -version $version"}
+        $newProcess.Arguments = $myInvocation.MyCommand.Definition, $parameters
+
+        # Specify the current working directory
+        $newProcess.WorkingDirectory = "$script_path"
+
+        # Indicate that the process should be elevated
+        $newProcess.Verb = "runas";
+
+        # Start the new process
+        [System.Diagnostics.Process]::Start($newProcess);
+
+        # Exit from the current, unelevated, process
+        Exit
+
+    } Else {
+        Throw "You must be administrator to run this script"
+    }
+}
+
 Write-Verbose "Parameters passed in:"
 Write-Verbose "version: $version"
 Write-Verbose "runservice: $runservice"
@@ -132,14 +185,14 @@ if (!$version) {
 }
 
 # Download minion setup file
-Write-Output -NoNewline "Downloading Salt minion installer Salt-Minion-$version-$arch-Setup.exe"
+Write-Output "Downloading Salt minion installer Salt-Minion-$version-$arch-Setup.exe"
 $webclient = New-Object System.Net.WebClient
 $url = "https://repo.saltstack.com/windows/Salt-Minion-$version-$arch-Setup.exe"
 $file = "C:\tmp\salt.exe"
 $webclient.DownloadFile($url, $file)
 
 # Install minion silently
-Write-Output -NoNewline "Installing Salt minion"
+Write-Output "Installing Salt minion"
 
 # Set the parameters for the installer
 # Unless specified, use the installer defaults
@@ -147,9 +200,9 @@ Write-Output -NoNewline "Installing Salt minion"
 # - master: salt
 # - Start the service
 $parameters = ""
-If($minion -ne "not-specified") {$parameters, "/minion-name=$minion" -join " "}
-If($master -ne "not-specified") {$parameters, "/master=$master" -join " "}
-If($runservice -eq $false) {$parameters, "/start-service=0" -join " "}
+If($minion -ne "not-specified") {$parameters = "/minion-name=$minion"}
+If($master -ne "not-specified") {$parameters = "$parameters /master=$master"}
+If($runservice -eq $false) {$parameters = "$parameters /start-service=0"}
 
 #Wait for process to exit before continuing.
 Start-Process C:\tmp\salt.exe -ArgumentList "/S $parameters" -Wait -NoNewWindow -PassThru | Out-Null