#!/usr/bin/env pwsh
#350 check if windows user run as admin

# this allows us to call ps1 files, which allows us to have spaces in filenames
# ('powershell "$Env:USERPROFILE\test.ps1" foo' will fail if it has a space in
# the path but '& "$Env:USERPROFILE\test.ps1" foo' will work even with a space)
Set-ExecutionPolicy -Scope Process Bypass

# If a command returns an error, halt the script.
$ErrorActionPreference = 'Stop'

# Ignore progress events from cmdlets so Invoke-WebRequest is not painfully slow
$ProgressPreference = 'SilentlyContinue'

$Env:WEBI_HOST = 'https://webi.run'
$Env:WEBI_PKG = 'wsl@0.0.0'
$Env:PKG_NAME = 'wsl'
$Env:WEBI_VERSION = '0.0.0'
$Env:WEBI_GIT_TAG = ''
$Env:WEBI_PKG_URL = 'https://example.com/doesntexist.ext'
$Env:WEBI_PKG_FILE = 'doesntexist.ext'
$Env:WEBI_PKG_PATHNAME = 'doesntexist.ext'

# Switch to userprofile
Push-Location $Env:USERPROFILE

# Make paths
New-Item -Path "$Env:USERPROFILE\Downloads" -ItemType Directory -Force | Out-Null
New-Item -Path "$Env:USERPROFILE\.local\bin" -ItemType Directory -Force | Out-Null
New-Item -Path "$Env:USERPROFILE\.local\opt" -ItemType Directory -Force | Out-Null

# {{ baseurl }}
# {{ version }}

# See
#   - <https://superuser.com/q/1264444>
#   - <https://stackoverflow.com/a/60572643/151312>
$Esc = [char]27
$TTask = "${Esc}[36m"
$TName = "${Esc}[1m${Esc}[32m"
$TUrl = "${Esc}[2m"
$TPath = "${Esc}[2m${Esc}[32m"
$TCmd = "${Esc}[2m${Esc}[35m"
$TDim = "${Esc}[2m"
$TReset = "${Esc}[0m"

function Invoke-DownloadUrl {
    Param (
        [string]$URL,
        [string]$Params,
        [string]$Path,
        [switch]$Force
    )

    IF (Test-Path -Path "$Path") {
        IF (-Not $Force.IsPresent) {
            Write-Host "    ${TDim}Found${TReset} $Path"
            return
        }
        Write-Host "    Updating ${TDim}${Path}${TDim}"
    }

    $TmpPath = "${Path}.part"
    Remove-Item -Path $TmpPath -Force -ErrorAction Ignore

    Write-Host "    Downloading ${TDim}from${TReset}"
    Write-Host "      ${TDim}${URL}${TReset}"
    IF ($Params.Length -ne 0) {
        Write-Host "        ?$Params"
        $URL = "${URL}?${Params}"
    }
    curl.exe '-#' -f -sS -A $Env:WEBI_UA -L $URL | Out-File $TmpPath

    Remove-Item -Path $Path -Force -ErrorAction Ignore
    Move-Item $TmpPath $Path
    Write-Host "      Saved ${TPath}${Path}${TReset}"
}

function Get-UserAgent {
    # This is the canonical CPU arch when the process is emulated
    $my_arch = "$Env:PROCESSOR_ARCHITEW6432"

    IF ($my_arch -eq $null -or $my_arch -eq "") {
        # This is the canonical CPU arch when the process is native
        $my_arch = "$Env:PROCESSOR_ARCHITECTURE"
    }

    IF ($my_arch -eq "AMD64") {
        # Because PowerShell is sometimes AMD64 on Windows 10 ARM
        # See https://oofhours.com/2020/02/04/powershell-on-windows-10-arm64/
        $my_os_arch = (Get-CimInstance -ClassName Win32_OperatingSystem).OSArchitecture

        # Using -clike because of the trailing newline
        IF ($my_os_arch -clike "ARM 64*") {
            $my_arch = "ARM64"
        }
    }

    "PowerShell+curl Windows/10+ $my_arch msvc"
}

function webi_path_add($pathname) {
    # C:\Users\me => C:/Users/me
    $my_home = $Env:UserProfile
    $my_home = $my_home.replace('\\', '/')
    $my_home_re = [regex]::escape($my_home)

    # ~/bin => %USERPROFILE%/bin
    $pathname = $pathname.replace('~/', "$Env:UserProfile/")

    # C:\Users\me\bin => %USERPROFILE%/bin
    $my_pathname = $pathname.replace('\\', '/')
    $my_pathname = $my_pathname -ireplace $my_home_re, "%USERPROFILE%"

    $all_user_paths = [Environment]::GetEnvironmentVariable("Path", "User")
    $user_paths = "${all_user_paths}".Trim(';').Split(';')
    $exists_in_path = $false
    foreach ($user_path in $user_paths) {
        # C:\Users\me\bin => %USERPROFILE%/bin
        $my_user_path = $user_path.replace('\\', '/')
        $my_user_path = $my_user_path -ireplace $my_home_re, "%USERPROFILE%"

        if ($my_user_path -ieq $my_pathname) {
            $exists_in_path = $true
        }
    }
    if (-Not $exists_in_path) {
        $all_user_paths = "${pathname};${all_user_paths}".Trim(';')
        [Environment]::SetEnvironmentVariable("Path", $all_user_paths, "User")
        $null = Sync-EnvPath
    }
}

