use activitystreams_types::activity; use chrono::NaiveDateTime; use diesel::{self, PgConnection, QueryDsl, RunQueryDsl, ExpressionMethods}; use activity_pub::{IntoId, actor::Actor, object::Object}; use models::{posts::Post, users::User}; use schema::reshares; #[derive(Serialize, Deserialize, Queryable, Identifiable)] pub struct Reshare { pub id: i32, pub user_id: i32, pub post_id: i32, pub ap_url: String, pub creation_date: NaiveDateTime } #[derive(Insertable)] #[table_name = "reshares"] pub struct NewReshare { pub user_id: i32, pub post_id: i32, pub ap_url: String } impl Reshare { pub fn insert(conn: &PgConnection, new: NewReshare) -> Reshare { diesel::insert_into(reshares::table) .values(new) .get_result(conn) .expect("Couldn't save reshare") } pub fn get(conn: &PgConnection, id: i32) -> Option { reshares::table.filter(reshares::id.eq(id)) .limit(1) .load::(conn) .expect("Could'nt load reshare") .into_iter().nth(0) } pub fn update_ap_url(&self, conn: &PgConnection) { if self.ap_url.len() == 0 { diesel::update(self) .set(reshares::ap_url.eq(format!( "{}/reshare/{}", User::get(conn, self.user_id).unwrap().compute_id(conn), Post::get(conn, self.post_id).unwrap().compute_id(conn) ))) .get_result::(conn).expect("Couldn't update AP URL"); } } pub fn find_by_ap_url(conn: &PgConnection, ap_url: String) -> Option { reshares::table.filter(reshares::ap_url.eq(ap_url)) .limit(1) .load::(conn) .expect("Error loading reshare by AP URL") .into_iter().nth(0) } pub fn find_by_user_on_post(conn: &PgConnection, user: &User, post: &Post) -> Option { reshares::table.filter(reshares::post_id.eq(post.id)) .filter(reshares::user_id.eq(user.id)) .limit(1) .load::(conn) .expect("Error loading reshare for user and post") .into_iter().nth(0) } pub fn get_recents_for_author(conn: &PgConnection, user: &User, limit: i64) -> Vec { reshares::table.filter(reshares::user_id.eq(user.id)) .order(reshares::creation_date.desc()) .limit(limit) .load::(conn) .expect("Error loading recent reshares for user") } pub fn get_post(&self, conn: &PgConnection) -> Option { Post::get(conn, self.post_id) } pub fn delete(&self, conn: &PgConnection) -> activity::Undo { diesel::delete(self).execute(conn).unwrap(); let mut act = activity::Undo::default(); act.set_actor_link(User::get(conn, self.user_id).unwrap().into_id()).unwrap(); act.set_object_object(self.into_activity(conn)).unwrap(); act } pub fn into_activity(&self, conn: &PgConnection) -> activity::Announce { let mut act = activity::Announce::default(); act.set_actor_link(User::get(conn, self.user_id).unwrap().into_id()).unwrap(); act.set_object_link(Post::get(conn, self.post_id).unwrap().into_id()).unwrap(); act.object_props.set_id_string(self.ap_url.clone()).unwrap(); act } }