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

use bytes::Bytes;

use crate::command::common::PbToBytes;
use crate::command::img_store::GroupImageStoreResp;
use crate::common::RQAddr;
use crate::{pb, RQError, RQResult};

impl super::super::super::Engine {
    pub fn decode_group_image_store_response(
        &self,
        payload: Bytes,
    ) -> RQResult<GroupImageStoreResp> {
        let mut rsp = pb::cmd0x388::D388RspBody::from_bytes(&payload)
            .map_err(|_| RQError::Decode("D388RspBody".into()))?;
        let rsp = rsp
            .tryup_img_rsp
            .pop()
            .ok_or_else(|| RQError::Other("tryup_img_rsp.len = 0".into()))?;
        if rsp.result() != 0 {
            return Err(RQError::Other(
                String::from_utf8_lossy(&rsp.fail_msg.unwrap_or_default()).to_string(),
            ));
        }
        Ok(if rsp.file_exit() {
            GroupImageStoreResp::Exist {
                file_id: rsp.fileid.unwrap_or_default(),
                addrs: rsp
                    .up_ip
                    .into_iter()
                    .zip(rsp.up_port)
                    .map(|(ip, port)| RQAddr(ip, port as u16))
                    .collect(),
            }
        } else {
            GroupImageStoreResp::NotExist {
                file_id: rsp.fileid.unwrap_or_default(),
                upload_key: rsp.up_ukey.unwrap_or_default(),
                upload_addrs: rsp
                    .up_ip
                    .into_iter()
                    .zip(rsp.up_port)
                    .map(|(ip, port)| RQAddr(ip, port as u16))
                    .collect(),
            }
        })
    }
}