install.ps1

function Install-Cargo-Shuttle {
    # Anonymous telemetry
    $TELEMETRY = "1"
    $PLATFORM = "windows"
    $NEW_INSTALL = "true"
    $INSTALL_METHOD = ""
    $OUTCOME = ""
    $STEP_FAILED = "N/A"
    $STARTED_AT = (Get-Date -Format "o")

    # Disable telemetry if any opt-out vars are set
    if ($env:DO_NOT_TRACK -eq "1" -or $env:DO_NOT_TRACK -eq "true" -or
        $env:DISABLE_TELEMETRY -eq "1" -or $env:DISABLE_TELEMETRY -eq "true" -or
        $env:SHUTTLE_DISABLE_TELEMETRY -eq "1" -or $env:SHUTTLE_DISABLE_TELEMETRY -eq "true" -or
        $env:CI -eq "1" -or $env:CI -eq "true") {
        $TELEMETRY = "0"
    }

    Write-Host "       ___                                  " -NoNewline -ForegroundColor Red -BackgroundColor Black; Write-Host "" -ForegroundColor White -BackgroundColor Black
    Write-Host "      /   \" -NoNewline -ForegroundColor Red -BackgroundColor Black; Write-Host "    _           _   _   _        " -ForegroundColor White -BackgroundColor Black
    Write-Host "   __/    /" -NoNewline -ForegroundColor Red -BackgroundColor Black; Write-Host "___| |__  _   _| |_| |_| | ___   " -ForegroundColor White -BackgroundColor Black
    Write-Host "  /_     /" -NoNewline -ForegroundColor Red -BackgroundColor Black; Write-Host "/ __| '_ \| | | | __| __| |/ _ \  " -ForegroundColor White -BackgroundColor Black
    Write-Host "   _|_  | " -NoNewline -ForegroundColor Red -BackgroundColor Black; Write-Host "\__ \ | | | |_| | |_| |_| |  __/  " -ForegroundColor White -BackgroundColor Black
    Write-Host "  |_| |/  " -NoNewline -ForegroundColor Red -BackgroundColor Black; Write-Host "|___/_| |_|\__,_|\__|\__|_|\___|  " -ForegroundColor White -BackgroundColor Black
    Write-Host "                                            " -ForegroundColor White -BackgroundColor Black
    Write-Host
    Write-Host @"
https://docs.shuttle.dev
https://discord.gg/shuttle
https://github.com/shuttle-hq/shuttle

Please open an issue if you encounter any problems!
"@
    if ($TELEMETRY -eq "1") {
        Write-Host "Anonymous telemetry enabled. More info and opt-out:" -ForegroundColor Gray
        Write-Host "https://docs.shuttle.dev/install-script" -ForegroundColor Gray
    }
    Write-Host "==================================================="
    Write-Host

    function Send-Telemetry {
        if ($TELEMETRY -eq "1") {
            $ENDED_AT = (Get-Date -Format "o")
            $telemetry_data = @{
                api_key = "phc_cQMQqF5QmcEzXEaVlrhv3yBSNRyaabXYAyiCV7xKHUH"
                distinct_id = "install_script"
                event = "install_script_completion"
                properties = @{
                    platform = $PLATFORM
                    new_install = $NEW_INSTALL
                    install_method = $INSTALL_METHOD
                    started_at = $STARTED_AT
                    ended_at = $ENDED_AT
                    outcome = $OUTCOME
                    step_failed = $STEP_FAILED
                    dont_track_ip = $true
                }
            } | ConvertTo-Json -Depth 3

            if ($env:SHUTTLE_DEBUG) {
                Write-Host "DEBUG: Sending telemetry data:`n$telemetry_data"
            }
            Invoke-RestMethod -Uri "https://console.shuttle.dev/ingest/i/v0/e" -Method Post -ContentType "application/json" -Body $telemetry_data -ErrorAction SilentlyContinue | Out-Null
        }
    }

    function Exit-Success {
        $OUTCOME = "success"
        Send-Telemetry
        Write-Host ""
        Write-Host "Thanks for installing Shuttle CLI!" -ForegroundColor Green
    }

    function Exit-Neutral {
        $OUTCOME = "neutral"
        Write-Host ""
        Write-Host "If you have any problems, please open an issue on GitHub or visit our Discord!"
        Send-Telemetry
    }

    function Exit-Failure {
        param($StepFailed)
        $STEP_FAILED = $StepFailed
        $OUTCOME = "failure"
        Write-Host ""
        Write-Host "Shuttle installation script failed with reason: $STEP_FAILED" -ForegroundColor Red
        Write-Host "If you have any problems, please open an issue on GitHub or visit our Discord!"
        Send-Telemetry
    }

    $RepoUrl = "https://github.com/shuttle-hq/shuttle"
    (Invoke-WebRequest "$RepoUrl/releases/latest" -Headers @{ "Accept" = "application/json" }).Content -match '"tag_name":"([^"]*)"' | Out-Null
    if (!$?) { return Exit-Failure "check-latest-release" }
    $LatestRelease = $Matches.1
    if ($LatestRelease -eq $null) { return Exit-Failure "parse-latest-version" }

    if (Get-Command -CommandType Application -ErrorAction SilentlyContinue cargo-shuttle.exe) {
        $NEW_INSTALL = "false"
        $LatestReleaseStripped = $LatestRelease -replace '^v', ''
        $CurrentVersion = & cargo-shuttle.exe -V
        $CurrentVersionStripped = $CurrentVersion -replace '^cargo-shuttle ', ''
        if ($LatestReleaseStripped -eq $CurrentVersionStripped) {
            Write-Host "Shuttle CLI is already at the latest version!" -ForegroundColor Green
            return
        }
        else {
            Write-Host "Updating Shuttle CLI to $LatestRelease"
        }
    }

    $Arch = [Environment]::GetEnvironmentVariable("PROCESSOR_ARCHITECTURE", [EnvironmentVariableTarget]::Machine)
    $TempDir = $Env:TEMP

    if (!(Get-Command -CommandType Application -ErrorAction SilentlyContinue cargo.exe)) {
        Write-Host "Could not find cargo.exe, Rust may not be installed." -ForegroundColor Red
        if ($Arch -inotin "AMD64", "x86") {
            Write-Host "Rustup is only provided for x86 and x86_64, not $Arch" -ForegroundColor Red
            Write-Host "Please install Rust manually, more info at: https://rust-lang.github.io/rustup/installation/other.html" -ForegroundColor Red
            return Exit-Failure "unsupported-arch"
        }
        $Confirm = Read-Host -Prompt "Install Rust and Cargo via rustup? [Y/n]"
        if ($Confirm -inotin "y", "Y", "yes", "") {
            Write-Host "Skipping rustup install, cargo-shuttle not installed"
            return Exit-Neutral
        }
        $RustupUrl = if ($Arch -eq "AMD64") { "https://win.rustup.rs/x86_64" } else { "https://win.rustup.rs/i686" }
        Invoke-WebRequest $RustupUrl -OutFile "$TempDir\rustup-init.exe"
        if (!$?) { return Exit-Failure "get-rustup" }
        & "$TempDir\rustup-init.exe"
        if (!$?) {
            Remove-Item -ErrorAction SilentlyContinue "$TempDir\rustup.exe"
            Write-Host "Rust install via Rustup failed, please install Rust manually: https://rustup.rs/" -ForegroundColor Red
            return Exit-Failure "install-rust"
        }
        Remove-Item -ErrorAction SilentlyContinue "$TempDir\rustup.exe"
        Write-Host "Rust installed via Rustup, please re-run this script, you probably need reopen your terminal" -ForegroundColor Green
        return Exit-Neutral
    }

    if (Get-Command -CommandType Application -ErrorAction SilentlyContinue cargo-binstall.exe) {
        Write-Host "Installing Shuttle CLI using cargo-binstall"
        $INSTALL_METHOD = "cargo-binstall"
        cargo-binstall.exe -y --force --locked cargo-shuttle
        if (!$?) {
            Write-Host "Could not install from release using cargo-binstall" -ForegroundColor Red
            return Exit-Failure "cargo-binstall"
        }
        return Exit-Success
    }
    Write-Host "cargo-binstall not found, trying manual binary download" -ForegroundColor Red

    if (($Arch -eq "AMD64") -and (Get-Command -CommandType Application -ErrorAction SilentlyContinue tar.exe)) {
        $INSTALL_METHOD = "binary-download"
        $BinaryUrl = "$RepoUrl/releases/download/$LatestRelease/cargo-shuttle-$LatestRelease-x86_64-pc-windows-msvc.tar.gz"
        Invoke-WebRequest $BinaryUrl -OutFile "$TempDir\cargo-shuttle.tar.gz"
        if (!$?) { return Exit-Failure "download-binary" }
        New-Item -ItemType Directory -Force "$TempDir\cargo-shuttle" | Out-Null
        if (!$?) { return Exit-Failure "temp-folder" }
        tar.exe -xzf "$TempDir\cargo-shuttle.tar.gz" -C "$TempDir\cargo-shuttle"
        if (!$?) { return Exit-Failure "tar-extract-binary" }
        $CargoDir = if ($null -ne $Env:CARGO_HOME) { $Env:CARGO_HOME } else { "$HOME\.cargo" }
        $CargoInstallDir = if ($null -ne $Env:CARGO_INSTALL_ROOT) { $Env:CARGO_INSTALL_ROOT } else { "$CargoDir" }
        $CargoInstallBinDir = "$CargoInstallDir\bin"
        New-Item -ItemType Directory -Force "$CargoInstallBinDir" | Out-Null
        if (!$?) { return Exit-Failure "binary-folder" }
        Write-Host "Installing to $CargoInstallBinDir\cargo-shuttle.exe"
        Move-Item -Force "$TempDir\cargo-shuttle\cargo-shuttle-x86_64-pc-windows-msvc-$LatestRelease\cargo-shuttle.exe" "$CargoInstallBinDir\cargo-shuttle.exe"
        if (!$?) { return Exit-Failure "move-binary" }
        Write-Host "Installing to $CargoInstallBinDir\shuttle.exe"
        Move-Item -Force "$TempDir\cargo-shuttle\cargo-shuttle-x86_64-pc-windows-msvc-$LatestRelease\shuttle.exe" "$CargoInstallBinDir\shuttle.exe"
        if (!$?) { return Exit-Failure "move-binary" }
        Remove-Item -Recurse -Force -ErrorAction SilentlyContinue "$TempDir\cargo-shuttle.tar.gz", "$TempDir\cargo-shuttle"
        return Exit-Success
    }
    elseif ($Arch -ne "AMD64") {
        Write-Host "Unsupported Architecture: Binaries are not currently built for $Arch, skipping manual binary download" -ForegroundColor Red
    }
    else {
        Write-Host "Could not find tar.exe, skipping manual binary download (required to extract the release asset)" -ForegroundColor Red
    }

    $INSTALL_METHOD = "cargo"
    if (!(Get-Command -CommandType Application -ErrorAction SilentlyContinue cargo.exe)) {
        return Exit-Failure "cargo-not-found"
    }

    Write-Host "Installing cargo-shuttle using cargo install (from source)"
    cargo.exe install --locked cargo-shuttle
    if (!$?) { return Exit-Failure "cargo-install" }
    return Exit-Success
}


$OldErrorAction = $ErrorActionPreference
$ErrorActionPreference = "Stop"
Install-Cargo-Shuttle
$ErrorActionPreference = $OldErrorAction