img-gen-spec 0.0.0

use crate::{Border, ColorKind, Corners, Rectangle};
use pyo3::{exceptions::PyValueError, prelude::*};

#[pymethods]
impl Corners {
    /// Translate a given string value into an enumeration of `Corners`.
    ///
    /// Throws `ValueError` if the string is not a supported value (see signature).
    #[staticmethod]
    #[pyo3(
        text_signature = "(val: Literal[\"top left\", \"top right\", \"bottom left\", \"bottom right\"]) -> Corners",
        name = "from_string"
    )]
    pub fn from_string_py(val: &str) -> PyResult<Corners> {
        Self::from_string(val).map_err(|e| PyValueError::new_err(e.to_string()))
    }

    #[staticmethod]
    pub fn all() -> Vec<Self> {
        Self::ALL.to_vec()
    }
}

#[pymethods]
impl Rectangle {
    /// Instantiate a `Rectangle` object.
    ///
    /// The `corners` parameter defaults to all `Corners` if `None` is specified.
    #[new]
    #[pyo3(
        text_signature = "(color: ColorKind, radius: float = 0, corners: list[Corners] | None = None, border: Border | None = None) -> Rectangle",
        signature = (color, radius = 0f32, corners = None, border = None)
    )]
    pub fn new(
        color: ColorKind,
        radius: f32,
        corners: Option<Vec<Corners>>,
        border: Option<Border>,
    ) -> Self {
        Self {
            color,
            radius,
            corners: match corners {
                None => Corners::ALL.to_vec(),
                Some(v) => {
                    if v.is_empty() {
                        vec![]
                    } else {
                        v
                    }
                }
            },
            border,
        }
    }
}