Refactor notifications

plume-models/src/notifications.rs

@@ -1,28 +1,39 @@
 use chrono::NaiveDateTime;
 use diesel::{self, PgConnection, RunQueryDsl, QueryDsl, ExpressionMethods};
+use serde_json;
 
+use comments::Comment;
+use follows::Follow;
+use likes::Like;
+use mentions::Mention;
+use posts::Post;
+use reshares::Reshare;
 use users::User;
 use schema::notifications;
 
+pub mod notification_kind {
+    pub const COMMENT: &'static str = "COMMENT";
+    pub const FOLLOW: &'static str = "FOLLOW";
+    pub const LIKE: &'static str = "LIKE";
+    pub const MENTION: &'static str = "MENTION";
+    pub const RESHARE: &'static str = "RESHARE";
+}
+
 #[derive(Queryable, Identifiable, Serialize)]
 pub struct Notification {
     pub id: i32,
-    pub title: String,
-    pub content: Option<String>,
-    pub link: Option<String>,
     pub user_id: i32,
     pub creation_date: NaiveDateTime,
-    pub data: Option<String>
+    pub kind: String,
+    pub object_id: i32
 }
 
 #[derive(Insertable)]
 #[table_name = "notifications"]
 pub struct NewNotification {
-    pub title: String,
-    pub content: Option<String>,
-    pub link: Option<String>,
     pub user_id: i32,
-    pub data: Option<String>
+    pub kind: String,
+    pub object_id: i32
 }
 
 impl Notification {
@@ -44,4 +55,47 @@ impl Notification {
             .load::<Notification>(conn)
             .expect("Couldn't load user notifications page")
     }
+
+    pub fn to_json(&self, conn: &PgConnection) -> serde_json::Value {
+        let mut json = json!(self);
+        json["object"] = json!(match self.kind.as_ref() {
+            notification_kind::COMMENT => Comment::get(conn, self.object_id).map(|comment|
+                json!({
+                    "post": comment.get_post(conn).to_json(conn),
+                    "user": comment.get_author(conn).to_json(conn),
+                    "id": comment.id
+                })
+            ),
+            notification_kind::FOLLOW => Follow::get(conn, self.object_id).map(|follow|
+                json!({
+                    "follower": User::get(conn, follow.follower_id).map(|u| u.to_json(conn))
+                })
+            ),
+            notification_kind::LIKE => Like::get(conn, self.object_id).map(|like|
+                json!({
+                    "post": Post::get(conn, like.post_id).map(|p| p.to_json(conn)),
+                    "user": User::get(conn, like.user_id).map(|u| u.to_json(conn))
+                })
+            ),
+            notification_kind::MENTION => Mention::get(conn, self.object_id).map(|mention|
+                json!({
+                    "user": mention.get_user(conn).map(|u| u.to_json(conn)),
+                    "url": mention.get_post(conn).map(|p| p.to_json(conn)["url"].clone())
+                        .unwrap_or_else(|| {
+                            let comment = mention.get_comment(conn).expect("No comment nor post for mention");
+                            let post = comment.get_post(conn).to_json(conn);
+                            json!(format!("{}#comment-{}", post["url"].as_str().unwrap(), comment.id))
+                        })
+                })
+            ),
+            notification_kind::RESHARE => Reshare::get(conn, self.object_id).map(|reshare|
+                json!({
+                    "post": reshare.get_post(conn).map(|p| p.to_json(conn)),
+                    "user": reshare.get_user(conn).map(|u| u.to_json(conn))
+                })
+            ),
+            _ => Some(json!({}))
+        });
+        json
+    }
 }