# Version: 11
# A script that installs the dependencies needed to build and test Bun.
# This should work on Windows 10 or newer with PowerShell.

# If this script does not work on your machine, please open an issue:
# https://github.com/oven-sh/bun/issues

# If you need to make a change to this script, such as upgrading a dependency,
# increment the version comment to indicate that a new image should be built.
# Otherwise, the existing image will be retroactively updated.

param (
  [Parameter(Mandatory = $false)]
  [switch]$CI = $false,
  [Parameter(Mandatory = $false)]
  [switch]$Optimize = $CI
)

$ErrorActionPreference = "Stop"
Set-ExecutionPolicy -Scope Process -ExecutionPolicy Bypass -Force

function Execute-Command {
  $command = $args -join ' '
  Write-Output "$ $command"

  & $args[0] $args[1..$args.Length]

  if ((-not $?) -or ($LASTEXITCODE -ne 0 -and $null -ne $LASTEXITCODE)) {
    throw "Command failed: $command"
  }
}

function Which {
  param ([switch]$Required = $false)

  foreach ($command in $args) {
    $result = Get-Command $command -ErrorAction SilentlyContinue
    if ($result -and $result.Path) {
      return $result.Path
    }
  }

  if ($Required) {
    $commands = $args -join ', '
    throw "Command not found: $commands"
  }
}

function Execute-Script {
  param (
    [Parameter(Mandatory = $true, Position = 0)]
    [string]$Path
  )

  $pwsh = Which pwsh powershell -Required
  Execute-Command $pwsh $Path
}

function Download-File {
  param (
    [Parameter(Mandatory = $true, Position = 0)]
    [string]$Url,
    [Parameter(Mandatory = $false)]
    [string]$Name,
    [Parameter(Mandatory = $false)]
    [string]$Path
  )

  if (-not $Name) {
    $Name = [System.IO.Path]::ChangeExtension([System.IO.Path]::GetRandomFileName(), [System.IO.Path]::GetExtension($Url))
  }

  if (-not $Path) {
    $Path = "$env:TEMP\$Name"
  }

  $client = New-Object System.Net.WebClient
  for ($i = 0; $i -lt 10 -and -not (Test-Path $Path); $i++) {
    try {
      $client.DownloadFile($Url, $Path)
    } catch {
      Write-Warning "Failed to download $Url, retry $i..."
      Start-Sleep -s $i
    }
  }

  return $Path
}

function Install-Chocolatey {
  if (Which choco) {
    return
  }

  Write-Output "Installing Chocolatey..."
  [System.Net.ServicePointManager]::SecurityProtocol = [System.Net.ServicePointManager]::SecurityProtocol -bor 3072
  $installScript = Download-File "https://community.chocolatey.org/install.ps1"
  Execute-Script $installScript
  Refresh-Path
}

function Refresh-Path {
  $paths = @(
    [System.Environment]::GetEnvironmentVariable("Path", "Machine"),
    [System.Environment]::GetEnvironmentVariable("Path", "User"),
    [System.Environment]::GetEnvironmentVariable("Path", "Process")
  )
  $uniquePaths = $paths |
    Where-Object { $_ } |
    ForEach-Object { $_.Split(';', [StringSplitOptions]::RemoveEmptyEntries) } |
    Where-Object { $_ -and (Test-Path $_) } |
    Select-Object -Unique
  $env:Path = ($uniquePaths -join ';').TrimEnd(';')

  if ($env:ChocolateyInstall) {
    Import-Module $env:ChocolateyInstall\helpers\chocolateyProfile.psm1 -ErrorAction SilentlyContinue
  }
}

function Add-To-Path {
  $absolutePath = Resolve-Path $args[0]
  $currentPath = [Environment]::GetEnvironmentVariable("Path", "Machine")
  if ($currentPath -like "*$absolutePath*") {
    return
  }

  $newPath = $currentPath.TrimEnd(";") + ";" + $absolutePath
  if ($newPath.Length -ge 2048) {
    Write-Warning "PATH is too long, removing duplicate and old entries..."

    $paths = $currentPath.Split(';', [StringSplitOptions]::RemoveEmptyEntries) |
      Where-Object { $_ -and (Test-Path $_) } |
      Select-Object -Unique

    $paths += $absolutePath
    $newPath = $paths -join ';'
    while ($newPath.Length -ge 2048 -and $paths.Count -gt 1) {
      $paths = $paths[1..$paths.Count]
      $newPath = $paths -join ';'
    }
  }

  Write-Output "Adding $absolutePath to PATH..."
  [Environment]::SetEnvironmentVariable("Path", "$newPath", "Machine")
  Refresh-Path
}

