use std::fmt;
use serde::{
de::{self, Visitor},
Deserialize, Deserializer, Serialize, Serializer,
};
#[derive(Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub enum Label {
Int(i64),
Text(String),
}
impl Label {
pub fn as_int(&self) -> Option<i64> {
match self {
Label::Int(i) => Some(*i),
Label::Text(_) => None,
}
}
pub fn as_text(&self) -> Option<&str> {
match self {
Label::Text(s) => Some(s),
Label::Int(_) => None,
}
}
}
impl fmt::Display for Label {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
match self {
Label::Int(i) => write!(f, "{i}"),
Label::Text(s) => write!(f, "{s}"),
}
}
}
impl From<i64> for Label {
fn from(value: i64) -> Self {
Label::Int(value)
}
}
impl From<i32> for Label {
fn from(value: i32) -> Self {
Label::Int(value as i64)
}
}
impl From<String> for Label {
fn from(value: String) -> Self {
Label::Text(value)
}
}
impl From<&str> for Label {
fn from(value: &str) -> Self {
Label::Text(value.to_string())
}
}
impl Serialize for Label {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
match self {
Label::Int(i) => serializer.serialize_i64(*i),
Label::Text(s) => serializer.serialize_str(s),
}
}
}
impl<'de> Deserialize<'de> for Label {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: Deserializer<'de>,
{
struct LabelVisitor;
impl Visitor<'_> for LabelVisitor {
type Value = Label;
fn expecting(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.write_str("an integer or text string COSE label")
}
fn visit_i64<E: de::Error>(self, v: i64) -> Result<Label, E> {
Ok(Label::Int(v))
}
fn visit_i128<E: de::Error>(self, v: i128) -> Result<Label, E> {
i64::try_from(v)
.map(Label::Int)
.map_err(|_| E::custom("integer label out of range"))
}
fn visit_u64<E: de::Error>(self, v: u64) -> Result<Label, E> {
i64::try_from(v)
.map(Label::Int)
.map_err(|_| E::custom("integer label out of range"))
}
fn visit_u128<E: de::Error>(self, v: u128) -> Result<Label, E> {
i64::try_from(v)
.map(Label::Int)
.map_err(|_| E::custom("integer label out of range"))
}
fn visit_str<E: de::Error>(self, v: &str) -> Result<Label, E> {
Ok(Label::Text(v.to_string()))
}
fn visit_string<E: de::Error>(self, v: String) -> Result<Label, E> {
Ok(Label::Text(v))
}
}
deserializer.deserialize_any(LabelVisitor)
}
}