crates/db_views_actor/src/community_follower_view.rs

   1 use diesel::{result::Error, *};
   2 use lemmy_db_queries::{ToSafe, ViewToVec};
   3 use lemmy_db_schema::{
   4   schema::{community, community_follower, user_},
   5   source::{
   6     community::{Community, CommunitySafe},
   7     user::{UserSafe, User_},
   8   },
   9 };
  10 use serde::Serialize;
  11
  12 #[derive(Debug, Serialize, Clone)]
  13 pub struct CommunityFollowerView {
  14   pub community: CommunitySafe,
  15   pub follower: UserSafe,
  16 }
  17
  18 type CommunityFollowerViewTuple = (CommunitySafe, UserSafe);
  19
  20 impl CommunityFollowerView {
  21   pub fn for_community(conn: &PgConnection, community_id: i32) -> Result<Vec<Self>, Error> {
  22     let res = community_follower::table
  23       .inner_join(community::table)
  24       .inner_join(user_::table)
  25       .select((Community::safe_columns_tuple(), User_::safe_columns_tuple()))
  26       .filter(community_follower::community_id.eq(community_id))
  27       .order_by(community_follower::published)
  28       .load::<CommunityFollowerViewTuple>(conn)?;
  29
  30     Ok(Self::from_tuple_to_vec(res))
  31   }
  32
  33   pub fn for_user(conn: &PgConnection, user_id: i32) -> Result<Vec<Self>, Error> {
  34     let res = community_follower::table
  35       .inner_join(community::table)
  36       .inner_join(user_::table)
  37       .select((Community::safe_columns_tuple(), User_::safe_columns_tuple()))
  38       .filter(community_follower::user_id.eq(user_id))
  39       .order_by(community_follower::published)
  40       .load::<CommunityFollowerViewTuple>(conn)?;
  41
  42     Ok(Self::from_tuple_to_vec(res))
  43   }
  44 }
  45
  46 impl ViewToVec for CommunityFollowerView {
  47   type DbTuple = CommunityFollowerViewTuple;
  48   fn from_tuple_to_vec(items: Vec<Self::DbTuple>) -> Vec<Self> {
  49     items
  50       .iter()
  51       .map(|a| Self {
  52         community: a.0.to_owned(),
  53         follower: a.1.to_owned(),
  54       })
  55       .collect::<Vec<Self>>()
  56   }
  57 }