function Set-Env {
  param (
    [Parameter(Mandatory = $true, Position = 0)]
    [string]$Name,
    [Parameter(Mandatory = $true, Position = 1)]
    [string]$Value
  )

  Write-Output "Setting environment variable $Name=$Value..."
  [System.Environment]::SetEnvironmentVariable("$Name", "$Value", "Machine")
  [System.Environment]::SetEnvironmentVariable("$Name", "$Value", "Process")
}

function Install-Package {
  param (
    [Parameter(Mandatory = $true, Position = 0)]
    [string]$Name,
    [Parameter(Mandatory = $false)]
    [string]$Command = $Name,
    [Parameter(Mandatory = $false)]
    [string]$Version,
    [Parameter(Mandatory = $false)]
    [switch]$Force = $false,
    [Parameter(Mandatory = $false)]
    [string[]]$ExtraArgs = @()
  )

  if (-not $Force `
    -and (Which $Command) `
    -and (-not $Version -or (& $Command --version) -like "*$Version*")) {
    return
  }

  Write-Output "Installing $Name..."
  $flags = @(
    "--yes",
    "--accept-license",
    "--no-progress",
    "--force"
  )
  if ($Version) {
    $flags += "--version=$Version"
  }

  Execute-Command choco install $Name @flags @ExtraArgs
  Refresh-Path
}

function Install-Packages {
  foreach ($package in $args) {
    Install-Package $package
  }
}

function Install-Common-Software {
  Install-Chocolatey
  Install-Pwsh
  Install-Git
  Install-Packages curl 7zip nssm
  Install-NodeJs
  Install-Bun
  Install-Cygwin
  if ($CI) {
    # FIXME: Installing tailscale causes the AWS metadata server to become unreachable
    # Install-Tailscale
    Install-Buildkite
  }
}

