Skip to content

Commit b88dbd8

Browse files
committed
Skip banned posts
1 parent e8b0ccc commit b88dbd8

File tree

3 files changed

+10
-1
lines changed

3 files changed

+10
-1
lines changed

Source/WetPicsRebirth/Commands/ServiceCommands/Posting/PostNext.cs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,9 @@ await _postedMediaRepository.Add(
150150
}
151151

152152
private static bool ShouldSkip(LoadedPost loadedPost)
153-
=> loadedPost.Post.FileSize > 50_000_000 || loadedPost.Post.FileName.EndsWith(".webm");
153+
=> loadedPost.Post is BannedPost
154+
|| loadedPost.Post.FileSize > 50_000_000
155+
|| loadedPost.Post.FileName.EndsWith(".webm");
154156

155157
private async Task<(Message sentPost, string fileId, MediaType fileType)> SentPostToTelegram(
156158
Actress actress,

Source/WetPicsRebirth/Infrastructure/Engines/BooruEngine.cs

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,11 @@ async Task<LoadedPost> LoadPostCore()
5454
var post = await _loader.GetPostAsync(postHeader.Id);
5555
mediaUrl = GetMediaUrl(post);
5656

57+
if (string.IsNullOrWhiteSpace(mediaUrl))
58+
{
59+
return new LoadedPost(new BannedPost(postHeader));
60+
}
61+
5762
var loaded = IsUgoira(post)
5863
? await LoadPostFromUgoira(post.OriginalUrl!)
5964
: await LoadRegularPost(mediaUrl, post);

Source/WetPicsRebirth/Infrastructure/Models/Post.cs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,3 +15,5 @@ public record Post(PostHeader PostHeader, string Url, Stream File, long FileSize
1515
? $"<a href=\"https://yande.re/post/show/{PostHeader.Id}\">yande.re</a>"
1616
: $"<a href=\"https://danbooru.donmai.us/posts/{PostHeader.Id}\">danbooru.donmai.us</a>";
1717
}
18+
19+
public record BannedPost(PostHeader PostHeader) : Post(PostHeader, string.Empty, Stream.Null, 0, null);

0 commit comments

Comments
 (0)