ci/scripts/python_wheel_windows_build_drivers.ps1

#!/usr/bin/env pwsh
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

param(
    [Parameter(Mandatory = $true)]
    [string]$SourceDir,
    [Parameter(Mandatory = $true)]
    [string]$BuildDir,
    [Parameter(Mandatory = $true)]
    [ValidateSet("amd64", "arm64")]
    [string]$Architecture
)

$ErrorActionPreference = "Stop"

if (-not $env:VCPKG_ROOT) {
    throw "Must set VCPKG_ROOT"
}

if ($Architecture -eq "amd64") {
    $CMakePlatform = "x64"
    $PlatformTag = "win_amd64"
    $VcpkgTriplet = "x64-windows-static-release"
} else {
    $CMakePlatform = "ARM64"
    $PlatformTag = "win_arm64"
    $VcpkgTriplet = "arm64-windows-static-release"
}

$BuildType = "RelWithDebInfo"
$env:VCPKG_FEATURE_FLAGS = "-manifests"
$env:VCPKG_TARGET_TRIPLET = $VcpkgTriplet
$env:VCPKG_DEFAULT_HOST_TRIPLET = $VcpkgTriplet

Write-Host "=== Building ADBC drivers for $Architecture ==="

& (Join-Path $env:VCPKG_ROOT "vcpkg.exe") install `
    "--triplet=$VcpkgTriplet" `
    "libpq" `
    "sqlite3[dbstat,fts3,fts4,fts5,geopoly,json1,limit,math,rtree,session,snapshot,soundex]" `
    --overlay-triplets $(Join-Path $SourceDir "ci\vcpkg\triplets")
if (-not $?) { throw "Failed to install vcpkg dependencies" }

New-Item -ItemType Directory -Force -Path $BuildDir | Out-Null

cmake `
    -S (Join-Path $SourceDir "c") `
    -B $BuildDir `
    -G "Visual Studio 18 2026" `
    -A $CMakePlatform `
    -DADBC_BUILD_SHARED=ON `
    "-DCMAKE_BUILD_TYPE=$BuildType" `
    "-DCMAKE_INSTALL_PREFIX=$BuildDir" `
    "-DCMAKE_TOOLCHAIN_FILE=$(Join-Path $env:VCPKG_ROOT 'scripts/buildsystems/vcpkg.cmake')" `
    -DCMAKE_UNITY_BUILD=ON `
    "-DVCPKG_OVERLAY_TRIPLETS=$(Join-Path $SourceDir 'ci\vcpkg\triplets')" `
    "-DVCPKG_TARGET_TRIPLET=$VcpkgTriplet" `
    -DADBC_DRIVER_FLIGHTSQL=ON `
    -DADBC_DRIVER_MANAGER=ON `
    -DADBC_DRIVER_POSTGRESQL=ON `
    -DADBC_DRIVER_SQLITE=ON
if (-not $?) { throw "Failed to configure the CMake build" }

cmake --build $BuildDir --config $BuildType --target install --verbose -j
if (-not $?) { throw "Failed to build and install the CMake project" }

$env:ADBC_FLIGHTSQL_LIBRARY = Join-Path $BuildDir "bin/adbc_driver_flightsql.dll"
$env:ADBC_POSTGRESQL_LIBRARY = Join-Path $BuildDir "bin/adbc_driver_postgresql.dll"
$env:ADBC_SQLITE_LIBRARY = Join-Path $BuildDir "bin/adbc_driver_sqlite.dll"

python -m pip install --upgrade pip delvewheel wheel
if (-not $?) { throw "Failed to install Python build dependencies" }

$env:ADBC_BUILD_TYPE = "debug"
$Components = @(
    "adbc_driver_flightsql",
    "adbc_driver_postgresql",
    "adbc_driver_sqlite"
)

foreach ($Component in $Components) {
    $ComponentDir = Join-Path $SourceDir "python/$Component"
    Push-Location $ComponentDir
    try {
        Remove-Item -Recurse -Force -ErrorAction SilentlyContinue build, dist, repaired_wheels

        Write-Host "=== Checking $Component version ==="
        python "$Component/_version.py"
        if (-not $?) { throw "Failed to check the $Component version" }

        Write-Host "=== Building $Component wheel ==="
        python -m pip wheel --no-deps -w dist -vvv .
        if (-not $?) { throw "Failed to build the $Component wheel" }

        $Wheels = @(Get-ChildItem -Path "dist/*.whl")
        foreach ($Wheel in $Wheels) {
            python (Join-Path $SourceDir "ci/scripts/python_wheel_fix_tag.py") `
                "--plat-name=$PlatformTag" $Wheel.FullName
            if (-not $?) { throw "Failed to update the $Component wheel tag" }
        }

        New-Item -ItemType Directory -Force -Path repaired_wheels | Out-Null
        $Wheels = @(Get-ChildItem -Path "dist/*.whl")
        foreach ($Wheel in $Wheels) {
            Copy-Item $Wheel.FullName repaired_wheels
            python -m delvewheel repair -w repaired_wheels $Wheel.FullName
            if (-not $?) { throw "Failed to repair the $Component wheel" }
        }
    } finally {
        Pop-Location
    }
}