use diesel::prelude::*; use crate::models; use crate::{errors, models::Password}; use bcrypt::{hash, verify, DEFAULT_COST}; use validators::prelude::*; pub fn find_user_by_uid( uid: i32, conn: &impl diesel::Connection, ) -> Result, errors::DomainError> { use crate::schema::users::dsl::*; let maybe_user = users .select(users::all_columns()) .find(uid) .first::(conn) .optional(); Ok(maybe_user?) } pub fn _find_user_by_name( user_name: String, conn: &impl diesel::Connection, ) -> Result, errors::DomainError> { use crate::schema::users::dsl::*; let maybe_user = query::_get_user_by_name() .filter(name.eq(user_name)) .first::(conn) .optional(); Ok(maybe_user?) } pub fn get_all( conn: &impl diesel::Connection, ) -> Result, errors::DomainError> { use crate::schema::users::dsl::*; Ok(users .select(users::all_columns()) .load::(conn)?) } pub fn insert_new_user( nu: models::NewUser, conn: &impl diesel::Connection, hash_cost: Option, ) -> Result { use crate::schema::users::dsl::*; let nu = { let mut nu2 = nu; let hash = hash(&nu2.password.as_str(), hash_cost.unwrap_or(DEFAULT_COST))?; nu2.password = Password::parse_string(hash).map_err(|err| { errors::DomainError::new_field_validation_error(err.to_string()) })?; nu2 }; diesel::insert_into(users).values(&nu).execute(conn)?; let user = query::_get_user_by_name() .filter(name.eq(nu.name.as_str())) .first::(conn)?; Ok(user) } //TODO: Add newtype here pub fn verify_password( user_name: &str, given_password: &str, conn: &impl diesel::Connection, ) -> Result { use crate::schema::users::dsl::*; let password_hash = users .select(password) .filter(name.eq(user_name)) .first::(conn)?; Ok(verify(given_password, password_hash.as_str())?) } mod query { use diesel::prelude::*; use diesel::sql_types::Integer; use diesel::sql_types::Text; use diesel::sql_types::Timestamp; use diesel::sqlite::Sqlite; /// <'a, B, T> where a = lifetime, B = Backend, T = SQL data types type Query<'a, B, T> = crate::schema::users::BoxedQuery<'a, B, T>; pub fn _get_user_by_name( ) -> Query<'static, Sqlite, (Integer, Text, Text, Timestamp)> { use crate::schema::users::dsl::*; users .select(users::all_columns()) // .filter(name.eq(user_name)) .into_boxed() } }