use chrono::NaiveDateTime; use diesel::{self, QueryDsl, RunQueryDsl, ExpressionMethods, PgConnection}; use activity_pub::activity::Activity; use activity_pub::actor::{Actor, ActorType}; use activity_pub::outbox::Outbox; use activity_pub::webfinger::*; use models::instance::Instance; use schema::blogs; #[derive(Queryable, Identifiable)] pub struct Blog { pub id: i32, pub actor_id: String, pub title: String, pub summary: String, pub outbox_url: String, pub inbox_url: String, pub instance_id: i32, pub creation_date: NaiveDateTime, pub ap_url: String } #[derive(Insertable)] #[table_name = "blogs"] pub struct NewBlog { pub actor_id: String, pub title: String, pub summary: String, pub outbox_url: String, pub inbox_url: String, pub instance_id: i32, pub ap_url: String } impl Blog { pub fn insert (conn: &PgConnection, new: NewBlog) -> Blog { diesel::insert_into(blogs::table) .values(new) .get_result(conn) .expect("Error saving new blog") } pub fn get(conn: &PgConnection, id: i32) -> Option { blogs::table.filter(blogs::id.eq(id)) .limit(1) .load::(conn) .expect("Error loading blog by id") .into_iter().nth(0) } pub fn find_by_actor_id(conn: &PgConnection, username: String) -> Option { blogs::table.filter(blogs::actor_id.eq(username)) .limit(1) .load::(conn) .expect("Error loading blog by actor_id") .into_iter().nth(0) } pub fn update_boxes(&self, conn: &PgConnection) { if self.outbox_url.len() == 0 { diesel::update(self) .set(blogs::outbox_url.eq(self.compute_outbox(conn))) .get_result::(conn).expect("Couldn't update outbox URL"); } if self.inbox_url.len() == 0 { diesel::update(self) .set(blogs::inbox_url.eq(self.compute_inbox(conn))) .get_result::(conn).expect("Couldn't update inbox URL"); } if self.ap_url.len() == 0 { diesel::update(self) .set(blogs::ap_url.eq(self.compute_id(conn))) .get_result::(conn).expect("Couldn't update AP URL"); } } pub fn outbox(&self, conn: &PgConnection) -> Outbox { Outbox::new(self.compute_outbox(conn), self.get_activities(conn)) } fn get_activities(&self, _conn: &PgConnection) -> Vec> { vec![] } } impl Actor for Blog { fn get_box_prefix() -> &'static str { "~" } fn get_actor_id(&self) -> String { self.actor_id.to_string() } fn get_instance(&self, conn: &PgConnection) -> Instance { Instance::get(conn, self.instance_id).unwrap() } fn get_actor_type () -> ActorType { ActorType::Blog } fn from_url(conn: &PgConnection, url: String) -> Option { blogs::table.filter(blogs::ap_url.eq(url)) .limit(1) .load::(conn) .expect("Error loading blog from url") .into_iter().nth(0) } } impl Webfinger for Blog { fn webfinger_subject(&self, conn: &PgConnection) -> String { format!("acct:{}@{}", self.actor_id, self.get_instance(conn).public_domain) } fn webfinger_aliases(&self, conn: &PgConnection) -> Vec { vec![self.compute_id(conn)] } fn webfinger_links(&self, conn: &PgConnection) -> Vec> { vec![ vec![ (String::from("rel"), String::from("http://webfinger.net/rel/profile-page")), (String::from("href"), self.compute_id(conn)) ], vec![ (String::from("rel"), String::from("http://schemas.google.com/g/2010#updates-from")), (String::from("type"), String::from("application/atom+xml")), (String::from("href"), self.compute_box(conn, "feed.atom")) ], vec![ (String::from("rel"), String::from("self")), (String::from("type"), String::from("application/activity+json")), (String::from("href"), self.compute_id(conn)) ] ] } } impl NewBlog { pub fn new_local( actor_id: String, title: String, summary: String, instance_id: i32 ) -> NewBlog { NewBlog { actor_id: actor_id, title: title, summary: summary, outbox_url: String::from(""), inbox_url: String::from(""), instance_id: instance_id, ap_url: String::from("") } } }