function Sync-EnvPath {
    $UserPath = [Environment]::GetEnvironmentVariable("Path", "User").Trim(';')
    $MachinePath = [Environment]::GetEnvironmentVariable("Path", "Machine").Trim(';')
    $Env:Path = "${UserPath};${MachinePath}"
    [Environment]::SetEnvironmentVariable("Path", $Env:Path)
    $Env:Path
}

$Env:WEBI_UA = Get-UserAgent

#$has_local_bin = echo "$Env:PATH" | Select-String -Pattern '\.local.bin'
#if (!$has_local_bin)
#{
webi_path_add ~/.local/bin
#}
        #!/usr/bin/env pwsh

Write-Output "Installing WSL (Windows Subsystem for Linux) ..."
Write-Output ""
Write-Output "Security: requires administrator approval to install"

IF ($null -eq $Env:WEBI_HOST -or $Env:WEBI_HOST -eq "") { $Env:WEBI_HOST = "https://webinstall.dev" }

# From https://devblogs.microsoft.com/scripting/use-a-powershell-function-to-see-if-a-command-exists/
Function Test-CommandExist {
    Param ($command)
    $oldPreference = $ErrorActionPreference
    $ErrorActionPreference = 'stop'
    try { if (Get-Command $command) { RETURN $true } }
    Catch { RETURN $false }
    Finally { $ErrorActionPreference = $oldPreference }
}


# $MYPWD = (Get-Item .).FullName
& curl.exe -fsSA "MS" -o "$Env:TEMP\install-wsl2.ps1" "$Env:WEBI_HOST/packages/wsl/install-wsl.ps1"
powershell -Command "Start-Process cmd -Wait -Verb RunAs -ArgumentList '/c cd /d %CD% && powershell -ExecutionPolicy Bypass $Env:TEMP\install-wsl2.ps1'"

IF (!(Test-CommandExists wsl)) {
    Write-Output "Warning: Skipping 3 of 5: Reboot Required to install WSL 2 !!"
}

Write-Output ""
IF ((Test-Path -Path "$Env:UserProfile\Downloads\webi\Ubuntu_2004_x64.appx" )) {
    Write-Output "Skipping 4 of 5: Ubuntu Linux 20.04 already installed"
}
ELSE {
    Write-Output "Installing 4 of 5 Ubuntu Linux 20.04 (for WSL 1 and WSL 2) ..."
    curl.exe -fL -o "$Env:UserProfile\Downloads\webi\Ubuntu_2004_x64.appx.part" https://aka.ms/wslubuntu2004
    & Move-Item "$Env:UserProfile\Downloads\webi\Ubuntu_2004_x64.appx.part" "$Env:UserProfile\Downloads\webi\Ubuntu_2004_x64.appx"
    Add-AppxPackage "$Env:UserProfile\Downloads\webi\Ubuntu_2004_x64.appx"
}

Write-Output ""
Write-Output ""
Write-Output ""
Write-Output "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!"
Write-Output "!!!                           !!!"
Write-Output "!!!      ACTION REQUIRED      !!!"
Write-Output "!!!      READ CAREFULLY!      !!!"
Write-Output "!!!                           !!!"
Write-Output "!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!"

IF (!(Test-CommandExists wsl)) {
    Write-Output ""
    Write-Output "WSL 2 was NOT installed yet. FOLLOW these instructions:"
    Write-Output ""
    Write-Output "    1. REBOOT you computer to finish the WSL 1 install"
    Write-Output "       (either click Start Menu => Restart, or run 'shutdown /r /t 5')"
    Write-Output ""
    Write-Output "    2. RE-RUN this WSL 2 installer"
    Write-Output "       (WSL 2 cannot finish installing until the WSL 1 install is complete)"
    Write-Output ""
    Write-Output "    3. WSL 2 must be enabled manually. See https://webinstall.dev/wsl2"
    Write-Output ""

    Exit
}

Write-Output ""
Write-Output "You must ALSO run UBUNTU LINUX from the START MENU to complete the install."
Write-Output ""
Write-Output "    -  Select Ubuntu Linux from the Search menu or Start Menu"
Write-Output "    -  Wait for the initialization to complete"
Write-Output "    -  Choose a username (we recommend 'app') and a password"
Write-Output ""


webi_path_add ~/.local/bin

# Done
Pop-Location