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

use std::collections::VecDeque;

use log::trace;
use uuid::Uuid;

use crate::{
    types::ByteMessage,
    util::{
        converter::{str_to_decque, uuid_to_decque},
        modes::Modes,
        tools::{u64_from_vec, uuid_from_vec},
        vec::{decque_to_vec, vec_to_decque, extract_vec},
    }, consts::MSG_DIGEST,
};

#[derive(Debug, Clone)]
pub struct FileQuestionMsg {
    pub filename: String,
    pub sender: Uuid,
    pub receiver: Uuid,
    pub uuid: Uuid,
    // Uses sha256 so 32 bytes
    pub hash: Vec<u8>,
    pub size: u64
}

impl ByteMessage for FileQuestionMsg {
    fn serialize(&self) -> Vec<u8> {
        let mut merged: VecDeque<u8> = VecDeque::new();
        let mut b_filename = str_to_decque(&self.filename);

        let mut b_uuid = uuid_to_decque(&self.uuid);
        let mut b_receiver = uuid_to_decque(&self.receiver);
        let mut b_sender = uuid_to_decque(&self.sender);
        let mut b_size = vec_to_decque(self.size.to_le_bytes().to_vec());
        let mut b_hash = vec_to_decque(self.hash.clone());

        merged.append(&mut b_uuid);
        merged.append(&mut b_receiver);
        merged.append(&mut b_sender);
        merged.append(&mut b_size);
        merged.append(&mut b_hash);
        merged.append(&mut b_filename);

        return Modes::SendFileQuestion.get_send(&decque_to_vec(merged));
    }

    fn deserialize(data: &Vec<u8>) -> anyhow::Result<Self> {
        trace!("Parsing file question...");
        let mut data = data.clone();

        let uuid = uuid_from_vec(&mut data)?;
        let receiver = uuid_from_vec(&mut data)?;
        let sender = uuid_from_vec(&mut data)?;
        let size = u64_from_vec(&mut data)?;
        let hash = extract_vec(0..MSG_DIGEST.size(), &mut data)?;

        let filename = String::from_utf8(data)?;

        let msg = FileQuestionMsg {
            filename,
            uuid,
            sender,
            receiver,
            size,
            hash
        };

        trace!("FileQuestion info is {:?}", msg);
        return Ok(msg);
    }
}