boilerplate

MANIFEST.in

@@ -1 +1,5 @@
-recursive-include actuator/ *.py *.txt py.typed MANIFEST.in
+include actuator/rust/Cargo.toml
+include actuator/rust/Cargo.lock
+include actuator/rust/*.so
+include actuator/rust/*.pyi
+recursive-include actuator/rust/src *.rs

actuator/.gitignore

@@ -0,0 +1,5 @@
+# .gitignore
+
+# Rust
+target/
+Cargo.lock

actuator/rust/.gitignore

@@ -0,0 +1,4 @@
+# .gitignore
+
+*.so
+*.pyi

actuator/rust/Cargo.toml

@@ -0,0 +1,11 @@
+[package]
+name = "lib"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+name = "lib"
+crate-type = ["cdylib"]
+
+[dependencies]
+pyo3 = { version = "0.17", features = ["extension-module"] }

actuator/rust/src/lib.rs

@@ -0,0 +1,7 @@
+use pyo3::prelude::*;
+
+#[pymodule]
+fn lib(_py: Python, _m: &PyModule) -> PyResult<()> {
+    println!("Hello, world!");
+    Ok(())
+}

pyproject.toml

@@ -70,3 +70,8 @@ combine-as-imports = true
 [tool.ruff.lint.pydocstyle]
 
 convention = "google"
+
+[build-system]
+
+requires = ["setuptools>=42", "wheel", "setuptools-rust>=1.5.2"]
+build-backend = "setuptools.build_meta"

setup.py

@@ -2,9 +2,24 @@
 #!/usr/bin/env python
 """Setup script for the project."""
 
+import os
 import re
+import subprocess
 
 from setuptools import setup
+from setuptools.command.build_ext import build_ext as _build_ext
+from setuptools_rust import Binding, RustExtension
+
+
+class BuildExtWithStubgen(_build_ext):
+    def run(self) -> None:
+        super().run()
+
+        module_name = "actuator.rust.lib"
+        output_dir = os.path.dirname(__file__)
+        os.makedirs(output_dir, exist_ok=True)
+        subprocess.run(["stubgen", "-m", module_name, "-o", output_dir], check=False)
+
 
 with open("README.md", "r", encoding="utf-8") as f:
     long_description: str = f.read()
@@ -30,6 +45,20 @@
     description="The actuator project",
     author="Benjamin Bolte",
     url="https://github.com/kscalelabs/actuator",
+    rust_extensions=[
+        RustExtension(
+            target="actuator.rust.lib",
+            path="actuator/rust/Cargo.toml",
+            binding=Binding.PyO3,
+        )
+    ],
+    cmdclass={"build_ext": BuildExtWithStubgen},
+    setup_requires=[
+        "setuptools-rust",
+        "mypy",  # For stubgen
+    ],
+    include_package_data=True,
+    zip_safe=False,
     long_description=long_description,
     long_description_content_type="text/markdown",
     python_requires=">=3.11",