lightning/.github/workflows/_build-packages.yml

name: Call building packages

on:
  workflow_call:
    inputs:
      artifact-name:
        description: 'Unique name for collecting artifacts'
        required: true
        type: string
      pkg-names:
        description: 'list package names to be build in json format'
        required: false
        type: string
        default: |
          ["lightning", "app", "fabric", "pytorch"]

defaults:
  run:
    shell: bash

jobs:

  init:
    runs-on: ubuntu-20.04
    steps:
    - uses: actions/checkout@v3
    - run: |
        mkdir dist && touch dist/.placeholder
    - name: Keep artifact
      id: keep-artifact
      run: python -c "print('DAYS=' + str(5 if '${{ github.event_name }}'.startswith('pull_request') else 0))" >> $GITHUB_OUTPUT
    - uses: actions/upload-artifact@v3
      with:
        name: ${{ inputs.artifact-name }}
        path: dist
        retention-days: ${{ steps.keep-artifact.outputs.DAYS }}


  build-packages:
    needs: init
    runs-on: ubuntu-20.04
    strategy:
      max-parallel: 1  # run sequential to prevent download/upload collisions
      matrix:
        pkg-name: ${{ fromJSON(inputs.pkg-names) }}
    steps:
    - uses: actions/checkout@v3
    - uses: actions/download-artifact@v3
      with:
        name: ${{ inputs.artifact-name }}
        path: pypi
    - uses: actions/setup-python@v4
      with:
        python-version: 3.9

    - run: python -c "print('NB_DIRS=' + str(2 if '${{ matrix.pkg-name }}' == 'pytorch' else 1))" >> $GITHUB_ENV
    - name: Build & check package
      uses: ./.github/actions/pkg-check
      with:
        pkg-name: ${{ matrix.pkg-name }}
        nb-dirs: ${{ env.NB_DIRS }}

    - run: |
        mkdir pypi/${{ matrix.pkg-name }}
        cp dist/* pypi/${{ matrix.pkg-name }}/

    - uses: actions/upload-artifact@v3
      with:
        name: ${{ inputs.artifact-name }}
        path: pypi