SocialPub/PrivaPub/Federation/Inbox/Handlers/UndoHandler.cs

using MongoDB.Entities;

using PrivaPub.Domain.Social;
using PrivaPub.Federation.Actors;
using PrivaPub.Models.Federation;
using PrivaPub.Models.Social;
using PrivaPub.Models.User;
using PrivaPub.StaticServices;

using System.Text.Json.Nodes;

using static PrivaPub.Federation.Inbox.ForeignMembers;
using static PrivaPub.Federation.Objects.ActivityJson;

using PostEntity = PrivaPub.Models.Post.Post;

namespace PrivaPub.Federation.Inbox.Handlers
{
	public class UndoHandler : IActivityHandler
	{
		readonly IReactions _reactions;
		readonly DbEntities _dbEntities;
		readonly ILocalActorService _localActors;

		public UndoHandler(DbEntities dbEntities, ILocalActorService localActors, IReactions reactions)
		{
			_reactions = reactions;
			_dbEntities = dbEntities;
			_localActors = localActors;
		}

		public string Type => "Undo";

		public async Task Handle(JsonNode activity, ForeignAvatar actor, CancellationToken token)
		{
			var inner = activity["object"];
			var innerType = inner is JsonObject ? Value(inner, "type") : default;
			var innerId = Id(inner);

			Arrival.About(innerType);
			var undone = false;
			if (innerType is null or "Follow")
				undone |= await UndoFollow(inner, innerId, actor, token);
			if (innerType is null or "Like")
				undone |= await UndoLike(inner, innerId, actor, token);
			if (innerType is null or "Announce")
				undone |= await UndoAnnounce(innerId, actor, token);
			if (innerType is null or "Dislike")
				undone |= await UndoDislike(inner, innerId, actor, token);
			if (innerType is null or "Like" or "EmojiReact")
				undone |= await _reactions.Withdraw(actor, innerId, inner is JsonObject ? Id(inner["object"]) : default,
					inner is JsonObject ? Value(inner, "_misskey_reaction") ?? Value(inner, "content") : default, token);
			if (undone)
				Arrival.Accept("undone");
			else
				Arrival.Drop("unknown-object");
		}

		async Task<bool> UndoDislike(JsonNode inner, string innerId, ForeignAvatar actor, CancellationToken token)
		{
			var downvote = innerId == default
				? default
				: await DB.Default.Find<Downvote>().Match(d => d.ActivityURI == innerId && d.ActorURI == actor.ActorURI).ExecuteFirstAsync(token);
			if (downvote == default && inner is JsonObject && Id(inner["object"]) is { } objectUri)
			{
				var post = await _dbEntities.Posts.Match(p => p.ObjectURI == objectUri).ExecuteFirstAsync(token);
				downvote = post == default
					? default
					: await DB.Default.Find<Downvote>().Match(d => d.PostId == post.ID && d.ActorURI == actor.ActorURI).ExecuteFirstAsync(token);
			}
			if (downvote == default)
				return false;
			await DB.Default.DeleteAsync<Downvote>(downvote.ID);
			await DB.Default.Update<PostEntity>().MatchID(downvote.PostId).Modify(b => b.Inc(p => p.DownvotesCount, -1)).ExecuteAsync(token);
			return true;
		}

		async Task<bool> UndoFollow(JsonNode inner, string innerId, ForeignAvatar actor, CancellationToken token)
		{
			var followers = await _dbEntities.Followers.Match(f => f.ActorURI == actor.ActorURI).ExecuteAsync(token);
			var targetUri = inner is JsonObject ? Id(inner["object"]) : default;
			var target = targetUri == default ? default : await _localActors.FindByUri(targetUri, token);
			var undone = false;
			foreach (var follower in followers)
			{
				var matchesActivity = innerId != default && follower.FollowActivityURI == innerId;
				var matchesTarget = target != default && target.Id == follower.LocalActorId && target.Kind == follower.LocalActorKind;
				if (!matchesActivity && !matchesTarget)
					continue;

				await DB.Default.DeleteAsync<Follower>(follower.ID);
				if (follower.LocalActorKind == LocalActorKind.Group)
					await RemoveForeignMember(follower.LocalActorId, actor.ActorURI, token);
				undone = true;
			}
			return undone;
		}

		async Task<bool> UndoLike(JsonNode inner, string innerId, ForeignAvatar actor, CancellationToken token)
		{
			var favourite = innerId == default
				? default
				: await _dbEntities.Favourites.Match(f => f.ActivityURI == innerId && f.ActorURI == actor.ActorURI).ExecuteFirstAsync(token);
			if (favourite == default && inner is JsonObject && Id(inner["object"]) is { } objectUri)
			{
				var post = await _dbEntities.Posts.Match(p => p.ObjectURI == objectUri).ExecuteFirstAsync(token);
				favourite = post == default
					? default
					: await _dbEntities.Favourites.Match(f => f.PostId == post.ID && f.ActorURI == actor.ActorURI).ExecuteFirstAsync(token);
			}
			if (favourite == default)
				return false;
			await DB.Default.DeleteAsync<Favourite>(favourite.ID);
			await DB.Default.Update<PostEntity>().MatchID(favourite.PostId).Modify(b => b.Inc(p => p.FavouritesCount, -1)).ExecuteAsync(token);
			return true;
		}

		async Task<bool> UndoAnnounce(string innerId, ForeignAvatar actor, CancellationToken token)
		{
			if (innerId == default)
				return false;
			var reblog = await _dbEntities.Posts
				.Match(p => p.ObjectURI == innerId && p.ActorURI == actor.ActorURI && p.ReblogOfPostId != null)
				.ExecuteFirstAsync(token);
			if (reblog == default)
				return false;
			await DB.Default.DeleteAsync<PostEntity>(reblog.ID);
			await DB.Default.DeleteAsync<TimelineEntry>(e => e.PostId == reblog.ID);
			await DB.Default.Update<PostEntity>().MatchID(reblog.ReblogOfPostId).Modify(b => b.Inc(p => p.ReblogsCount, -1)).ExecuteAsync(token);
			return true;
		}
	}
}