# bootstrap-windows.ps1
#
# One-shot setup for building RustDesk 1.4.7 (our fork) on Windows 11.
# Run as Administrator from a PowerShell window.
#
# What it does:
#   1. Verify / install Visual Studio 2022 Build Tools (Desktop C++ workload)
#   2. Install Rust (stable-msvc) via rustup
#   3. Install Flutter 3.44+ (Chinese mirror)
#   4. Install VCPKG and the C++ deps (libvpx, libyuv, opus, aom) for x64-windows-static
#   5. Clone the fork from git.cardsoon.com
#   6. Apply the post-clone patches (google_fonts const -> final)
#
# This script is idempotent: re-running it skips already-completed steps.
# Run-time: ~30-60 minutes (mostly VCPKG building C++ deps).
#
# Usage (from an "x64 Native Tools Command Prompt for VS 2022"):
#   powershell -ExecutionPolicy Bypass -File .\res\bootstrap-windows.ps1
#
# After this finishes, run:
#   .\res\build-windows.ps1

$ErrorActionPreference = "Stop"
Set-StrictMode -Version Latest

# Pretty logging
function Write-Section { param($msg) Write-Host "`n=== $msg ===" -ForegroundColor Cyan }
function Write-Ok { param($msg) Write-Host "  [OK] $msg" -ForegroundColor Green }
function Write-Warn { param($msg) Write-Host "  [WARN] $msg" -ForegroundColor Yellow }
function Write-Err { param($msg) Write-Host "  [ERR] $msg" -ForegroundColor Red }

# ------- Config -------
$PROJECT_ROOT = "C:\prj\rustdesk"
$VCPKG_ROOT = "C:\vcpkg"
$FLUTTER_ROOT = "C:\flutter"
$FLUTTER_VERSION = "3.44.1-stable"
$RUST_TOOLCHAIN = "stable-msvc"
$GIT_URL = "https://git.cardsoon.com/xuwenwei/RustDesk.git"

# Chinese mirrors (faster in CN; remove if you're elsewhere)
$PUB_HOSTED_URL = "https://pub.flutter-io.cn"
$FLUTTER_STORAGE_BASE_URL = "https://storage.flutter-io.cn"

# ------- 1. Verify Visual Studio 2022 Build Tools -------
Write-Section "1/7  Visual Studio 2022 Build Tools"

$vsWhere = "${env:ProgramFiles(x86)}\Microsoft Visual Studio\Installer\vswhere.exe"
if (Test-Path $vsWhere) {
    $vs = & $vsWhere -latest -property installationPath 2>$null
    if ($vs) {
        Write-Ok "Found VS install: $vs"
    } else {
        Write-Err "vswhere.exe exists but returned no installation."
        Write-Host "Please install Visual Studio 2022 Build Tools with 'Desktop development with C++' workload:"
        Write-Host "  https://visualstudio.microsoft.com/visual-studio-build-tools/"
        exit 1
    }
} else {
    Write-Err "Visual Studio 2022 Build Tools not detected."
    Write-Host "Please install it with 'Desktop development with C++' workload:"
    Write-Host "  https://visualstudio.microsoft.com/visual-studio-build-tools/"
    Write-Host "  Required components: MSVC v143, Windows 11 SDK, C++ CMake tools"
    exit 1
}

# Detect the vcvars batch file (needed for cl.exe / link.exe on PATH)
$vcvars = Get-ChildItem -Path $vs -Recurse -Filter "vcvars64.bat" -ErrorAction SilentlyContinue | Select-Object -First 1
if (-not $vcvars) {
    Write-Err "vcvars64.bat not found under $vs"
    exit 1
}
$vcvarsPath = $vcvars.FullName
Write-Ok "Found vcvars64: $vcvarsPath"

# ------- 2. Install Rust -------
Write-Section "2/7  Rust (stable-msvc)"

$rustupExe = "$env:USERPROFILE\.cargo\bin\rustup.exe"
$cargoExe = "$env:USERPROFILE\.cargo\bin\cargo.exe"
if (Test-Path $cargoExe) {
    Write-Ok "cargo already installed: $cargoExe"
} else {
    Write-Host "  Installing rustup..."
    Invoke-WebRequest -Uri "https://win.rustup.rs/x86_64" -OutFile "$env:TEMP\rustup-init.exe"
    & "$env:TEMP\rustup-init.exe" -y --default-toolchain $RUST_TOOLCHAIN
    Remove-Item "$env:TEMP\rustup-init.exe"
    Write-Ok "rustup installed"
}
$env:Path = "$env:USERPROFILE\.cargo\bin;$env:Path"
& rustc --version
Write-Ok "rustc OK"

# ------- 3. Install Flutter -------
Write-Section "3/7  Flutter $FLUTTER_VERSION (Chinese mirror)"

if (Test-Path "$FLUTTER_ROOT\bin\flutter.bat") {
    Write-Ok "Flutter already installed at $FLUTTER_ROOT"
} else {
    Write-Host "  Downloading Flutter SDK..."
    $zipUrl = "$FLUTTER_STORAGE_BASE_URL/flutter_infra_release/releases/stable/windows/flutter_windows_$FLUTTER_VERSION.zip"
    $zipPath = "$env:TEMP\flutter.zip"
    Invoke-WebRequest -Uri $zipUrl -OutFile $zipPath
    Write-Host "  Extracting to $FLUTTER_ROOT..."
    Expand-Archive -Path $zipPath -DestinationPath "C:\"
    Remove-Item $zipPath
    Write-Ok "Flutter installed at $FLUTTER_ROOT"
}

