bucky_objects/objects/
people.rs

1use crate::*;
2use serde::Serialize;
3
4use std::convert::TryFrom;
5use std::str::FromStr;
6
7#[repr(u8)]
8#[derive(Clone, Debug, Eq, PartialEq, RawEncode, RawDecode, Serialize)]
9pub enum OODWorkMode {
10    Standalone = 0,
11    ActiveStandby = 1,
12}
13
14impl OODWorkMode {
15    pub fn as_str(&self) -> &str {
16        match &self {
17            Self::Standalone => "standalone",
18            Self::ActiveStandby => "active-standby",
19        }
20    }
21}
22
23impl std::fmt::Display for OODWorkMode {
24    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
25        write!(f, "{}", self.as_str())
26    }
27}
28
29impl FromStr for OODWorkMode {
30    type Err = BuckyError;
31    fn from_str(value: &str) -> Result<Self, Self::Err> {
32        let ret = match value {
33            "standalone" => Self::Standalone,
34            "active-standby" => Self::ActiveStandby,
35            _ => {
36                let msg = format!("unknown OODWorkMode: {}", value);
37                error!("{}", msg);
38
39                return Err(BuckyError::new(BuckyErrorCode::UnSupport, msg));
40            }
41        };
42
43        Ok(ret)
44    }
45}
46
47#[derive(Clone, Debug, RawEncode, RawDecode)]
48pub struct PeopleDescContent {}
49
50impl PeopleDescContent {
51    pub fn new() -> Self {
52        Self {}
53    }
54}
55
56impl DescContent for PeopleDescContent {
57    fn obj_type() -> u16 {
58        ObjectTypeCode::People.into()
59    }
60
61    type OwnerType = Option<ObjectId>;
62    type AreaType = Option<Area>;
63    type AuthorType = SubDescNone;
64    type PublicKeyType = PublicKey;
65}
66
67#[derive(Clone, Debug)]
68pub struct PeopleBodyContent {
69    ood_work_mode: Option<OODWorkMode>,
70    ood_list: Vec<DeviceId>,
71    name: Option<String>,
72    icon: Option<FileId>,
73}
74
75impl BodyContent for PeopleBodyContent {
76    fn format(&self) -> u8 {
77        OBJECT_CONTENT_CODEC_FORMAT_PROTOBUF
78    }
79}
80
81// body使用protobuf编解码
82impl TryFrom<protos::PeopleBodyContent> for PeopleBodyContent {
83    type Error = BuckyError;
84
85    fn try_from(mut value: protos::PeopleBodyContent) -> BuckyResult<Self> {
86        let mut ret = Self {
87            ood_work_mode: None,
88            ood_list: ProtobufCodecHelper::decode_buf_list(value.take_ood_list())?,
89            name: None,
90            icon: None,
91        };
92
93        if value.has_ood_work_mode() {
94            ret.ood_work_mode = Some(OODWorkMode::from_str(value.get_ood_work_mode())?);
95        }
96
97        if value.has_name() {
98            ret.name = Some(value.take_name());
99        }
100        if value.has_icon() {
101            ret.icon = Some(ProtobufCodecHelper::decode_buf(value.take_icon())?);
102        }
103
104        Ok(ret)
105    }
106}
107
108impl TryFrom<&PeopleBodyContent> for protos::PeopleBodyContent {
109    type Error = BuckyError;
110
111    fn try_from(value: &PeopleBodyContent) -> BuckyResult<Self> {
112        let mut ret = Self::new();
113        ret.set_ood_list(ProtobufCodecHelper::encode_buf_list(&value.ood_list)?);
114
115        if let Some(ood_work_mode) = &value.ood_work_mode {
116            ret.set_ood_work_mode(ood_work_mode.to_string());
117        }
118
119        if let Some(name) = &value.name {
120            ret.set_name(name.to_owned());
121        }
122        if let Some(icon) = &value.icon {
123            ret.set_icon(icon.to_vec()?);
124        }
125
126        Ok(ret)
127    }
128}
129
130crate::inner_impl_default_protobuf_raw_codec!(PeopleBodyContent);
131
132impl PeopleBodyContent {
133    pub fn new(
134        ood_work_mode: OODWorkMode,
135        ood_list: Vec<DeviceId>,
136        name: Option<String>,
137        icon: Option<FileId>,
138    ) -> Self {
139        Self {
140            ood_work_mode: Some(ood_work_mode),
141            ood_list,
142            name,
143            icon,
144        }
145    }
146
147    pub fn ood_work_mode(&self) -> OODWorkMode {
148        self.ood_work_mode.clone().unwrap_or(OODWorkMode::Standalone)
149    }
150
151    pub fn set_ood_work_mode(&mut self, ood_work_mode: OODWorkMode) {
152        self.ood_work_mode = Some(ood_work_mode);
153    }
154
155    pub fn ood_list(&self) -> &Vec<DeviceId> {
156        &self.ood_list
157    }
158
159    pub fn ood_list_mut(&mut self) -> &mut Vec<DeviceId> {
160        &mut self.ood_list
161    }
162
163    pub fn name(&self) -> Option<&str> {
164        self.name.as_ref().map(|f| f.as_str())
165    }
166    pub fn icon(&self) -> Option<&FileId> {
167        self.icon.as_ref()
168    }
169
170    pub fn set_name(&mut self, name: String) {
171        self.name = Some(name)
172    }
173    pub fn set_icon(&mut self, icon: FileId) {
174        self.icon = Some(icon)
175    }
176}
177
178pub type PeopleType = NamedObjType<PeopleDescContent, PeopleBodyContent>;
179pub type PeopleBuilder = NamedObjectBuilder<PeopleDescContent, PeopleBodyContent>;
180
181pub type PeopleDesc = NamedObjectDesc<PeopleDescContent>;
182pub type PeopleId = NamedObjectId<PeopleType>;
183pub type People = NamedObjectBase<PeopleType>;
184
185impl PeopleDesc {
186    pub fn people_id(&self) -> PeopleId {
187        PeopleId::try_from(self.calculate_id()).unwrap()
188    }
189}
190
191impl People {
192    pub fn new(
193        owner: Option<ObjectId>,
194        ood_list: Vec<DeviceId>,
195        public_key: PublicKey,
196        area: Option<Area>,
197        name: Option<String>,
198        icon: Option<FileId>,
199    ) -> PeopleBuilder {
200        let desc_content = PeopleDescContent::new();
201
202        let body_content = PeopleBodyContent::new(OODWorkMode::Standalone, ood_list, name, icon);
203
204        PeopleBuilder::new(desc_content, body_content)
205            .option_owner(owner)
206            .option_area(area)
207            .public_key(public_key)
208    }
209
210    pub fn ood_work_mode(&self) -> OODWorkMode {
211        self.body().as_ref().unwrap().content().ood_work_mode()
212    }
213
214    pub fn set_ood_work_mode(&mut self, ood_work_mode: OODWorkMode) {
215        self.body_mut()
216            .as_mut()
217            .unwrap()
218            .content_mut()
219            .set_ood_work_mode(ood_work_mode)
220    }
221
222    pub fn ood_list(&self) -> &Vec<DeviceId> {
223        self.body().as_ref().unwrap().content().ood_list()
224    }
225
226    pub fn ood_list_mut(&mut self) -> &mut Vec<DeviceId> {
227        self.body_mut()
228            .as_mut()
229            .unwrap()
230            .content_mut()
231            .ood_list_mut()
232    }
233
234    pub fn name(&self) -> Option<&str> {
235        self.body().as_ref().unwrap().content().name()
236    }
237    pub fn icon(&self) -> Option<&FileId> {
238        self.body().as_ref().unwrap().content().icon()
239    }
240
241    pub fn set_name(&mut self, name: String) {
242        self.body_mut()
243            .as_mut()
244            .unwrap()
245            .content_mut()
246            .set_name(name)
247    }
248    pub fn set_icon(&mut self, icon: FileId) {
249        self.body_mut()
250            .as_mut()
251            .unwrap()
252            .content_mut()
253            .set_icon(icon)
254    }
255}
256
257#[cfg(test)]
258mod test {
259    use crate::*;
260    // use std::str::FromStr;
261
262    // 必须要跑的测试用以，用以测试新老编码的编码兼容性
263    const OLD_LIST: &'static [&'static str] = &[
264        //"0002520e002f3536f92b62c00000000000010030818902818100b81a66cc846de21354844cd3e3df991f45f4747c05b488027eacfb520ef3b3d4c2e58f2b75ad90bcf4a436a95ccd2cdd1b3bc0155fb92945524ec33c1ff263852288b44a75f430ebd9e3cdaffef59c075a3f16d43c3df30b1fdf167d696ec2f5ddbe0b1b69ab5397e1d46f2224cec8629dbd1e5e5fd9e569dde12077edf9c0c10203010001000000000000000000000000000000000000000000000000000000002f3536f92b62c0000140630a2044000000006c7e539b49ee3ad1cb1765b1afb344fecde8719416da84be391ad3123370656f706c652d3572344d59664637676355375a6a427a735079574748434e50734d3376514a7639324a506965516a576b6979220a7374616e64616c6f6e650100ff002f3536f92f2eac0061e7aac92e3ee196a9a1c278507da287d0ba872eff6de6abd7eeba3877d7c3e795c1cbe8ea524f58203c31f11a80e802afc5b00a5d5ceb9ce7d9fd2b6328dfeafd2bbfb9bb0217f378d4b04f5f230206573cf67a53b6298a62e2e86eda7d7769d93d64f550b85160d0f6c12b79a5b83aaf4a8c3a716a2b622bd059941f66dec70100ff002f3536f92f2fe900adb22dffd705deb589c0973a722f27d537cc2be48a365e90bc11007b4b6337dd402f550c9634043a2f25e723248aa1bfaa1b7880e6880789dc124ab801fd41b28551775ecb4b3bbdf6183a27c771a15a7459e1a809db89207cd1c9e66b9c80487dfef57ca5aa547063778bbf6ad9310df39e0de5dcfff6a11796c0e1f6723dd9",
265        "0002500e0000000000010030818902818100e0252144cac6aa8493f252c1c7d288afd9d01f04430a24f19bbd1f0fec428278b149f3b748e26a532c7e238dcdde6fb60d3820727f53b7ae090ce1bb04f637d43aea4551043a06535ded73e6a7de845e6a6187cfcd4def56b841fd098afc0671f659bfbabd1fbceb268b6fa0f47b8c7e3cb698a2d6ba120e54b6df9064c889ed0203010001000000000000000000000000000000000000000000000000000000002f3b2f6e3acd900001390a2045c40d30000cd65e863aa69f59d818f2090e3fa3b3d646dadc87568f773da50c12096275636b7962616c6c220a7374616e64616c6f6e650100ff002f3b2f6e3ad56000c661eeddb115b2b1cc8f6a0abe871b83c3108379c766303457676e1beca4836220767e72cac8fa53b3addff7686a604ee5537b4593d7ef363dcfd827dace32a51fb46c527330d6cb1a2bf37a4fcc4d6bae9ed5acf0289c7f6fa3e957c4a11362bf237746a253edd574c59acf85705d36747dd83ac65acd0995c201e76be7db5f0100ff002f3b2f6e3e09b000bb6a8d783a3be84580323e7e70b7aeb0c277c60c09705218fe77eb5a527df5cfdfe738d05c0661c9e06f59c883d7b314d4709d56675cecdde81bbb72dc692c5413c9a39f81aaf7fd928319f7bd4183132ab3c383b6e39a924a87ba1608133cbd2a6bdfeb2e613752971d24c944ed666ed5d50c9a77177ab4077143c5354f90c2",
266        "0002500e0000000000010030818902818100e0252144cac6aa8493f252c1c7d288afd9d01f04430a24f19bbd1f0fec428278b149f3b748e26a532c7e238dcdde6fb60d3820727f53b7ae090ce1bb04f637d43aea4551043a06535ded73e6a7de845e6a6187cfcd4def56b841fd098afc0671f659bfbabd1fbceb268b6fa0f47b8c7e3cb698a2d6ba120e54b6df9064c889ed0203010001000000000000000000000000000000000000000000000000000000002f3b2f6e3acd9000013f0a2045c40d30000cd65e863aa69f59d818f2090e3fa3b3d646dadc87568f773da50c120fe7bab3e696afe8b59be58d9ae4bcaf220a7374616e64616c6f6e650100ff002f3b2f6e3ad56000c661eeddb115b2b1cc8f6a0abe871b83c3108379c766303457676e1beca4836220767e72cac8fa53b3addff7686a604ee5537b4593d7ef363dcfd827dace32a51fb46c527330d6cb1a2bf37a4fcc4d6bae9ed5acf0289c7f6fa3e957c4a11362bf237746a253edd574c59acf85705d36747dd83ac65acd0995c201e76be7db5f0100ff002f3b2f6e3e09b000bb6a8d783a3be84580323e7e70b7aeb0c277c60c09705218fe77eb5a527df5cfdfe738d05c0661c9e06f59c883d7b314d4709d56675cecdde81bbb72dc692c5413c9a39f81aaf7fd928319f7bd4183132ab3c383b6e39a924a87ba1608133cbd2a6bdfeb2e613752971d24c944ed666ed5d50c9a77177ab4077143c5354f90c2",
267    ];
268
269    #[test]
270    fn test_codec_with_author() {
271        let data = "000262220564656369640000000000000000000000000000000000000000000000000000002f4f8b051af10806617574686f7200000000000000000000000000000000000000000000000000010030818902818100e6f0a1f6d3ca953f315039103eea4f3bf4f4e4acab178a3f6552d45bb3352fb875a386c51ef0eeed2ed4c49faabcf47c252333cd8d809fcc237dcb537276e8ee71a6b96a881c1c177bce51ecd5222dc08f37bdf9d461d6f751fe49d3070faaf07708bb82e59e0ec839351fdd88c4bb344e4366895f12444608e8b2160112e6bd0203010001000000000000000000000000000000000000000000000000000000002f4f8b051af108000100";
272        let buf = hex::decode(data).unwrap();
273        People::clone_from_slice(&buf).unwrap_err();
274    }
275
276    #[test]
277    fn test_codec() {
278        for code in OLD_LIST {
279            let code = hex::decode(code).unwrap();
280            let people = People::clone_from_slice(&code).unwrap();
281
282            println!("ood list: {:?}", people.ood_list());
283            println!("ood_work_mode: {}, name={:?}", people.ood_work_mode(), people.name());
284            let hash = people.body().as_ref().unwrap().raw_hash_value().unwrap();
285            println!("desc hash: {}", hash);
286
287            let new_buf = people.to_vec().unwrap();
288            assert_eq!(code.len(), new_buf.len());
289
290            println!("code: {:?}", code);
291            assert_eq!(code, new_buf);
292        }
293    }
294
295    #[test]
296    fn people() {
297        let private_key = PrivateKey::generate_rsa(1024).unwrap();
298
299        let pubic_key = private_key.public();
300
301        let mut p = People::new(None, Vec::new(), pubic_key.clone(), None, None, None)
302            .no_create_time()
303            .build();
304
305        let p2 = People::new(None, Vec::new(), pubic_key, None, None, None)
306            .no_create_time()
307            .build();
308
309        assert!(p.desc().people_id() == p2.desc().people_id());
310
311        p.set_name("people".to_owned());
312
313        // let path = Path::new("f:\\temp\\people.obj");
314        // if path.parent().unwrap().exists() {
315        //     p.encode_to_file(path, false);
316        // }
317
318        let user_data = vec![0u8; 100];
319        let _ = p.body_mut().as_mut().unwrap().set_userdata(&user_data);
320
321        let buf = p.to_vec().unwrap();
322        let pp = People::clone_from_slice(&buf).unwrap();
323
324        assert_eq!(p.desc().people_id(), pp.desc().people_id());
325        assert_eq!(p.name(), pp.name());
326    }
327}
bucky_objects/objects/people.rs

bucky_objects/objects/
people.rs
