use crate::structs::PrivateMessageView; use diesel::{pg::Pg, result::Error, *}; use lemmy_db_schema::{ newtypes::{PersonId, PrivateMessageId}, schema::{person, private_message}, source::{ person::{Person, PersonSafe}, private_message::PrivateMessage, }, traits::{ToSafe, ViewToVec}, utils::limit_and_offset, }; use tracing::debug; use typed_builder::TypedBuilder; type PrivateMessageViewTuple = (PrivateMessage, PersonSafe, PersonSafe); impl PrivateMessageView { pub fn read( conn: &mut PgConnection, private_message_id: PrivateMessageId, ) -> Result { let person_alias_1 = diesel::alias!(person as person1); let (private_message, creator, recipient) = private_message::table .find(private_message_id) .inner_join(person::table.on(private_message::creator_id.eq(person::id))) .inner_join( person_alias_1.on(private_message::recipient_id.eq(person_alias_1.field(person::id))), ) .order_by(private_message::published.desc()) .select(( private_message::all_columns, Person::safe_columns_tuple(), person_alias_1.fields(Person::safe_columns_tuple()), )) .first::(conn)?; Ok(PrivateMessageView { private_message, creator, recipient, }) } /// Gets the number of unread messages pub fn get_unread_messages( conn: &mut PgConnection, my_person_id: PersonId, ) -> Result { use diesel::dsl::*; private_message::table .filter(private_message::read.eq(false)) .filter(private_message::recipient_id.eq(my_person_id)) .filter(private_message::deleted.eq(false)) .select(count(private_message::id)) .first::(conn) } } #[derive(TypedBuilder)] #[builder(field_defaults(default))] pub struct PrivateMessageQuery<'a> { #[builder(!default)] conn: &'a mut PgConnection, #[builder(!default)] recipient_id: PersonId, unread_only: Option, page: Option, limit: Option, } impl<'a> PrivateMessageQuery<'a> { pub fn list(self) -> Result, Error> { let person_alias_1 = diesel::alias!(person as person1); let mut query = private_message::table .inner_join(person::table.on(private_message::creator_id.eq(person::id))) .inner_join( person_alias_1.on(private_message::recipient_id.eq(person_alias_1.field(person::id))), ) .select(( private_message::all_columns, Person::safe_columns_tuple(), person_alias_1.fields(Person::safe_columns_tuple()), )) .into_boxed(); // If its unread, I only want the ones to me if self.unread_only.unwrap_or(false) { query = query .filter(private_message::read.eq(false)) .filter(private_message::recipient_id.eq(self.recipient_id)); } // Otherwise, I want the ALL view to show both sent and received else { query = query.filter( private_message::recipient_id .eq(self.recipient_id) .or(private_message::creator_id.eq(self.recipient_id)), ) } let (limit, offset) = limit_and_offset(self.page, self.limit)?; query = query .filter(private_message::deleted.eq(false)) .limit(limit) .offset(offset) .order_by(private_message::published.desc()); debug!( "Private Message View Query: {:?}", debug_query::(&query) ); let res = query.load::(self.conn)?; Ok(PrivateMessageView::from_tuple_to_vec(res)) } } impl ViewToVec for PrivateMessageView { type DbTuple = PrivateMessageViewTuple; fn from_tuple_to_vec(items: Vec) -> Vec { items .into_iter() .map(|a| Self { private_message: a.0, creator: a.1, recipient: a.2, }) .collect::>() } }