2023-09-18 13:44:48 +00:00
use crate ::structs ::{ LocalUserView , PaginationCursor , PostView } ;
2022-11-09 10:05:00 +00:00
use diesel ::{
debug_query ,
2023-09-18 13:44:48 +00:00
dsl ::{ self , exists , not , IntervalDsl } ,
expression ::AsExpression ,
2022-11-09 10:05:00 +00:00
pg ::Pg ,
result ::Error ,
sql_function ,
2023-09-18 13:44:48 +00:00
sql_types ::{ self , SingleValue , SqlType , Timestamptz } ,
2022-11-09 10:05:00 +00:00
BoolExpressionMethods ,
2023-09-04 09:05:00 +00:00
BoxableExpression ,
2023-09-18 13:44:48 +00:00
Expression ,
2022-11-09 10:05:00 +00:00
ExpressionMethods ,
2023-08-24 15:27:00 +00:00
IntoSql ,
2023-09-18 13:44:48 +00:00
JoinOnDsl ,
2022-11-09 10:05:00 +00:00
NullableExpressionMethods ,
2023-09-18 13:44:48 +00:00
OptionalExtension ,
2022-11-09 10:05:00 +00:00
PgTextExpressionMethods ,
QueryDsl ,
} ;
use diesel_async ::RunQueryDsl ;
2021-10-16 13:33:38 +00:00
use lemmy_db_schema ::{
2023-09-18 13:44:48 +00:00
aggregates ::structs ::PostAggregates ,
2023-04-12 14:40:59 +00:00
newtypes ::{ CommunityId , LocalUserId , PersonId , PostId } ,
2020-12-10 20:53:49 +00:00
schema ::{
community ,
2021-08-19 20:54:15 +00:00
community_block ,
2020-12-10 20:53:49 +00:00
community_follower ,
2023-07-26 17:51:11 +00:00
community_moderator ,
2021-03-10 22:33:55 +00:00
community_person_ban ,
2022-08-18 19:11:19 +00:00
local_user_language ,
2021-03-11 04:43:11 +00:00
person ,
2021-08-19 20:54:15 +00:00
person_block ,
2022-09-27 16:45:46 +00:00
person_post_aggregates ,
2020-12-10 20:53:49 +00:00
post ,
2023-09-18 13:44:48 +00:00
post_aggregates ::{ self , newest_comment_time } ,
2020-12-10 20:53:49 +00:00
post_like ,
post_read ,
post_saved ,
} ,
2023-09-18 13:44:48 +00:00
utils ::{ fuzzy_search , get_conn , limit_and_offset , DbConn , DbPool , ListFn , Queries , ReadFn } ,
2022-05-06 20:55:07 +00:00
ListingType ,
SortType ,
2020-12-10 20:53:49 +00:00
} ;
2021-11-23 12:16:47 +00:00
use tracing ::debug ;
2020-12-10 20:53:49 +00:00
2022-09-27 16:45:46 +00:00
sql_function! ( fn coalesce ( x : sql_types ::Nullable < sql_types ::BigInt > , y : sql_types ::BigInt ) -> sql_types ::BigInt ) ;
2023-09-18 13:44:48 +00:00
fn order_and_page_filter_desc < Q , C , T > (
query : Q ,
column : C ,
options : & PostQuery ,
getter : impl Fn ( & PostAggregates ) -> T ,
) -> Q
where
Q : diesel ::query_dsl ::methods ::ThenOrderDsl < dsl ::Desc < C > , Output = Q >
+ diesel ::query_dsl ::methods ::ThenOrderDsl < dsl ::Asc < C > , Output = Q >
+ diesel ::query_dsl ::methods ::FilterDsl < dsl ::GtEq < C , T > , Output = Q >
+ diesel ::query_dsl ::methods ::FilterDsl < dsl ::LtEq < C , T > , Output = Q > ,
C : Expression + Copy ,
C ::SqlType : SingleValue + SqlType ,
T : AsExpression < C ::SqlType > ,
{
let mut query = query . then_order_by ( column . desc ( ) ) ;
if let Some ( before ) = & options . page_before_or_equal {
query = query . filter ( column . ge ( getter ( & before . 0 ) ) ) ;
}
if let Some ( after ) = & options . page_after {
query = query . filter ( column . le ( getter ( & after . 0 ) ) ) ;
}
query
}
fn order_and_page_filter_asc < Q , C , T > (
query : Q ,
column : C ,
options : & PostQuery ,
getter : impl Fn ( & PostAggregates ) -> T ,
) -> Q
where
Q : diesel ::query_dsl ::methods ::ThenOrderDsl < dsl ::Asc < C > , Output = Q >
+ diesel ::query_dsl ::methods ::FilterDsl < dsl ::LtEq < C , T > , Output = Q >
+ diesel ::query_dsl ::methods ::FilterDsl < dsl ::GtEq < C , T > , Output = Q > ,
C : Expression + Copy ,
C ::SqlType : SingleValue + SqlType ,
T : AsExpression < C ::SqlType > ,
{
let mut query = query . then_order_by ( column . asc ( ) ) ;
if let Some ( before ) = & options . page_before_or_equal {
query = query . filter ( column . le ( getter ( & before . 0 ) ) ) ;
}
if let Some ( after ) = & options . page_after {
query = query . filter ( column . ge ( getter ( & after . 0 ) ) ) ;
}
query
}
2023-07-28 08:36:50 +00:00
fn queries < ' a > ( ) -> Queries <
2023-08-11 09:13:14 +00:00
impl ReadFn < ' a , PostView , ( PostId , Option < PersonId > , bool ) > ,
2023-07-28 08:36:50 +00:00
impl ListFn < ' a , PostView , PostQuery < ' a > > ,
> {
2023-09-04 09:05:00 +00:00
let is_creator_banned_from_community = exists (
community_person_ban ::table . filter (
post_aggregates ::community_id
. eq ( community_person_ban ::community_id )
. and ( community_person_ban ::person_id . eq ( post_aggregates ::creator_id ) ) ,
) ,
) ;
let is_saved = | person_id | {
exists (
post_saved ::table . filter (
post_aggregates ::post_id
. eq ( post_saved ::post_id )
. and ( post_saved ::person_id . eq ( person_id ) ) ,
) ,
)
} ;
let is_read = | person_id | {
exists (
post_read ::table . filter (
post_aggregates ::post_id
. eq ( post_read ::post_id )
. and ( post_read ::person_id . eq ( person_id ) ) ,
) ,
)
} ;
let is_creator_blocked = | person_id | {
exists (
person_block ::table . filter (
post_aggregates ::creator_id
. eq ( person_block ::target_id )
. and ( person_block ::person_id . eq ( person_id ) ) ,
) ,
)
} ;
let score = | person_id | {
post_like ::table
. filter (
post_aggregates ::post_id
. eq ( post_like ::post_id )
. and ( post_like ::person_id . eq ( person_id ) ) ,
)
. select ( post_like ::score . nullable ( ) )
. single_value ( )
} ;
let all_joins = move | query : post_aggregates ::BoxedQuery < ' a , Pg > ,
my_person_id : Option < PersonId > ,
saved_only : bool | {
let is_saved_selection : Box < dyn BoxableExpression < _ , Pg , SqlType = sql_types ::Bool > > =
if saved_only {
Box ::new ( true . into_sql ::< sql_types ::Bool > ( ) )
} else if let Some ( person_id ) = my_person_id {
Box ::new ( is_saved ( person_id ) )
} else {
Box ::new ( false . into_sql ::< sql_types ::Bool > ( ) )
} ;
let is_read_selection : Box < dyn BoxableExpression < _ , Pg , SqlType = sql_types ::Bool > > =
if let Some ( person_id ) = my_person_id {
Box ::new ( is_read ( person_id ) )
} else {
Box ::new ( false . into_sql ::< sql_types ::Bool > ( ) )
} ;
let is_creator_blocked_selection : Box < dyn BoxableExpression < _ , Pg , SqlType = sql_types ::Bool > > =
if let Some ( person_id ) = my_person_id {
Box ::new ( is_creator_blocked ( person_id ) )
} else {
Box ::new ( false . into_sql ::< sql_types ::Bool > ( ) )
} ;
let subscribed_type_selection : Box <
dyn BoxableExpression < _ , Pg , SqlType = sql_types ::Nullable < sql_types ::Bool > > ,
> = if let Some ( person_id ) = my_person_id {
Box ::new (
community_follower ::table
. filter (
post_aggregates ::community_id
. eq ( community_follower ::community_id )
. and ( community_follower ::person_id . eq ( person_id ) ) ,
)
. select ( community_follower ::pending . nullable ( ) )
. single_value ( ) ,
)
} else {
Box ::new ( None ::< bool > . into_sql ::< sql_types ::Nullable < sql_types ::Bool > > ( ) )
} ;
let score_selection : Box <
dyn BoxableExpression < _ , Pg , SqlType = sql_types ::Nullable < sql_types ::SmallInt > > ,
> = if let Some ( person_id ) = my_person_id {
Box ::new ( score ( person_id ) )
} else {
Box ::new ( None ::< i16 > . into_sql ::< sql_types ::Nullable < sql_types ::SmallInt > > ( ) )
} ;
let read_comments : Box <
dyn BoxableExpression < _ , Pg , SqlType = sql_types ::Nullable < sql_types ::BigInt > > ,
> = if let Some ( person_id ) = my_person_id {
Box ::new (
person_post_aggregates ::table
. filter (
post_aggregates ::post_id
. eq ( person_post_aggregates ::post_id )
. and ( person_post_aggregates ::person_id . eq ( person_id ) ) ,
)
. select ( person_post_aggregates ::read_comments . nullable ( ) )
. single_value ( ) ,
)
} else {
Box ::new ( None ::< i64 > . into_sql ::< sql_types ::Nullable < sql_types ::BigInt > > ( ) )
} ;
2023-07-28 08:36:50 +00:00
query
2021-03-10 22:33:55 +00:00
. inner_join ( person ::table )
2020-12-11 15:27:33 +00:00
. inner_join ( community ::table )
2023-07-20 15:13:21 +00:00
. inner_join ( post ::table )
2023-09-04 09:05:00 +00:00
. select ( (
post ::all_columns ,
person ::all_columns ,
community ::all_columns ,
is_creator_banned_from_community ,
post_aggregates ::all_columns ,
subscribed_type_selection ,
is_saved_selection ,
is_read_selection ,
is_creator_blocked_selection ,
score_selection ,
coalesce (
post_aggregates ::comments . nullable ( ) - read_comments ,
post_aggregates ::comments ,
2020-12-11 15:27:33 +00:00
) ,
2023-09-04 09:05:00 +00:00
) )
2023-07-28 08:36:50 +00:00
} ;
2023-08-11 09:13:14 +00:00
let read =
move | mut conn : DbConn < ' a > ,
( post_id , my_person_id , is_mod_or_admin ) : ( PostId , Option < PersonId > , bool ) | async move {
// The left join below will return None in this case
let person_id_join = my_person_id . unwrap_or ( PersonId ( - 1 ) ) ;
let mut query = all_joins (
post_aggregates ::table
. filter ( post_aggregates ::post_id . eq ( post_id ) )
. into_boxed ( ) ,
my_person_id ,
2023-09-04 09:05:00 +00:00
false ,
) ;
2023-03-01 03:46:15 +00:00
2023-08-11 09:13:14 +00:00
// Hide deleted and removed for non-admins or mods
if ! is_mod_or_admin {
query = query
. filter ( community ::removed . eq ( false ) )
. filter ( post ::removed . eq ( false ) )
// users can see their own deleted posts
. filter (
community ::deleted
. eq ( false )
. or ( post ::creator_id . eq ( person_id_join ) ) ,
)
. filter (
post ::deleted
. eq ( false )
. or ( post ::creator_id . eq ( person_id_join ) ) ,
) ;
}
2023-03-01 03:46:15 +00:00
2023-08-31 13:26:10 +00:00
query . first ::< PostView > ( & mut conn ) . await
2023-08-11 09:13:14 +00:00
} ;
2021-01-31 15:29:21 +00:00
2023-07-28 08:36:50 +00:00
let list = move | mut conn : DbConn < ' a > , options : PostQuery < ' a > | async move {
let person_id = options . local_user . map ( | l | l . person . id ) ;
let local_user_id = options . local_user . map ( | l | l . local_user . id ) ;
2020-12-10 20:53:49 +00:00
2020-12-16 18:59:43 +00:00
// The left join below will return None in this case
2023-07-28 08:36:50 +00:00
let person_id_join = person_id . unwrap_or ( PersonId ( - 1 ) ) ;
let local_user_id_join = local_user_id . unwrap_or ( LocalUserId ( - 1 ) ) ;
2020-12-16 18:59:43 +00:00
2023-09-04 09:05:00 +00:00
let mut query = all_joins (
post_aggregates ::table . into_boxed ( ) ,
person_id ,
options . saved_only ,
) ;
2020-12-10 20:53:49 +00:00
2023-07-28 08:36:50 +00:00
let is_creator = options . creator_id = = options . local_user . map ( | l | l . person . id ) ;
2023-07-20 14:36:16 +00:00
// only show deleted posts to creator
if is_creator {
2023-03-01 03:46:15 +00:00
query = query
2023-06-23 10:53:46 +00:00
. filter ( community ::deleted . eq ( false ) )
. filter ( post ::deleted . eq ( false ) ) ;
2023-03-01 03:46:15 +00:00
}
2023-08-24 09:40:08 +00:00
let is_admin = options
. local_user
. map ( | l | l . local_user . admin )
. unwrap_or ( false ) ;
2023-07-20 14:36:16 +00:00
// only show removed posts to admin when viewing user profile
2023-08-04 15:36:36 +00:00
if ! ( options . is_profile_view & & is_admin ) {
2023-07-20 14:36:16 +00:00
query = query
. filter ( community ::removed . eq ( false ) )
. filter ( post ::removed . eq ( false ) ) ;
}
2023-09-18 13:44:48 +00:00
if options . community_id . is_none ( ) | | options . community_id_just_for_prefetch {
query = order_and_page_filter_desc ( query , post_aggregates ::featured_local , & options , | e | {
e . featured_local
} ) ;
} else {
query =
order_and_page_filter_desc ( query , post_aggregates ::featured_community , & options , | e | {
e . featured_community
} ) ;
}
if let Some ( community_id ) = options . community_id {
query = query . filter ( post_aggregates ::community_id . eq ( community_id ) ) ;
2023-04-25 23:28:06 +00:00
}
2023-07-28 08:36:50 +00:00
if let Some ( creator_id ) = options . creator_id {
2023-07-20 15:13:21 +00:00
query = query . filter ( post_aggregates ::creator_id . eq ( creator_id ) ) ;
2023-04-25 23:28:06 +00:00
}
2023-09-04 09:05:00 +00:00
if let ( Some ( listing_type ) , Some ( person_id ) ) = ( options . listing_type , person_id ) {
let is_subscribed = exists (
community_follower ::table . filter (
post_aggregates ::community_id
. eq ( community_follower ::community_id )
. and ( community_follower ::person_id . eq ( person_id ) ) ,
) ,
) ;
2022-02-18 02:30:47 +00:00
match listing_type {
2023-09-04 09:05:00 +00:00
ListingType ::Subscribed = > query = query . filter ( is_subscribed ) ,
2022-02-18 02:30:47 +00:00
ListingType ::Local = > {
2023-09-04 09:05:00 +00:00
query = query
. filter ( community ::local . eq ( true ) )
. filter ( community ::hidden . eq ( false ) . or ( is_subscribed ) ) ;
2022-02-18 02:30:47 +00:00
}
2023-09-04 09:05:00 +00:00
ListingType ::All = > query = query . filter ( community ::hidden . eq ( false ) . or ( is_subscribed ) ) ,
2023-08-31 11:07:45 +00:00
ListingType ::ModeratorView = > {
2023-09-04 09:05:00 +00:00
query = query . filter ( exists (
community_moderator ::table . filter (
post ::community_id
. eq ( community_moderator ::community_id )
. and ( community_moderator ::person_id . eq ( person_id ) ) ,
) ,
) ) ;
2023-08-31 11:07:45 +00:00
}
2022-02-18 02:30:47 +00:00
}
2020-12-10 20:53:49 +00:00
}
2022-07-29 10:57:39 +00:00
2023-09-18 13:44:48 +00:00
if let Some ( url_search ) = & options . url_search {
2020-12-10 20:53:49 +00:00
query = query . filter ( post ::url . eq ( url_search ) ) ;
}
2023-09-18 13:44:48 +00:00
if let Some ( search_term ) = & options . search_term {
let searcher = fuzzy_search ( search_term ) ;
2020-12-10 20:53:49 +00:00
query = query . filter (
post ::name
2022-11-19 04:33:54 +00:00
. ilike ( searcher . clone ( ) )
2020-12-10 20:53:49 +00:00
. or ( post ::body . ilike ( searcher ) ) ,
) ;
}
2023-07-28 08:36:50 +00:00
if ! options
2023-07-20 14:36:16 +00:00
. local_user
. map ( | l | l . local_user . show_nsfw )
. unwrap_or ( false )
{
2020-12-16 18:59:43 +00:00
query = query
. filter ( post ::nsfw . eq ( false ) )
. filter ( community ::nsfw . eq ( false ) ) ;
} ;
2023-07-28 08:36:50 +00:00
if ! options
2023-07-20 14:36:16 +00:00
. local_user
. map ( | l | l . local_user . show_bot_accounts )
. unwrap_or ( true )
{
2021-04-21 21:41:14 +00:00
query = query . filter ( person ::bot_account . eq ( false ) ) ;
} ;
2023-09-04 09:05:00 +00:00
if let ( true , Some ( person_id ) ) = ( options . saved_only , person_id ) {
query = query . filter ( is_saved ( person_id ) ) ;
2021-10-14 17:03:12 +00:00
}
// Only hide the read posts, if the saved_only is false. Otherwise ppl with the hide_read
// setting wont be able to see saved posts.
2023-07-28 08:36:50 +00:00
else if ! options
2023-07-20 14:36:16 +00:00
. local_user
. map ( | l | l . local_user . show_read_posts )
. unwrap_or ( true )
{
2023-08-02 16:31:28 +00:00
// Do not hide read posts when it is a user profile view
2023-09-04 09:05:00 +00:00
if let ( false , Some ( person_id ) ) = ( options . is_profile_view , person_id ) {
query = query . filter ( not ( is_read ( person_id ) ) ) ;
2023-08-02 16:31:28 +00:00
}
2021-10-14 17:03:12 +00:00
}
2020-12-16 18:59:43 +00:00
2023-09-04 09:05:00 +00:00
if let Some ( person_id ) = person_id {
if options . liked_only {
query = query . filter ( score ( person_id ) . eq ( 1 ) ) ;
} else if options . disliked_only {
query = query . filter ( score ( person_id ) . eq ( - 1 ) ) ;
}
} ;
2023-08-08 09:40:28 +00:00
2023-08-31 11:07:45 +00:00
// Dont filter blocks or missing languages for moderator view type
2023-09-04 09:05:00 +00:00
if let ( Some ( person_id ) , false ) = (
person_id ,
options . listing_type . unwrap_or_default ( ) = = ListingType ::ModeratorView ,
) {
2022-08-19 14:27:39 +00:00
// Filter out the rows with missing languages
2023-09-04 09:05:00 +00:00
query = query . filter ( exists (
local_user_language ::table . filter (
post ::language_id
. eq ( local_user_language ::language_id )
. and ( local_user_language ::local_user_id . eq ( local_user_id_join ) ) ,
) ,
) ) ;
2022-08-18 19:11:19 +00:00
2022-08-19 14:27:39 +00:00
// Don't show blocked communities or persons
2023-09-04 09:05:00 +00:00
query = query . filter ( not ( exists (
community_block ::table . filter (
post_aggregates ::community_id
. eq ( community_block ::community_id )
. and ( community_block ::person_id . eq ( person_id_join ) ) ,
) ,
) ) ) ;
query = query . filter ( not ( is_creator_blocked ( person_id ) ) ) ;
2021-08-19 20:54:15 +00:00
}
2023-08-24 15:27:00 +00:00
let now = diesel ::dsl ::now . into_sql ::< Timestamptz > ( ) ;
2021-08-19 20:54:15 +00:00
2023-09-18 13:44:48 +00:00
{
use post_aggregates ::{
comments ,
controversy_rank ,
hot_rank ,
hot_rank_active ,
published ,
scaled_rank ,
score ,
} ;
match options . sort . as_ref ( ) . unwrap_or ( & SortType ::Hot ) {
SortType ::Active = > {
query =
order_and_page_filter_desc ( query , hot_rank_active , & options , | e | e . hot_rank_active ) ;
query = order_and_page_filter_desc ( query , published , & options , | e | e . published ) ;
}
SortType ::Hot = > {
query = order_and_page_filter_desc ( query , hot_rank , & options , | e | e . hot_rank ) ;
query = order_and_page_filter_desc ( query , published , & options , | e | e . published ) ;
}
SortType ::Scaled = > {
query = order_and_page_filter_desc ( query , scaled_rank , & options , | e | e . scaled_rank ) ;
query = order_and_page_filter_desc ( query , published , & options , | e | e . published ) ;
}
SortType ::Controversial = > {
query =
order_and_page_filter_desc ( query , controversy_rank , & options , | e | e . controversy_rank ) ;
query = order_and_page_filter_desc ( query , published , & options , | e | e . published ) ;
}
SortType ::New = > {
query = order_and_page_filter_desc ( query , published , & options , | e | e . published )
}
SortType ::Old = > {
query = order_and_page_filter_asc ( query , published , & options , | e | e . published )
}
SortType ::NewComments = > {
query = order_and_page_filter_desc ( query , newest_comment_time , & options , | e | {
e . newest_comment_time
} )
}
SortType ::MostComments = > {
query = order_and_page_filter_desc ( query , comments , & options , | e | e . comments ) ;
query = order_and_page_filter_desc ( query , published , & options , | e | e . published ) ;
}
SortType ::TopAll = > {
query = order_and_page_filter_desc ( query , score , & options , | e | e . score ) ;
query = order_and_page_filter_desc ( query , published , & options , | e | e . published ) ;
}
o @ ( SortType ::TopYear
| SortType ::TopMonth
| SortType ::TopWeek
| SortType ::TopDay
| SortType ::TopHour
| SortType ::TopSixHour
| SortType ::TopTwelveHour
| SortType ::TopThreeMonths
| SortType ::TopSixMonths
| SortType ::TopNineMonths ) = > {
let interval = match o {
SortType ::TopYear = > 1. years ( ) ,
SortType ::TopMonth = > 1. months ( ) ,
SortType ::TopWeek = > 1. weeks ( ) ,
SortType ::TopDay = > 1. days ( ) ,
SortType ::TopHour = > 1. hours ( ) ,
SortType ::TopSixHour = > 6. hours ( ) ,
SortType ::TopTwelveHour = > 12. hours ( ) ,
SortType ::TopThreeMonths = > 3. months ( ) ,
SortType ::TopSixMonths = > 6. months ( ) ,
SortType ::TopNineMonths = > 9. months ( ) ,
_ = > return Err ( Error ::NotFound ) ,
} ;
query = query . filter ( post_aggregates ::published . gt ( now - interval ) ) ;
query = order_and_page_filter_desc ( query , score , & options , | e | e . score ) ;
query = order_and_page_filter_desc ( query , published , & options , | e | e . published ) ;
}
}
2020-12-10 20:53:49 +00:00
} ;
2023-09-18 13:44:48 +00:00
let ( limit , mut offset ) = limit_and_offset ( options . page , options . limit ) ? ;
if options . page_after . is_some ( ) {
// always skip exactly one post because that's the last post of the previous page
// fixing the where clause is more difficult because we'd have to change only the last order-by-where clause
// e.g. WHERE (featured_local<=, hot_rank<=, published<=) to WHERE (<=, <=, <)
offset = 1 ;
}
2023-03-01 03:46:15 +00:00
query = query . limit ( limit ) . offset ( offset ) ;
2021-01-06 18:23:05 +00:00
debug! ( " Post View Query: {:?} " , debug_query ::< Pg , _ > ( & query ) ) ;
2023-08-31 13:26:10 +00:00
query . load ::< PostView > ( & mut conn ) . await
2023-07-28 08:36:50 +00:00
} ;
Queries ::new ( read , list )
}
2020-12-10 20:53:49 +00:00
2023-07-28 08:36:50 +00:00
impl PostView {
pub async fn read (
pool : & mut DbPool < '_ > ,
post_id : PostId ,
my_person_id : Option < PersonId > ,
2023-08-11 09:13:14 +00:00
is_mod_or_admin : bool ,
2023-07-28 08:36:50 +00:00
) -> Result < Self , Error > {
let mut res = queries ( )
. read ( pool , ( post_id , my_person_id , is_mod_or_admin ) )
. await ? ;
// If a person is given, then my_vote, if None, should be 0, not null
// Necessary to differentiate between other person's votes
if my_person_id . is_some ( ) & & res . my_vote . is_none ( ) {
res . my_vote = Some ( 0 )
} ;
Ok ( res )
}
}
2023-09-18 13:44:48 +00:00
impl PaginationCursor {
// get cursor for page that starts immediately after the given post
pub fn after_post ( view : & PostView ) -> PaginationCursor {
// hex encoding to prevent ossification
PaginationCursor ( format! ( " P {:x} " , view . counts . post_id . 0 ) )
}
pub async fn read ( & self , pool : & mut DbPool < '_ > ) -> Result < PaginationCursorData , Error > {
Ok ( PaginationCursorData (
PostAggregates ::read (
pool ,
PostId (
self
. 0
. get ( 1 .. )
. and_then ( | e | i32 ::from_str_radix ( e , 16 ) . ok ( ) )
. ok_or_else ( | | Error ::QueryBuilderError ( " Could not parse pagination token " . into ( ) ) ) ? ,
) ,
)
. await ? ,
) )
}
}
// currently we use a postaggregates struct as the pagination token.
// we only use some of the properties of the post aggregates, depending on which sort type we page by
#[ derive(Clone) ]
pub struct PaginationCursorData ( PostAggregates ) ;
#[ derive(Default, Clone) ]
2023-07-28 08:36:50 +00:00
pub struct PostQuery < ' a > {
pub listing_type : Option < ListingType > ,
pub sort : Option < SortType > ,
pub creator_id : Option < PersonId > ,
pub community_id : Option < CommunityId > ,
2023-09-18 13:44:48 +00:00
// if true, the query should be handled as if community_id was not given except adding the literal filter
pub community_id_just_for_prefetch : bool ,
2023-07-28 08:36:50 +00:00
pub local_user : Option < & ' a LocalUserView > ,
pub search_term : Option < String > ,
pub url_search : Option < String > ,
2023-08-11 09:13:14 +00:00
pub saved_only : bool ,
pub liked_only : bool ,
pub disliked_only : bool ,
pub moderator_view : bool ,
2023-08-04 15:36:36 +00:00
pub is_profile_view : bool ,
2023-07-28 08:36:50 +00:00
pub page : Option < i64 > ,
pub limit : Option < i64 > ,
2023-09-18 13:44:48 +00:00
pub page_after : Option < PaginationCursorData > ,
pub page_before_or_equal : Option < PaginationCursorData > ,
2023-07-28 08:36:50 +00:00
}
impl < ' a > PostQuery < ' a > {
2023-09-18 13:44:48 +00:00
async fn prefetch_upper_bound_for_page_before (
& self ,
pool : & mut DbPool < '_ > ,
) -> Result < Option < PostQuery < ' a > > , Error > {
// first get one page for the most popular community to get an upper bound for the the page end for the real query
// the reason this is needed is that when fetching posts for a single community PostgreSQL can optimize
// the query to use an index on e.g. (=, >=, >=, >=) and fetch only LIMIT rows
// but for the followed-communities query it has to query the index on (IN, >=, >=, >=)
// which it currently can't do at all (as of PG 16). see the discussion here:
// https://github.com/LemmyNet/lemmy/issues/2877#issuecomment-1673597190
//
// the results are correct no matter which community we fetch these for, since it basically covers the "worst case" of the whole page consisting of posts from one community
// but using the largest community decreases the pagination-frame so make the real query more efficient.
use lemmy_db_schema ::schema ::{
community_aggregates ::dsl ::{ community_aggregates , community_id , users_active_month } ,
community_follower ::dsl ::{
community_follower ,
community_id as follower_community_id ,
person_id ,
} ,
} ;
let ( limit , offset ) = limit_and_offset ( self . page , self . limit ) ? ;
if offset ! = 0 {
return Err ( Error ::QueryBuilderError (
" legacy pagination cannot be combined with v2 pagination " . into ( ) ,
) ) ;
}
let self_person_id = self
. local_user
. expect ( " part of the above if " )
. local_user
. person_id ;
let largest_subscribed = {
let conn = & mut get_conn ( pool ) . await ? ;
community_follower
. filter ( person_id . eq ( self_person_id ) )
. inner_join ( community_aggregates . on ( community_id . eq ( follower_community_id ) ) )
. order_by ( users_active_month . desc ( ) )
. select ( community_id )
. limit ( 1 )
. get_result ::< CommunityId > ( conn )
. await
. optional ( ) ?
} ;
let Some ( largest_subscribed ) = largest_subscribed else {
// nothing subscribed to? no posts
return Ok ( None ) ;
} ;
let mut v = queries ( )
. list (
pool ,
PostQuery {
community_id : Some ( largest_subscribed ) ,
community_id_just_for_prefetch : true ,
.. self . clone ( )
} ,
)
. await ? ;
// take last element of array. if this query returned less than LIMIT elements,
// the heuristic is invalid since we can't guarantee the full query will return >= LIMIT results (return original query)
if ( v . len ( ) as i64 ) < limit {
Ok ( Some ( self . clone ( ) ) )
} else {
let page_before_or_equal = Some ( PaginationCursorData ( v . pop ( ) . expect ( " else case " ) . counts ) ) ;
Ok ( Some ( PostQuery {
page_before_or_equal ,
.. self . clone ( )
} ) )
}
}
2023-07-28 08:36:50 +00:00
pub async fn list ( self , pool : & mut DbPool < '_ > ) -> Result < Vec < PostView > , Error > {
2023-09-18 13:44:48 +00:00
if self . listing_type = = Some ( ListingType ::Subscribed )
& & self . community_id . is_none ( )
& & self . local_user . is_some ( )
& & self . page_before_or_equal . is_none ( )
{
if let Some ( query ) = self . prefetch_upper_bound_for_page_before ( pool ) . await ? {
queries ( ) . list ( pool , query ) . await
} else {
Ok ( vec! [ ] )
}
} else {
queries ( ) . list ( pool , self ) . await
}
2020-12-10 20:53:49 +00:00
}
}
2020-12-11 15:27:33 +00:00
#[ cfg(test) ]
mod tests {
2023-07-17 15:04:14 +00:00
#![ allow(clippy::unwrap_used) ]
#![ allow(clippy::indexing_slicing) ]
2023-07-20 14:36:16 +00:00
use crate ::{
post_view ::{ PostQuery , PostView } ,
structs ::LocalUserView ,
} ;
2021-10-16 13:33:38 +00:00
use lemmy_db_schema ::{
2022-05-03 17:44:13 +00:00
aggregates ::structs ::PostAggregates ,
2023-03-01 00:49:31 +00:00
impls ::actor_language ::UNDETERMINED_ID ,
2022-08-18 19:11:19 +00:00
newtypes ::LanguageId ,
2021-10-16 13:33:38 +00:00
source ::{
2022-10-06 18:27:58 +00:00
actor_language ::LocalUserLanguage ,
2023-03-01 17:19:46 +00:00
community ::{ Community , CommunityInsertForm } ,
2021-10-16 13:33:38 +00:00
community_block ::{ CommunityBlock , CommunityBlockForm } ,
2022-10-27 09:24:07 +00:00
instance ::Instance ,
2022-08-18 19:11:19 +00:00
language ::Language ,
2022-10-27 09:24:07 +00:00
local_user ::{ LocalUser , LocalUserInsertForm , LocalUserUpdateForm } ,
2023-03-01 17:19:46 +00:00
person ::{ Person , PersonInsertForm } ,
2021-10-16 13:33:38 +00:00
person_block ::{ PersonBlock , PersonBlockForm } ,
2023-06-23 10:53:46 +00:00
post ::{ Post , PostInsertForm , PostLike , PostLikeForm , PostUpdateForm } ,
2021-10-16 13:33:38 +00:00
} ,
traits ::{ Blockable , Crud , Likeable } ,
2022-11-09 10:05:00 +00:00
utils ::{ build_db_pool_for_tests , DbPool } ,
2022-05-06 20:55:07 +00:00
SortType ,
2022-06-22 12:05:41 +00:00
SubscribedType ,
2020-12-11 15:27:33 +00:00
} ;
2021-02-25 19:43:39 +00:00
use serial_test ::serial ;
2020-12-11 15:27:33 +00:00
2022-08-18 19:11:19 +00:00
struct Data {
2022-10-27 09:24:07 +00:00
inserted_instance : Instance ,
2023-07-20 14:36:16 +00:00
local_user_view : LocalUserView ,
2022-08-18 19:11:19 +00:00
inserted_blocked_person : Person ,
inserted_bot : Person ,
inserted_community : Community ,
inserted_post : Post ,
}
2020-12-11 15:27:33 +00:00
2023-07-11 13:09:59 +00:00
async fn init_data ( pool : & mut DbPool < '_ > ) -> Data {
2023-03-01 02:36:57 +00:00
let inserted_instance = Instance ::read_or_create ( pool , " my_domain.tld " . to_string ( ) )
. await
. unwrap ( ) ;
2022-10-27 09:24:07 +00:00
2021-03-10 22:33:55 +00:00
let person_name = " tegan " . to_string ( ) ;
2020-12-11 15:27:33 +00:00
2022-10-27 09:24:07 +00:00
let new_person = PersonInsertForm ::builder ( )
2022-11-19 04:33:54 +00:00
. name ( person_name . clone ( ) )
2022-10-27 09:24:07 +00:00
. public_key ( " pubkey " . to_string ( ) )
. instance_id ( inserted_instance . id )
. build ( ) ;
2020-12-11 15:27:33 +00:00
2022-11-09 10:05:00 +00:00
let inserted_person = Person ::create ( pool , & new_person ) . await . unwrap ( ) ;
2020-12-11 15:27:33 +00:00
2022-10-27 09:24:07 +00:00
let local_user_form = LocalUserInsertForm ::builder ( )
. person_id ( inserted_person . id )
2022-11-19 04:33:54 +00:00
. password_encrypted ( String ::new ( ) )
2022-10-27 09:24:07 +00:00
. build ( ) ;
2022-11-09 10:05:00 +00:00
let inserted_local_user = LocalUser ::create ( pool , & local_user_form ) . await . unwrap ( ) ;
2022-08-19 14:27:39 +00:00
2022-10-27 09:24:07 +00:00
let new_bot = PersonInsertForm ::builder ( )
. name ( " mybot " . to_string ( ) )
. bot_account ( Some ( true ) )
. public_key ( " pubkey " . to_string ( ) )
. instance_id ( inserted_instance . id )
. build ( ) ;
2021-04-21 21:41:14 +00:00
2022-11-09 10:05:00 +00:00
let inserted_bot = Person ::create ( pool , & new_bot ) . await . unwrap ( ) ;
2021-04-21 21:41:14 +00:00
2022-10-27 09:24:07 +00:00
let new_community = CommunityInsertForm ::builder ( )
. name ( " test_community_3 " . to_string ( ) )
. title ( " nada " . to_owned ( ) )
. public_key ( " pubkey " . to_string ( ) )
. instance_id ( inserted_instance . id )
. build ( ) ;
2020-12-11 15:27:33 +00:00
2022-11-09 10:05:00 +00:00
let inserted_community = Community ::create ( pool , & new_community ) . await . unwrap ( ) ;
2020-12-11 15:27:33 +00:00
2021-08-19 20:54:15 +00:00
// Test a person block, make sure the post query doesn't include their post
2022-10-27 09:24:07 +00:00
let blocked_person = PersonInsertForm ::builder ( )
. name ( person_name )
. public_key ( " pubkey " . to_string ( ) )
. instance_id ( inserted_instance . id )
. build ( ) ;
2021-08-19 20:54:15 +00:00
2022-11-09 10:05:00 +00:00
let inserted_blocked_person = Person ::create ( pool , & blocked_person ) . await . unwrap ( ) ;
2021-08-19 20:54:15 +00:00
2022-10-27 09:24:07 +00:00
let post_from_blocked_person = PostInsertForm ::builder ( )
. name ( " blocked_person_post " . to_string ( ) )
. creator_id ( inserted_blocked_person . id )
. community_id ( inserted_community . id )
. language_id ( Some ( LanguageId ( 1 ) ) )
. build ( ) ;
2021-08-19 20:54:15 +00:00
2022-11-09 10:05:00 +00:00
Post ::create ( pool , & post_from_blocked_person ) . await . unwrap ( ) ;
2021-08-19 20:54:15 +00:00
// block that person
let person_block = PersonBlockForm {
person_id : inserted_person . id ,
target_id : inserted_blocked_person . id ,
} ;
2022-11-09 10:05:00 +00:00
PersonBlock ::block ( pool , & person_block ) . await . unwrap ( ) ;
2021-08-19 20:54:15 +00:00
// A sample post
2022-10-27 09:24:07 +00:00
let new_post = PostInsertForm ::builder ( )
. name ( " test post 3 " . to_string ( ) )
. creator_id ( inserted_person . id )
. community_id ( inserted_community . id )
. language_id ( Some ( LanguageId ( 47 ) ) )
. build ( ) ;
2020-12-11 15:27:33 +00:00
2022-11-09 10:05:00 +00:00
let inserted_post = Post ::create ( pool , & new_post ) . await . unwrap ( ) ;
2020-12-11 15:27:33 +00:00
2022-10-27 09:24:07 +00:00
let new_bot_post = PostInsertForm ::builder ( )
. name ( " test bot post " . to_string ( ) )
. creator_id ( inserted_bot . id )
. community_id ( inserted_community . id )
. build ( ) ;
2021-04-21 21:41:14 +00:00
2022-11-09 10:05:00 +00:00
let _inserted_bot_post = Post ::create ( pool , & new_bot_post ) . await . unwrap ( ) ;
2023-07-20 14:36:16 +00:00
let local_user_view = LocalUserView {
local_user : inserted_local_user ,
person : inserted_person ,
counts : Default ::default ( ) ,
} ;
2020-12-11 15:27:33 +00:00
2022-08-18 19:11:19 +00:00
Data {
2022-10-27 09:24:07 +00:00
inserted_instance ,
2023-07-20 14:36:16 +00:00
local_user_view ,
2022-08-18 19:11:19 +00:00
inserted_blocked_person ,
inserted_bot ,
inserted_community ,
inserted_post ,
}
}
2020-12-11 15:27:33 +00:00
2022-11-09 10:05:00 +00:00
#[ tokio::test ]
2022-08-18 19:11:19 +00:00
#[ serial ]
2022-11-09 10:05:00 +00:00
async fn post_listing_with_person ( ) {
let pool = & build_db_pool_for_tests ( ) . await ;
2023-07-11 13:09:59 +00:00
let pool = & mut pool . into ( ) ;
2023-07-20 14:36:16 +00:00
let mut data = init_data ( pool ) . await ;
2021-08-19 20:54:15 +00:00
2023-08-08 09:41:41 +00:00
let local_user_form = LocalUserUpdateForm {
show_bot_accounts : Some ( false ) ,
.. Default ::default ( )
} ;
2022-08-19 14:27:39 +00:00
let inserted_local_user =
2023-07-20 14:36:16 +00:00
LocalUser ::update ( pool , data . local_user_view . local_user . id , & local_user_form )
2022-11-09 10:05:00 +00:00
. await
. unwrap ( ) ;
2023-07-20 14:36:16 +00:00
data . local_user_view . local_user = inserted_local_user ;
2022-08-19 14:27:39 +00:00
2023-07-17 10:20:25 +00:00
let read_post_listing = PostQuery {
sort : ( Some ( SortType ::New ) ) ,
community_id : ( Some ( data . inserted_community . id ) ) ,
2023-07-20 14:36:16 +00:00
local_user : ( Some ( & data . local_user_view ) ) ,
2023-07-17 10:20:25 +00:00
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
2020-12-11 15:27:33 +00:00
2023-03-01 03:46:15 +00:00
let post_listing_single_with_person = PostView ::read (
pool ,
data . inserted_post . id ,
2023-07-20 14:36:16 +00:00
Some ( data . local_user_view . person . id ) ,
2023-08-11 09:13:14 +00:00
false ,
2023-03-01 03:46:15 +00:00
)
. await
. unwrap ( ) ;
2020-12-11 15:27:33 +00:00
2022-11-09 10:05:00 +00:00
let mut expected_post_listing_with_user = expected_post_view ( & data , pool ) . await ;
2020-12-11 15:27:33 +00:00
2022-08-18 19:11:19 +00:00
// Should be only one person, IE the bot post, and blocked should be missing
assert_eq! ( 1 , read_post_listing . len ( ) ) ;
assert_eq! ( expected_post_listing_with_user , read_post_listing [ 0 ] ) ;
expected_post_listing_with_user . my_vote = Some ( 0 ) ;
2021-03-11 04:43:11 +00:00
assert_eq! (
expected_post_listing_with_user ,
2022-08-18 19:11:19 +00:00
post_listing_single_with_person
2021-03-11 04:43:11 +00:00
) ;
2021-04-21 21:41:14 +00:00
2023-08-08 09:41:41 +00:00
let local_user_form = LocalUserUpdateForm {
show_bot_accounts : Some ( true ) ,
.. Default ::default ( )
} ;
2022-08-19 14:27:39 +00:00
let inserted_local_user =
2023-07-20 14:36:16 +00:00
LocalUser ::update ( pool , data . local_user_view . local_user . id , & local_user_form )
2022-11-09 10:05:00 +00:00
. await
. unwrap ( ) ;
2023-07-20 14:36:16 +00:00
data . local_user_view . local_user = inserted_local_user ;
2022-08-19 14:27:39 +00:00
2023-07-17 10:20:25 +00:00
let post_listings_with_bots = PostQuery {
sort : ( Some ( SortType ::New ) ) ,
community_id : ( Some ( data . inserted_community . id ) ) ,
2023-07-20 14:36:16 +00:00
local_user : ( Some ( & data . local_user_view ) ) ,
2023-07-17 10:20:25 +00:00
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
2022-08-18 19:11:19 +00:00
// should include bot post which has "undetermined" language
assert_eq! ( 2 , post_listings_with_bots . len ( ) ) ;
2022-11-09 10:05:00 +00:00
cleanup ( data , pool ) . await ;
2022-08-18 19:11:19 +00:00
}
2022-11-09 10:05:00 +00:00
#[ tokio::test ]
2022-08-18 19:11:19 +00:00
#[ serial ]
2022-11-09 10:05:00 +00:00
async fn post_listing_no_person ( ) {
let pool = & build_db_pool_for_tests ( ) . await ;
2023-07-11 13:09:59 +00:00
let pool = & mut pool . into ( ) ;
2022-11-09 10:05:00 +00:00
let data = init_data ( pool ) . await ;
2022-08-18 19:11:19 +00:00
2023-07-17 10:20:25 +00:00
let read_post_listing_multiple_no_person = PostQuery {
sort : ( Some ( SortType ::New ) ) ,
community_id : ( Some ( data . inserted_community . id ) ) ,
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
2022-08-18 19:11:19 +00:00
2023-03-01 03:46:15 +00:00
let read_post_listing_single_no_person =
2023-08-11 09:13:14 +00:00
PostView ::read ( pool , data . inserted_post . id , None , false )
2023-03-01 03:46:15 +00:00
. await
. unwrap ( ) ;
2020-12-11 15:27:33 +00:00
2022-11-09 10:05:00 +00:00
let expected_post_listing_no_person = expected_post_view ( & data , pool ) . await ;
2022-08-18 19:11:19 +00:00
// Should be 2 posts, with the bot post, and the blocked
assert_eq! ( 3 , read_post_listing_multiple_no_person . len ( ) ) ;
assert_eq! (
expected_post_listing_no_person ,
read_post_listing_multiple_no_person [ 1 ]
) ;
2021-03-11 04:43:11 +00:00
assert_eq! (
expected_post_listing_no_person ,
2022-08-18 19:11:19 +00:00
read_post_listing_single_no_person
2021-03-11 04:43:11 +00:00
) ;
2020-12-11 15:27:33 +00:00
2022-11-09 10:05:00 +00:00
cleanup ( data , pool ) . await ;
2022-08-18 19:11:19 +00:00
}
2022-11-09 10:05:00 +00:00
#[ tokio::test ]
2022-08-18 19:11:19 +00:00
#[ serial ]
2022-11-09 10:05:00 +00:00
async fn post_listing_block_community ( ) {
let pool = & build_db_pool_for_tests ( ) . await ;
2023-07-11 13:09:59 +00:00
let pool = & mut pool . into ( ) ;
2022-11-09 10:05:00 +00:00
let data = init_data ( pool ) . await ;
2022-08-18 19:11:19 +00:00
let community_block = CommunityBlockForm {
2023-07-20 14:36:16 +00:00
person_id : data . local_user_view . person . id ,
2022-08-18 19:11:19 +00:00
community_id : data . inserted_community . id ,
} ;
2022-11-09 10:05:00 +00:00
CommunityBlock ::block ( pool , & community_block ) . await . unwrap ( ) ;
2021-08-19 20:54:15 +00:00
2023-07-17 10:20:25 +00:00
let read_post_listings_with_person_after_block = PostQuery {
sort : ( Some ( SortType ::New ) ) ,
community_id : ( Some ( data . inserted_community . id ) ) ,
2023-07-20 14:36:16 +00:00
local_user : ( Some ( & data . local_user_view ) ) ,
2023-07-17 10:20:25 +00:00
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
2021-08-19 20:54:15 +00:00
// Should be 0 posts after the community block
assert_eq! ( 0 , read_post_listings_with_person_after_block . len ( ) ) ;
2021-04-21 21:41:14 +00:00
2022-11-09 10:05:00 +00:00
CommunityBlock ::unblock ( pool , & community_block )
. await
. unwrap ( ) ;
cleanup ( data , pool ) . await ;
2022-08-18 19:11:19 +00:00
}
2022-11-09 10:05:00 +00:00
#[ tokio::test ]
2022-08-18 19:11:19 +00:00
#[ serial ]
2022-11-09 10:05:00 +00:00
async fn post_listing_like ( ) {
let pool = & build_db_pool_for_tests ( ) . await ;
2023-07-11 13:09:59 +00:00
let pool = & mut pool . into ( ) ;
2023-07-20 14:36:16 +00:00
let mut data = init_data ( pool ) . await ;
2022-08-18 19:11:19 +00:00
let post_like_form = PostLikeForm {
post_id : data . inserted_post . id ,
2023-07-20 14:36:16 +00:00
person_id : data . local_user_view . person . id ,
2022-08-18 19:11:19 +00:00
score : 1 ,
} ;
2022-11-09 10:05:00 +00:00
let inserted_post_like = PostLike ::like ( pool , & post_like_form ) . await . unwrap ( ) ;
2022-08-18 19:11:19 +00:00
let expected_post_like = PostLike {
id : inserted_post_like . id ,
post_id : data . inserted_post . id ,
2023-07-20 14:36:16 +00:00
person_id : data . local_user_view . person . id ,
2022-08-18 19:11:19 +00:00
published : inserted_post_like . published ,
score : 1 ,
} ;
2020-12-11 15:27:33 +00:00
assert_eq! ( expected_post_like , inserted_post_like ) ;
2022-08-18 19:11:19 +00:00
2023-06-21 08:48:39 +00:00
let post_listing_single_with_person = PostView ::read (
pool ,
data . inserted_post . id ,
2023-07-20 14:36:16 +00:00
Some ( data . local_user_view . person . id ) ,
2023-08-11 09:13:14 +00:00
false ,
2023-06-21 08:48:39 +00:00
)
. await
. unwrap ( ) ;
let mut expected_post_with_upvote = expected_post_view ( & data , pool ) . await ;
expected_post_with_upvote . my_vote = Some ( 1 ) ;
expected_post_with_upvote . counts . score = 1 ;
expected_post_with_upvote . counts . upvotes = 1 ;
assert_eq! ( expected_post_with_upvote , post_listing_single_with_person ) ;
2023-08-08 09:41:41 +00:00
let local_user_form = LocalUserUpdateForm {
show_bot_accounts : Some ( false ) ,
.. Default ::default ( )
} ;
2023-06-21 08:48:39 +00:00
let inserted_local_user =
2023-07-20 14:36:16 +00:00
LocalUser ::update ( pool , data . local_user_view . local_user . id , & local_user_form )
2023-06-21 08:48:39 +00:00
. await
. unwrap ( ) ;
2023-07-20 14:36:16 +00:00
data . local_user_view . local_user = inserted_local_user ;
2023-06-21 08:48:39 +00:00
2023-07-17 10:20:25 +00:00
let read_post_listing = PostQuery {
sort : ( Some ( SortType ::New ) ) ,
community_id : ( Some ( data . inserted_community . id ) ) ,
2023-07-20 14:36:16 +00:00
local_user : ( Some ( & data . local_user_view ) ) ,
2023-07-17 10:20:25 +00:00
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
2023-06-21 08:48:39 +00:00
assert_eq! ( 1 , read_post_listing . len ( ) ) ;
assert_eq! ( expected_post_with_upvote , read_post_listing [ 0 ] ) ;
2023-08-08 09:40:28 +00:00
let read_liked_post_listing = PostQuery {
community_id : ( Some ( data . inserted_community . id ) ) ,
local_user : ( Some ( & data . local_user_view ) ) ,
2023-08-11 09:13:14 +00:00
liked_only : ( true ) ,
2023-08-08 09:40:28 +00:00
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
assert_eq! ( read_post_listing , read_liked_post_listing ) ;
let read_disliked_post_listing = PostQuery {
community_id : ( Some ( data . inserted_community . id ) ) ,
local_user : ( Some ( & data . local_user_view ) ) ,
2023-08-11 09:13:14 +00:00
disliked_only : ( true ) ,
2023-08-08 09:40:28 +00:00
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
assert! ( read_disliked_post_listing . is_empty ( ) ) ;
2023-07-20 14:36:16 +00:00
let like_removed =
PostLike ::remove ( pool , data . local_user_view . person . id , data . inserted_post . id )
. await
. unwrap ( ) ;
2020-12-11 15:27:33 +00:00
assert_eq! ( 1 , like_removed ) ;
2022-11-09 10:05:00 +00:00
cleanup ( data , pool ) . await ;
2022-08-18 19:11:19 +00:00
}
2022-11-09 10:05:00 +00:00
#[ tokio::test ]
2022-08-18 19:11:19 +00:00
#[ serial ]
2022-11-09 10:05:00 +00:00
async fn post_listing_person_language ( ) {
let pool = & build_db_pool_for_tests ( ) . await ;
2023-07-11 13:09:59 +00:00
let pool = & mut pool . into ( ) ;
2022-11-09 10:05:00 +00:00
let data = init_data ( pool ) . await ;
2022-08-18 19:11:19 +00:00
2023-02-05 05:38:08 +00:00
let spanish_id = Language ::read_id_from_code ( pool , Some ( " es " ) )
. await
. unwrap ( )
. unwrap ( ) ;
2022-10-27 09:24:07 +00:00
let post_spanish = PostInsertForm ::builder ( )
. name ( " asffgdsc " . to_string ( ) )
2023-07-20 14:36:16 +00:00
. creator_id ( data . local_user_view . person . id )
2022-10-27 09:24:07 +00:00
. community_id ( data . inserted_community . id )
. language_id ( Some ( spanish_id ) )
. build ( ) ;
2022-08-18 19:11:19 +00:00
2022-11-09 10:05:00 +00:00
Post ::create ( pool , & post_spanish ) . await . unwrap ( ) ;
2022-08-18 19:11:19 +00:00
2023-07-17 10:20:25 +00:00
let post_listings_all = PostQuery {
sort : ( Some ( SortType ::New ) ) ,
2023-07-20 14:36:16 +00:00
local_user : ( Some ( & data . local_user_view ) ) ,
2023-07-17 10:20:25 +00:00
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
2022-08-18 19:11:19 +00:00
// no language filters specified, all posts should be returned
2022-08-19 14:27:39 +00:00
assert_eq! ( 3 , post_listings_all . len ( ) ) ;
2022-08-18 19:11:19 +00:00
2023-02-05 05:38:08 +00:00
let french_id = Language ::read_id_from_code ( pool , Some ( " fr " ) )
. await
. unwrap ( )
. unwrap ( ) ;
2023-07-20 14:36:16 +00:00
LocalUserLanguage ::update ( pool , vec! [ french_id ] , data . local_user_view . local_user . id )
2022-11-09 10:05:00 +00:00
. await
. unwrap ( ) ;
2022-08-18 19:11:19 +00:00
2023-07-17 10:20:25 +00:00
let post_listing_french = PostQuery {
sort : ( Some ( SortType ::New ) ) ,
2023-07-20 14:36:16 +00:00
local_user : ( Some ( & data . local_user_view ) ) ,
2023-07-17 10:20:25 +00:00
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
2022-08-18 19:11:19 +00:00
2023-05-18 14:34:21 +00:00
// only one post in french and one undetermined should be returned
assert_eq! ( 2 , post_listing_french . len ( ) ) ;
assert! ( post_listing_french
. iter ( )
. any ( | p | p . post . language_id = = french_id ) ) ;
2022-08-18 19:11:19 +00:00
2022-10-06 18:27:58 +00:00
LocalUserLanguage ::update (
2022-11-09 10:05:00 +00:00
pool ,
2023-03-01 00:49:31 +00:00
vec! [ french_id , UNDETERMINED_ID ] ,
2023-07-20 14:36:16 +00:00
data . local_user_view . local_user . id ,
2022-08-18 19:11:19 +00:00
)
2022-11-09 10:05:00 +00:00
. await
2022-08-18 19:11:19 +00:00
. unwrap ( ) ;
2023-07-17 10:20:25 +00:00
let post_listings_french_und = PostQuery {
sort : ( Some ( SortType ::New ) ) ,
2023-07-20 14:36:16 +00:00
local_user : ( Some ( & data . local_user_view ) ) ,
2023-07-17 10:20:25 +00:00
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
2022-08-18 19:11:19 +00:00
// french post and undetermined language post should be returned
assert_eq! ( 2 , post_listings_french_und . len ( ) ) ;
assert_eq! (
2023-03-01 00:49:31 +00:00
UNDETERMINED_ID ,
2022-08-18 19:11:19 +00:00
post_listings_french_und [ 0 ] . post . language_id
) ;
assert_eq! ( french_id , post_listings_french_und [ 1 ] . post . language_id ) ;
2023-06-23 10:53:46 +00:00
cleanup ( data , pool ) . await ;
}
#[ tokio::test ]
#[ serial ]
2023-07-20 14:36:16 +00:00
async fn post_listings_removed ( ) {
2023-06-23 10:53:46 +00:00
let pool = & build_db_pool_for_tests ( ) . await ;
2023-07-11 13:09:59 +00:00
let pool = & mut pool . into ( ) ;
2023-07-20 14:36:16 +00:00
let mut data = init_data ( pool ) . await ;
2023-06-23 10:53:46 +00:00
2023-07-20 14:36:16 +00:00
// Remove the post
2023-06-23 10:53:46 +00:00
Post ::update (
pool ,
data . inserted_post . id ,
2023-08-08 09:41:41 +00:00
& PostUpdateForm {
removed : Some ( true ) ,
.. Default ::default ( )
} ,
2023-06-23 10:53:46 +00:00
)
. await
. unwrap ( ) ;
2023-07-20 14:36:16 +00:00
// Make sure you don't see the removed post in the results
2023-07-17 10:20:25 +00:00
let post_listings_no_admin = PostQuery {
2023-07-20 14:36:16 +00:00
sort : Some ( SortType ::New ) ,
local_user : Some ( & data . local_user_view ) ,
2023-07-17 10:20:25 +00:00
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
2023-06-23 10:53:46 +00:00
assert_eq! ( 1 , post_listings_no_admin . len ( ) ) ;
2023-07-20 14:36:16 +00:00
// Removed post is shown to admins on profile page
2023-08-24 09:40:08 +00:00
data . local_user_view . local_user . admin = true ;
2023-07-17 10:20:25 +00:00
let post_listings_is_admin = PostQuery {
2023-07-20 14:36:16 +00:00
sort : Some ( SortType ::New ) ,
local_user : Some ( & data . local_user_view ) ,
2023-08-04 15:36:36 +00:00
is_profile_view : true ,
2023-07-17 10:20:25 +00:00
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
2023-06-23 10:53:46 +00:00
assert_eq! ( 2 , post_listings_is_admin . len ( ) ) ;
2022-11-09 10:05:00 +00:00
cleanup ( data , pool ) . await ;
2020-12-11 15:27:33 +00:00
}
2022-08-22 20:55:10 +00:00
2023-07-20 14:36:16 +00:00
#[ tokio::test ]
#[ serial ]
async fn post_listings_deleted ( ) {
let pool = & build_db_pool_for_tests ( ) . await ;
let pool = & mut pool . into ( ) ;
let data = init_data ( pool ) . await ;
// Delete the post
Post ::update (
pool ,
data . inserted_post . id ,
2023-08-08 09:41:41 +00:00
& PostUpdateForm {
deleted : Some ( true ) ,
.. Default ::default ( )
} ,
2023-07-20 14:36:16 +00:00
)
. await
. unwrap ( ) ;
// Make sure you don't see the deleted post in the results
let post_listings_no_creator = PostQuery {
sort : Some ( SortType ::New ) ,
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
let not_contains_deleted = post_listings_no_creator
. iter ( )
. map ( | p | p . post . id )
. all ( | p | p ! = data . inserted_post . id ) ;
assert! ( not_contains_deleted ) ;
// Deleted post is shown to creator
let post_listings_is_creator = PostQuery {
sort : Some ( SortType ::New ) ,
local_user : Some ( & data . local_user_view ) ,
.. Default ::default ( )
}
. list ( pool )
. await
. unwrap ( ) ;
let contains_deleted = post_listings_is_creator
. iter ( )
. map ( | p | p . post . id )
. any ( | p | p = = data . inserted_post . id ) ;
assert! ( contains_deleted ) ;
cleanup ( data , pool ) . await ;
}
2023-07-11 13:09:59 +00:00
async fn cleanup ( data : Data , pool : & mut DbPool < '_ > ) {
2022-11-09 10:05:00 +00:00
let num_deleted = Post ::delete ( pool , data . inserted_post . id ) . await . unwrap ( ) ;
Community ::delete ( pool , data . inserted_community . id )
. await
. unwrap ( ) ;
2023-07-20 14:36:16 +00:00
Person ::delete ( pool , data . local_user_view . person . id )
. await
. unwrap ( ) ;
2022-11-09 10:05:00 +00:00
Person ::delete ( pool , data . inserted_bot . id ) . await . unwrap ( ) ;
Person ::delete ( pool , data . inserted_blocked_person . id )
. await
. unwrap ( ) ;
Instance ::delete ( pool , data . inserted_instance . id )
. await
. unwrap ( ) ;
2022-08-22 20:55:10 +00:00
assert_eq! ( 1 , num_deleted ) ;
}
2023-07-11 13:09:59 +00:00
async fn expected_post_view ( data : & Data , pool : & mut DbPool < '_ > ) -> PostView {
2022-08-22 20:55:10 +00:00
let ( inserted_person , inserted_community , inserted_post ) = (
2023-07-20 14:36:16 +00:00
& data . local_user_view . person ,
2022-08-22 20:55:10 +00:00
& data . inserted_community ,
& data . inserted_post ,
) ;
2022-11-09 10:05:00 +00:00
let agg = PostAggregates ::read ( pool , inserted_post . id ) . await . unwrap ( ) ;
2022-08-22 20:55:10 +00:00
PostView {
post : Post {
id : inserted_post . id ,
name : inserted_post . name . clone ( ) ,
creator_id : inserted_person . id ,
url : None ,
body : None ,
published : inserted_post . published ,
updated : None ,
community_id : inserted_community . id ,
removed : false ,
deleted : false ,
locked : false ,
nsfw : false ,
embed_title : None ,
embed_description : None ,
embed_video_url : None ,
thumbnail_url : None ,
2022-11-19 04:33:54 +00:00
ap_id : inserted_post . ap_id . clone ( ) ,
2022-08-22 20:55:10 +00:00
local : true ,
language_id : LanguageId ( 47 ) ,
2022-12-12 11:17:10 +00:00
featured_community : false ,
featured_local : false ,
2022-08-22 20:55:10 +00:00
} ,
my_vote : None ,
2022-09-27 16:45:46 +00:00
unread_comments : 0 ,
2023-03-01 17:19:46 +00:00
creator : Person {
2022-08-22 20:55:10 +00:00
id : inserted_person . id ,
name : inserted_person . name . clone ( ) ,
display_name : None ,
published : inserted_person . published ,
avatar : None ,
2022-11-19 04:33:54 +00:00
actor_id : inserted_person . actor_id . clone ( ) ,
2022-08-22 20:55:10 +00:00
local : true ,
bot_account : false ,
banned : false ,
deleted : false ,
bio : None ,
banner : None ,
updated : None ,
2022-11-19 04:33:54 +00:00
inbox_url : inserted_person . inbox_url . clone ( ) ,
2022-08-22 20:55:10 +00:00
shared_inbox_url : None ,
matrix_user_id : None ,
ban_expires : None ,
2022-10-27 09:24:07 +00:00
instance_id : data . inserted_instance . id ,
2023-03-01 17:19:46 +00:00
private_key : inserted_person . private_key . clone ( ) ,
public_key : inserted_person . public_key . clone ( ) ,
last_refreshed_at : inserted_person . last_refreshed_at ,
2022-08-22 20:55:10 +00:00
} ,
creator_banned_from_community : false ,
2023-03-01 17:19:46 +00:00
community : Community {
2022-08-22 20:55:10 +00:00
id : inserted_community . id ,
name : inserted_community . name . clone ( ) ,
icon : None ,
removed : false ,
deleted : false ,
nsfw : false ,
2022-11-19 04:33:54 +00:00
actor_id : inserted_community . actor_id . clone ( ) ,
2022-08-22 20:55:10 +00:00
local : true ,
title : " nada " . to_owned ( ) ,
description : None ,
updated : None ,
banner : None ,
hidden : false ,
posting_restricted_to_mods : false ,
published : inserted_community . published ,
2022-10-27 09:24:07 +00:00
instance_id : data . inserted_instance . id ,
2023-03-01 17:19:46 +00:00
private_key : inserted_community . private_key . clone ( ) ,
public_key : inserted_community . public_key . clone ( ) ,
last_refreshed_at : inserted_community . last_refreshed_at ,
followers_url : inserted_community . followers_url . clone ( ) ,
inbox_url : inserted_community . inbox_url . clone ( ) ,
shared_inbox_url : inserted_community . shared_inbox_url . clone ( ) ,
moderators_url : inserted_community . moderators_url . clone ( ) ,
featured_url : inserted_community . featured_url . clone ( ) ,
2022-08-22 20:55:10 +00:00
} ,
counts : PostAggregates {
id : agg . id ,
post_id : inserted_post . id ,
comments : 0 ,
score : 0 ,
upvotes : 0 ,
downvotes : 0 ,
published : agg . published ,
newest_comment_time_necro : inserted_post . published ,
newest_comment_time : inserted_post . published ,
2022-12-12 11:17:10 +00:00
featured_community : false ,
featured_local : false ,
2023-09-06 17:43:27 +00:00
hot_rank : 0.1728 ,
hot_rank_active : 0.1728 ,
2023-07-26 17:07:05 +00:00
controversy_rank : 0.0 ,
2023-09-06 17:43:27 +00:00
scaled_rank : 0.3621 ,
2023-07-20 15:13:21 +00:00
community_id : inserted_post . community_id ,
creator_id : inserted_post . creator_id ,
2022-08-22 20:55:10 +00:00
} ,
subscribed : SubscribedType ::NotSubscribed ,
read : false ,
saved : false ,
creator_blocked : false ,
}
}
2020-12-11 15:27:33 +00:00
}