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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

use jsonwebtoken::{Algorithm, DecodingKey, EncodingKey, Header, Validation};
use serde::{Deserialize, Serialize};
use uuid::Uuid;

use crate::{error::MessageError, policy::Policy, publish::PublishHost};

#[derive(Debug, Serialize, Deserialize)]
pub struct ClientToken {
    pub uid: Uuid,
    pub name: String,
    pub exp: usize,
    #[serde(skip_serializing_if = "<[_]>::is_empty")]
    #[serde(default = "Vec::new")]
    pub policies: Vec<u32>,
}
//Todo: replace to from/into if possible//Todo: replace to from/into if possible
impl ClientToken {
    pub fn from_str(s: &str, token: &[u8]) -> Result<ClientToken, MessageError> {
        Ok(jsonwebtoken::decode::<ClientToken>(
            s,
            &DecodingKey::from_secret(token),
            &Validation::new(Algorithm::default()),
        )?
        .claims)
    }

    pub fn to_string(&self, token: &[u8]) -> Result<String, MessageError> {
        Ok(jsonwebtoken::encode(
            &Header::new(Algorithm::default()),
            self,
            &EncodingKey::from_secret(token),
        )?)
    }
}
#[derive(Debug, Serialize, Deserialize)]
pub struct PolicyToken {
    pub uid: Uuid,
    pub name: String,
    pub exp: usize,
    pub pid: u32,
    pub policy: Policy,
}

impl PolicyToken {
    pub fn from_str(s: &str, token: &[u8]) -> Result<Self, MessageError> {
        Ok(jsonwebtoken::decode::<PolicyToken>(
            s,
            &DecodingKey::from_secret(token),
            &Validation::new(Algorithm::default()),
        )?
        .claims)
    }

    pub fn to_string(&self, token: &[u8]) -> Result<String, MessageError> {
        Ok(jsonwebtoken::encode(
            &Header::new(Algorithm::default()),
            self,
            &EncodingKey::from_secret(token),
        )?)
    }
}

#[derive(Debug, Serialize, Deserialize)]
pub struct AgentToken {
    pub uid: Uuid,
    pub name: String,
    pub exp: usize,
}

impl AgentToken {
    pub fn from_str(s: &str, token: &[u8]) -> Result<AgentToken, MessageError> {
        Ok(jsonwebtoken::decode::<AgentToken>(
            s,
            &DecodingKey::from_secret(token),
            &Validation::new(Algorithm::default()),
        )?
        .claims)
    }

    pub fn to_string(&self, token: &[u8]) -> Result<String, MessageError> {
        Ok(jsonwebtoken::encode(
            &Header::new(Algorithm::default()),
            self,
            &EncodingKey::from_secret(token),
        )?)
    }
}

#[derive(Debug, Serialize, Deserialize)]
pub struct AgentPublishToken {
    pub uid: Uuid,
    pub name: String,
    pub exp: usize,
    pub publish_hosts: Vec<PublishHost>,
}

impl AgentPublishToken {
    pub fn from_str(s: &str, token: &[u8]) -> Result<AgentPublishToken, MessageError> {
        Ok(jsonwebtoken::decode::<AgentPublishToken>(
            s,
            &DecodingKey::from_secret(token),
            &Validation::new(Algorithm::default()),
        )?
        .claims)
    }

    pub fn to_string(&self, token: &[u8]) -> Result<String, MessageError> {
        Ok(jsonwebtoken::encode(
            &Header::new(Algorithm::default()),
            self,
            &EncodingKey::from_secret(token),
        )?)
    }
}