img-gen-renderer 0.1.0

use crate::{
    Layout,
    generator::{Generator, Image},
};
use pyo3::{exceptions::PyOSError, prelude::*};
use std::{borrow::Cow, path::PathBuf};

#[pymethods]
impl Generator {
    /// Instantiate a `Generator` object for a given `layout`.
    #[new]
    #[pyo3(
        text_signature = "(external_resource_paths: list[Path| str] | None = None, cache_root: Path | str | None = None) -> Generator",
        signature = (external_resource_paths=None, cache_root=None)
    )]
    pub fn new_py(
        external_resource_paths: Option<Vec<PathBuf>>,
        cache_root: Option<PathBuf>,
    ) -> PyResult<Self> {
        Generator::new(external_resource_paths.unwrap_or_default(), cache_root)
            .map_err(|e| PyOSError::new_err(format!("{e:?}")))
    }

    /// Render the layout and return the `Image`.
    #[pyo3(name = "render", text_signature = "(layout: Layout) -> Image")]
    pub fn render_py<'py>(&self, py: Python<'py>, layout: Layout) -> PyResult<Bound<'py, PyAny>> {
        let this = self.clone();
        pyo3_async_runtimes::tokio::future_into_py(py, async move {
            this.render(layout)
                .await
                .map_err(|e| PyOSError::new_err(format!("{e:?}")))
        })
    }
}

#[pymethods]
impl Image {
    /// The image data as a `bytes` object
    #[getter(bytes)]
    pub fn get_bytes_py(&'_ self) -> PyResult<Cow<'_, [u8]>> {
        self.get_bytes()
            .map_err(|e| PyOSError::new_err(e.to_string()))
    }

    /// Save the image data to a file.
    ///
    /// Does not support SVG output.
    /// The image format is determined from the file extension in the given ``name``.
    #[pyo3(text_signature = "(name: Path | str) -> None", name = "save")]
    pub fn save_py(&self, name: PathBuf) -> PyResult<()> {
        self.save(&name)
            .map_err(|e| PyOSError::new_err(e.to_string()))
    }

    /// A hexadecimal string representing the SHA256 hash digest of the image data.
    #[getter(sha256)]
    pub fn get_sha256_py(&self) -> PyResult<String> {
        self.get_sha256()
            .map_err(|e| PyOSError::new_err(e.to_string()))
    }
}