Ptrskay3/PySprint

// This file basically sets up the panic hook for PySprint-CLI.
// Currently the other functions are just for testing purposes.
#![warn(clippy::all, clippy::pedantic)]

use pyo3::ffi::Py_FinalizeEx;
use pyo3::prelude::*;
use pyo3::wrap_pyfunction;

/// A placeholder function to check if the Rust <-> Python connection works.
#[pyfunction]
fn blank(a: usize) -> usize {
    a + 1
}

/// This function sets up the CTRL+C exit hook, which may be used
/// with PySprint-CLI. This calls `Py_FinalizeEx`, which makes sure
/// that all Python interpreters are destroyed correctly.
#[pyfunction]
fn set_panic_hook() {
    ctrlc::set_handler(|| {
        // destroy all (sub)interpreters, and free all the memory allocated
        // by the Python interpreter
        unsafe {
            Py_FinalizeEx();
        }
        std::process::exit(130)
    })
    .unwrap();
}

#[pymodule]
fn pysprint(_py: Python<'_>, m: &PyModule) -> PyResult<()> {
    m.add_function(wrap_pyfunction!(blank, m)?)?;
    m.add_function(wrap_pyfunction!(set_panic_hook, m)?)?;
    Ok(())
}

// This definitely needs a little bit of extra work to be able to test like this..

// #[test]
// fn test_blank_embedded() {
//     let gil = Python::acquire_gil();
//     let py = gil.python();
//     let result = py.eval("blank(1)", None, None).unwrap();
//     assert_eq!(result.extract::<usize>().unwrap(), 2);
// }

// #[test]
// #[should_panic]
// fn test_blank_embedded_panic() {
//     let gil = Python::acquire_gil();
//     let py = gil.python();
//     let result = py.eval("blank('notvalid')", None, None).unwrap();
//     assert_eq!(result.extract::<usize>().unwrap(), 2);
// }