67 lines
2.2 KiB
C#
67 lines
2.2 KiB
C#
 
|
using MongoDB.Entities;
|
||
|
|
|
||
|
|
using PrivaPub.Federation.Actors;
|
||
|
|
using PrivaPub.Models.Social;
|
||
|
|
using PrivaPub.Models.User;
|
||
|
|
using PrivaPub.StaticServices;
|
||
|
|
|
||
|
|
using System.Text.Json.Nodes;
|
||
|
|
|
||
|
|
using static PrivaPub.Federation.Objects.ActivityJson;
|
||
|
|
|
||
|
|
namespace PrivaPub.Federation.Inbox.Handlers
|
||
|
|
{
|
||
|
|
public class AcceptHandler : IActivityHandler
|
||
|
|
{
|
||
|
|
readonly DbEntities _dbEntities;
|
||
|
|
readonly ILocalActorService _localActors;
|
||
|
|
|
||
|
|
public AcceptHandler(DbEntities dbEntities, ILocalActorService localActors)
|
||
|
|
{
|
||
|
|
_dbEntities = dbEntities;
|
||
|
|
_localActors = localActors;
|
||
|
|
}
|
||
|
|
|
||
|
|
public virtual string Type => "Accept";
|
||
|
|
|
||
|
|
public async Task Handle(JsonNode activity, ForeignAvatar actor, CancellationToken token)
|
||
|
|
{
|
||
|
|
var following = await FindFollowing(activity["object"], actor, _dbEntities, _localActors, token);
|
||
|
|
if (following == default)
|
||
|
|
return;
|
||
|
|
await Apply(following, token);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected virtual Task Apply(Following following, CancellationToken token) =>
|
||
|
|
DB.Default.Update<Following>().MatchID(following.ID).Modify(f => f.State, FollowState.Accepted).ExecuteAsync(token);
|
||
|
|
|
||
|
|
public static async Task<Following> FindFollowing(JsonNode follow, ForeignAvatar target, DbEntities dbEntities, ILocalActorService localActors,
|
||
|
|
CancellationToken token)
|
||
|
|
{
|
||
|
|
if (follow is JsonObject && Value(follow, "type") is { } type && type != "Follow")
|
||
|
|
return default;
|
||
|
|
var followId = Id(follow);
|
||
|
|
var byId = followId == default
|
||
|
|
? default
|
||
|
|
: await dbEntities.Followings.Match(f => f.FollowActivityURI == followId && f.TargetActorURI == target.ActorURI).ExecuteFirstAsync(token);
|
||
|
|
if (byId != default || follow is not JsonObject)
|
||
|
|
return byId;
|
||
|
|
|
||
|
|
var follower = await localActors.FindByUri(Id(follow["actor"]), token);
|
||
|
|
return follower == default
|
||
|
|
? default
|
||
|
|
: await dbEntities.Followings.Match(f => f.AvatarId == follower.Id && f.TargetActorURI == target.ActorURI).ExecuteFirstAsync(token);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public class RejectHandler : AcceptHandler
|
||
|
|
{
|
||
|
|
public RejectHandler(DbEntities dbEntities, ILocalActorService localActors) : base(dbEntities, localActors)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
public override string Type => "Reject";
|
||
|
|
|
||
|
|
protected override Task Apply(Following following, CancellationToken token) => DB.Default.DeleteAsync<Following>(following.ID);
|
||
|
|
}
|
||
|
|
}
|