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

$repoRoot = Resolve-Path (Join-Path $PSScriptRoot '..\..')
$protoRoot = Join-Path $repoRoot 'src\ZB.MOM.WW.MxGateway.Contracts\Protos'
$outputRoot = Join-Path $PSScriptRoot 'internal\generated'
$modulePath = 'gitea.dohertylan.com/dohertj2/mxaccessgw/clients/go/internal/generated'

function Resolve-Tool {
    # Resolve a codegen tool from PATH first (portable), then the documented Windows fallbacks,
    # instead of the previous hard-coded per-machine paths. See docs/ToolchainLinks.md.
    param(
        [string[]]$Names,
        [string[]]$FallbackPaths
    )

    foreach ($name in $Names) {
        $cmd = Get-Command $name -ErrorAction SilentlyContinue
        if ($null -ne $cmd) {
            return $cmd.Source
        }
    }

    foreach ($fallback in $FallbackPaths) {
        if ($fallback -and (Test-Path $fallback)) {
            return $fallback
        }
    }

    throw "Could not find $($Names -join '/') on PATH. See docs/ToolchainLinks.md."
}

$wingetProtoc = if ($env:LOCALAPPDATA) {
    Join-Path $env:LOCALAPPDATA 'Microsoft\WinGet\Packages\Google.Protobuf_Microsoft.Winget.Source_8wekyb3d8bbwe\bin\protoc.exe'
} else { $null }
$goBin = if ($env:USERPROFILE) { Join-Path $env:USERPROFILE 'go\bin' } elseif ($env:HOME) { Join-Path $env:HOME 'go/bin' } else { $null }

$protoc = Resolve-Tool -Names @('protoc', 'protoc.exe') -FallbackPaths @($wingetProtoc)
$protocGenGo = Resolve-Tool -Names @('protoc-gen-go', 'protoc-gen-go.exe') -FallbackPaths @((if ($goBin) { Join-Path $goBin 'protoc-gen-go.exe' }), (if ($goBin) { Join-Path $goBin 'protoc-gen-go' }))
$protocGenGoGrpc = Resolve-Tool -Names @('protoc-gen-go-grpc', 'protoc-gen-go-grpc.exe') -FallbackPaths @((if ($goBin) { Join-Path $goBin 'protoc-gen-go-grpc.exe' }), (if ($goBin) { Join-Path $goBin 'protoc-gen-go-grpc' }))

# protoc discovers the plugins on PATH; prepend the directories the resolved plugins live in.
$env:Path = (Split-Path $protocGenGo -Parent) + [System.IO.Path]::PathSeparator + (Split-Path $protocGenGoGrpc -Parent) + [System.IO.Path]::PathSeparator + $env:Path

New-Item -ItemType Directory -Path $outputRoot -Force | Out-Null
Get-ChildItem -Path $outputRoot -Filter '*.pb.go' -File | Remove-Item

& $protoc `
    --proto_path=$protoRoot `
    --go_out=$outputRoot `
    --go_opt=paths=source_relative `
    "--go_opt=Mmxaccess_gateway.proto=$modulePath;generated" `
    "--go_opt=Mmxaccess_worker.proto=$modulePath;generated" `
    "--go_opt=Mgalaxy_repository.proto=$modulePath;generated" `
    mxaccess_gateway.proto `
    mxaccess_worker.proto `
    galaxy_repository.proto

if ($LASTEXITCODE -ne 0) {
    throw "protoc (go) failed with exit code $LASTEXITCODE."
}

& $protoc `
    --proto_path=$protoRoot `
    --go-grpc_out=$outputRoot `
    --go-grpc_opt=paths=source_relative `
    "--go-grpc_opt=Mmxaccess_gateway.proto=$modulePath;generated" `
    "--go-grpc_opt=Mgalaxy_repository.proto=$modulePath;generated" `
    mxaccess_gateway.proto `
    galaxy_repository.proto

if ($LASTEXITCODE -ne 0) {
    throw "protoc (go-grpc) failed with exit code $LASTEXITCODE."
}