aboutsummaryrefslogtreecommitdiffstats
path: root/src/models/user.rs
blob: 82a359b3481bd0500b0a216efa3552f2dd266a43 (plain) (blame)
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
use diesel;
use diesel::prelude::*;
use schema::users;
use utils;

#[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: utils::DbConn) -> QueryResult<Vec<User>> {
        use schema::users::dsl::*;
        users.load::<User>(&*conn)
    }

    pub fn by_id(user_id: i32, conn: &utils::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: utils::DbConn) -> QueryResult<User> {
        use schema::users::dsl::*;
        users.filter(email.eq(user_email))
            .get_result::<User>(&*conn)
    }

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

    pub fn update(user: &User, conn: &utils::DbConn) -> QueryResult<User> {
        use ::schema::users::dsl::*;
        user.save_changes(&*conn)
    }

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

    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()
    }
}