diff --git a/src/scripts/win32.ps1 b/src/scripts/win32.ps1 index 9e51df01..b35c8894 100644 --- a/src/scripts/win32.ps1 +++ b/src/scripts/win32.ps1 @@ -22,13 +22,35 @@ Function Add-Log($mark, $subject, $message) { printf "\033[%s;1m%s \033[0m\033[34;1m%s \033[0m\033[90;1m%s \033[0m\n" $code $mark $subject $message } +# Function to add a line to a powershell profile safely. +Function Add-ToProfile { + param( + [Parameter(Position = 0, Mandatory = $true)] + [ValidateNotNull()] + [ValidateLength(1, [int]::MaxValue)] + [string] + $input_profile, + [Parameter(Position = 1, Mandatory = $true)] + [ValidateNotNull()] + [ValidateLength(1, [int]::MaxValue)] + [string] + $search, + [Parameter(Position = 2, Mandatory = $true)] + [ValidateNotNull()] + [ValidateLength(1, [int]::MaxValue)] + [string] + $value + ) + if($null -eq (Get-Content $input_profile | findstr $search)) { + Add-Content -Path $input_profile -Value $value + } +} + # Function to fetch PATH from the registry. Function Get-PathFromRegistry { - $env:Path = [System.Environment]::GetEnvironmentVariable("Path","Machine") + ";" + - [System.Environment]::GetEnvironmentVariable("Path","User") - if($null -eq (Get-Content $current_profile | findstr 'Get-PathFromRegistry')) { - Add-Content -Path $current_profile -Value 'Function Get-PathFromRegistry { $env:Path = [System.Environment]::GetEnvironmentVariable("Path", "Machine") + ";" + [System.Environment]::GetEnvironmentVariable("Path", "User") }; Get-PathFromRegistry' - } + $env:Path = [System.Environment]::GetEnvironmentVariable("Path","User") + ";" + + [System.Environment]::GetEnvironmentVariable("Path","Machine") + Add-ToProfile $current_profile 'Get-PathFromRegistry' 'Function Get-PathFromRegistry { $env:Path = [System.Environment]::GetEnvironmentVariable("Path", "User") + ";" + [System.Environment]::GetEnvironmentVariable("Path", "Machine") }; Get-PathFromRegistry' } # Function to add a location to PATH. @@ -48,23 +70,21 @@ Function Get-CleanPSProfile { New-Item -Path $profile -ItemType "file" -Force } Set-Content $current_profile -Value '' - if ($null -eq (Get-Content $profile | FindStr $current_profile.replace('\', '\\'))) { - Add-Content $profile -Value ". $current_profile" - } + Add-ToProfile $profile $current_profile.replace('\', '\\') ". $current_profile" } # Function to install PhpManager. Function Install-PhpManager() { $repo = "mlocati/powershell-phpmanager" - $zip_file = "$php_dir\PhpManager.zip" $tag = (Invoke-RestMethod https://api.github.com/repos/$repo/tags)[0].Name - $module_path = "$php_dir\PhpManager\powershell-phpmanager-$tag\PhpManager" - Invoke-WebRequest -UseBasicParsing -Uri https://github.com/$repo/archive/$tag.zip -OutFile $zip_file - Expand-Archive -Path $zip_file -DestinationPath $php_dir\PhpManager - Import-Module $module_path - if($null -eq (Get-Content $current_profile | findstr 'powershell-phpmanager')) { - Add-Content -Path $current_profile -Value "Import-Module $module_path" + $module_path = "$bin_dir\PhpManager\powershell-phpmanager-$tag\PhpManager" + if(-not (Test-Path $module_path\PhpManager.psm1 -PathType Leaf)) { + $zip_file = "$bin_dir\PhpManager.zip" + Invoke-WebRequest -UseBasicParsing -Uri https://github.com/$repo/archive/$tag.zip -OutFile $zip_file + Expand-Archive -Path $zip_file -DestinationPath $bin_dir\PhpManager -Force } + Import-Module $module_path + Add-ToProfile $current_profile 'powershell-phpmanager' "Import-Module $module_path" } # Function to add PHP extensions. @@ -155,22 +175,22 @@ Function Add-Tool() { [string] $tool ) - if (Test-Path $php_dir\$tool) { - Remove-Item $php_dir\$tool + if (Test-Path $bin_dir\$tool) { + Remove-Item $bin_dir\$tool } if ($tool -eq "symfony") { - Invoke-WebRequest -UseBasicParsing -Uri $url -OutFile $php_dir\$tool.exe - Add-Content -Path $current_profile -Value "New-Alias $tool $php_dir\$tool.exe" >$null 2>&1 + Invoke-WebRequest -UseBasicParsing -Uri $url -OutFile $bin_dir\$tool.exe + Add-ToProfile $current_profile $tool "New-Alias $tool $bin_dir\$tool.exe" >$null 2>&1 } else { try { - Invoke-WebRequest -UseBasicParsing -Uri $url -OutFile $php_dir\$tool + Invoke-WebRequest -UseBasicParsing -Uri $url -OutFile $bin_dir\$tool $bat_content = @() $bat_content += "@ECHO off" $bat_content += "setlocal DISABLEDELAYEDEXPANSION" $bat_content += "SET BIN_TARGET=%~dp0/" + $tool $bat_content += "php %BIN_TARGET% %*" - Set-Content -Path $php_dir\$tool.bat -Value $bat_content - Add-Content -Path $current_profile -Value "New-Alias $tool $php_dir\$tool.bat" >$null 2>&1 + Set-Content -Path $bin_dir\$tool.bat -Value $bat_content + Add-ToProfile $current_profile $tool "New-Alias $tool $bin_dir\$tool.bat" >$null 2>&1 } catch { } } if($tool -eq "phan") { @@ -179,7 +199,7 @@ Function Add-Tool() { } elseif($tool -eq "phive") { Add-Extension xml >$null 2>&1 } elseif($tool -eq "cs2pr") { - (Get-Content $php_dir/cs2pr).replace('exit(9)', 'exit(0)') | Set-Content $php_dir/cs2pr + (Get-Content $bin_dir/cs2pr).replace('exit(9)', 'exit(0)') | Set-Content $bin_dir/cs2pr } elseif($tool -eq "composer") { composer -q global config process-timeout 0 Write-Output "::add-path::$env:APPDATA\Composer\vendor\bin" @@ -191,9 +211,9 @@ Function Add-Tool() { composer -q global config repos.packagist composer https://repo-ca-bhs-1.packagist.org } } elseif($tool -eq "wp-cli") { - Copy-Item $php_dir\wp-cli.bat -Destination $php_dir\wp.bat + Copy-Item $bin_dir\wp-cli.bat -Destination $bin_dir\wp.bat } - if (((Get-ChildItem -Path $php_dir/* | Where-Object Name -Match "^$tool(.exe|.phar)*$").Count -gt 0)) { + if (((Get-ChildItem -Path $bin_dir/* | Where-Object Name -Match "^$tool(.exe|.phar)*$").Count -gt 0)) { Add-Log $tick $tool "Added" } else { Add-Log $cross $tool "Could not add $tool" @@ -237,10 +257,10 @@ Function Add-Blackfire() { $agent_data = Invoke-WebRequest https://blackfire.io/docs/up-and-running/update | ForEach-Object { $_.tostring() -split "[`r`n]" | Select-String '