Refactor entrypoint more (#2371)

Signed-off-by: Yee Hing Tong <[email protected]>
flyteorg · Apr 22, 2024 · d34ce56 · d34ce56
1 parent 4ec68ea
commit d34ce56
Show file tree

Hide file tree

Showing 13 changed files with 148 additions and 0 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -0,0 +1,131 @@
+name: Rust CI
+
+on:
+  push:
+    branches:
+      - master
+      - refactor-entrypoint
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+jobs:
+  linux:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: ubuntu-latest
+            target: x86_64
+          - runner: ubuntu-latest
+            target: x86
+          - runner: ubuntu-latest
+            target: aarch64
+          - runner: ubuntu-latest
+            target: armv7
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '3.10'
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          working-directory: rust/flyrs
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter
+          sccache: 'true'
+          manylinux: auto
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-linux-${{ matrix.platform.target }}
+          path: dist
+
+  windows:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: windows-latest
+            target: x64
+          - runner: windows-latest
+            target: x86
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '3.10'
+          architecture: ${{ matrix.platform.target }}
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          working-directory: rust/flyrs
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter
+          sccache: 'true'
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-windows-${{ matrix.platform.target }}
+          path: dist
+
+  macos:
+    runs-on: ${{ matrix.platform.runner }}
+    strategy:
+      matrix:
+        platform:
+          - runner: macos-latest
+            target: x86_64
+          - runner: macos-14
+            target: aarch64
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '3.10'
+      - name: Build wheels
+        uses: PyO3/maturin-action@v1
+        with:
+          working-directory: rust/flyrs
+          target: ${{ matrix.platform.target }}
+          args: --release --out dist --find-interpreter
+          sccache: 'true'
+      - name: Upload wheels
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-macos-${{ matrix.platform.target }}
+          path: dist
+
+  sdist:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - name: Build sdist
+        uses: PyO3/maturin-action@v1
+        with:
+          working-directory: rust/flyrs
+          command: sdist
+          args: --out dist
+      - name: Upload sdist
+        uses: actions/upload-artifact@v4
+        with:
+          name: wheels-sdist
+          path: dist
+
+  release:
+    name: Release
+    runs-on: ubuntu-latest
+    if: "startsWith(github.ref, 'refs/tags/')"
+    needs: [linux, windows, macos, sdist]
+    steps:
+      - uses: actions/download-artifact@v4
+      - name: Publish to PyPI
+        uses: PyO3/maturin-action@v1
+        env:
+          MATURIN_PYPI_TOKEN: ${{ secrets.PYPI_FLYRS_SECRET }}
+        with:
+          working-directory: rust/flyrs
+          command: upload
+          args: --non-interactive --skip-existing wheels-*/*
diff --git a/Cargo.lock → rust/flyrs/Cargo.lock b/Cargo.lock → rust/flyrs/Cargo.lock
diff --git a/Cargo.toml → rust/flyrs/Cargo.toml b/Cargo.toml → rust/flyrs/Cargo.toml
diff --git a/flyrs_test/README.md → rust/flyrs/flyrs_test/README.md b/flyrs_test/README.md → rust/flyrs/flyrs_test/README.md
diff --git a/flyrs_test/test.sh → rust/flyrs/flyrs_test/test.sh b/flyrs_test/test.sh → rust/flyrs/flyrs_test/test.sh
diff --git a/flyrs_test/testdata/expected_outputs.pb → ...s/flyrs_test/testdata/expected_outputs.pb b/flyrs_test/testdata/expected_outputs.pb → ...s/flyrs_test/testdata/expected_outputs.pb
diff --git a/flyrs_test/testdata/inputs.pb → rust/flyrs/flyrs_test/testdata/inputs.pb b/flyrs_test/testdata/inputs.pb → rust/flyrs/flyrs_test/testdata/inputs.pb
diff --git a/flyrs_test/testdata/outputs.pb → rust/flyrs/flyrs_test/testdata/outputs.pb b/flyrs_test/testdata/outputs.pb → rust/flyrs/flyrs_test/testdata/outputs.pb
diff --git a/flyrs_test/testdata/schedule.tar.gz → ...flyrs/flyrs_test/testdata/schedule.tar.gz b/flyrs_test/testdata/schedule.tar.gz → ...flyrs/flyrs_test/testdata/schedule.tar.gz
diff --git a/rust/flyrs/pyproject.toml b/rust/flyrs/pyproject.toml
@@ -0,0 +1,17 @@
+[build-system]
+requires = ["maturin>=1.5,<2.0"]
+build-backend = "maturin"
+
+[project]
+name = "flyrs"
+requires-python = ">=3.8"
+classifiers = [
+    "Programming Language :: Rust",
+    "Programming Language :: Python :: Implementation :: CPython",
+    "Programming Language :: Python :: Implementation :: PyPy",
+]
+
+dynamic = ["version"]
+
+[tool.maturin]
+bindings = "bin"
diff --git a/src/distribution.rs → rust/flyrs/src/distribution.rs b/src/distribution.rs → rust/flyrs/src/distribution.rs
diff --git a/src/executor.rs → rust/flyrs/src/executor.rs b/src/executor.rs → rust/flyrs/src/executor.rs
diff --git a/src/main.rs → rust/flyrs/src/main.rs b/src/main.rs → rust/flyrs/src/main.rs