selene-plugins.yml

name: Selene Plugins

env:
  CARGO_TERM_COLOR: always
  RUSTFLAGS: -C debuginfo=0

on:
  push:
    branches: [master, development, dev]
    paths:
      - 'python/selene-plugins/**'
      - 'crates/pecos-simulators/**'
      - 'crates/pecos-core/**'
      - 'exp/pecos-stab-tn/**'
      - '.github/workflows/selene-plugins.yml'
  pull_request:
    branches: [master, development, dev]
    paths:
      - 'python/selene-plugins/**'
      - 'crates/pecos-simulators/**'
      - 'crates/pecos-core/**'
      - 'exp/pecos-stab-tn/**'
      - '.github/workflows/selene-plugins.yml'
  workflow_dispatch:

concurrency:
  group: ${{ github.workflow }}-${{ github.ref }}
  cancel-in-progress: true

defaults:
  run:
    shell: bash

jobs:
  # Build and test on the current platform (fast feedback)
  test:
    runs-on: ${{ matrix.os }}
    timeout-minutes: 30
    strategy:
      fail-fast: false
      matrix:
        os: [ubuntu-latest, macos-latest, macos-15-intel, windows-latest]

    steps:
      - uses: actions/checkout@v6

      - name: Set up Python
        uses: actions/setup-python@v6
        with:
          python-version: "3.12"

      - name: Install uv
        uses: astral-sh/setup-uv@v7
        with:
          enable-cache: true

      - name: Set up Rust
        run: rustup show

      - name: Cache Rust
        uses: Swatinem/rust-cache@v2

      - name: Build and install Selene plugins (Unix)
        if: runner.os != 'Windows'
        run: |
          if [[ "${{ runner.os }}" == "macOS" ]]; then
            EXT="dylib"
          else
            EXT="so"
          fi

          # Discover all selene plugins, build, copy libraries, and install
          CARGO_ARGS=""
          for DIR in python/selene-plugins/pecos-selene-*/; do
            PLUGIN=$(basename "$DIR")
            CARGO_ARGS="$CARGO_ARGS -p $PLUGIN"
          done
          cargo build --release $CARGO_ARGS

          for DIR in python/selene-plugins/pecos-selene-*/; do
            PLUGIN=$(basename "$DIR")
            LIB_NAME=$(echo "$PLUGIN" | tr '-' '_')
            DEST="python/selene-plugins/$PLUGIN/python/${LIB_NAME}/_dist/lib"
            mkdir -p "$DEST"
            cp "target/release/lib${LIB_NAME}.$EXT" "$DEST/"
            uv pip install --system -e "${DIR}[test]"
          done

      - name: Build and install Selene plugins (Windows)
        if: runner.os == 'Windows'
        shell: pwsh
        run: |
          $plugins = Get-ChildItem -Directory python/selene-plugins/pecos-selene-* | Select-Object -ExpandProperty Name
          $cargoArgs = $plugins | ForEach-Object { "-p"; $_ }
          cargo build --release @cargoArgs

          foreach ($plugin in $plugins) {
            $libName = $plugin -replace '-', '_'
            $dest = "python/selene-plugins/$plugin/python/$libName/_dist/lib"
            New-Item -ItemType Directory -Force -Path $dest
            Copy-Item "target/release/$libName.dll" "$dest/"
          }

          foreach ($pluginDir in (Get-ChildItem -Directory python/selene-plugins/pecos-selene-*)) {
            uv pip install --system -e "$($pluginDir.FullName)[test]"
          }

      - name: Run tests
        run: |
          pytest python/selene-plugins/ -v --tb=short

  # Build wheels for distribution
  build-wheels:
    needs: test
    runs-on: ${{ matrix.os }}
    strategy:
      fail-fast: false
      matrix:
        os: [ubuntu-latest, macos-latest, macos-15-intel, windows-latest]
        plugin:
          - name: pecos-selene-stabilizer
            package: pecos_selene_stabilizer
          - name: pecos-selene-statevec
            package: pecos_selene_statevec
          - name: pecos-selene-stab-vec
            package: pecos_selene_stab_vec
          - name: pecos-selene-stab-mps
            package: pecos_selene_stab_mps
          - name: pecos-selene-mast
            package: pecos_selene_mast

    steps:
      - uses: actions/checkout@v6

      - name: Set up Python
        uses: actions/setup-python@v6
        with:
          python-version: "3.12"

      - name: Install uv
        uses: astral-sh/setup-uv@v7
        with:
          enable-cache: true

      - name: Set up Rust
        run: rustup show

      - name: Cache Rust
        uses: Swatinem/rust-cache@v2

      - name: Build Rust library (Unix)
        if: runner.os != 'Windows'
        run: |
          cargo build --release -p ${{ matrix.plugin.name }}

      - name: Build Rust library (Windows)
        if: runner.os == 'Windows'
        shell: pwsh
        run: |
          cargo build --release -p ${{ matrix.plugin.name }}

      - name: Copy library to Python package (Unix)
        if: runner.os != 'Windows'
        run: |
          if [[ "${{ runner.os }}" == "macOS" ]]; then
            EXT="dylib"
          else
            EXT="so"
          fi
          PLUGIN_DIR="python/selene-plugins/${{ matrix.plugin.name }}"
          mkdir -p "$PLUGIN_DIR/python/${{ matrix.plugin.package }}/_dist/lib"
          cp "target/release/lib${{ matrix.plugin.package }}.$EXT" "$PLUGIN_DIR/python/${{ matrix.plugin.package }}/_dist/lib/"

      - name: Copy library to Python package (Windows)
        if: runner.os == 'Windows'
        shell: pwsh
        run: |
          $pluginDir = "python/selene-plugins/${{ matrix.plugin.name }}"
          New-Item -ItemType Directory -Force -Path "$pluginDir/python/${{ matrix.plugin.package }}/_dist/lib"
          Copy-Item "target/release/${{ matrix.plugin.package }}.dll" "$pluginDir/python/${{ matrix.plugin.package }}/_dist/lib/"

      - name: Build wheel
        run: |
          cd python/selene-plugins/${{ matrix.plugin.name }}
          uv pip install --system build
          python -m build --wheel --outdir dist

      - name: Upload wheel
        uses: actions/upload-artifact@v7
        with:
          name: wheel-${{ matrix.plugin.name }}-${{ matrix.os }}
          path: python/selene-plugins/${{ matrix.plugin.name }}/dist/*.whl

  # Collect all wheels into a single artifact
  collect-wheels:
    needs: build-wheels
    runs-on: ubuntu-latest
    steps:
      - name: Download all wheel artifacts
        uses: actions/download-artifact@v7
        with:
          path: wheels/
          pattern: wheel-*
          merge-multiple: true

      - name: List wheels
        run: |
          echo "=== All Selene plugin wheels ==="
          ls -la wheels/
          echo ""
          echo "=== Summary ==="
          echo "Total wheels: $(ls -1 wheels/*.whl 2>/dev/null | wc -l)"

      - name: Upload combined wheels
        uses: actions/upload-artifact@v7
        with:
          name: selene-plugin-wheels
          path: wheels/*.whl