mirror of
https://github.com/ankitects/anki.git
synced 2025-09-18 14:02:21 -04:00
b241ab9492
* Bump Python deps Primarily for flask-cors CVE * Bump TS deps; pin license checker Current checker is missing the binary https://github.com/RSeidelsohn/license-checker-rseidelsohn/issues/118 * Update Rust deps Hyper and axum are held back as we currently make use of the older http library that reqwest pulls in
92 lines
2.6 KiB
Rust
92 lines
2.6 KiB
Rust
// Copyright: Ankitects Pty Ltd and contributors
|
|
// License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
use anki::backend::init_backend;
|
|
use anki::backend::Backend as RustBackend;
|
|
use anki::log::set_global_logger;
|
|
use anki::sync::http_server::SimpleServer;
|
|
use pyo3::create_exception;
|
|
use pyo3::exceptions::PyException;
|
|
use pyo3::prelude::*;
|
|
use pyo3::types::PyBytes;
|
|
use pyo3::wrap_pyfunction;
|
|
|
|
#[pyclass(module = "_rsbridge")]
|
|
struct Backend {
|
|
backend: RustBackend,
|
|
}
|
|
|
|
create_exception!(_rsbridge, BackendError, PyException);
|
|
|
|
#[pyfunction]
|
|
fn buildhash() -> &'static str {
|
|
anki::version::buildhash()
|
|
}
|
|
|
|
#[pyfunction]
|
|
#[pyo3(signature = (path=None))]
|
|
fn initialize_logging(path: Option<&str>) -> PyResult<()> {
|
|
set_global_logger(path).map_err(|e| PyException::new_err(e.to_string()))
|
|
}
|
|
|
|
#[pyfunction]
|
|
fn syncserver() -> PyResult<()> {
|
|
set_global_logger(None).unwrap();
|
|
let err = SimpleServer::run();
|
|
Err(PyException::new_err(err.to_string()))
|
|
}
|
|
|
|
#[pyfunction]
|
|
fn open_backend(init_msg: &Bound<'_, PyBytes>) -> PyResult<Backend> {
|
|
match init_backend(init_msg.as_bytes()) {
|
|
Ok(backend) => Ok(Backend { backend }),
|
|
Err(e) => Err(PyException::new_err(e)),
|
|
}
|
|
}
|
|
|
|
#[pymethods]
|
|
impl Backend {
|
|
fn command(
|
|
&self,
|
|
py: Python,
|
|
service: u32,
|
|
method: u32,
|
|
input: &Bound<'_, PyBytes>,
|
|
) -> PyResult<PyObject> {
|
|
let in_bytes = input.as_bytes();
|
|
py.allow_threads(|| self.backend.run_service_method(service, method, in_bytes))
|
|
.map(|out_bytes| {
|
|
let out_obj = PyBytes::new_bound(py, &out_bytes);
|
|
out_obj.into()
|
|
})
|
|
.map_err(BackendError::new_err)
|
|
}
|
|
|
|
/// This takes and returns JSON, due to Python's slow protobuf
|
|
/// encoding/decoding.
|
|
fn db_command(&self, py: Python, input: &Bound<'_, PyBytes>) -> PyResult<PyObject> {
|
|
let in_bytes = input.as_bytes();
|
|
let out_res = py.allow_threads(|| {
|
|
self.backend
|
|
.run_db_command_bytes(in_bytes)
|
|
.map_err(BackendError::new_err)
|
|
});
|
|
let out_bytes = out_res?;
|
|
let out_obj = PyBytes::new_bound(py, &out_bytes);
|
|
Ok(out_obj.into())
|
|
}
|
|
}
|
|
|
|
// Module definition
|
|
//////////////////////////////////
|
|
|
|
#[pymodule]
|
|
fn _rsbridge(_py: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
|
|
m.add_class::<Backend>()?;
|
|
m.add_wrapped(wrap_pyfunction!(buildhash)).unwrap();
|
|
m.add_wrapped(wrap_pyfunction!(open_backend)).unwrap();
|
|
m.add_wrapped(wrap_pyfunction!(initialize_logging)).unwrap();
|
|
m.add_wrapped(wrap_pyfunction!(syncserver)).unwrap();
|
|
|
|
Ok(())
|
|
}
|