function Install-Pwsh {
  Install-Package powershell-core -Command pwsh

  if ($CI) {
    $shellPath = (Which pwsh -Required)
    New-ItemProperty `
      -Path "HKLM:\\SOFTWARE\\OpenSSH" `
      -Name DefaultShell `
      -Value $shellPath `
      -PropertyType String `
      -Force
  }
}

function Install-Git {
  Install-Packages git

  if ($CI) {
    Execute-Command git config --system --add safe.directory "*"
    Execute-Command git config --system core.autocrlf false
    Execute-Command git config --system core.eol lf
    Execute-Command git config --system core.longpaths true
  }
}

function Install-NodeJs {
  Install-Package nodejs -Command node -Version "24.3.0"
}

function Install-Bun {
  Install-Package bun -Version "1.3.1"
}

function Install-Cygwin {
  Install-Package cygwin
  Add-To-Path "C:\tools\cygwin\bin"
}

function Install-Tailscale {
  Install-Package tailscale
}

function Create-Buildkite-Environment-Hooks {
  param (
    [Parameter(Mandatory = $true)]
    [string]$BuildkiteHome
  )

  Write-Output "Creating Buildkite environment hooks..."
  $hooksDir = Join-Path $BuildkiteHome "hooks"

  if (-not (Test-Path $hooksDir)) {
    New-Item -Path $hooksDir -ItemType Directory -Force | Out-Null
  }

  $environmentHook = Join-Path $hooksDir "environment.ps1"
  $buildPath = Join-Path $BuildkiteHome "build"

  @"
# Buildkite environment hook
`$env:BUILDKITE_BUILD_CHECKOUT_PATH = "$buildPath"
"@ | Set-Content -Path $environmentHook -Encoding UTF8

  Write-Output "Environment hook created at $environmentHook"
}

function Install-Buildkite {
  if (Which buildkite-agent) {
    return
  }

  Write-Output "Installing Buildkite agent..."
  $env:buildkiteAgentToken = "xxx"
  $installScript = Download-File "https://raw.githubusercontent.com/buildkite/agent/main/install.ps1"
  Execute-Script $installScript
  Refresh-Path

  if ($CI) {
    $buildkiteHome = "C:\buildkite-agent"
    if (-not (Test-Path $buildkiteHome)) {
      New-Item -Path $buildkiteHome -ItemType Directory -Force | Out-Null
    }
    Create-Buildkite-Environment-Hooks -BuildkiteHome $buildkiteHome
  }
}

function Install-Build-Essentials {
  Install-Visual-Studio
  Install-Packages `
    cmake `
    make `
    ninja `
    python `
    golang `
    nasm `
    ruby `
    strawberryperl `
    mingw
  Install-Rust
  Install-Ccache
  # Needed to remap stack traces
  Install-PdbAddr2line
  Install-Llvm
}

function Install-Visual-Studio {
  param (
    [Parameter(Mandatory = $false)]
    [string]$Edition = "community"
  )

  Write-Output "Downloading Visual Studio installer..."
  $vsInstaller = Download-File "https://aka.ms/vs/17/release/vs_$Edition.exe"

  Write-Output "Installing Visual Studio..."
  $vsInstallArgs = @(
    "--passive",
    "--norestart",
    "--wait",
    "--force",
    "--locale en-US",
    "--add Microsoft.VisualStudio.Workload.NativeDesktop",
    "--includeRecommended"
  )
  $startInfo = New-Object System.Diagnostics.ProcessStartInfo
  $startInfo.FileName = $vsInstaller
  $startInfo.Arguments = $vsInstallArgs -join ' '
  $startInfo.CreateNoWindow = $true
  $process = New-Object System.Diagnostics.Process
  $process.StartInfo = $startInfo
  $process.Start()
  $process.WaitForExit()
  if ($process.ExitCode -ne 0) {
    throw "Failed to install Visual Studio: code $($process.ExitCode)"
  }
}

function Install-Rust {
  if (Which rustc) {
    return
  }

  Write-Output "Installing Rustup..."
  $rustupInit = Download-File "https://win.rustup.rs/" -Name "rustup-init.exe"

  Write-Output "Installing Rust..."
  Execute-Command $rustupInit -y

  Write-Output "Moving Rust to $env:ProgramFiles..."
  $rustPath = Join-Path $env:ProgramFiles "Rust"
  if (-not (Test-Path $rustPath)) {
    New-Item -Path $rustPath -ItemType Directory
  }
  Move-Item "$env:UserProfile\.cargo" "$rustPath\cargo" -Force
  Move-Item "$env:UserProfile\.rustup" "$rustPath\rustup" -Force

  Write-Output "Setting environment variables for Rust..."
  Set-Env "CARGO_HOME" "$rustPath\cargo"
  Set-Env "RUSTUP_HOME" "$rustPath\rustup"
  Add-To-Path "$rustPath\cargo\bin"
}

function Install-Ccache {
  Install-Package ccache
}

function Install-PdbAddr2line {
  Execute-Command cargo install --examples "pdb-addr2line@0.11.2"
}

function Install-Llvm {
  Install-Package llvm `
    -Command clang-cl `
    -Version "19.1.7"
  Add-To-Path "$env:ProgramFiles\LLVM\bin"
}

function Optimize-System {
  Disable-Windows-Defender
  Disable-Windows-Threat-Protection
  Disable-Windows-Services
  Disable-Power-Management
}

function Optimize-System-Needs-Reboot {
  Uninstall-Windows-Defender
}

function Disable-Windows-Defender {
  Write-Output "Disabling Windows Defender..."
  Set-MpPreference -DisableRealtimeMonitoring $true
  Add-MpPreference -ExclusionPath "C:\", "D:\"
}

function Disable-Windows-Threat-Protection {
  $itemPath = "HKLM:\SOFTWARE\Policies\Microsoft\Windows Advanced Threat Protection"
  if (Test-Path $itemPath) {
    Write-Output "Disabling Windows Threat Protection..."
    Set-ItemProperty -Path $itemPath -Name "ForceDefenderPassiveMode" -Value 1 -Type DWORD
  }
}

function Uninstall-Windows-Defender {
  Write-Output "Uninstalling Windows Defender..."
  Uninstall-WindowsFeature -Name Windows-Defender
}

function Disable-Windows-Services {
  $services = @(
    "WSearch",          # Windows Search
    "wuauserv",         # Windows Update
    "DiagTrack",        # Connected User Experiences and Telemetry
    "dmwappushservice", # WAP Push Message Routing Service
    "PcaSvc",           # Program Compatibility Assistant
    "SysMain"           # Superfetch
  )

  foreach ($service in $services) {
    Stop-Service $service -Force
    Set-Service $service -StartupType Disabled
  }
}

function Disable-Power-Management {
  Write-Output "Disabling Power Management..."
  powercfg /setactive 8c5e7fda-e8bf-4a96-9a85-a6e23a8c635c # High performance
  powercfg /change monitor-timeout-ac 0
  powercfg /change monitor-timeout-dc 0
  powercfg /change standby-timeout-ac 0
  powercfg /change standby-timeout-dc 0
  powercfg /change hibernate-timeout-ac 0
  powercfg /change hibernate-timeout-dc 0
}

if ($Optimize) {
  Optimize-System
}

Install-Common-Software
Install-Build-Essentials

if ($Optimize) {
  Optimize-System-Needs-Reboot
}