# Persist Chinese mirror env vars for the user
[Environment]::SetEnvironmentVariable("PUB_HOSTED_URL", $PUB_HOSTED_URL, "User")
[Environment]::SetEnvironmentVariable("FLUTTER_STORAGE_BASE_URL", $FLUTTER_STORAGE_BASE_URL, "User")
$env:PUB_HOSTED_URL = $PUB_HOSTED_URL
$env:FLUTTER_STORAGE_BASE_URL = $FLUTTER_STORAGE_BASE_URL
$env:Path = "$FLUTTER_ROOT\bin;$env:Path"

# Use the absolute path to flutter.bat so the rest of the script can find it
$flutterBat = "$FLUTTER_ROOT\bin\flutter.bat"
& $flutterBat --version
& $flutterBat config --no-analytics | Out-Null
& $flutterBat doctor | Out-Null
Write-Ok "Flutter doctor OK"

# ------- 4. Install VCPKG and C++ deps -------
Write-Section "4/7  VCPKG + C++ dependencies (libvpx, libyuv, opus, aom)"

if (-not (Test-Path $VCPKG_ROOT)) {
    Write-Host "  Cloning VCPKG to $VCPKG_ROOT..."
    git clone https://github.com/microsoft/vcpkg $VCPKG_ROOT
    & "$VCPKG_ROOT\bootstrap-vcpkg.bat" -disableMetrics
    Write-Ok "VCPKG installed"
} else {
    Write-Ok "VCPKG already at $VCPKG_ROOT"
}
[Environment]::SetEnvironmentVariable("VCPKG_ROOT", $VCPKG_ROOT, "User")
$env:VCPKG_ROOT = $VCPKG_ROOT
$env:Path = "$VCPKG_ROOT;$env:Path"

Write-Host "  Installing C++ deps (x64-windows-static, ~5-10 minutes)..."
& "$VCPKG_ROOT\vcpkg.exe" install --triplet=x64-windows-static libvpx libyuv opus aom
Write-Ok "C++ deps installed"

# ------- 5. Install flutter_rust_bridge_codegen -------
Write-Section "5/7  flutter_rust_bridge_codegen (1.80.1)"

# cargo install outputs the bin to %USERPROFILE%\.cargo\bin
$frbcBin = "$env:USERPROFILE\.cargo\bin\flutter_rust_bridge_codegen.exe"
if (Test-Path $frbcBin) {
    Write-Ok "flutter_rust_bridge_codegen already installed"
} else {
    Write-Host "  Installing (5-10 minutes)..."
    & cargo install flutter_rust_bridge_codegen --version 1.80.1 --features uuid --locked
    Write-Ok "Installed"
}

# ------- 6. Clone the fork -------
Write-Section "6/7  Clone the fork from git.cardsoon.com"

if (Test-Path $PROJECT_ROOT) {
    Write-Ok "Project already cloned at $PROJECT_ROOT"
    Push-Location $PROJECT_ROOT
    & git fetch origin | Out-Null
    & git status
    Pop-Location
} else {
    New-Item -ItemType Directory -Path (Split-Path $PROJECT_ROOT) -Force | Out-Null
    git clone $GIT_URL $PROJECT_ROOT
    Push-Location $PROJECT_ROOT
    & git checkout main
    Pop-Location
    Write-Ok "Fork cloned to $PROJECT_ROOT"
}

# ------- 7. Patch google_fonts (one-liner, per-machine) -------
Write-Section "7/7  Patch google_fonts pub cache (per-machine)"

$googleFontsFile = Join-Path $env:USERPROFILE ".pub-cache\hosted\pub.dev\google_fonts-5.0.0\lib\src\google_fonts_variant.dart"
if (Test-Path $googleFontsFile) {
    $content = Get-Content $googleFontsFile -Raw
    if ($content -match '^final _fontWeightToFilenameWeightParts') {
        Write-Ok "google_fonts already patched"
    } else {
        $patched = $content -replace '^const _fontWeightToFilenameWeightParts', 'final _fontWeightToFilenameWeightParts'
        Set-Content -Path $googleFontsFile -Value $patched -NoNewline
        Write-Ok "Patched google_fonts_variant.dart (const -> final)"
    }
} else {
    Write-Warn "google_fonts_variant.dart not found at $googleFontsFile"
    Write-Host "  It will be created on first 'flutter pub get'. Re-run this script after that, OR patch it manually:"
    Write-Host "  change 'const _fontWeightToFilenameWeightParts' to 'final _fontWeightToFilenameWeightParts'"
}

# ------- Done -------
Write-Section "Bootstrap complete"
Write-Host "Next step: open a new shell with vcvars64 sourced, then run:"
Write-Host "  cd $PROJECT_ROOT"
Write-Host "  .\res\build-windows.ps1"
Write-Host ""
Write-Host "If the build is interrupted, just re-run build-windows.ps1 - it's incremental."
Write-Host ""
Write-Host "Output: $PROJECT_ROOT\flutter\build\windows\x64\runner\Release\"
Write-Host "  (zip the Release\ folder for distribution)"

Write-Host ""
Write-Host "Note: This script must be re-run any time you 'flutter pub cache clean'."