use crate::{
newtypes::{CommunityId, DbUrl, PersonId},
- schema::community::dsl::*,
+ schema::{community, community_follower, instance},
source::{
- actor_language::{CommunityLanguage, SiteLanguage},
+ actor_language::CommunityLanguage,
community::{
Community,
CommunityFollower,
CommunityModeratorForm,
CommunityPersonBan,
CommunityPersonBanForm,
- CommunitySafe,
CommunityUpdateForm,
},
},
- traits::{ApubActor, Bannable, Crud, DeleteableOrRemoveable, Followable, Joinable},
+ traits::{ApubActor, Bannable, Crud, Followable, Joinable},
utils::{functions::lower, get_conn, DbPool},
SubscribedType,
};
-use diesel::{dsl::*, result::Error, ExpressionMethods, QueryDsl, TextExpressionMethods};
+use diesel::{
+ deserialize,
+ dsl,
+ dsl::insert_into,
+ pg::Pg,
+ result::Error,
+ sql_types,
+ ExpressionMethods,
+ NullableExpressionMethods,
+ QueryDsl,
+ Queryable,
+};
use diesel_async::RunQueryDsl;
-mod safe_type {
- use crate::{schema::community::*, source::community::Community, traits::ToSafe};
-
- type Columns = (
- id,
- name,
- title,
- description,
- removed,
- published,
- updated,
- deleted,
- nsfw,
- actor_id,
- local,
- icon,
- banner,
- hidden,
- posting_restricted_to_mods,
- instance_id,
- );
-
- impl ToSafe for Community {
- type SafeColumns = Columns;
- fn safe_columns_tuple() -> Self::SafeColumns {
- (
- id,
- name,
- title,
- description,
- removed,
- published,
- updated,
- deleted,
- nsfw,
- actor_id,
- local,
- icon,
- banner,
- hidden,
- posting_restricted_to_mods,
- instance_id,
- )
- }
- }
-}
-
#[async_trait]
impl Crud for Community {
type InsertForm = CommunityInsertForm;
type UpdateForm = CommunityUpdateForm;
type IdType = CommunityId;
- async fn read(pool: &DbPool, community_id: CommunityId) -> Result<Self, Error> {
- let conn = &mut get_conn(pool).await?;
- community.find(community_id).first::<Self>(conn).await
- }
- async fn delete(pool: &DbPool, community_id: CommunityId) -> Result<usize, Error> {
+ async fn create(pool: &mut DbPool<'_>, form: &Self::InsertForm) -> Result<Self, Error> {
+ let is_new_community = match &form.actor_id {
+ Some(id) => Community::read_from_apub_id(pool, id).await?.is_none(),
+ None => true,
+ };
let conn = &mut get_conn(pool).await?;
- diesel::delete(community.find(community_id))
- .execute(conn)
- .await
- }
- async fn create(pool: &DbPool, form: &Self::InsertForm) -> Result<Self, Error> {
- let conn = &mut get_conn(pool).await?;
- let community_ = insert_into(community)
+ // Can't do separate insert/update commands because InsertForm/UpdateForm aren't convertible
+ let community_ = insert_into(community::table)
.values(form)
- .on_conflict(actor_id)
+ .on_conflict(community::actor_id)
.do_update()
.set(form)
.get_result::<Self>(conn)
.await?;
- let site_languages = SiteLanguage::read_local(pool).await;
- if let Ok(langs) = site_languages {
- // if site exists, init user with site languages
- CommunityLanguage::update(pool, langs, community_.id).await?;
- } else {
- // otherwise, init with all languages (this only happens during tests)
+ // Initialize languages for new community
+ if is_new_community {
CommunityLanguage::update(pool, vec![], community_.id).await?;
}
}
async fn update(
- pool: &DbPool,
+ pool: &mut DbPool<'_>,
community_id: CommunityId,
form: &Self::UpdateForm,
) -> Result<Self, Error> {
let conn = &mut get_conn(pool).await?;
- diesel::update(community.find(community_id))
+ diesel::update(community::table.find(community_id))
.set(form)
.get_result::<Self>(conn)
.await
impl Joinable for CommunityModerator {
type Form = CommunityModeratorForm;
async fn join(
- pool: &DbPool,
+ pool: &mut DbPool<'_>,
community_moderator_form: &CommunityModeratorForm,
) -> Result<Self, Error> {
- use crate::schema::community_moderator::dsl::*;
+ use crate::schema::community_moderator::dsl::community_moderator;
let conn = &mut get_conn(pool).await?;
insert_into(community_moderator)
.values(community_moderator_form)
}
async fn leave(
- pool: &DbPool,
+ pool: &mut DbPool<'_>,
community_moderator_form: &CommunityModeratorForm,
) -> Result<usize, Error> {
- use crate::schema::community_moderator::dsl::*;
+ use crate::schema::community_moderator::dsl::{community_id, community_moderator, person_id};
let conn = &mut get_conn(pool).await?;
diesel::delete(
community_moderator
}
}
-impl DeleteableOrRemoveable for CommunitySafe {
- fn blank_out_deleted_or_removed_info(mut self) -> Self {
- self.title = "".into();
- self.description = None;
- self.icon = None;
- self.banner = None;
- self
- }
+pub enum CollectionType {
+ Moderators,
+ Featured,
}
-impl DeleteableOrRemoveable for Community {
- fn blank_out_deleted_or_removed_info(mut self) -> Self {
- self.title = "".into();
- self.description = None;
- self.icon = None;
- self.banner = None;
- self
+impl Community {
+ /// Get the community which has a given moderators or featured url, also return the collection type
+ pub async fn get_by_collection_url(
+ pool: &mut DbPool<'_>,
+ url: &DbUrl,
+ ) -> Result<(Community, CollectionType), Error> {
+ use crate::schema::community::dsl::{featured_url, moderators_url};
+ use CollectionType::*;
+ let conn = &mut get_conn(pool).await?;
+ let res = community::table
+ .filter(moderators_url.eq(url))
+ .first::<Self>(conn)
+ .await;
+ if let Ok(c) = res {
+ return Ok((c, Moderators));
+ }
+ let res = community::table
+ .filter(featured_url.eq(url))
+ .first::<Self>(conn)
+ .await;
+ if let Ok(c) = res {
+ return Ok((c, Featured));
+ }
+ Err(diesel::NotFound)
}
}
impl CommunityModerator {
pub async fn delete_for_community(
- pool: &DbPool,
+ pool: &mut DbPool<'_>,
for_community_id: CommunityId,
) -> Result<usize, Error> {
- use crate::schema::community_moderator::dsl::*;
+ use crate::schema::community_moderator::dsl::{community_id, community_moderator};
let conn = &mut get_conn(pool).await?;
diesel::delete(community_moderator.filter(community_id.eq(for_community_id)))
.await
}
+ pub async fn leave_all_communities(
+ pool: &mut DbPool<'_>,
+ for_person_id: PersonId,
+ ) -> Result<usize, Error> {
+ use crate::schema::community_moderator::dsl::{community_moderator, person_id};
+ let conn = &mut get_conn(pool).await?;
+ diesel::delete(community_moderator.filter(person_id.eq(for_person_id)))
+ .execute(conn)
+ .await
+ }
+
pub async fn get_person_moderated_communities(
- pool: &DbPool,
+ pool: &mut DbPool<'_>,
for_person_id: PersonId,
) -> Result<Vec<CommunityId>, Error> {
- use crate::schema::community_moderator::dsl::*;
+ use crate::schema::community_moderator::dsl::{community_id, community_moderator, person_id};
let conn = &mut get_conn(pool).await?;
community_moderator
.filter(person_id.eq(for_person_id))
impl Bannable for CommunityPersonBan {
type Form = CommunityPersonBanForm;
async fn ban(
- pool: &DbPool,
+ pool: &mut DbPool<'_>,
community_person_ban_form: &CommunityPersonBanForm,
) -> Result<Self, Error> {
- use crate::schema::community_person_ban::dsl::*;
+ use crate::schema::community_person_ban::dsl::{community_id, community_person_ban, person_id};
let conn = &mut get_conn(pool).await?;
insert_into(community_person_ban)
.values(community_person_ban_form)
}
async fn unban(
- pool: &DbPool,
+ pool: &mut DbPool<'_>,
community_person_ban_form: &CommunityPersonBanForm,
) -> Result<usize, Error> {
- use crate::schema::community_person_ban::dsl::*;
+ use crate::schema::community_person_ban::dsl::{community_id, community_person_ban, person_id};
let conn = &mut get_conn(pool).await?;
diesel::delete(
community_person_ban
pub fn to_subscribed_type(follower: &Option<Self>) -> SubscribedType {
match follower {
Some(f) => {
- if f.pending.unwrap_or(false) {
+ if f.pending {
SubscribedType::Pending
} else {
SubscribedType::Subscribed
None => SubscribedType::NotSubscribed,
}
}
+
+ pub fn select_subscribed_type() -> dsl::Nullable<community_follower::pending> {
+ community_follower::pending.nullable()
+ }
+}
+
+impl Queryable<sql_types::Nullable<sql_types::Bool>, Pg> for SubscribedType {
+ type Row = Option<bool>;
+ fn build(row: Self::Row) -> deserialize::Result<Self> {
+ Ok(match row {
+ Some(true) => SubscribedType::Pending,
+ Some(false) => SubscribedType::Subscribed,
+ None => SubscribedType::NotSubscribed,
+ })
+ }
}
#[async_trait]
impl Followable for CommunityFollower {
type Form = CommunityFollowerForm;
- async fn follow(
- pool: &DbPool,
- community_follower_form: &CommunityFollowerForm,
- ) -> Result<Self, Error> {
- use crate::schema::community_follower::dsl::*;
+ async fn follow(pool: &mut DbPool<'_>, form: &CommunityFollowerForm) -> Result<Self, Error> {
+ use crate::schema::community_follower::dsl::{community_follower, community_id, person_id};
let conn = &mut get_conn(pool).await?;
insert_into(community_follower)
- .values(community_follower_form)
+ .values(form)
.on_conflict((community_id, person_id))
.do_update()
- .set(community_follower_form)
+ .set(form)
.get_result::<Self>(conn)
.await
}
async fn follow_accepted(
- pool: &DbPool,
+ pool: &mut DbPool<'_>,
community_id_: CommunityId,
person_id_: PersonId,
) -> Result<Self, Error> {
- use crate::schema::community_follower::dsl::*;
+ use crate::schema::community_follower::dsl::{
+ community_follower,
+ community_id,
+ pending,
+ person_id,
+ };
let conn = &mut get_conn(pool).await?;
diesel::update(
community_follower
.get_result::<Self>(conn)
.await
}
- async fn unfollow(
- pool: &DbPool,
- community_follower_form: &CommunityFollowerForm,
- ) -> Result<usize, Error> {
- use crate::schema::community_follower::dsl::*;
+ async fn unfollow(pool: &mut DbPool<'_>, form: &CommunityFollowerForm) -> Result<usize, Error> {
+ use crate::schema::community_follower::dsl::{community_follower, community_id, person_id};
let conn = &mut get_conn(pool).await?;
diesel::delete(
community_follower
- .filter(community_id.eq(&community_follower_form.community_id))
- .filter(person_id.eq(&community_follower_form.person_id)),
+ .filter(community_id.eq(&form.community_id))
+ .filter(person_id.eq(&form.person_id)),
)
.execute(conn)
.await
}
- // TODO: this function name only makes sense if you call it with a remote community. for a local
- // community, it will also return true if only remote followers exist
- async fn has_local_followers(pool: &DbPool, community_id_: CommunityId) -> Result<bool, Error> {
- use crate::schema::community_follower::dsl::*;
- let conn = &mut get_conn(pool).await?;
- diesel::select(exists(
- community_follower.filter(community_id.eq(community_id_)),
- ))
- .get_result(conn)
- .await
- }
}
#[async_trait]
impl ApubActor for Community {
- async fn read_from_apub_id(pool: &DbPool, object_id: &DbUrl) -> Result<Option<Self>, Error> {
+ async fn read_from_apub_id(
+ pool: &mut DbPool<'_>,
+ object_id: &DbUrl,
+ ) -> Result<Option<Self>, Error> {
let conn = &mut get_conn(pool).await?;
Ok(
- community
- .filter(actor_id.eq(object_id))
+ community::table
+ .filter(community::actor_id.eq(object_id))
.first::<Community>(conn)
.await
.ok()
}
async fn read_from_name(
- pool: &DbPool,
+ pool: &mut DbPool<'_>,
community_name: &str,
include_deleted: bool,
) -> Result<Community, Error> {
let conn = &mut get_conn(pool).await?;
- let mut q = community
+ let mut q = community::table
.into_boxed()
- .filter(local.eq(true))
- .filter(lower(name).eq(lower(community_name)));
+ .filter(community::local.eq(true))
+ .filter(lower(community::name).eq(community_name.to_lowercase()));
if !include_deleted {
- q = q.filter(deleted.eq(false)).filter(removed.eq(false));
+ q = q
+ .filter(community::deleted.eq(false))
+ .filter(community::removed.eq(false));
}
q.first::<Self>(conn).await
}
async fn read_from_name_and_domain(
- pool: &DbPool,
+ pool: &mut DbPool<'_>,
community_name: &str,
- protocol_domain: &str,
+ for_domain: &str,
) -> Result<Community, Error> {
let conn = &mut get_conn(pool).await?;
- community
- .filter(lower(name).eq(lower(community_name)))
- .filter(actor_id.like(format!("{}%", protocol_domain)))
+ community::table
+ .inner_join(instance::table)
+ .filter(lower(community::name).eq(community_name.to_lowercase()))
+ .filter(instance::domain.eq(for_domain))
+ .select(community::all_columns)
.first::<Self>(conn)
.await
}
#[cfg(test)]
mod tests {
+ #![allow(clippy::unwrap_used)]
+ #![allow(clippy::indexing_slicing)]
+
use crate::{
- source::{community::*, instance::Instance, person::*},
+ source::{
+ community::{
+ Community,
+ CommunityFollower,
+ CommunityFollowerForm,
+ CommunityInsertForm,
+ CommunityModerator,
+ CommunityModeratorForm,
+ CommunityPersonBan,
+ CommunityPersonBanForm,
+ CommunityUpdateForm,
+ },
+ instance::Instance,
+ person::{Person, PersonInsertForm},
+ },
traits::{Bannable, Crud, Followable, Joinable},
utils::build_db_pool_for_tests,
};
#[serial]
async fn test_crud() {
let pool = &build_db_pool_for_tests().await;
+ let pool = &mut pool.into();
- let inserted_instance = Instance::create(pool, "my_domain.tld").await.unwrap();
+ let inserted_instance = Instance::read_or_create(pool, "my_domain.tld".to_string())
+ .await
+ .unwrap();
let new_person = PersonInsertForm::builder()
.name("bobbee".into())
deleted: false,
published: inserted_community.published,
updated: None,
- actor_id: inserted_community.actor_id.to_owned(),
+ actor_id: inserted_community.actor_id.clone(),
local: true,
private_key: None,
public_key: "pubkey".to_owned(),
last_refreshed_at: inserted_community.published,
icon: None,
banner: None,
- followers_url: inserted_community.followers_url.to_owned(),
- inbox_url: inserted_community.inbox_url.to_owned(),
+ followers_url: inserted_community.followers_url.clone(),
+ inbox_url: inserted_community.inbox_url.clone(),
shared_inbox_url: None,
+ moderators_url: None,
+ featured_url: None,
hidden: false,
posting_restricted_to_mods: false,
instance_id: inserted_instance.id,
id: inserted_community_follower.id,
community_id: inserted_community.id,
person_id: inserted_person.id,
- pending: Some(false),
+ pending: false,
published: inserted_community_follower.published,
};