$VER=$args[0]

function println($msg) {
    Write-Host $msg
}

println "VER=$VER"

$AX_ROOT=(Resolve-Path "$PSScriptRoot/../..").Path
println "AX_ROOT=$AX_ROOT"

$pwsh_ver = $PSVersionTable.PSVersion.ToString()

function mkdirs([string]$path) {
    if (!(Test-Path $path -PathType Container)) {
        New-Item $path -ItemType Directory 1>$null
    }
}

function download_file($url, $out, $force = $false) {
    if(Test-Path $out -PathType Leaf) { 
        if (!$force) {
            return
        }
        Remove-Item $out
    }
    Write-Host "Downloading $url to $out ..."
    if ($pwsh_ver -ge '7.0') {
        curl -L $url -o $out
    }
    else {
        Invoke-WebRequest -Uri $url -OutFile $out
    }
}

function download_and_expand($url, $out, $dest) {
    download_file $url $out
    if($out.EndsWith('.zip')) {
        Expand-Archive -Path $out -DestinationPath $dest
    } elseif($out.EndsWith('.tar.gz')) {
        if (!$dest.EndsWith('/')) {
            mkdirs $dest
        }
        tar xvf "$out" -C $dest
    } elseif($out.EndsWith('.sh')) {
        chmod 'u+x' "$out"
        mkdirs $dest
    }
}

# download version manifest
cd $AX_ROOT

$cwd = $(Get-Location).Path

$prefix = Join-Path $cwd "tmp/1kdist_$VER"

mkdirs $prefix

# ensure yaml parser module
if ($null -eq (Get-Module -ListAvailable -Name powershell-yaml)) {
    Install-Module -Name powershell-yaml -Force -Repository PSGallery -Scope CurrentUser
}

# check upstream prebuilts version
download_file "https://github.com/axmolengine/buildware/releases/download/$VER/verlist.yml" "./tmp/verlist.yml" $true
$newVerList = ConvertFrom-Yaml -Yaml (Get-Content './tmp/verlist.yml' -raw)
if ($newVerList.GetType() -eq [string]) {
    throw "Download version manifest file verlist.yml fail"
}

$myVerList = ConvertFrom-Yaml -Yaml (Get-Content './thirdparty/prebuilts.yml' -raw)

function update_lib
{
    $lib_name=$args[0]
    $lib_folder=$args[1]

    $myVer = $myVerList[$lib_name]
    if (($newVerList[$lib_name] -eq $myVer) -and ("$env:FORCE_UPDATE" -ne 'true')) {
        println "No update for lib: $lib_name, version: $myVer, skip it"
        return 0
    }

    $lib_dir="./thirdparty/$lib_folder$lib_name"
    $prebuilt_dir="$lib_dir/prebuilt"
    $inc_dir="$lib_dir/include"
    
    println "Updating lib files for ${lib_dir} from $prefix/$lib_name ..."

    # https://github.com/axmolengine/build1k/releases/download/v57/angle.zip
    download_and_expand "https://github.com/axmolengine/build1k/releases/download/$VER/$lib_name.zip" "$prefix/$lib_name.zip" "$prefix"

    Remove-Item $prebuilt_dir -Recurse -Force
    Copy-Item $prefix/$lib_name/prebuilt $lib_dir/ -Container -Recurse
    
    if (Test-Path "$prefix/$lib_name/include" -PathType Container) {
        println "Update inc files for ${lib_dir}"
        Remove-Item $inc_dir -Recurse -Force
        Copy-Item $prefix/$lib_name/include $lib_dir/ -Container -Recurse
    }

    return 1
}

println "Updating libs ..."

$libs_list = @('angle', 'curl', 'jpeg-turbo', 'openssl', 'zlib')

# update libs
$updateCount = 0
foreach($lib_name in $libs_list) {
    $updateCount += $(update_lib $lib_name)
}

$updateCount += $(update_lib luajit lua/)

# update README.md
$content = $(Get-Content -Path ./thirdparty/README.md.in -raw)
foreach ($item in $newVerList.GetEnumerator() )
{
    $key = ([Regex]::Replace($item.Name, '-', '_')).ToUpper()
    $key = "${key}_VERSION"
    $content = $content -replace "\$\{$key\}",$item.Value
}
Set-Content -Path ./thirdparty/README.md -Value "$content"
Copy-Item -Path './tmp/verlist.yml' './thirdparty/prebuilts.yml' -Force

println "Total update lib count $updateCount"

if ($updateCount -eq 0) {
    if ("$env.RUNNER_OS" -ne "") {
        Write-Output "AX_PREBUILTS_NO_UPDATE=true" >> $GITHUB_ENV
    }
}