Upload and download attachments, and added License file

src/db/models/attachment.rs

@@ -0,0 +1,93 @@
+use serde_json::Value as JsonValue;
+
+use super::Cipher;
+use CONFIG;
+
+#[derive(Debug, Identifiable, Queryable, Insertable, Associations)]
+#[table_name = "attachments"]
+#[belongs_to(Cipher, foreign_key = "cipher_uuid")]
+#[primary_key(id)]
+pub struct Attachment {
+    pub id: String,
+    pub cipher_uuid: String,
+    pub file_name: String,
+    pub file_size: i32,
+}
+
+/// Local methods
+impl Attachment {
+    pub fn new(id: String, cipher_uuid: String, file_name: String, file_size: i32) -> Self {
+        Self {
+            id,
+            cipher_uuid,
+            file_name,
+            file_size,
+        }
+    }
+
+    pub fn get_file_path(&self) -> String {
+        format!("{}/{}/{}", CONFIG.attachments_folder, self.cipher_uuid, self.id)
+    }
+
+    pub fn to_json(&self) -> JsonValue {
+        use util::get_display_size;
+        use CONFIG;
+
+        // TODO: Change all references to localhost (maybe put it in .env?)
+        let host = "http://localhost:8000";
+
+        let web_path = format!("{}/attachments/{}/{}", host, self.cipher_uuid, self.id);
+
+        let file_path = self.get_file_path();
+        let display_size = get_display_size(self.file_size);
+
+        json!({
+            "Id": self.id,
+            "Url": web_path,
+            "FileName": self.file_name,
+            "Size": self.file_size.to_string(),
+            "SizeName": display_size,
+            "Object": "attachment"
+        })
+    }
+}
+
+use diesel;
+use diesel::prelude::*;
+use db::DbConn;
+use db::schema::attachments;
+
+/// Database methods
+impl Attachment {
+    pub fn save(&self, conn: &DbConn) -> bool {
+        // TODO: Update modified date
+
+        match diesel::replace_into(attachments::table)
+            .values(self)
+            .execute(&**conn) {
+            Ok(1) => true, // One row inserted
+            _ => false,
+        }
+    }
+
+    pub fn delete(self, conn: &DbConn) -> bool {
+        match diesel::delete(attachments::table.filter(
+            attachments::id.eq(self.id)))
+            .execute(&**conn) {
+            Ok(1) => true, // One row deleted
+            _ => false,
+        }
+    }
+
+    pub fn find_by_id(id: &str, conn: &DbConn) -> Option<Self> {
+        attachments::table
+            .filter(attachments::id.eq(id))
+            .first::<Self>(&**conn).ok()
+    }
+
+    pub fn find_by_cipher(cipher_uuid: &str, conn: &DbConn) -> Vec<Self> {
+        attachments::table
+            .filter(attachments::cipher_uuid.eq(cipher_uuid))
+            .load::<Self>(&**conn).expect("Error loading attachments")
+    }
+}