feature: support pyo3

Cargo.toml

@@ -1,9 +1,9 @@
 [workspace]
-members  = ["node-binding", "algo"]
+members = ["algo", "node-binding", "python-binding"]
 resolver = "2"
 
-  [workspace.dependencies]
-  algo = { path = "algo" }
+[workspace.dependencies]
+algo = { path = "algo" }
 
 [profile.release]
 lto = true

algo/src/algo.rs

@@ -17,7 +17,7 @@ impl Algo {
         Self { r#type }
     }
 
-    pub fn hash(&self, v: String) -> String {
+    pub fn hash(&self, v: &str) -> String {
         match self.r#type {
             AlgoType::Blake3 => {
                 let hash = blake3::hash(v.as_bytes());
@@ -47,11 +47,11 @@ mod tests {
     fn hash_should_work() {
         let algo = Algo::new(AlgoType::Blake3);
         assert_eq!(
-            algo.hash("hello".to_string()),
+            algo.hash("hello"),
             "ea8f163db38682925e4491c5e58d4bb3506ef8c14eb78a86e908c5624a67200f"
         );
 
         let algo = Algo::new(AlgoType::Default);
-        assert_eq!(algo.hash("hello".to_string()), "16156531084128653017");
+        assert_eq!(algo.hash("hello"), "16156531084128653017");
     }
 }

deny.toml

@@ -73,6 +73,7 @@ unlicensed = "allow"
 allow = [
   "MIT",
   "Apache-2.0",
+  "Apache-2.0 WITH LLVM-exception",
   "Unicode-DFS-2016",
   "MPL-2.0",
   "BSD-2-Clause",

node-binding/src/algo.rs

@@ -19,7 +19,7 @@ impl JsAlgo {
 
   #[napi]
   pub fn hash(&self, v: String) -> String {
-    self.inner.hash(v)
+    self.inner.hash(&v)
   }
 
   #[napi]

python-binding/.gitignore

@@ -0,0 +1,13 @@
+# python generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+# Rust
+target/
+
+# venv
+.venv
+*.so

python-binding/.python-version

@@ -0,0 +1 @@
+3.12.3

python-binding/Cargo.toml

@@ -0,0 +1,13 @@
+[package]
+name = "python-binding"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+[lib]
+name = "python_binding"
+crate-type = ["cdylib"]
+
+[dependencies]
+algo = { workspace = true }
+pyo3 = { version = "0.22.2", features = ["extension-module"] }

python-binding/README.md

@@ -0,0 +1,3 @@
+# python-binding
+
+Describe your project here.

python-binding/pyproject.toml

@@ -0,0 +1,21 @@
+[project]
+name = "algo"
+version = "0.1.0"
+description = "Add your description here"
+authors = [{ name = "Tyr Chen", email = "[email protected]" }]
+dependencies = []
+readme = "README.md"
+requires-python = ">= 3.8"
+
+[build-system]
+requires = ["maturin>=1.2,<2.0"]
+build-backend = "maturin"
+
+[tool.rye]
+managed = true
+dev-dependencies = ["pip>=24.2", "ipython>=8.26.0"]
+
+[tool.maturin]
+python-source = "python"
+module-name = "algo._lowlevel"
+features = ["pyo3/extension-module"]

python-binding/python/algo/__init__.py

@@ -0,0 +1,4 @@
+from algo._lowlevel import hello
+from algo._lowlevel import Algo
+
+__all__ = ["hello", "Algo"]

python-binding/requirements-dev.lock

@@ -0,0 +1,43 @@
+# generated by rye
+# use `rye lock` or `rye sync` to update this lockfile
+#
+# last locked with the following flags:
+#   pre: false
+#   features: []
+#   all-features: false
+#   with-sources: false
+
+-e file:.
+asttokens==2.4.1
+    # via stack-data
+decorator==5.1.1
+    # via ipython
+executing==2.0.1
+    # via stack-data
+ipython==8.26.0
+jedi==0.19.1
+    # via ipython
+matplotlib-inline==0.1.7
+    # via ipython
+parso==0.8.4
+    # via jedi
+pexpect==4.9.0
+    # via ipython
+pip==24.2
+prompt-toolkit==3.0.47
+    # via ipython
+ptyprocess==0.7.0
+    # via pexpect
+pure-eval==0.2.3
+    # via stack-data
+pygments==2.18.0
+    # via ipython
+six==1.16.0
+    # via asttokens
+stack-data==0.6.3
+    # via ipython
+traitlets==5.14.3
+    # via ipython
+    # via matplotlib-inline
+wcwidth==0.2.13
+    # via prompt-toolkit

python-binding/requirements.lock

@@ -0,0 +1,10 @@
+# generated by rye
+# use `rye lock` or `rye sync` to update this lockfile
+#
+# last locked with the following flags:
+#   pre: false
+#   features: []
+#   all-features: false
+#   with-sources: false
+
+-e file:.

python-binding/src/hasher.rs

@@ -0,0 +1,44 @@
+use std::fmt;
+
+use algo::{Algo, AlgoType};
+use pyo3::prelude::*;
+
+#[pyclass(name = "Algo")]
+pub struct PyAlgo {
+    inner: Algo,
+}
+
+#[pymethods]
+impl PyAlgo {
+    #[new]
+    #[pyo3(signature = (name=""))]
+    pub fn new(name: &str) -> Self {
+        let algo = match name {
+            "blake3" => Algo::new(AlgoType::Blake3),
+            _ => Algo::new(AlgoType::Default),
+        };
+        Self { inner: algo }
+    }
+
+    pub fn hash(&self, v: &str) -> String {
+        self.inner.hash(v)
+    }
+
+    pub fn get_name(&self) -> &str {
+        self.inner.get_name()
+    }
+
+    pub fn __repr__(&self) -> String {
+        self.to_string()
+    }
+
+    pub fn __str__(&self) -> String {
+        self.to_string()
+    }
+}
+
+impl fmt::Display for PyAlgo {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        write!(f, "<Algo: {}>", self.inner.get_name())
+    }
+}

python-binding/src/lib.rs

@@ -0,0 +1,18 @@
+mod hasher;
+
+use hasher::PyAlgo;
+use pyo3::prelude::*;
+
+/// Prints a message.
+#[pyfunction]
+fn hello() -> PyResult<String> {
+    Ok("Hello from python-binding!".into())
+}
+
+/// A Python module implemented in Rust.
+#[pymodule]
+fn _lowlevel(_py: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
+    m.add_function(wrap_pyfunction!(hello, m)?)?;
+    m.add_class::<PyAlgo>()?;
+    Ok(())
+}