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

use std::convert::TryFrom;

use serde_derive::{Deserialize, Serialize};
use sha2::Digest;
use thiserror::Error;

use super::default_none;
use crate::db::models as db_models;

#[derive(Debug, Error)]
pub enum UserError {
    #[error("error parsing date: {0}")]
    Chrono(#[from] chrono::ParseError),
    #[error("error parsing uuid: {0}")]
    Uuid(#[from] uuid::Error),
    #[error("error decoding hex data: {0}")]
    Hex(#[from] hex::FromHexError),
    #[error("error decoding UTF-8 data: {0}")]
    Utf8(#[from] std::string::FromUtf8Error),
}

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(rename_all = "camelCase", deny_unknown_fields)]
pub struct User {
    pub name: String,
    #[serde(
        serialize_with = "hex::serialize",
        deserialize_with = "hex::deserialize"
    )]
    pub password: Vec<u8>,
    #[serde(
        serialize_with = "hex::serialize",
        deserialize_with = "hex::deserialize"
    )]
    pub token: Vec<u8>,
    pub salt: String,
    #[serde(default = "default_none")]
    pub comment: Option<String>,
    #[serde(default = "default_none")]
    pub id: Option<String>,
    #[serde(default = "chrono::Utc::now")]
    pub created_at: chrono::DateTime<chrono::Utc>,
    #[serde(default = "chrono::Utc::now")]
    pub last_use: chrono::DateTime<chrono::Utc>,
}

impl User {
    pub fn hyperion() -> Self {
        let name = "Hyperion".to_owned();
        let salt = Self::generate_salt();
        let token = Self::generate_token();
        let password = Self::hash_password("hyperion", salt.as_bytes());
        let created_at = chrono::Utc::now();
        let last_use = created_at;

        Self {
            name,
            password,
            token,
            salt,
            comment: None,
            id: None,
            created_at,
            last_use,
        }
    }

    pub fn generate_token() -> Vec<u8> {
        let mut hasher = sha2::Sha512::default();
        hasher.update(uuid::Uuid::new_v4().as_bytes());
        hasher.finalize().to_vec()
    }

    pub fn generate_salt() -> String {
        hex::encode(Self::generate_token())
    }

    pub fn hash_password(password: &str, salt: &[u8]) -> Vec<u8> {
        let mut hasher = sha2::Sha512::default();
        hasher.update(password.as_bytes());
        hasher.update(salt);
        hasher.finalize().to_vec()
    }
}

impl TryFrom<db_models::DbUser> for User {
    type Error = UserError;

    fn try_from(db: db_models::DbUser) -> Result<Self, Self::Error> {
        Ok(Self {
            name: db.user,
            password: hex::decode(db.password)?,
            token: hex::decode(db.token)?,
            salt: String::from_utf8(db.salt)?,
            comment: db.comment,
            id: db.id,
            created_at: chrono::DateTime::parse_from_rfc3339(&db.created_at)?
                .with_timezone(&chrono::Utc),
            last_use: chrono::DateTime::parse_from_rfc3339(&db.last_use)?
                .with_timezone(&chrono::Utc),
        })
    }
}