.github/actions/build-wheel/action.yml

# 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.

# Composite action that builds a datafusion-python wheel with maturin.
# Centralises the abi3-vs-free-threaded argument logic so platform jobs
# stay short and changes to wheel-build flags happen in one place.

name: "Build wheel"
description: "Build datafusion-python wheel with maturin (abi3 or free-threaded)"

inputs:
  target:
    description: "Rust target triple (e.g. x86_64-unknown-linux-gnu). Required when manylinux is set; ignored for native builds."
    required: false
    default: ""
  python-tag:
    description: "abi3 (covers 3.10..3.14 GIL builds) or a free-threaded interpreter such as 3.13t / 3.14t"
    required: true
  build-mode:
    description: "release or debug"
    required: true
  features:
    description: "Comma-separated extra features (in addition to those implied by the python-tag)"
    required: false
    default: "substrait"
  manylinux:
    description: "manylinux tag for maturin-action (e.g. 2_28). Leave empty to use uv-run maturin natively."
    required: false
    default: ""
  out-dir:
    description: "Output directory for built wheels"
    required: false
    default: "dist"

outputs:
  args:
    description: "Computed maturin args (for debugging)"
    value: ${{ steps.args.outputs.args }}

runs:
  using: "composite"
  steps:
    - name: Compute maturin args
      id: args
      shell: bash
      run: |
        set -euo pipefail
        FEATURES="${{ inputs.features }}"
        TAG="${{ inputs.python-tag }}"
        if [ "$TAG" = "abi3" ]; then
          # Default features include the `abi3` cargo feature.
          # One wheel covers Python 3.10..3.14 (GIL builds only).
          BUILD_ARGS="--features ${FEATURES}"
        else
          # Free-threaded build: disable abi3, force mimalloc back in, pin interpreter.
          if [ "${RUNNER_OS:-}" = "Windows" ]; then
            # Windows free-threaded builds ship as `python.exe` (no `tN`
            # suffix). Resolve sys.executable so the path is independent of
            # PATH ordering, and assert the interpreter is actually
            # free-threaded before we hand the wheel off.
            INTERP=$(python -c 'import sys; print(sys.executable)')
            python -c "import sysconfig, sys; \
              v = sysconfig.get_config_var('Py_GIL_DISABLED'); \
              sys.exit(0 if v == 1 else f'expected free-threaded interpreter, got Py_GIL_DISABLED={v!r} at {sys.executable}')"
            # Backslashes in BUILD_ARGS would be parsed as escapes when the
            # output is re-expanded in the next step; use forward slashes
            # (maturin/Rust accept them on Windows).
            INTERP="${INTERP//\\//}"
          else
            INTERP="python${TAG}"
          fi
          BUILD_ARGS="--no-default-features --features mimalloc,${FEATURES} --interpreter ${INTERP}"
        fi
        if [ "${{ inputs.build-mode }}" = "release" ]; then
          BUILD_ARGS="--release --strip ${BUILD_ARGS}"
        fi
        BUILD_ARGS="${BUILD_ARGS} --out ${{ inputs.out-dir }}"
        echo "args=${BUILD_ARGS}" >> "$GITHUB_OUTPUT"
        echo "maturin args: ${BUILD_ARGS}"

    - name: Build via maturin-action (manylinux container)
      if: inputs.manylinux != ''
      uses: PyO3/maturin-action@v1
      with:
        target: ${{ inputs.target }}
        manylinux: ${{ inputs.manylinux }}
        maturin-version: "1.13.3"
        args: ${{ steps.args.outputs.args }}
        rustup-components: rust-std

    - name: Build via native maturin
      if: inputs.manylinux == ''
      shell: bash
      # Use `uvx` so maturin is available even when `uv sync` was skipped
      # (free-threaded matrix entries don't pre-populate the project venv).
      run: uvx maturin@1.13.3 build ${{ steps.args.outputs.args }}