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

use std::collections::HashMap;

use bolt_proto_derive::*;

use crate::{impl_try_from_message, message::SIGNATURE_INIT, Value};

#[bolt_structure(SIGNATURE_INIT)]
#[derive(Debug, Clone, Eq, PartialEq)]
pub struct Init {
    pub(crate) user_agent: String,
    pub(crate) auth_token: HashMap<String, Value>,
}

impl Init {
    pub fn new(user_agent: String, auth_token: HashMap<String, Value>) -> Self {
        Self {
            user_agent,
            auth_token,
        }
    }

    pub fn user_agent(&self) -> &str {
        &self.user_agent
    }

    pub fn auth_token(&self) -> &HashMap<String, Value> {
        &self.auth_token
    }
}

impl_try_from_message!(Init, Init);