add extra steps to release

Author: nathanhhughes

.github/workflows/wheels.yaml

@@ -1,8 +1,9 @@
 ---
 # Adapted from:
 # https://github.com/pypa/cibuildwheel?tab=readme-ov-file#example-setup
+# https://github.com/pypa/cibuildwheel/blob/main/examples/github-deploy.yml
 name: Spark-DSG Wheels
-on: {release: {types: [published]}}
+on: {workflow_dispatch: {}, push: {branches: {main}} release: {types: [published]}}
 jobs:
   build_wheels:
     name: Build wheels on ${{ matrix.os }}
@@ -11,7 +12,7 @@ jobs:
       matrix:
         os: [ubuntu-latest, ubuntu-24.04-arm, windows-latest, macos-latest]
     steps:
-      - uses: actions/checkout@v4
+      - uses: actions/checkout@v5
       - uses: actions/setup-python@v5
       - name: Install cibuildwheel
         run: python -m pip install cibuildwheel
@@ -21,3 +22,31 @@ jobs:
         with:
           name: cibw-wheels-${{ matrix.os }}-${{ strategy.job-index }}
           path: ./wheelhouse/*.whl
+  build_sdist:
+    name: Build source distribution
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v5
+      - name: Build sdist
+        run: pipx run build --sdist
+      - uses: actions/upload-artifact@v4
+        with:
+          name: cibw-sdist
+          path: dist/*.tar.gz
+  upload_pypi:
+    needs: [build_wheels, build_sdist]
+    runs-on: ubuntu-latest
+    environment: pypi
+    permissions:
+      id-token: write
+    if: github.event_name == 'release' && github.event.action == 'published'
+    steps:
+      - uses: actions/download-artifact@v5
+        with:
+          pattern: cibw-*
+          path: dist
+          merge-multiple: true
+      #- uses: pypa/gh-action-pypi-publish@release/v1
+        # To test uploads to TestPyPI, uncomment the following:
+        # with:
+        #   repository-url: https://test.pypi.org/legacy/