notfire/Iceshrimp.NET
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[backend/core] Add ForceFollowerOnly rewrite policy

Browse source
This commit is contained in:
Laura Hausmann 2024-10-11 04:26:45 +02:00
parent 58d4dab21e
commit a7535cfcf8
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
1 changed files with 44 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
Iceshrimp.Backend/Core/Policies/ForceFollowerOnlyRewritePolicy.cs Normal file
View file

@ -0,0 +1,44 @@
using Iceshrimp.Backend.Core.Database.Tables;
using Iceshrimp.Backend.Core.Federation.ActivityStreams.Types;
using Iceshrimp.Backend.Core.Services;
namespace Iceshrimp.Backend.Core.Policies;
public class ForceFollowerOnlyRewritePolicy(bool enabled, int priority, string[] instances) : IRewritePolicy
{
public string Name => nameof(ForceFollowerOnlyRewritePolicy);
public bool Enabled => enabled;
public int Priority => priority;
public IRewritePolicy.HookLocationEnum HookLocation => IRewritePolicy.HookLocationEnum.PreLogic;
public void Apply(NoteService.NoteCreationData data)
{
if (IsApplicable(data))
data.Visibility = Note.NoteVisibility.Followers;
}
public void Apply(NoteService.NoteUpdateData data) { }
public void Apply(ASNote note, User actor) { }
private bool IsApplicable(NoteService.NoteCreationData data)
{
return instances.Contains(data.User.Host) && data.Visibility < Note.NoteVisibility.Followers;
}
}
public class ForceFollowerOnlyRewritePolicyConfiguration : IPolicyConfiguration<ForceFollowerOnlyRewritePolicy>
{
private string[] _instances = [];
public ForceFollowerOnlyRewritePolicy Apply() => new(Enabled, Priority, Instances);
IPolicy IPolicyConfiguration. Apply() => Apply();
public bool Enabled { get; set; }
public int Priority { get; set; }
public string[] Instances
{
get => _instances;
set => _instances = value.Select(p => p.ToLowerInvariant()).ToArray();
}
}