summaryrefslogtreecommitdiff
path: root/serde_dhall/src/options/ser.rs
blob: 026dd21efa9bad476ded8e28c66e37ca48b0a13a (plain)
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
use crate::{Result, SimpleType, StaticType, ToDhall};

#[derive(Debug, Clone, Copy)]
pub struct NoAnnot;
#[derive(Debug, Clone, Copy)]
pub struct ManualAnnot<'ty>(&'ty SimpleType);
#[derive(Debug, Clone, Copy)]
pub struct StaticAnnot;

pub trait RequiredAnnot<A> {
    fn get_annot(a: &A) -> SimpleType;
}
impl<'ty, T> RequiredAnnot<ManualAnnot<'ty>> for T {
    fn get_annot(a: &ManualAnnot<'ty>) -> SimpleType {
        a.0.clone()
    }
}
impl<T: StaticType> RequiredAnnot<StaticAnnot> for T {
    fn get_annot(_: &StaticAnnot) -> SimpleType {
        T::static_type()
    }
}

#[derive(Debug, Clone)]
pub struct Serializer<'a, T, A> {
    data: &'a T,
    annot: A,
}

impl<'a, T> Serializer<'a, T, NoAnnot> {
    pub fn type_annotation<'ty>(
        self,
        ty: &'ty SimpleType,
    ) -> Serializer<'a, T, ManualAnnot<'ty>> {
        Serializer {
            annot: ManualAnnot(ty),
            data: self.data,
        }
    }

    pub fn static_type_annotation(self) -> Serializer<'a, T, StaticAnnot> {
        Serializer {
            annot: StaticAnnot,
            data: self.data,
        }
    }
}

impl<'a, T, A> Serializer<'a, T, A> {
    pub fn to_string(&self) -> Result<String>
    where
        T: ToDhall + RequiredAnnot<A>,
    {
        let val = self.data.to_dhall(&T::get_annot(&self.annot))?;
        Ok(val.to_string())
    }
}

pub fn serialize<'a, T>(data: &'a T) -> Serializer<'a, T, NoAnnot>
where
    T: ToDhall,
{
    Serializer {
        data,
        annot: NoAnnot,
    }
}