cipherstash-client 0.34.1-alpha.1

The official CipherStash SDK
Documentation 
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138

use super::const_args::*;
use crate::encryption::{EncryptionError, Plaintext};

#[derive(Debug, Clone, PartialEq)]
pub enum ComposablePlaintext {
    Unit(Plaintext),
    ConstArg2(ConstArg2),
    ConstArg3(ConstArg3),
    ConstArg4(ConstArg4),
}

impl ComposablePlaintext {
    pub fn new(plaintext: impl Into<Plaintext>) -> Self {
        Self::Unit(plaintext.into())
    }

    pub fn as_plaintext(self) -> Option<Plaintext> {
        match self {
            ComposablePlaintext::Unit(inner) => Some(inner),
            _ => None,
        }
    }

    pub fn try_compose(self, plaintext: impl Into<Plaintext>) -> Result<Self, EncryptionError> {
        let plaintext: Plaintext = plaintext.into();
        match self {
            Self::Unit(a) => Ok(Self::ConstArg2(ConstArg2::new(a, plaintext))),
            Self::ConstArg2(inner) => Ok(Self::ConstArg3(ConstArg3(plaintext, inner))),
            Self::ConstArg3(inner) => Ok(Self::ConstArg4(ConstArg4(plaintext, inner))),
            Self::ConstArg4(_) => Err(EncryptionError::TooManyArguments),
        }
    }

    /// Returns the head of the cons list and the tail of the cons list (if it exists)
    pub fn pop(self) -> (Self, Option<Self>) {
        match self {
            Self::Unit(head) => (Self::Unit(head), None),
            Self::ConstArg2(ConstArg2(head, tail)) => (Self::Unit(head), Some(Self::Unit(tail))),
            Self::ConstArg3(ConstArg3(head, tail)) => {
                (Self::Unit(head), Some(Self::ConstArg2(tail)))
            }
            Self::ConstArg4(ConstArg4(head, tail)) => {
                (Self::Unit(head), Some(Self::ConstArg3(tail)))
            }
        }
    }
}

impl<T: Into<Plaintext>> From<T> for ComposablePlaintext {
    fn from(plaintext: T) -> Self {
        Self::Unit(plaintext.into())
    }
}

impl<A, B> TryFrom<(A, B)> for ComposablePlaintext
where
    A: Into<Plaintext>,
    B: Into<Plaintext>,
{
    type Error = EncryptionError;

    fn try_from((a, b): (A, B)) -> Result<Self, Self::Error> {
        Self::Unit(a.into()).try_compose(b)
    }
}

impl<A, B, C> TryFrom<(A, B, C)> for ComposablePlaintext
where
    A: Into<Plaintext>,
    B: Into<Plaintext>,
    C: Into<Plaintext>,
{
    type Error = EncryptionError;

    fn try_from((a, b, c): (A, B, C)) -> Result<Self, Self::Error> {
        Self::Unit(a.into()).try_compose(b)?.try_compose(c)
    }
}

impl<A, B, C, D> TryFrom<(A, B, C, D)> for ComposablePlaintext
where
    A: Into<Plaintext>,
    B: Into<Plaintext>,
    C: Into<Plaintext>,
    D: Into<Plaintext>,
{
    type Error = EncryptionError;

    fn try_from((a, b, c, d): (A, B, C, D)) -> Result<Self, Self::Error> {
        Self::Unit(a.into())
            .try_compose(b)?
            .try_compose(c)?
            .try_compose(d)
    }
}

impl TryFrom<ComposablePlaintext> for Plaintext {
    type Error = EncryptionError;

    fn try_from(value: ComposablePlaintext) -> Result<Self, Self::Error> {
        value
            .as_plaintext()
            .ok_or_else(|| EncryptionError::TooManyArguments)
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_pops_in_reverse_insert_order() -> Result<(), Box<dyn std::error::Error>> {
        let output = ComposablePlaintext::new("First")
            .try_compose("Second")?
            .try_compose("Third")?;

        let (head, output) = output.pop();
        assert_eq!(head, ComposablePlaintext::new("Third"));

        let (head, output) = output.unwrap().pop();
        assert_eq!(head, ComposablePlaintext::new("Second"));

        let (head, _) = output.unwrap().pop();
        assert_eq!(head, ComposablePlaintext::new("First"));

        Ok(())
    }

    #[test]
    fn test_single_plaintext_convers() -> Result<(), Box<dyn std::error::Error>> {
        let output = ComposablePlaintext::new("First");
        let plaintext = output.as_plaintext().expect("expected plaintext");

        assert_eq!(plaintext, Plaintext::from("First"));

        Ok(())
    }
}