build-python-wheels.yml

name: Build Python Wheels

on:
  workflow_call:
    inputs:
      upload-artifacts:
        description: 'Upload wheel artifacts'
        required: false
        default: true
        type: boolean
      test-wheels:
        description: 'Test the built wheels'
        required: false
        default: true
        type: boolean

jobs:
  build-wheels:
    name: Build wheel (${{ matrix.platform }})
    runs-on: ${{ matrix.os }}
    strategy:
      fail-fast: false
      matrix:
        include:
          # macOS Apple Silicon
          - os: macos-14
            platform: darwin-aarch64
            wheel_platform: macosx_14_0_arm64
            rust_target: aarch64-apple-darwin

          # macOS Intel
          - os: macos-15-intel
            platform: darwin-x86_64
            wheel_platform: macosx_15_0_x86_64
            rust_target: x86_64-apple-darwin

          # Linux x86_64 glibc
          - os: ubuntu-22.04
            platform: linux-x86_64-gnu
            wheel_platform: manylinux_2_35_x86_64
            rust_target: x86_64-unknown-linux-gnu

          # Linux ARM64 glibc
          - os: ubuntu-22.04-arm
            platform: linux-aarch64-gnu
            wheel_platform: manylinux_2_35_aarch64
            rust_target: aarch64-unknown-linux-gnu

          # Windows x64
          - os: windows-latest
            platform: windows-x86_64
            wheel_platform: win_amd64
            rust_target: x86_64-pc-windows-msvc

    steps:
      - uses: actions/checkout@v4

      - name: Install Rust
        uses: dtolnay/rust-toolchain@stable
        with:
          targets: ${{ matrix.rust_target }}

      - name: Build pg0 CLI (Unix)
        if: runner.os != 'Windows'
        env:
          BUNDLE_POSTGRESQL: "true"
        run: |
          cargo build --release --target ${{ matrix.rust_target }}
          ls -la target/${{ matrix.rust_target }}/release/pg0*

      - name: Build pg0 CLI (Windows)
        if: runner.os == 'Windows'
        env:
          BUNDLE_POSTGRESQL: "true"
        run: |
          cargo build --release --target ${{ matrix.rust_target }}
          dir target\${{ matrix.rust_target }}\release\pg0*

      - name: Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '3.11'

      - name: Install uv
        uses: astral-sh/setup-uv@v4

      - name: Build wheel (Unix)
        if: runner.os != 'Windows'
        working-directory: sdk/python
        env:
          PG0_BINARY_PATH: ${{ github.workspace }}/target/${{ matrix.rust_target }}/release/pg0
        run: uv build --wheel

      - name: Build wheel (Windows)
        if: runner.os == 'Windows'
        working-directory: sdk/python
        env:
          PG0_BINARY_PATH: ${{ github.workspace }}\target\${{ matrix.rust_target }}\release\pg0.exe
        run: uv build --wheel

      - name: Rename wheel to platform-specific
        working-directory: sdk/python/dist
        run: |
          for f in *.whl; do
            newname=$(echo "$f" | sed 's/py3-none-any/py3-none-${{ matrix.wheel_platform }}/g')
            if [ "$f" != "$newname" ]; then
              echo "Renaming $f -> $newname"
              mv "$f" "$newname"
            fi
          done
          ls -la
        shell: bash

      - name: Test wheel (Unix)
        if: inputs.test-wheels && runner.os != 'Windows'
        working-directory: sdk/python
        run: |
          uv venv test-env
          source test-env/bin/activate
          pip install dist/*.whl

          python -c "
          from pg0 import _get_bundled_binary, Pg0
          bundled = _get_bundled_binary()
          assert bundled is not None, 'Bundled binary not found!'
          assert bundled.exists(), f'Bundled binary does not exist: {bundled}'
          print(f'Bundled binary found: {bundled}')

          pg = Pg0(name='wheel-test')
          info = pg.start()
          print(f'Started PostgreSQL: {info.uri}')
          pg.stop()
          pg.drop()
          print('Wheel test passed!')
          "

      - name: Test wheel (Windows)
        if: inputs.test-wheels && runner.os == 'Windows'
        working-directory: sdk/python
        run: |
          uv venv test-env
          test-env\Scripts\activate
          pip install (Get-ChildItem dist\*.whl).FullName

          python -c "
          from pg0 import _get_bundled_binary, Pg0
          bundled = _get_bundled_binary()
          assert bundled is not None, 'Bundled binary not found!'
          assert bundled.exists(), f'Bundled binary does not exist: {bundled}'
          print(f'Bundled binary found: {bundled}')

          pg = Pg0(name='wheel-test')
          info = pg.start()
          print(f'Started PostgreSQL: {info.uri}')
          pg.stop()
          pg.drop()
          print('Wheel test passed!')
          "
        shell: pwsh

      - name: Upload wheel artifact
        if: inputs.upload-artifacts
        uses: actions/upload-artifact@v4
        with:
          name: wheel-${{ matrix.platform }}
          path: sdk/python/dist/*.whl

  build-sdist:
    name: Build source distribution
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v4

      - name: Set up Python
        uses: actions/setup-python@v5
        with:
          python-version: '3.11'

      - name: Install uv
        uses: astral-sh/setup-uv@v4

      - name: Build sdist
        working-directory: sdk/python
        run: uv build --sdist

      - name: Upload sdist artifact
        if: inputs.upload-artifacts
        uses: actions/upload-artifact@v4
        with:
          name: sdist
          path: sdk/python/dist/*.tar.gz