2022-04-23 18:44:02 +00:00
|
|
|
use rust_utilities::crypto::sha::{encode, Algorithm, CryptographicHash};
|
2022-04-17 22:07:37 +00:00
|
|
|
use uuid::Uuid;
|
|
|
|
|
2022-04-24 12:50:25 +00:00
|
|
|
/// SQL `user` Table
|
2022-04-21 18:53:30 +00:00
|
|
|
#[derive(Debug, sqlx::FromRow)]
|
2022-04-16 22:09:23 +00:00
|
|
|
pub struct User {
|
|
|
|
pub id: String,
|
|
|
|
pub username: String,
|
|
|
|
pub password: String,
|
|
|
|
}
|
2022-04-17 22:07:37 +00:00
|
|
|
|
|
|
|
impl User {
|
2022-04-23 10:52:56 +00:00
|
|
|
/// **Note this doesn't create a new user in the database!**
|
2022-04-17 22:07:37 +00:00
|
|
|
///
|
2022-04-23 10:52:56 +00:00
|
|
|
/// This function creates a unique UUID for a user and creates a password hash using SHA-512
|
|
|
|
/// and returns in the User type
|
2022-04-17 22:07:37 +00:00
|
|
|
/// ```
|
2022-04-24 12:50:25 +00:00
|
|
|
/// use homedisk_types::database::User;
|
2022-04-17 22:07:37 +00:00
|
|
|
///
|
2022-04-19 11:08:30 +00:00
|
|
|
/// let user = User::new("medzik", "SuperSecretPassword123!");
|
2022-04-17 22:07:37 +00:00
|
|
|
/// ```
|
2022-04-19 11:05:47 +00:00
|
|
|
pub fn new(username: &str, password: &str) -> Self {
|
2022-04-23 10:52:56 +00:00
|
|
|
// change username to lowercase
|
|
|
|
let username = username.to_lowercase();
|
|
|
|
|
2022-04-17 22:07:37 +00:00
|
|
|
// create user UUID
|
2022-04-23 10:52:56 +00:00
|
|
|
let sha1_name = CryptographicHash::hash(Algorithm::SHA1, username.as_bytes());
|
2022-04-17 22:07:37 +00:00
|
|
|
let id = Uuid::new_v5(&Uuid::NAMESPACE_X500, &sha1_name).to_string();
|
|
|
|
|
2022-04-23 18:44:02 +00:00
|
|
|
// hash password using SHA-512
|
|
|
|
let password = encode(CryptographicHash::hash(
|
|
|
|
Algorithm::SHA512,
|
|
|
|
password.as_bytes(),
|
|
|
|
));
|
2022-04-17 22:07:37 +00:00
|
|
|
|
2022-04-19 11:05:47 +00:00
|
|
|
Self {
|
2022-04-17 22:07:37 +00:00
|
|
|
id,
|
2022-04-21 18:53:30 +00:00
|
|
|
username,
|
2022-04-17 22:07:37 +00:00
|
|
|
password,
|
2022-04-19 11:05:47 +00:00
|
|
|
}
|
2022-04-17 22:07:37 +00:00
|
|
|
}
|
2022-05-01 18:34:28 +00:00
|
|
|
|
|
|
|
/// User directory
|
|
|
|
/// function returns the directory where the user file is located
|
|
|
|
/// e.g.
|
|
|
|
/// ```
|
|
|
|
/// use homedisk_types::database::User;
|
|
|
|
///
|
|
|
|
/// let user = User::new("medzik", "whatever");
|
|
|
|
///
|
2022-05-01 20:44:28 +00:00
|
|
|
/// let dir = user.user_dir("/home/homedisk"); // will return `/home/homedisk/medzik`
|
|
|
|
///
|
|
|
|
/// assert_eq!(dir, "/home/homedisk/medzik")
|
2022-05-01 18:34:28 +00:00
|
|
|
/// ```
|
|
|
|
pub fn user_dir(&self, storage: &str) -> String {
|
|
|
|
let path = format!(
|
|
|
|
"{path}/{username}",
|
|
|
|
path = storage,
|
|
|
|
username = self.username,
|
|
|
|
);
|
|
|
|
|
|
|
|
path
|
|
|
|
}
|
2022-04-17 22:07:37 +00:00
|
|
|
}
|
2022-04-21 18:53:30 +00:00
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use super::User;
|
|
|
|
|
2022-06-07 20:36:26 +00:00
|
|
|
/// Check if the username has been changed to lowercase
|
2022-04-21 18:53:30 +00:00
|
|
|
#[test]
|
|
|
|
fn check_username_is_in_lowercase() {
|
|
|
|
let user = User::new("MEdzIk", "SuperSecretPassword123!");
|
|
|
|
|
|
|
|
assert_eq!(user.username, "medzik")
|
|
|
|
}
|
|
|
|
|
2022-06-07 20:36:26 +00:00
|
|
|
/// Check that the password is a checksum
|
2022-04-21 18:53:30 +00:00
|
|
|
#[test]
|
2022-04-23 18:44:02 +00:00
|
|
|
fn check_if_password_is_hashed() {
|
2022-04-23 10:52:56 +00:00
|
|
|
let password = "password";
|
2022-04-21 18:53:30 +00:00
|
|
|
let user = User::new("test", password);
|
|
|
|
|
|
|
|
assert!(user.password != password)
|
|
|
|
}
|
|
|
|
}
|