shex_ast/ast/
iri_ref_or_wildcard.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
use iri_s::IriSError;
use prefixmap::IriRef;
use serde::ser::SerializeMap;
use serde::{Serialize, Serializer};
use std::{result, str::FromStr};

#[derive(Debug, PartialEq, Clone)]
pub enum IriRefOrWildcard {
    IriRef(IriRef),
    Wildcard,
}

impl FromStr for IriRefOrWildcard {
    type Err = IriSError;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        let iri_ref = IriRef::try_from(s)?;
        Ok(IriRefOrWildcard::IriRef(iri_ref))
    }
}

impl Serialize for IriRefOrWildcard {
    fn serialize<S>(&self, serializer: S) -> result::Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        match self {
            IriRefOrWildcard::IriRef(iri) => serializer.serialize_str(iri.to_string().as_str()),
            IriRefOrWildcard::Wildcard => {
                let mut map = serializer.serialize_map(Some(1))?;
                map.serialize_entry("type", "Wildcard")?;
                map.end()
            }
        }
    }
}