use super::users::User; use crate::schema::{sessions, users}; use base64; use diesel::PgConnection; use diesel::{insert_into, prelude::*, Insertable, RunQueryDsl}; use rand::{self, Rng}; #[derive(Insertable)] #[table_name = "sessions"] struct NewSession { token: String, user_id: i32, } #[derive(Queryable, Debug, PartialEq)] pub struct Session { pub id: i32, pub user_id: i32, pub token: String, } pub fn create_session(user: &User, conn: &PgConnection) -> Session { let new_session = NewSession { token: gen_session_token(), user_id: user.id, }; let session = insert_into(sessions::table) .values(&new_session) .get_result::(conn) .unwrap(); return session; } pub fn find_user_by_session(token: &str, conn: &PgConnection) -> QueryResult<(Session, User)> { sessions::table .inner_join(users::table) .filter(sessions::token.eq(&token)) .first::<(Session, User)>(conn) } pub fn gen_session_token() -> String { let mut rng = rand::thread_rng(); let token: [u8; 32] = rng.gen(); return base64::encode(&token); }