Skip to content

Commit

Permalink
feat: add install dir check
Browse files Browse the repository at this point in the history 
Signed-off-by: Chawye Hsu <[email protected]>
  • Loading branch information
@chawyehsu
chawyehsu committed Jan 10, 2024
1 parent 656e17b commit 5a53f9c
Showing 1 changed file with 46 additions and 41 deletions.
87 changes: 46 additions & 41 deletions install.ps1
View file
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ function Deny-Install {
)

Write-InstallInfo -String $message -ForegroundColor DarkRed
Write-InstallInfo "Abort."
Write-InstallInfo 'Abort.'

# Don't abort if invoked with iex that would close the PS session
if ($IS_EXECUTED_FROM_IEX) {
Expand All @@ -107,12 +107,17 @@ function Deny-Install {

function Test-ValidateParameter {
if ($null -eq $Proxy -and ($null -ne $ProxyCredential -or $ProxyUseDefaultCredentials)) {
Deny-Install "Provide a valid proxy URI for the -Proxy parameter when using the -ProxyCredential or -ProxyUseDefaultCredentials."
Deny-Install 'Provide a valid proxy URI for the -Proxy parameter when using the -ProxyCredential or -ProxyUseDefaultCredentials.'
}

if ($ProxyUseDefaultCredentials -and $null -ne $ProxyCredential) {
Deny-Install "ProxyUseDefaultCredentials is conflict with ProxyCredential. Don't use the -ProxyCredential and -ProxyUseDefaultCredentials together."
}

# Ensure the directory to install scoop is empty
if ((Test-Path $SCOOP_DIR -PathType Container) -and [bool](Get-ChildItem $SCOOP_DIR -Force)) {
Deny-Install "You are trying to install Scoop to a non-empty directory '$SCOOP_DIR', please choose another directory."
}
}

function Test-IsAdministrator {
Expand All @@ -124,12 +129,12 @@ function Test-IsAdministrator {
function Test-Prerequisite {
# Scoop requires PowerShell 5 at least
if (($PSVersionTable.PSVersion.Major) -lt 5) {
Deny-Install "PowerShell 5 or later is required to run Scoop. Go to https://microsoft.com/powershell to get the latest version of PowerShell."
Deny-Install 'PowerShell 5 or later is required to run Scoop. Go to https://microsoft.com/powershell to get the latest version of PowerShell.'
}

# Scoop requires TLS 1.2 SecurityProtocol, which exists in .NET Framework 4.5+
if ([System.Enum]::GetNames([System.Net.SecurityProtocolType]) -notcontains 'Tls12') {
Deny-Install "Scoop requires .NET Framework 4.5+ to work. Go to https://microsoft.com/net/download to get the latest version of .NET Framework."
Deny-Install 'Scoop requires .NET Framework 4.5+ to work. Go to https://microsoft.com/net/download to get the latest version of .NET Framework.'
}

# Ensure Robocopy.exe is accessible
Expand All @@ -139,13 +144,13 @@ function Test-Prerequisite {

# Detect if RunAsAdministrator, there is no need to run as administrator when installing Scoop.
if (!$RunAsAdmin -and (Test-IsAdministrator)) {
Deny-Install "Running the installer as administrator is disabled by default, see https://github.com/ScoopInstaller/Install#for-admin for details."
Deny-Install 'Running the installer as administrator is disabled by default, see https://github.com/ScoopInstaller/Install#for-admin for details.'
}

# Show notification to change execution policy
$allowedExecutionPolicy = @('Unrestricted', 'RemoteSigned', 'ByPass')
if ((Get-ExecutionPolicy).ToString() -notin $allowedExecutionPolicy) {
Deny-Install "PowerShell requires an execution policy in [$($allowedExecutionPolicy -join ", ")] to run Scoop. For example, to set the execution policy to 'RemoteSigned' please run 'Set-ExecutionPolicy RemoteSigned -Scope CurrentUser'."
Deny-Install "PowerShell requires an execution policy in [$($allowedExecutionPolicy -join ', ')] to run Scoop. For example, to set the execution policy to 'RemoteSigned' please run 'Set-ExecutionPolicy RemoteSigned -Scope CurrentUser'."
}

# Test if scoop is installed, by checking if scoop command exists.
Expand All @@ -168,7 +173,7 @@ function Optimize-SecurityProtocol {
# Set to TLS 1.2 (3072), then TLS 1.1 (768), and TLS 1.0 (192). Ssl3 has been superseded,
# https://docs.microsoft.com/en-us/dotnet/api/system.net.securityprotocoltype?view=netframework-4.5
[System.Net.ServicePointManager]::SecurityProtocol = 3072 -bor 768 -bor 192
Write-Verbose "SecurityProtocol has been updated to support TLS 1.2"
Write-Verbose 'SecurityProtocol has been updated to support TLS 1.2'
}
}

Expand All @@ -181,7 +186,7 @@ function Get-Downloader {
} elseif ($Proxy) {
# Prepend protocol if not provided
if (!$Proxy.IsAbsoluteUri) {
$Proxy = New-Object System.Uri("http://" + $Proxy.OriginalString)
$Proxy = New-Object System.Uri('http://' + $Proxy.OriginalString)
}

$Proxy = New-Object System.Net.WebProxy($Proxy)
Expand Down Expand Up @@ -268,7 +273,7 @@ function Expand-ZipArchive {
function Out-UTF8File {
param(
[Parameter(Mandatory = $True, Position = 0)]
[Alias("Path")]
[Alias('Path')]
[String] $FilePath,
[Switch] $Append,
[Switch] $NoNewLine,
Expand All @@ -293,7 +298,7 @@ function Out-UTF8File {
}

function Import-ScoopShim {
Write-InstallInfo "Creating shim..."
Write-InstallInfo 'Creating shim...'
# The scoop executable
$path = "$SCOOP_APP_DIR\bin\scoop.ps1"

Expand Down Expand Up @@ -331,31 +336,31 @@ function Import-ScoopShim {
# make ps1 accessible from cmd.exe
@(
"@rem $absolutePath",
"@echo off",
"setlocal enabledelayedexpansion",
"set args=%*",
":: replace problem characters in arguments",
'@echo off',
'setlocal enabledelayedexpansion',
'set args=%*',
':: replace problem characters in arguments',
"set args=%args:`"='%",
"set args=%args:(=``(%",
"set args=%args:)=``)%",
"set invalid=`"='",
"if !args! == !invalid! ( set args= )",
"where /q pwsh.exe",
"if %errorlevel% equ 0 (",
'if !args! == !invalid! ( set args= )',
'where /q pwsh.exe',
'if %errorlevel% equ 0 (',
" pwsh -noprofile -ex unrestricted -file `"$absolutePath`" $arg %args%",
") else (",
') else (',
" powershell -noprofile -ex unrestricted -file `"$absolutePath`" $arg %args%",
")"
')'
) -join "`r`n" | Out-UTF8File "$shim.cmd"

@(
"#!/bin/sh",
'#!/bin/sh',
"# $absolutePath",
"if command -v pwsh.exe > /dev/null 2>&1; then",
'if command -v pwsh.exe > /dev/null 2>&1; then',
" pwsh.exe -noprofile -ex unrestricted -file `"$absolutePath`" $arg `"$@`"",
"else",
'else',
" powershell.exe -noprofile -ex unrestricted -file `"$absolutePath`" $arg `"$@`"",
"fi"
'fi'
) -join "`n" | Out-UTF8File $shim -NoNewLine
}

Expand All @@ -377,13 +382,13 @@ function Get-Env {
}

function Publish-Env {
if (-not ("Win32.NativeMethods" -as [Type])) {
Add-Type -Namespace Win32 -Name NativeMethods -MemberDefinition @"
if (-not ('Win32.NativeMethods' -as [Type])) {
Add-Type -Namespace Win32 -Name NativeMethods -MemberDefinition @'
[DllImport("user32.dll", SetLastError = true, CharSet = CharSet.Auto)]
public static extern IntPtr SendMessageTimeout(
IntPtr hWnd, uint Msg, UIntPtr wParam, string lParam,
uint fuFlags, uint uTimeout, out UIntPtr lpdwResult);
"@
'@
}

$HWND_BROADCAST = [IntPtr] 0xffff
Expand All @@ -393,7 +398,7 @@ public static extern IntPtr SendMessageTimeout(
[Win32.Nativemethods]::SendMessageTimeout($HWND_BROADCAST,
$WM_SETTINGCHANGE,
[UIntPtr]::Zero,
"Environment",
'Environment',
2,
5000,
[ref] $result
Expand Down Expand Up @@ -440,7 +445,7 @@ function Add-ShimsDirToPath {
}

if (!($h -eq '\')) {
$friendlyPath = "$SCOOP_SHIMS_DIR" -Replace ([Regex]::Escape($h)), "~\"
$friendlyPath = "$SCOOP_SHIMS_DIR" -Replace ([Regex]::Escape($h)), '~\'
Write-InstallInfo "Adding $friendlyPath to your path."
} else {
Write-InstallInfo "Adding $SCOOP_SHIMS_DIR to your path."
Expand Down Expand Up @@ -552,7 +557,7 @@ function Test-CommandAvailable {
}

function Install-Scoop {
Write-InstallInfo "Initializing..."
Write-InstallInfo 'Initializing...'
# Validate install parameters
Test-ValidateParameter
# Check prerequisites
Expand All @@ -561,7 +566,7 @@ function Install-Scoop {
Optimize-SecurityProtocol

# Download scoop from GitHub
Write-InstallInfo "Downloading ..."
Write-InstallInfo 'Downloading ...'
$downloader = Get-Downloader
[bool]$downloadZipsRequired = $True

Expand All @@ -577,12 +582,12 @@ function Install-Scoop {
Write-Verbose "Cloning $SCOOP_PACKAGE_GIT_REPO to $SCOOP_APP_DIR"
git clone -q $SCOOP_PACKAGE_GIT_REPO $SCOOP_APP_DIR
if (-Not $?) {
throw "Cloning failed. Falling back to downloading zip files."
throw 'Cloning failed. Falling back to downloading zip files.'
}
Write-Verbose "Cloning $SCOOP_MAIN_BUCKET_GIT_REPO to $SCOOP_MAIN_BUCKET_DIR"
git clone -q $SCOOP_MAIN_BUCKET_GIT_REPO $SCOOP_MAIN_BUCKET_DIR
if (-Not $?) {
throw "Cloning failed. Falling back to downloading zip files."
throw 'Cloning failed. Falling back to downloading zip files.'
}
$downloadZipsRequired = $False
} catch {
Expand Down Expand Up @@ -611,7 +616,7 @@ function Install-Scoop {
$downloader.downloadFile($SCOOP_MAIN_BUCKET_REPO, $scoopMainZipfile)

# Extract files from downloaded zip
Write-InstallInfo "Extracting..."
Write-InstallInfo 'Extracting...'
# 1. extract scoop
$scoopUnzipTempDir = "$SCOOP_APP_DIR\_tmp"
Write-Verbose "Extracting $scoopZipfile to $scoopUnzipTempDir"
Expand All @@ -636,22 +641,22 @@ function Install-Scoop {
# Setup initial configuration of Scoop
Add-DefaultConfig

Write-InstallInfo "Scoop was installed successfully!" -ForegroundColor DarkGreen
Write-InstallInfo 'Scoop was installed successfully!' -ForegroundColor DarkGreen
Write-InstallInfo "Type 'scoop help' for instructions."
}

function Write-DebugInfo {
param($BoundArgs)

Write-Verbose "-------- PSBoundParameters --------"
Write-Verbose '-------- PSBoundParameters --------'
$BoundArgs.GetEnumerator() | ForEach-Object { Write-Verbose $_ }
Write-Verbose "-------- Environment Variables --------"
Write-Verbose '-------- Environment Variables --------'
Write-Verbose "`$env:USERPROFILE: $env:USERPROFILE"
Write-Verbose "`$env:ProgramData: $env:ProgramData"
Write-Verbose "`$env:SCOOP: $env:SCOOP"
Write-Verbose "`$env:SCOOP_CACHE: $SCOOP_CACHE"
Write-Verbose "`$env:SCOOP_GLOBAL: $env:SCOOP_GLOBAL"
Write-Verbose "-------- Selected Variables --------"
Write-Verbose '-------- Selected Variables --------'
Write-Verbose "SCOOP_DIR: $SCOOP_DIR"
Write-Verbose "SCOOP_CACHE_DIR: $SCOOP_CACHE_DIR"
Write-Verbose "SCOOP_GLOBAL_DIR: $SCOOP_GLOBAL_DIR"
Expand All @@ -678,11 +683,11 @@ $SCOOP_CONFIG_HOME = $env:XDG_CONFIG_HOME, "$env:USERPROFILE\.config" | Select-O
$SCOOP_CONFIG_FILE = "$SCOOP_CONFIG_HOME\scoop\config.json"

# TODO: Use a specific version of Scoop and the main bucket
$SCOOP_PACKAGE_REPO = "https://github.com/ScoopInstaller/Scoop/archive/master.zip"
$SCOOP_MAIN_BUCKET_REPO = "https://github.com/ScoopInstaller/Main/archive/master.zip"
$SCOOP_PACKAGE_REPO = 'https://github.com/ScoopInstaller/Scoop/archive/master.zip'
$SCOOP_MAIN_BUCKET_REPO = 'https://github.com/ScoopInstaller/Main/archive/master.zip'

$SCOOP_PACKAGE_GIT_REPO = "https://github.com/ScoopInstaller/Scoop.git"
$SCOOP_MAIN_BUCKET_GIT_REPO = "https://github.com/ScoopInstaller/Main.git"
$SCOOP_PACKAGE_GIT_REPO = 'https://github.com/ScoopInstaller/Scoop.git'
$SCOOP_MAIN_BUCKET_GIT_REPO = 'https://github.com/ScoopInstaller/Main.git'

# Quit if anything goes wrong
$oldErrorActionPreference = $ErrorActionPreference
Expand Down

0 comments on commit 5a53f9c

Please sign in to comment.