kanidmd_lib/valueset/
secret.rs

1use crate::prelude::*;
2use crate::schema::SchemaAttribute;
3use crate::valueset::{DbValueSetV2, ScimResolveStatus, ValueSet};
4
5use smolset::SmolSet;
6
7#[derive(Debug, Clone)]
8pub struct ValueSetSecret {
9    set: SmolSet<[String; 1]>,
10}
11
12impl ValueSetSecret {
13    pub fn new(b: String) -> Box<Self> {
14        let mut set = SmolSet::new();
15        set.insert(b);
16        Box::new(ValueSetSecret { set })
17    }
18
19    pub fn push(&mut self, b: String) -> bool {
20        self.set.insert(b)
21    }
22
23    pub fn from_dbvs2(data: Vec<String>) -> Result<ValueSet, OperationError> {
24        let set = data.into_iter().collect();
25        Ok(Box::new(ValueSetSecret { set }))
26    }
27
28    // We need to allow this, because rust doesn't allow us to impl FromIterator on foreign
29    // types, and String is foreign.
30    #[allow(clippy::should_implement_trait)]
31    pub fn from_iter<T>(iter: T) -> Option<Box<Self>>
32    where
33        T: IntoIterator<Item = String>,
34    {
35        let set = iter.into_iter().collect();
36        Some(Box::new(ValueSetSecret { set }))
37    }
38}
39
40impl ValueSetT for ValueSetSecret {
41    fn insert_checked(&mut self, value: Value) -> Result<bool, OperationError> {
42        match value {
43            Value::SecretValue(u) => Ok(self.set.insert(u)),
44            _ => {
45                debug_assert!(false);
46                Err(OperationError::InvalidValueState)
47            }
48        }
49    }
50
51    fn clear(&mut self) {
52        self.set.clear();
53    }
54
55    fn remove(&mut self, _pv: &PartialValue, _cid: &Cid) -> bool {
56        false
57    }
58
59    fn contains(&self, _pv: &PartialValue) -> bool {
60        false
61    }
62
63    fn substring(&self, _pv: &PartialValue) -> bool {
64        false
65    }
66
67    fn startswith(&self, _pv: &PartialValue) -> bool {
68        false
69    }
70
71    fn endswith(&self, _pv: &PartialValue) -> bool {
72        false
73    }
74
75    fn lessthan(&self, _pv: &PartialValue) -> bool {
76        false
77    }
78
79    fn len(&self) -> usize {
80        self.set.len()
81    }
82
83    fn generate_idx_eq_keys(&self) -> Vec<String> {
84        Vec::with_capacity(0)
85    }
86
87    fn syntax(&self) -> SyntaxType {
88        SyntaxType::SecretUtf8String
89    }
90
91    fn validate(&self, _schema_attr: &SchemaAttribute) -> bool {
92        true
93    }
94
95    fn to_proto_string_clone_iter(&self) -> Box<dyn Iterator<Item = String> + '_> {
96        Box::new(self.set.iter().map(|_| "hidden".to_string()))
97    }
98
99    fn to_scim_value(&self) -> Option<ScimResolveStatus> {
100        None
101    }
102
103    fn to_db_valueset_v2(&self) -> DbValueSetV2 {
104        DbValueSetV2::SecretValue(self.set.iter().cloned().collect())
105    }
106
107    fn to_partialvalue_iter(&self) -> Box<dyn Iterator<Item = PartialValue> + '_> {
108        Box::new(self.set.iter().map(|_| PartialValue::SecretValue))
109    }
110
111    fn to_value_iter(&self) -> Box<dyn Iterator<Item = Value> + '_> {
112        Box::new(self.set.iter().cloned().map(Value::SecretValue))
113    }
114
115    fn equal(&self, other: &ValueSet) -> bool {
116        if let Some(other) = other.as_secret_set() {
117            &self.set == other
118        } else {
119            debug_assert!(false);
120            false
121        }
122    }
123
124    fn merge(&mut self, other: &ValueSet) -> Result<(), OperationError> {
125        if let Some(b) = other.as_secret_set() {
126            mergesets!(self.set, b)
127        } else {
128            debug_assert!(false);
129            Err(OperationError::InvalidValueState)
130        }
131    }
132
133    fn to_secret_single(&self) -> Option<&str> {
134        if self.set.len() == 1 {
135            self.set.iter().map(|s| s.as_str()).take(1).next()
136        } else {
137            None
138        }
139    }
140
141    fn as_secret_set(&self) -> Option<&SmolSet<[String; 1]>> {
142        Some(&self.set)
143    }
144}
145
146#[cfg(test)]
147mod tests {
148    use crate::valueset::{ValueSet, ValueSetSecret};
149
150    #[test]
151    fn test_scim_secret() {
152        let vs: ValueSet = ValueSetSecret::new("super secret special awesome value".to_string());
153
154        assert!(vs.to_scim_value().is_none());
155    }
156}