diff --git a/api_tests/package.json b/api_tests/package.json index 06c2c46ae..e57d37d08 100644 --- a/api_tests/package.json +++ b/api_tests/package.json @@ -19,7 +19,7 @@ "eslint": "^8.40.0", "eslint-plugin-prettier": "^4.0.0", "jest": "^29.5.0", - "lemmy-js-client": "0.19.0-rc.3", + "lemmy-js-client": "0.19.0-rc.5", "prettier": "^3.0.0", "ts-jest": "^29.1.0", "typescript": "^5.0.4" diff --git a/api_tests/src/community.spec.ts b/api_tests/src/community.spec.ts index a2af84440..effd2169e 100644 --- a/api_tests/src/community.spec.ts +++ b/api_tests/src/community.spec.ts @@ -24,6 +24,7 @@ import { getComments, createComment, getCommunityByName, + blockInstance, waitUntil, delay, } from "./shared"; @@ -348,3 +349,39 @@ test("Get community for different casing on domain", async () => { .community_view; assertCommunityFederation(betaCommunity, communityRes.community_view); }); + +test("User blocks instance, communities are hidden", async () => { + // create community and post on beta + let communityRes = await createCommunity(beta); + expect(communityRes.community_view.community.name).toBeDefined(); + let postRes = await createPost( + beta, + communityRes.community_view.community.id, + ); + expect(postRes.post_view.post.id).toBeDefined(); + + // fetch post to alpha + let alphaPost = await resolvePost(alpha, postRes.post_view.post); + expect(alphaPost.post?.post).toBeDefined(); + + // post should be included in listing + let listing = await getPosts(alpha, "All"); + let listing_ids = listing.posts.map(p => p.post.ap_id); + expect(listing_ids).toContain(postRes.post_view.post.ap_id); + + // block the beta instance + await blockInstance(alpha, alphaPost.post!.community.instance_id, true); + + // after blocking, post should not be in listing + let listing2 = await getPosts(alpha, "All"); + let listing_ids2 = listing2.posts.map(p => p.post.ap_id); + expect(listing_ids2.indexOf(postRes.post_view.post.ap_id)).toBe(-1); + + // unblock instance again + await blockInstance(alpha, alphaPost.post!.community.instance_id, false); + + // post should be included in listing + let listing3 = await getPosts(alpha, "All"); + let listing_ids3 = listing3.posts.map(p => p.post.ap_id); + expect(listing_ids3).toContain(postRes.post_view.post.ap_id); +}); diff --git a/api_tests/src/shared.ts b/api_tests/src/shared.ts index d116d329f..f7039942d 100644 --- a/api_tests/src/shared.ts +++ b/api_tests/src/shared.ts @@ -1,8 +1,11 @@ import { + BlockInstance, + BlockInstanceResponse, GetReplies, GetRepliesResponse, GetUnreadCount, GetUnreadCountResponse, + InstanceId, LemmyHttp, } from "lemmy-js-client"; import { CreatePost } from "lemmy-js-client/dist/types/CreatePost"; @@ -817,6 +820,19 @@ export function getPosts( return api.client.getPosts(form); } +export function blockInstance( + api: API, + instance_id: InstanceId, + block: boolean, +): Promise { + let form: BlockInstance = { + instance_id, + block, + auth: api.auth, + }; + return api.client.blockInstance(form); +} + export function delay(millis = 500) { return new Promise(resolve => setTimeout(resolve, millis)); } diff --git a/api_tests/yarn.lock b/api_tests/yarn.lock index 17c5a2849..3b6a85146 100644 --- a/api_tests/yarn.lock +++ b/api_tests/yarn.lock @@ -2174,10 +2174,10 @@ kleur@^3.0.3: resolved "https://registry.yarnpkg.com/kleur/-/kleur-3.0.3.tgz#a79c9ecc86ee1ce3fa6206d1216c501f147fc07e" integrity sha512-eTIzlVOSUR+JxdDFepEYcBMtZ9Qqdef+rnzWdRZuMbOywu5tO2w2N7rqjoANZ5k9vywhL6Br1VRjUIgTQx4E8w== -lemmy-js-client@0.19.0-rc.3: - version "0.19.0-rc.3" - resolved "https://registry.yarnpkg.com/lemmy-js-client/-/lemmy-js-client-0.19.0-rc.3.tgz#1efbfd5ce492319227a41cb020fc1cf9b2e7c075" - integrity sha512-RmibQ3+YTvqsQ89II2I29pfPmVAWiSObGAU9Nc/AGYfyvaCya7f5+TirKwHdKA2eWDWLOTnD4rm6WgcgAwvhWw== +lemmy-js-client@0.19.0-rc.5: + version "0.19.0-rc.5" + resolved "https://registry.yarnpkg.com/lemmy-js-client/-/lemmy-js-client-0.19.0-rc.5.tgz#a0d3e0ac829b1e46124edcf3a0343ae04317d51a" + integrity sha512-Z1T95Ht1VZNvWlLH9XpVnO2oC7LhMT81jTiU5BhYPIkEtGhOwN91jk5uI1oyI6/d4v9lwbrsyzFYPsiuMmXTFQ== dependencies: cross-fetch "^3.1.5" form-data "^4.0.0" diff --git a/crates/api/src/site/block.rs b/crates/api/src/site/block.rs new file mode 100644 index 000000000..3278767cf --- /dev/null +++ b/crates/api/src/site/block.rs @@ -0,0 +1,41 @@ +use activitypub_federation::config::Data; +use actix_web::web::Json; +use lemmy_api_common::{ + context::LemmyContext, + site::{BlockInstance, BlockInstanceResponse}, + utils::local_user_view_from_jwt, +}; +use lemmy_db_schema::{ + source::instance_block::{InstanceBlock, InstanceBlockForm}, + traits::Blockable, +}; +use lemmy_utils::error::{LemmyError, LemmyErrorExt, LemmyErrorType}; + +#[tracing::instrument(skip(context))] +pub async fn block_instance( + data: Json, + context: Data, +) -> Result, LemmyError> { + let local_user_view = local_user_view_from_jwt(&data.auth, &context).await?; + + let instance_id = data.instance_id; + let person_id = local_user_view.person.id; + let instance_block_form = InstanceBlockForm { + person_id, + instance_id, + }; + + if data.block { + InstanceBlock::block(&mut context.pool(), &instance_block_form) + .await + .with_lemmy_type(LemmyErrorType::InstanceBlockAlreadyExists)?; + } else { + InstanceBlock::unblock(&mut context.pool(), &instance_block_form) + .await + .with_lemmy_type(LemmyErrorType::InstanceBlockAlreadyExists)?; + } + + Ok(Json(BlockInstanceResponse { + blocked: data.block, + })) +} diff --git a/crates/api/src/site/mod.rs b/crates/api/src/site/mod.rs index 85c20e272..d63c77ad9 100644 --- a/crates/api/src/site/mod.rs +++ b/crates/api/src/site/mod.rs @@ -1,3 +1,4 @@ +pub mod block; pub mod federated_instances; pub mod leave_admin; pub mod mod_log; diff --git a/crates/api_common/src/site.rs b/crates/api_common/src/site.rs index 0dc796ed8..7ed94f2de 100644 --- a/crates/api_common/src/site.rs +++ b/crates/api_common/src/site.rs @@ -1,6 +1,6 @@ use crate::sensitive::Sensitive; use lemmy_db_schema::{ - newtypes::{CommentId, CommunityId, LanguageId, PersonId, PostId}, + newtypes::{CommentId, CommunityId, InstanceId, LanguageId, PersonId, PostId}, source::{instance::Instance, language::Language, tagline::Tagline}, ListingType, ModlogActionType, @@ -21,6 +21,7 @@ use lemmy_db_views_actor::structs::{ CommunityFollowerView, CommunityModeratorView, CommunityView, + InstanceBlockView, PersonBlockView, PersonView, }; @@ -320,6 +321,7 @@ pub struct MyUserInfo { pub follows: Vec, pub moderates: Vec, pub community_blocks: Vec, + pub instance_blocks: Vec, pub person_blocks: Vec, pub discussion_languages: Vec, } @@ -450,3 +452,21 @@ pub struct GetUnreadRegistrationApplicationCount { pub struct GetUnreadRegistrationApplicationCountResponse { pub registration_applications: i64, } + +#[derive(Debug, Serialize, Deserialize, Clone, Default)] +#[cfg_attr(feature = "full", derive(TS))] +#[cfg_attr(feature = "full", ts(export))] +/// Block an instance as user +pub struct BlockInstance { + pub instance_id: InstanceId, + pub block: bool, + pub auth: Sensitive, +} + +#[skip_serializing_none] +#[derive(Debug, Serialize, Deserialize, Clone)] +#[cfg_attr(feature = "full", derive(TS))] +#[cfg_attr(feature = "full", ts(export))] +pub struct BlockInstanceResponse { + pub blocked: bool, +} diff --git a/crates/api_crud/src/site/read.rs b/crates/api_crud/src/site/read.rs index 62d96492a..d435b8684 100644 --- a/crates/api_crud/src/site/read.rs +++ b/crates/api_crud/src/site/read.rs @@ -18,6 +18,7 @@ use lemmy_db_views_actor::structs::{ CommunityBlockView, CommunityFollowerView, CommunityModeratorView, + InstanceBlockView, PersonBlockView, PersonView, }; @@ -52,6 +53,10 @@ pub async fn get_site( .await .with_lemmy_type(LemmyErrorType::SystemErrLogin)?; + let instance_blocks = InstanceBlockView::for_person(&mut context.pool(), person_id) + .await + .with_lemmy_type(LemmyErrorType::SystemErrLogin)?; + let person_id = local_user_view.person.id; let person_blocks = PersonBlockView::for_person(&mut context.pool(), person_id) .await @@ -70,6 +75,7 @@ pub async fn get_site( follows, moderates, community_blocks, + instance_blocks, person_blocks, discussion_languages, }) diff --git a/crates/db_schema/src/aggregates/structs.rs b/crates/db_schema/src/aggregates/structs.rs index f24f02f49..03ff9a640 100644 --- a/crates/db_schema/src/aggregates/structs.rs +++ b/crates/db_schema/src/aggregates/structs.rs @@ -1,4 +1,4 @@ -use crate::newtypes::{CommentId, CommunityId, PersonId, PostId, SiteId}; +use crate::newtypes::{CommentId, CommunityId, InstanceId, PersonId, PostId, SiteId}; #[cfg(feature = "full")] use crate::schema::{ comment_aggregates, @@ -100,6 +100,7 @@ pub struct PostAggregates { pub community_id: CommunityId, pub creator_id: PersonId, pub controversy_rank: f64, + pub instance_id: InstanceId, /// A rank that amplifies smaller communities pub scaled_rank: f64, } diff --git a/crates/db_schema/src/impls/instance_block.rs b/crates/db_schema/src/impls/instance_block.rs new file mode 100644 index 000000000..f7f81a234 --- /dev/null +++ b/crates/db_schema/src/impls/instance_block.rs @@ -0,0 +1,36 @@ +use crate::{ + schema::instance_block::dsl::{instance_block, instance_id, person_id}, + source::instance_block::{InstanceBlock, InstanceBlockForm}, + traits::Blockable, + utils::{get_conn, DbPool}, +}; +use diesel::{dsl::insert_into, result::Error, ExpressionMethods, QueryDsl}; +use diesel_async::RunQueryDsl; + +#[async_trait] +impl Blockable for InstanceBlock { + type Form = InstanceBlockForm; + async fn block(pool: &mut DbPool<'_>, instance_block_form: &Self::Form) -> Result { + let conn = &mut get_conn(pool).await?; + insert_into(instance_block) + .values(instance_block_form) + .on_conflict((person_id, instance_id)) + .do_update() + .set(instance_block_form) + .get_result::(conn) + .await + } + async fn unblock( + pool: &mut DbPool<'_>, + instance_block_form: &Self::Form, + ) -> Result { + let conn = &mut get_conn(pool).await?; + diesel::delete( + instance_block + .filter(person_id.eq(instance_block_form.person_id)) + .filter(instance_id.eq(instance_block_form.instance_id)), + ) + .execute(conn) + .await + } +} diff --git a/crates/db_schema/src/impls/mod.rs b/crates/db_schema/src/impls/mod.rs index a8aabf18d..a46131f83 100644 --- a/crates/db_schema/src/impls/mod.rs +++ b/crates/db_schema/src/impls/mod.rs @@ -12,6 +12,7 @@ pub mod federation_allowlist; pub mod federation_blocklist; pub mod image_upload; pub mod instance; +pub mod instance_block; pub mod language; pub mod local_site; pub mod local_site_rate_limit; diff --git a/crates/db_schema/src/schema.rs b/crates/db_schema/src/schema.rs index 62b6ea80a..bec1ab40e 100644 --- a/crates/db_schema/src/schema.rs +++ b/crates/db_schema/src/schema.rs @@ -337,6 +337,15 @@ diesel::table! { } } +diesel::table! { + instance_block (id) { + id -> Int4, + person_id -> Int4, + instance_id -> Int4, + published -> Timestamptz, + } +} + diesel::table! { language (id) { id -> Int4, @@ -712,6 +721,7 @@ diesel::table! { community_id -> Int4, creator_id -> Int4, controversy_rank -> Float8, + instance_id -> Int4, scaled_rank -> Float8, } } @@ -928,6 +938,8 @@ diesel::joinable!(federation_allowlist -> instance (instance_id)); diesel::joinable!(federation_blocklist -> instance (instance_id)); diesel::joinable!(federation_queue_state -> instance (instance_id)); diesel::joinable!(image_upload -> local_user (local_user_id)); +diesel::joinable!(instance_block -> instance (instance_id)); +diesel::joinable!(instance_block -> person (person_id)); diesel::joinable!(local_site -> site (site_id)); diesel::joinable!(local_site_rate_limit -> local_site (local_site_id)); diesel::joinable!(local_user -> person (person_id)); @@ -960,6 +972,7 @@ diesel::joinable!(post -> community (community_id)); diesel::joinable!(post -> language (language_id)); diesel::joinable!(post -> person (creator_id)); diesel::joinable!(post_aggregates -> community (community_id)); +diesel::joinable!(post_aggregates -> instance (instance_id)); diesel::joinable!(post_aggregates -> person (creator_id)); diesel::joinable!(post_aggregates -> post (post_id)); diesel::joinable!(post_like -> person (person_id)); @@ -1005,6 +1018,7 @@ diesel::allow_tables_to_appear_in_same_query!( federation_queue_state, image_upload, instance, + instance_block, language, local_site, local_site_rate_limit, diff --git a/crates/db_schema/src/source/instance_block.rs b/crates/db_schema/src/source/instance_block.rs new file mode 100644 index 000000000..1aa215e45 --- /dev/null +++ b/crates/db_schema/src/source/instance_block.rs @@ -0,0 +1,26 @@ +use crate::newtypes::{InstanceId, PersonId}; +#[cfg(feature = "full")] +use crate::schema::instance_block; +use chrono::{DateTime, Utc}; +use serde::{Deserialize, Serialize}; + +#[derive(Clone, PartialEq, Eq, Debug, Serialize, Deserialize)] +#[cfg_attr(feature = "full", derive(Queryable, Associations, Identifiable))] +#[cfg_attr( + feature = "full", + diesel(belongs_to(crate::source::instance::Instance)) +)] +#[cfg_attr(feature = "full", diesel(table_name = instance_block))] +pub struct InstanceBlock { + pub id: i32, + pub person_id: PersonId, + pub instance_id: InstanceId, + pub published: DateTime, +} + +#[cfg_attr(feature = "full", derive(Insertable, AsChangeset))] +#[cfg_attr(feature = "full", diesel(table_name = instance_block))] +pub struct InstanceBlockForm { + pub person_id: PersonId, + pub instance_id: InstanceId, +} diff --git a/crates/db_schema/src/source/mod.rs b/crates/db_schema/src/source/mod.rs index a200806cd..8288126ec 100644 --- a/crates/db_schema/src/source/mod.rs +++ b/crates/db_schema/src/source/mod.rs @@ -17,6 +17,7 @@ pub mod federation_allowlist; pub mod federation_blocklist; pub mod image_upload; pub mod instance; +pub mod instance_block; pub mod language; pub mod local_site; pub mod local_site_rate_limit; diff --git a/crates/db_views/src/comment_view.rs b/crates/db_views/src/comment_view.rs index b93f12842..a3f35b0a9 100644 --- a/crates/db_views/src/comment_view.rs +++ b/crates/db_views/src/comment_view.rs @@ -23,6 +23,7 @@ use lemmy_db_schema::{ community_follower, community_moderator, community_person_ban, + instance_block, local_user_language, person, person_block, @@ -121,6 +122,13 @@ fn queries<'a>() -> Queries< let local_user_id_join = local_user_id.unwrap_or(LocalUserId(-1)); let mut query = all_joins(comment::table.into_boxed(), person_id) + .left_join( + instance_block::table.on( + community::instance_id + .eq(instance_block::instance_id) + .and(instance_block::person_id.eq(person_id_join)), + ), + ) .left_join( community_block::table.on( community::id @@ -221,6 +229,7 @@ fn queries<'a>() -> Queries< // Don't show blocked communities or persons if options.post_id.is_none() { + query = query.filter(instance_block::person_id.is_null()); query = query.filter(community_block::person_id.is_null()); } query = query.filter(person_block::person_id.is_null()); diff --git a/crates/db_views/src/post_view.rs b/crates/db_views/src/post_view.rs index 607af986d..87c31d7e5 100644 --- a/crates/db_views/src/post_view.rs +++ b/crates/db_views/src/post_view.rs @@ -28,6 +28,7 @@ use lemmy_db_schema::{ community_follower, community_moderator, community_person_ban, + instance_block, local_user_language, person, person_block, @@ -420,7 +421,7 @@ fn queries<'a>() -> Queries< ), )); - // Don't show blocked communities or persons + // Don't show blocked instances, communities or persons query = query.filter(not(exists( community_block::table.filter( post_aggregates::community_id @@ -428,6 +429,13 @@ fn queries<'a>() -> Queries< .and(community_block::person_id.eq(person_id_join)), ), ))); + query = query.filter(not(exists( + instance_block::table.filter( + post_aggregates::instance_id + .eq(instance_block::instance_id) + .and(instance_block::person_id.eq(person_id_join)), + ), + ))); query = query.filter(not(is_creator_blocked(person_id))); } let now = diesel::dsl::now.into_sql::(); @@ -706,6 +714,7 @@ mod tests { community::{Community, CommunityInsertForm}, community_block::{CommunityBlock, CommunityBlockForm}, instance::Instance, + instance_block::{InstanceBlock, InstanceBlockForm}, language::Language, local_user::{LocalUser, LocalUserInsertForm, LocalUserUpdateForm}, person::{Person, PersonInsertForm}, @@ -1239,6 +1248,84 @@ mod tests { cleanup(data, pool).await; } + #[tokio::test] + #[serial] + async fn post_listing_instance_block() { + let pool = &build_db_pool_for_tests().await; + let pool = &mut pool.into(); + let data = init_data(pool).await; + + let blocked_instance = Instance::read_or_create(pool, "another_domain.tld".to_string()) + .await + .unwrap(); + + let community_form = CommunityInsertForm::builder() + .name("test_community_4".to_string()) + .title("none".to_owned()) + .public_key("pubkey".to_string()) + .instance_id(blocked_instance.id) + .build(); + let inserted_community = Community::create(pool, &community_form).await.unwrap(); + + let post_form = PostInsertForm::builder() + .name("blocked instance post".to_string()) + .creator_id(data.inserted_bot.id) + .community_id(inserted_community.id) + .language_id(Some(LanguageId(1))) + .build(); + + let post_from_blocked_instance = Post::create(pool, &post_form).await.unwrap(); + + // no instance block, should return all posts + let post_listings_all = PostQuery { + local_user: Some(&data.local_user_view), + ..Default::default() + } + .list(pool) + .await + .unwrap(); + assert_eq!(post_listings_all.len(), 3); + + // block the instance + let block_form = InstanceBlockForm { + person_id: data.local_user_view.person.id, + instance_id: blocked_instance.id, + }; + InstanceBlock::block(pool, &block_form).await.unwrap(); + + // now posts from communities on that instance should be hidden + let post_listings_blocked = PostQuery { + local_user: Some(&data.local_user_view), + ..Default::default() + } + .list(pool) + .await + .unwrap(); + assert_eq!(post_listings_blocked.len(), 2); + assert_ne!( + post_listings_blocked[0].post.id, + post_from_blocked_instance.id + ); + assert_ne!( + post_listings_blocked[1].post.id, + post_from_blocked_instance.id + ); + + // after unblocking it should return all posts again + InstanceBlock::unblock(pool, &block_form).await.unwrap(); + let post_listings_blocked = PostQuery { + local_user: Some(&data.local_user_view), + ..Default::default() + } + .list(pool) + .await + .unwrap(); + assert_eq!(post_listings_blocked.len(), 3); + + Instance::delete(pool, blocked_instance.id).await.unwrap(); + cleanup(data, pool).await; + } + async fn cleanup(data: Data, pool: &mut DbPool<'_>) { let num_deleted = Post::delete(pool, data.inserted_post.id).await.unwrap(); Community::delete(pool, data.inserted_community.id) @@ -1359,6 +1446,7 @@ mod tests { scaled_rank: 0.3621, community_id: inserted_post.community_id, creator_id: inserted_post.creator_id, + instance_id: data.inserted_instance.id, }, subscribed: SubscribedType::NotSubscribed, read: false, diff --git a/crates/db_views_actor/src/community_view.rs b/crates/db_views_actor/src/community_view.rs index 974267061..f646e21f8 100644 --- a/crates/db_views_actor/src/community_view.rs +++ b/crates/db_views_actor/src/community_view.rs @@ -12,7 +12,14 @@ use diesel::{ use diesel_async::RunQueryDsl; use lemmy_db_schema::{ newtypes::{CommunityId, PersonId}, - schema::{community, community_aggregates, community_block, community_follower, local_user}, + schema::{ + community, + community_aggregates, + community_block, + community_follower, + instance_block, + local_user, + }, source::{community::CommunityFollower, local_user::LocalUser}, utils::{fuzzy_search, limit_and_offset, DbConn, DbPool, ListFn, Queries, ReadFn}, ListingType, @@ -36,6 +43,13 @@ fn queries<'a>() -> Queries< .and(community_follower::person_id.eq(person_id_join)), ), ) + .left_join( + instance_block::table.on( + community::instance_id + .eq(instance_block::instance_id) + .and(instance_block::person_id.eq(person_id_join)), + ), + ) .left_join( community_block::table.on( community::id @@ -131,8 +145,10 @@ fn queries<'a>() -> Queries< }; } - // Don't show blocked communities or nsfw communities if not enabled in profile + // Don't show blocked communities and communities on blocked instances. nsfw communities are + // also hidden (based on profile setting) if options.local_user.is_some() { + query = query.filter(instance_block::person_id.is_null()); query = query.filter(community_block::person_id.is_null()); query = query.filter(community::nsfw.eq(false).or(local_user::show_nsfw.eq(true))); } else { diff --git a/crates/db_views_actor/src/instance_block_view.rs b/crates/db_views_actor/src/instance_block_view.rs new file mode 100644 index 000000000..05820862a --- /dev/null +++ b/crates/db_views_actor/src/instance_block_view.rs @@ -0,0 +1,27 @@ +use crate::structs::InstanceBlockView; +use diesel::{result::Error, ExpressionMethods, JoinOnDsl, NullableExpressionMethods, QueryDsl}; +use diesel_async::RunQueryDsl; +use lemmy_db_schema::{ + newtypes::PersonId, + schema::{instance, instance_block, person, site}, + utils::{get_conn, DbPool}, +}; + +impl InstanceBlockView { + pub async fn for_person(pool: &mut DbPool<'_>, person_id: PersonId) -> Result, Error> { + let conn = &mut get_conn(pool).await?; + instance_block::table + .inner_join(person::table) + .inner_join(instance::table) + .left_join(site::table.on(site::instance_id.eq(instance::id))) + .select(( + person::all_columns, + instance::all_columns, + site::all_columns.nullable(), + )) + .filter(instance_block::person_id.eq(person_id)) + .order_by(instance_block::published) + .load::(conn) + .await + } +} diff --git a/crates/db_views_actor/src/lib.rs b/crates/db_views_actor/src/lib.rs index 686945505..e9f8e4189 100644 --- a/crates/db_views_actor/src/lib.rs +++ b/crates/db_views_actor/src/lib.rs @@ -11,6 +11,8 @@ pub mod community_person_ban_view; #[cfg(feature = "full")] pub mod community_view; #[cfg(feature = "full")] +pub mod instance_block_view; +#[cfg(feature = "full")] pub mod person_block_view; #[cfg(feature = "full")] pub mod person_mention_view; diff --git a/crates/db_views_actor/src/structs.rs b/crates/db_views_actor/src/structs.rs index 3728bb02c..bdc9e6bbd 100644 --- a/crates/db_views_actor/src/structs.rs +++ b/crates/db_views_actor/src/structs.rs @@ -6,9 +6,11 @@ use lemmy_db_schema::{ comment::Comment, comment_reply::CommentReply, community::Community, + instance::Instance, person::Person, person_mention::PersonMention, post::Post, + site::Site, }, SubscribedType, }; @@ -26,6 +28,17 @@ pub struct CommunityBlockView { pub community: Community, } +#[skip_serializing_none] +#[derive(Debug, Serialize, Deserialize, Clone)] +#[cfg_attr(feature = "full", derive(TS, Queryable))] +#[cfg_attr(feature = "full", ts(export))] +/// An instance block by a user. +pub struct InstanceBlockView { + pub person: Person, + pub instance: Instance, + pub site: Option, +} + #[derive(Debug, Serialize, Deserialize, Clone)] #[cfg_attr(feature = "full", derive(TS, Queryable))] #[cfg_attr(feature = "full", ts(export))] diff --git a/crates/utils/src/error.rs b/crates/utils/src/error.rs index 9326805fb..8c3523363 100644 --- a/crates/utils/src/error.rs +++ b/crates/utils/src/error.rs @@ -209,6 +209,7 @@ pub enum LemmyErrorType { InvalidUrlScheme, CouldntSendWebmention, ContradictingFilters, + InstanceBlockAlreadyExists, AuthCookieInsecure, Unknown(String), } diff --git a/migrations/2023-08-09-101305_user_instance_block/down.sql b/migrations/2023-08-09-101305_user_instance_block/down.sql new file mode 100644 index 000000000..e24a6ea0f --- /dev/null +++ b/migrations/2023-08-09-101305_user_instance_block/down.sql @@ -0,0 +1,21 @@ +DROP TABLE instance_block; + +ALTER TABLE post_aggregates + DROP COLUMN instance_id; + +CREATE OR REPLACE FUNCTION post_aggregates_post () + RETURNS TRIGGER + LANGUAGE plpgsql + AS $$ +BEGIN + IF (TG_OP = 'INSERT') THEN + INSERT INTO post_aggregates (post_id, published, newest_comment_time, newest_comment_time_necro, community_id, creator_id) + VALUES (NEW.id, NEW.published, NEW.published, NEW.published, NEW.community_id, NEW.creator_id); + ELSIF (TG_OP = 'DELETE') THEN + DELETE FROM post_aggregates + WHERE post_id = OLD.id; + END IF; + RETURN NULL; +END +$$; + diff --git a/migrations/2023-08-09-101305_user_instance_block/up.sql b/migrations/2023-08-09-101305_user_instance_block/up.sql new file mode 100644 index 000000000..7b3668a82 --- /dev/null +++ b/migrations/2023-08-09-101305_user_instance_block/up.sql @@ -0,0 +1,51 @@ +CREATE TABLE instance_block ( + id serial PRIMARY KEY, + person_id int REFERENCES person ON UPDATE CASCADE ON DELETE CASCADE NOT NULL, + instance_id int REFERENCES instance ON UPDATE CASCADE ON DELETE CASCADE NOT NULL, + published timestamptz NOT NULL DEFAULT now(), + UNIQUE (person_id, instance_id) +); + +ALTER TABLE post_aggregates + ADD COLUMN instance_id integer REFERENCES instance ON UPDATE CASCADE ON DELETE CASCADE; + +CREATE OR REPLACE FUNCTION post_aggregates_post () + RETURNS TRIGGER + LANGUAGE plpgsql + AS $$ +BEGIN + IF (TG_OP = 'INSERT') THEN + INSERT INTO post_aggregates (post_id, published, newest_comment_time, newest_comment_time_necro, community_id, creator_id, instance_id) + SELECT + NEW.id, + NEW.published, + NEW.published, + NEW.published, + NEW.community_id, + NEW.creator_id, + community.instance_id + FROM + community + WHERE + NEW.community_id = community.id; + ELSIF (TG_OP = 'DELETE') THEN + DELETE FROM post_aggregates + WHERE post_id = OLD.id; + END IF; + RETURN NULL; +END +$$; + +UPDATE + post_aggregates +SET + instance_id = community.instance_id +FROM + post + JOIN community ON post.community_id = community.id +WHERE + post.id = post_aggregates.post_id; + +ALTER TABLE post_aggregates + ALTER COLUMN instance_id SET NOT NULL; + diff --git a/src/api_routes_http.rs b/src/api_routes_http.rs index 74003b8d8..183034d05 100644 --- a/src/api_routes_http.rs +++ b/src/api_routes_http.rs @@ -56,6 +56,7 @@ use lemmy_api::{ resolve::resolve_pm_report, }, site::{ + block::block_instance, federated_instances::get_federated_instances, leave_admin::leave_admin, mod_log::get_mod_log, @@ -129,7 +130,8 @@ pub fn config(cfg: &mut web::ServiceConfig, rate_limit: &RateLimitCell) { .route("", web::get().to(get_site)) // Admin Actions .route("", web::post().to(create_site)) - .route("", web::put().to(update_site)), + .route("", web::put().to(update_site)) + .route("/block", web::post().to(block_instance)), ) .service( web::resource("/modlog")