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

$repoRoot = (Resolve-Path (Join-Path $PSScriptRoot "..")).Path
Set-Location $repoRoot

function Invoke-NativeCommand {
    param(
        [Parameter(Mandatory = $true)]
        [string]$FilePath,
        [string[]]$Arguments = @()
    )

    & $FilePath @Arguments
    if ($LASTEXITCODE -ne 0) {
        $argText = if ($Arguments.Count -gt 0) { " $($Arguments -join ' ')" } else { "" }
        throw "$FilePath$argText failed with exit code $LASTEXITCODE"
    }
}

function Assert-RequiredSourceFiles {
    param(
        [Parameter(Mandatory = $true)]
        [string]$Action,
        [Parameter(Mandatory = $true)]
        [string[]]$Paths
    )

    $missing = @()
    foreach ($relativePath in $Paths) {
        $fullPath = Join-Path $repoRoot $relativePath
        if (-not (Test-Path -LiteralPath $fullPath -PathType Leaf)) {
            $missing += $relativePath
        }
    }

    if ($missing.Count -gt 0) {
        throw "$Action requires a complete source tree. Missing files: $($missing -join ', ')"
    }
}

try {
    Write-Host "========================================"
    Write-Host "  Ant Browser - Build Script"
    Write-Host "========================================"
    Write-Host ""
    Write-Host "Current workdir: $repoRoot"
    Write-Host ""

    $proxyHost = "127.0.0.1"
    $proxyPort = "7890"
    $useProxy = $true

    if ($useProxy) {
        Write-Host "[0/7] Configuring proxy..."
        $proxyValue = "http://${proxyHost}:${proxyPort}"
        $env:HTTP_PROXY = $proxyValue
        $env:HTTPS_PROXY = $proxyValue
        $env:http_proxy = $proxyValue
        $env:https_proxy = $proxyValue
        $env:GOPROXY = "https://goproxy.cn,direct"

        & npm config set proxy $proxyValue | Out-Null
        & npm config set https-proxy $proxyValue | Out-Null

        Write-Host "OK proxy configured: ${proxyHost}:${proxyPort}"
        Write-Host ""
    }

    Assert-RequiredSourceFiles -Action "Building from source" -Paths @(
        "go.mod",
        "go.sum",
        "main.go",
        "wails.json"
    )

    Write-Host "[1/7] Installing frontend dependencies..."
    Push-Location (Join-Path $repoRoot "frontend")
    try {
        Invoke-NativeCommand -FilePath "npm" -Arguments @("install")
        Invoke-NativeCommand -FilePath "npm" -Arguments @("run", "ensure:native")
    }
    finally {
        Pop-Location
    }

    Write-Host ""
    Write-Host "[2/7] Installing Go dependencies..."
    Invoke-NativeCommand -FilePath "go" -Arguments @("mod", "download")
    Invoke-NativeCommand -FilePath "go" -Arguments @("mod", "tidy")

    Write-Host ""
    Write-Host "[3/7] Ensuring frontend\dist exists..."
    $frontendDist = Join-Path $repoRoot "frontend/dist"
    $tempDistCreated = $false
    if (-not (Test-Path -LiteralPath $frontendDist)) {
        New-Item -ItemType Directory -Path $frontendDist -Force | Out-Null
        Set-Content -LiteralPath (Join-Path $frontendDist "index.html") -Value "" -Encoding ascii
        $tempDistCreated = $true
        Write-Host "OK temporary dist directory created"
    } else {
        Write-Host "OK dist directory already exists"
    }

    Write-Host ""
    Write-Host "[4/7] Generating Wails bindings..."
    Invoke-NativeCommand -FilePath "cmd" -Arguments @("/c", "call bat\generate-bindings.bat --no-pause")

    $binaryPath = Join-Path $repoRoot "build/bin/ant-chrome.exe"

    Write-Host ""
    Write-Host "[5/7] Building frontend..."
    if ($tempDistCreated -and (Test-Path -LiteralPath $frontendDist)) {
        Remove-Item -LiteralPath $frontendDist -Recurse -Force -ErrorAction SilentlyContinue
    }
    Push-Location (Join-Path $repoRoot "frontend")
    try {
        Invoke-NativeCommand -FilePath "npm" -Arguments @("run", "build:clean")
    }
    finally {
        Pop-Location
    }

    Write-Host ""
    Write-Host "[6/7] Building app..."
    Invoke-NativeCommand -FilePath "wails" -Arguments @("build")

    if ($tempDistCreated -and (Test-Path -LiteralPath $frontendDist)) {
        Remove-Item -LiteralPath $frontendDist -Recurse -Force -ErrorAction SilentlyContinue
    }

    Write-Host ""
    Write-Host "[7/7] Copying runtime dependencies..."
    $binDir = Join-Path $repoRoot "bin"
    $targetDir = Join-Path $repoRoot "build/bin/bin"
    if (Test-Path -LiteralPath $binDir -PathType Container) {
        Copy-Item -LiteralPath $binDir -Destination $targetDir -Recurse -Force
        Write-Host "OK copied bin directory to build\bin\bin\"
    } else {
        Write-Host "[WARN] bin directory not found, skipping copy"
    }

    Write-Host ""
    Write-Host "========================================"
    Write-Host "  OK build completed"
    Write-Host "========================================"
    Write-Host ""
    Write-Host "Executable: build\bin\ant-chrome.exe"
    exit 0
}
catch {
    Write-Host ""
    Write-Host "[ERROR] $($_.Exception.Message)"
    exit 1
}
finally {
    & npm config delete proxy 2>$null | Out-Null
    & npm config delete https-proxy 2>$null | Out-Null
}