path: root/src/models/user.rs

use diesel;
use diesel::prelude::*;
use schema::users;
use utils::DbConn;

#[derive(AsChangeset, Clone, FromForm, Identifiable, Serialize, Queryable)]
pub struct User {
    pub id: i32,
    pub username: String,
    pub realname: Option<String>,
    pub email: Option<String>,
    pub password: String,
}

impl User {
    pub fn all(conn: &DbConn) -> QueryResult<Vec<User>> {
        use schema::users::dsl::*;
        users.load::<User>(&**conn)
    }

    pub fn by_id(user_id: i32, conn: &DbConn) -> QueryResult<User> {
        use schema::users::dsl::*;
        users
            .filter(id.eq(user_id))
            .load::<User>(&**conn)
            .map(|ref v| v[0].clone())
    }

    pub fn by_email(user_email: &str, conn: &DbConn) -> QueryResult<User> {
        use schema::users::dsl::*;
        users
            .filter(email.eq(user_email))
            .get_result::<User>(&**conn)
    }

    pub fn create(new_user: &NewUser, conn: &DbConn) -> QueryResult<User> {
        use schema::users::dsl::*;
        diesel::insert_into(users)
            .values(new_user)
            .get_result(&**conn)
    }

    pub fn update(user: &User, conn: &DbConn) -> QueryResult<User> {
        user.save_changes(&*conn)
    }

    pub fn delete(user_id: i32, conn: &DbConn) -> QueryResult<User> {
        use schema::users::dsl::*;
        diesel::delete(users.filter(id.eq(user_id))).get_result::<User>(&**conn)
    }

    // Display name of user is real name (if it exists)
    // or username if it does not.
    pub fn displayname(&self) -> &str {
        match self.realname {
            Some(ref n) => &n,
            None => &self.username,
        }
    }

    pub fn realname(&self) -> String {
        self.realname.as_ref().unwrap_or(&String::new()).clone()
    }

    pub fn email(&self) -> String {
        self.email.as_ref().unwrap_or(&String::new()).clone()
    }
}

#[derive(Default, Insertable)]
#[table_name = "users"]
pub struct NewUser {
    pub username: String,
    pub realname: Option<String>,
    pub email: Option<String>,
    pub password: String,
}

impl NewUser {
    pub fn realname(&self) -> String {
        self.realname.as_ref().unwrap_or(&String::new()).clone()
    }

    pub fn email(&self) -> String {
        self.email.as_ref().unwrap_or(&String::new()).clone()
    }
}