python-release.yml

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

name: Python Release

on:
  release:
    types:
      - published
  pull_request:
    paths:
      - .github/workflows/python-release.yml
  workflow_dispatch:
    inputs:
      mode:
        description: "dry_run: build & test only, release: build & publish"
        required: true
        default: "dry_run"
        type: choice
        options:
          - dry_run
          - release
      repo:
        description: "Target repository for publishing"
        required: true
        default: "fury"
        type: choice
        options:
          - fury
          - pypi
      ref:
        description: 'The branch, tag or SHA to checkout'
        required: false
        type: string

jobs:
  publish:
    runs-on: ubuntu-latest
    permissions:
      id-token: write  # Required for PyPI trusted publishing
      contents: read
    steps:
      - name: Checkout code
        uses: actions/checkout@v4
        with:
          # When triggered by a release, use the release tag
          # When triggered manually, use the provided ref
          ref: ${{ github.event.release.tag_name || inputs.ref }}

      - 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 lance-ray
        run: |
          uv build

      - name: Get package version
        id: get_version
        run: |
          VERSION=$(python -c "import tomllib; print(tomllib.load(open('pyproject.toml', 'rb'))['project']['version'])")
          echo "version=$VERSION" >> $GITHUB_OUTPUT
          echo "Package version: $VERSION"

      - name: Determine release type
        id: release_type
        run: |
          if [[ "${{ github.event_name }}" == "workflow_dispatch" ]]; then
            # Use the input for manual workflow dispatch
            echo "repo=${{ inputs.repo }}" >> $GITHUB_OUTPUT
            echo "Release type (manual): ${{ inputs.repo }}"
          else
            # For release events, check the tag
            TAG="${{ github.event.release.tag_name }}"
            if [[ "$TAG" == *-beta.* ]] || [[ "$TAG" == *-rc.* ]]; then
              echo "repo=fury" >> $GITHUB_OUTPUT
              echo "Release type: beta/rc -> fury.io"
            else
              echo "repo=pypi" >> $GITHUB_OUTPUT
              echo "Release type: stable -> PyPI"
            fi
          fi

      - name: Publish to PyPI
        if: |
          steps.release_type.outputs.repo == 'pypi' &&
          (github.event_name == 'release' ||
          (github.event_name == 'workflow_dispatch' && inputs.mode == 'release'))
        run: |
          uv publish --trusted-publishing always
          echo "✅ Successfully published version ${{ steps.get_version.outputs.version }} to PyPI!"

      - name: Publish to Fury (beta/rc)
        if: |
          steps.release_type.outputs.repo == 'fury' &&
          (github.event_name == 'release' ||
          (github.event_name == 'workflow_dispatch' && inputs.mode == 'release'))
        env:
          FURY_TOKEN: ${{ secrets.FURY_TOKEN }}
        run: |
          WHEEL=$(ls dist/lance_ray-*.whl 2> /dev/null | head -n 1)
          echo "Uploading $WHEEL to Fury"
          curl -f -F package=@$WHEEL https://$FURY_TOKEN@push.fury.io/lance-format/
          echo "✅ Successfully published version ${{ steps.get_version.outputs.version }} to fury.io!"