grammers_session/
session_data.rs

1// Copyright 2020 - developers of the `grammers` project.
2//
3// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
4// https://www.apache.org/licenses/LICENSE-2.0> or the MIT license
5// <LICENSE-MIT or https://opensource.org/licenses/MIT>, at your
6// option. This file may not be copied, modified, or distributed
7// except according to those terms.
8
9use std::collections::HashMap;
10
11use crate::types::{DcOption, PeerId, PeerInfo, UpdateState, UpdatesState};
12use crate::{DEFAULT_DC, KNOWN_DC_OPTIONS, Session};
13
14/// In-memory representation of the entire [`Session`] state.
15///
16/// This type can be used for conversions `From` any [`Session`],
17/// and be [`SessionData::import_to`] any other [`Session`].
18pub struct SessionData {
19    /// The identifier of the datacenter option determined
20    /// to be the primary one for the logged-in user, or
21    /// the identifier of an arbitrary datacenter otherwise.
22    pub home_dc: i32,
23    /// List of all known datacenter options, along with their
24    /// Authorization Key if an encrypted connection has been
25    /// made to them previously. Indexed by their identifier.
26    pub dc_options: HashMap<i32, DcOption>,
27    /// List of all peer informations cached in the session.
28    /// Indexed by their identifier.
29    pub peer_infos: HashMap<PeerId, PeerInfo>,
30    /// Entirety of the update state for the logged-in user.
31    pub updates_state: UpdatesState,
32}
33
34impl Default for SessionData {
35    /// Constructs a default instance of the session data, with an arbitrary
36    /// [`Self::home_dc`] and the list of statically-known [`Self::dc_options`].
37    fn default() -> Self {
38        Self {
39            home_dc: DEFAULT_DC,
40            dc_options: KNOWN_DC_OPTIONS
41                .iter()
42                .cloned()
43                .map(|dc_option| (dc_option.id, dc_option))
44                .collect(),
45            peer_infos: HashMap::new(),
46            updates_state: UpdatesState::default(),
47        }
48    }
49}
50
51impl SessionData {
52    /// Imports all information from this session data to a type implementing `Session`.
53    pub async fn import_to<S: Session>(&self, session: &S) {
54        session.set_home_dc_id(self.home_dc).await;
55        for dc_option in self.dc_options.values() {
56            session.set_dc_option(dc_option).await;
57        }
58        for peer in self.peer_infos.values() {
59            session.cache_peer(peer).await;
60        }
61        session
62            .set_update_state(UpdateState::All(self.updates_state.clone()))
63            .await;
64    }
65}
grammers_session/session_data.rs

grammers_session/
session_data.rs
