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
use crate::{IdentityError, IdentityIdentifier};
use ockam_core::compat::vec::Vec;
use ockam_core::{Decodable, Encodable, LocalInfo, LocalMessage, Result};
use serde::{Deserialize, Serialize};
pub const IDENTITY_SECURE_CHANNEL_IDENTIFIER: &str = "IDENTITY_SECURE_CHANNEL_IDENTIFIER";
#[derive(Serialize, Deserialize)]
pub struct IdentitySecureChannelLocalInfo {
their_identity_id: IdentityIdentifier,
}
impl IdentitySecureChannelLocalInfo {
pub fn from_local_info(value: &LocalInfo) -> Result<Self> {
if value.type_identifier() != IDENTITY_SECURE_CHANNEL_IDENTIFIER {
return Err(IdentityError::InvalidLocalInfoType.into());
}
if let Ok(info) = IdentitySecureChannelLocalInfo::decode(value.data()) {
return Ok(info);
}
Err(IdentityError::InvalidLocalInfoType.into())
}
pub fn to_local_info(&self) -> Result<LocalInfo> {
Ok(LocalInfo::new(
IDENTITY_SECURE_CHANNEL_IDENTIFIER.into(),
self.encode()?,
))
}
pub fn find_info(local_msg: &LocalMessage) -> Result<Self> {
if let Some(local_info) = local_msg
.local_info()
.iter()
.find(|x| x.type_identifier() == IDENTITY_SECURE_CHANNEL_IDENTIFIER)
{
Self::from_local_info(local_info)
} else {
Err(IdentityError::InvalidLocalInfoType.into())
}
}
}
impl IdentitySecureChannelLocalInfo {
pub fn their_identity_id(&self) -> &IdentityIdentifier {
&self.their_identity_id
}
}
impl IdentitySecureChannelLocalInfo {
pub fn mark(
mut local_info: Vec<LocalInfo>,
their_identity_id: IdentityIdentifier,
) -> Result<Vec<LocalInfo>> {
local_info.retain(|x| x.type_identifier() != IDENTITY_SECURE_CHANNEL_IDENTIFIER);
local_info.push(Self { their_identity_id }.to_local_info()?);
Ok(local_info)
}
}