charming 0.6.0

A visualization library for Rust
Documentation
use serde::{de::Visitor, Deserialize, Serialize};

#[derive(Debug, PartialEq, PartialOrd, Clone)]
pub enum Step {
    True,
    False,
    Start,
    Middle,
    End,
}

impl From<bool> for Step {
    fn from(value: bool) -> Self {
        if value {
            Step::True
        } else {
            Step::False
        }
    }
}

impl Serialize for Step {
    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: serde::Serializer,
    {
        match self {
            Step::True => serializer.serialize_bool(true),
            Step::False => serializer.serialize_bool(false),
            Step::Start => serializer.serialize_str("start"),
            Step::Middle => serializer.serialize_str("middle"),
            Step::End => serializer.serialize_str("end"),
        }
    }
}

impl<'de> Deserialize<'de> for Step {
    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
    where
        D: serde::Deserializer<'de>,
    {
        struct StepVisitor;

        impl Visitor<'_> for StepVisitor {
            type Value = Step;

            fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
                formatter.write_str(
                    "a boolean or a string containing either \"start\", \"middle\" or \"end\"",
                )
            }

            fn visit_bool<E>(self, v: bool) -> Result<Self::Value, E>
            where
                E: serde::de::Error,
            {
                match v {
                    true => Ok(Step::True),
                    false => Ok(Step::False),
                }
            }

            fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
            where
                E: serde::de::Error,
            {
                match v {
                    "start" => Ok(Step::Start),
                    "middle" => Ok(Step::Middle),
                    "end" => Ok(Step::End),
                    _ => Err(E::custom("unable to parse step type, invalid string")),
                }
            }
        }
        deserializer.deserialize_any(StepVisitor)
    }
}