|
- use lazy_static::lazy_static;
- use pyo3::prelude::*;
- use pyo3_log::ResetHandle;
-
- pub mod acl;
- pub mod push;
-
- lazy_static! {
- static ref LOGGING_HANDLE: ResetHandle = pyo3_log::init();
- }
-
- /// Returns the hash of all the rust source files at the time it was compiled.
- ///
- /// Used by python to detect if the rust library is outdated.
- #[pyfunction]
- fn get_rust_file_digest() -> &'static str {
- env!("SYNAPSE_RUST_DIGEST")
- }
-
- /// Formats the sum of two numbers as string.
- #[pyfunction]
- #[pyo3(text_signature = "(a, b, /)")]
- fn sum_as_string(a: usize, b: usize) -> PyResult<String> {
- Ok((a + b).to_string())
- }
-
- /// Reset the cached logging configuration of pyo3-log to pick up any changes
- /// in the Python logging configuration.
- ///
- #[pyfunction]
- fn reset_logging_config() {
- LOGGING_HANDLE.reset();
- }
-
- /// The entry point for defining the Python module.
- #[pymodule]
- fn synapse_rust(py: Python<'_>, m: &PyModule) -> PyResult<()> {
- m.add_function(wrap_pyfunction!(sum_as_string, m)?)?;
- m.add_function(wrap_pyfunction!(get_rust_file_digest, m)?)?;
- m.add_function(wrap_pyfunction!(reset_logging_config, m)?)?;
-
- acl::register_module(py, m)?;
- push::register_module(py, m)?;
-
- Ok(())
- }
|