lemmy/crates/db_views_actor/src/community_view.rs

239 lines
7.2 KiB
Rust
Raw Normal View History

use crate::structs::{CommunityModeratorView, CommunityView, PersonView};
2022-11-09 10:05:00 +00:00
use diesel::{
result::Error,
BoolExpressionMethods,
ExpressionMethods,
JoinOnDsl,
NullableExpressionMethods,
PgTextExpressionMethods,
QueryDsl,
};
use diesel_async::RunQueryDsl;
2021-10-16 13:33:38 +00:00
use lemmy_db_schema::{
aggregates::structs::CommunityAggregates,
2021-10-16 13:33:38 +00:00
newtypes::{CommunityId, PersonId},
schema::{community, community_aggregates, community_block, community_follower, local_user},
source::{
community::{Community, CommunityFollower},
community_block::CommunityBlock,
local_user::LocalUser,
},
traits::JoinView,
utils::{fuzzy_search, get_conn, limit_and_offset, DbPool},
ListingType,
SortType,
};
use typed_builder::TypedBuilder;
2020-12-04 16:29:44 +00:00
2020-12-11 01:39:42 +00:00
type CommunityViewTuple = (
Community,
2020-12-11 01:39:42 +00:00
CommunityAggregates,
Option<CommunityFollower>,
Option<CommunityBlock>,
2020-12-11 01:39:42 +00:00
);
2020-12-04 16:29:44 +00:00
impl CommunityView {
2022-11-09 10:05:00 +00:00
pub async fn read(
pool: &DbPool,
community_id: CommunityId,
my_person_id: Option<PersonId>,
is_mod_or_admin: Option<bool>,
2020-12-04 16:29:44 +00:00
) -> Result<Self, Error> {
2022-11-09 10:05:00 +00:00
let conn = &mut get_conn(pool).await?;
2020-12-06 03:49:15 +00:00
// The left join below will return None in this case
let person_id_join = my_person_id.unwrap_or(PersonId(-1));
2020-12-04 16:29:44 +00:00
let mut query = community::table
2020-12-04 16:29:44 +00:00
.find(community_id)
2020-12-04 21:35:46 +00:00
.inner_join(community_aggregates::table)
2020-12-06 03:49:15 +00:00
.left_join(
community_follower::table.on(
community::id
.eq(community_follower::community_id)
2021-03-10 22:33:55 +00:00
.and(community_follower::person_id.eq(person_id_join)),
2020-12-06 03:49:15 +00:00
),
)
.left_join(
community_block::table.on(
community::id
.eq(community_block::community_id)
.and(community_block::person_id.eq(person_id_join)),
),
)
2020-12-05 04:18:30 +00:00
.select((
community::all_columns,
2020-12-05 04:18:30 +00:00
community_aggregates::all_columns,
2020-12-06 03:49:15 +00:00
community_follower::all_columns.nullable(),
community_block::all_columns.nullable(),
2020-12-05 04:18:30 +00:00
))
.into_boxed();
// Hide deleted and removed for non-admins or mods
if !is_mod_or_admin.unwrap_or(true) {
query = query
.filter(community::removed.eq(false))
.filter(community::deleted.eq(false));
}
let (community, counts, follower, blocked) = query.first::<CommunityViewTuple>(conn).await?;
2020-12-04 16:29:44 +00:00
Ok(CommunityView {
community,
subscribed: CommunityFollower::to_subscribed_type(&follower),
blocked: blocked.is_some(),
2020-12-04 21:35:46 +00:00
counts,
2020-12-04 16:29:44 +00:00
})
}
2022-11-09 10:05:00 +00:00
pub async fn is_mod_or_admin(
pool: &DbPool,
person_id: PersonId,
community_id: CommunityId,
2022-11-09 10:05:00 +00:00
) -> Result<bool, Error> {
let is_mod = CommunityModeratorView::for_community(pool, community_id)
.await
.map(|v| {
v.into_iter()
.map(|m| m.moderator.id)
.collect::<Vec<PersonId>>()
})
.unwrap_or_default()
.contains(&person_id);
if is_mod {
2022-11-09 10:05:00 +00:00
return Ok(true);
}
let is_admin = PersonView::admins(pool)
2022-11-09 10:05:00 +00:00
.await
.map(|v| {
v.into_iter()
.map(|a| a.person.id)
.collect::<Vec<PersonId>>()
})
.unwrap_or_default()
2022-11-09 10:05:00 +00:00
.contains(&person_id);
Ok(is_admin)
}
2020-12-04 16:29:44 +00:00
}
2020-12-06 03:49:15 +00:00
#[derive(TypedBuilder)]
#[builder(field_defaults(default))]
pub struct CommunityQuery<'a> {
#[builder(!default)]
2022-11-09 10:05:00 +00:00
pool: &'a DbPool,
listing_type: Option<ListingType>,
sort: Option<SortType>,
local_user: Option<&'a LocalUser>,
2020-12-06 03:49:15 +00:00
search_term: Option<String>,
is_mod_or_admin: Option<bool>,
2020-12-06 03:49:15 +00:00
page: Option<i64>,
limit: Option<i64>,
}
impl<'a> CommunityQuery<'a> {
2022-11-09 10:05:00 +00:00
pub async fn list(self) -> Result<Vec<CommunityView>, Error> {
let conn = &mut get_conn(self.pool).await?;
// The left join below will return None in this case
let person_id_join = self.local_user.map(|l| l.person_id).unwrap_or(PersonId(-1));
let mut query = community::table
.inner_join(community_aggregates::table)
.left_join(local_user::table.on(local_user::person_id.eq(person_id_join)))
.left_join(
community_follower::table.on(
community::id
.eq(community_follower::community_id)
2021-03-10 22:33:55 +00:00
.and(community_follower::person_id.eq(person_id_join)),
),
)
.left_join(
community_block::table.on(
community::id
.eq(community_block::community_id)
.and(community_block::person_id.eq(person_id_join)),
),
)
.select((
community::all_columns,
community_aggregates::all_columns,
community_follower::all_columns.nullable(),
community_block::all_columns.nullable(),
))
.into_boxed();
2020-12-06 03:49:15 +00:00
if let Some(search_term) = self.search_term {
let searcher = fuzzy_search(&search_term);
query = query
.filter(community::name.ilike(searcher.clone()))
.or_filter(community::title.ilike(searcher));
2020-12-06 03:49:15 +00:00
};
// Hide deleted and removed for non-admins or mods
if !self.is_mod_or_admin.unwrap_or(true) {
query = query
.filter(community::removed.eq(false))
.filter(community::deleted.eq(false));
}
match self.sort.unwrap_or(SortType::Hot) {
2020-12-06 03:49:15 +00:00
SortType::New => query = query.order_by(community::published.desc()),
SortType::TopAll => query = query.order_by(community_aggregates::subscribers.desc()),
SortType::TopMonth => query = query.order_by(community_aggregates::users_active_month.desc()),
SortType::Hot => {
query = query.order_by(community_aggregates::hot_rank.desc());
// Don't show hidden communities in Hot (trending)
query = query.filter(
community::hidden
.eq(false)
.or(community_follower::person_id.eq(person_id_join)),
);
}
// Covers all other sorts
_ => query = query.order_by(community_aggregates::users_active_month.desc()),
2020-12-06 03:49:15 +00:00
};
if let Some(listing_type) = self.listing_type {
query = match listing_type {
ListingType::Subscribed => query.filter(community_follower::person_id.is_not_null()), // TODO could be this: and(community_follower::person_id.eq(person_id_join)),
ListingType::Local => query.filter(community::local.eq(true)),
_ => query,
};
}
// Don't show blocked communities or nsfw communities if not enabled in profile
if self.local_user.is_some() {
query = query.filter(community_block::person_id.is_null());
query = query.filter(community::nsfw.eq(false).or(local_user::show_nsfw.eq(true)));
} else {
// No person in request, only show nsfw communities if show_nsfw passed into request
if !self.local_user.map(|l| l.show_nsfw).unwrap_or(false) {
query = query.filter(community::nsfw.eq(false));
}
}
let (limit, offset) = limit_and_offset(self.page, self.limit)?;
2020-12-06 03:49:15 +00:00
let res = query
.limit(limit)
.offset(offset)
.filter(community::removed.eq(false))
.filter(community::deleted.eq(false))
2022-11-09 10:05:00 +00:00
.load::<CommunityViewTuple>(conn)
.await?;
2020-12-06 03:49:15 +00:00
Ok(res.into_iter().map(CommunityView::from_tuple).collect())
2020-12-06 03:49:15 +00:00
}
}
impl JoinView for CommunityView {
type JoinTuple = CommunityViewTuple;
fn from_tuple(a: Self::JoinTuple) -> Self {
Self {
community: a.0,
counts: a.1,
subscribed: CommunityFollower::to_subscribed_type(&a.2),
blocked: a.3.is_some(),
}
2020-12-11 01:39:42 +00:00
}
2020-12-06 03:49:15 +00:00
}