First working version

src/db/models/device.rs

@@ -0,0 +1,117 @@
+use chrono::{NaiveDate, NaiveDateTime, Utc};
+use time::Duration;
+use serde_json::Value as JsonValue;
+
+use uuid::Uuid;
+
+#[derive(Queryable, Insertable, Identifiable)]
+#[table_name = "devices"]
+#[primary_key(uuid)]
+pub struct Device {
+    pub uuid: String,
+    pub created_at: NaiveDateTime,
+    pub updated_at: NaiveDateTime,
+
+    pub user_uuid: String,
+
+    pub name: String,
+    /// https://github.com/bitwarden/core/tree/master/src/Core/Enums
+    pub type_: i32,
+    pub push_token: Option<String>,
+
+    pub refresh_token: String,
+}
+
+/// Local methods
+impl Device {
+    pub fn new(uuid: String, user_uuid: String, name: String, type_: i32) -> Device {
+        let now = Utc::now().naive_utc();
+
+        Device {
+            uuid,
+            created_at: now,
+            updated_at: now,
+
+            user_uuid,
+            name,
+            type_,
+
+            push_token: None,
+            refresh_token: String::new(),
+        }
+    }
+
+    pub fn refresh_tokens(&mut self, user: &super::User) -> (String, i64) {
+        // If there is no refresh token, we create one
+        if self.refresh_token.is_empty() {
+            use data_encoding::BASE64URL;
+            use crypto;
+
+            self.refresh_token = BASE64URL.encode(&crypto::get_random_64());
+        }
+
+        // Update the expiration of the device and the last update date
+        let time_now = Utc::now().naive_utc();
+
+        self.updated_at = time_now;
+
+        // Create the JWT claims struct, to send to the client
+        use auth::{encode_jwt, JWTClaims, DEFAULT_VALIDITY, JWT_ISSUER};
+        let claims = JWTClaims {
+            nbf: time_now.timestamp(),
+            exp: (time_now + *DEFAULT_VALIDITY).timestamp(),
+            iss: JWT_ISSUER.to_string(),
+            sub: user.uuid.to_string(),
+            premium: true,
+            name: user.name.to_string(),
+            email: user.email.to_string(),
+            email_verified: true,
+            sstamp: user.security_stamp.to_string(),
+            device: self.uuid.to_string(),
+            scope: vec!["api".into(), "offline_access".into()],
+            amr: vec!["Application".into()],
+        };
+
+        (encode_jwt(&claims), DEFAULT_VALIDITY.num_seconds())
+    }
+}
+
+use diesel;
+use diesel::prelude::*;
+use db::DbConn;
+use db::schema::devices;
+
+/// Database methods
+impl Device {
+    pub fn save(&self, conn: &DbConn) -> bool {
+        // TODO: Update modified date
+
+        match diesel::replace_into(devices::table)
+            .values(self)
+            .execute(&**conn) {
+            Ok(1) => true, // One row inserted
+            _ => false,
+        }
+    }
+
+    pub fn delete(self, conn: &DbConn) -> bool {
+        match diesel::delete(devices::table.filter(
+            devices::uuid.eq(self.uuid)))
+            .execute(&**conn) {
+            Ok(1) => true, // One row deleted
+            _ => false,
+        }
+    }
+
+    pub fn find_by_uuid(uuid: &str, conn: &DbConn) -> Option<Device> {
+        devices::table
+            .filter(devices::uuid.eq(uuid))
+            .first::<Device>(&**conn).ok()
+    }
+
+    pub fn find_by_refresh_token(refresh_token: &str, conn: &DbConn) -> Option<Device> {
+        devices::table
+            .filter(devices::refresh_token.eq(refresh_token))
+            .first::<Device>(&**conn).ok()
+    }
+}