[Milky] Support suppressing self-generated message events

Author: NoirHare

Lagrange.Milky/Configurations/MilkyConfiguration.cs

@@ -12,8 +12,9 @@ public class MilkyConfiguration(MilkyHttpServerConfiguration httpServer, string?
     public MilkyEventConfiguration Event => Event_;
 }
 
-public class MilkyEventConfiguration(MilkyWebSocketEventConfiguration? webSocket = null, MilkySSEEventConfiguration? sse = null, MilkyWebHookEventConfiguration? webHook = null)
+public class MilkyEventConfiguration(bool suppressSelfMessageEvents = false, MilkyWebSocketEventConfiguration? webSocket = null, MilkySSEEventConfiguration? sse = null, MilkyWebHookEventConfiguration? webHook = null)
 {
+    public bool SuppressSelfMessageEvents { get; } = suppressSelfMessageEvents;
     public MilkyWebSocketEventConfiguration? WebSocket { get; } = webSocket;
     public MilkySSEEventConfiguration? SSE { get; } = sse;
     public MilkyWebHookEventConfiguration? WebHook { get; } = webHook;

Lagrange.Milky/Events/SSEEventHandler.cs

@@ -8,13 +8,15 @@
 using System.Threading.Tasks;
 using Lagrange.Core;
 using Lagrange.Core.Events;
+using Lagrange.Core.Events.EventArgs;
 using Lagrange.Milky.Configurations;
 using Lagrange.Milky.Events.Converters;
 using Lagrange.Milky.Events.Extensions;
 using Lagrange.Milky.Http;
 using Lagrange.Milky.Serialization;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
+using LogLevel = Microsoft.Extensions.Logging.LogLevel;
 
 namespace Lagrange.Milky.Events;
 
@@ -33,6 +35,7 @@ public sealed class SSEEventHandler : IHttpHandler, IGenericEventHandler, IDispo
     private readonly string? _token;
     private readonly string? _allowCorsOrigins;
     private readonly ulong _heartbeatIntervalSeconds;
+    private readonly bool _suppressSelfMessageEvents;
 
     private readonly ConcurrentDictionary<Guid, (HttpListenerResponse Response, SemaphoreSlim SendLock)> _connections = new();
     private readonly CancellationTokenSource _cts = new();
@@ -49,6 +52,7 @@ public SSEEventHandler(IServiceScopeFactory scopeFactory, ILogger<SSEEventHandle
         _token = configuration.AccessToken;
         _allowCorsOrigins = sseConfiguration.AllowCorsOrigins;
         _heartbeatIntervalSeconds = sseConfiguration.HeartbeatIntervalSeconds;
+        _suppressSelfMessageEvents = configuration.Event.SuppressSelfMessageEvents;
 
         _lagrange.RegisterConvertibleEvents(this);
     }
@@ -179,6 +183,8 @@ private async Task SendAllAsync(byte[] payload, CancellationToken ct = default)
 
     public async Task OnEvent<TEvent>(BotContext lagrange, TEvent @event) where TEvent : EventBase
     {
+        if (_suppressSelfMessageEvents && @event is BotMessageEvent message && message.Message.Contact.Uin == _lagrange.BotUin) return;
+
         await using var scope = _scopeFactory.CreateAsyncScope();
         var converter = scope.ServiceProvider.GetRequiredService<IEventConverter<TEvent>>();
         byte[] bytes = Serializer.JsonSerializeToUtf8Bytes(new MilkyEvent

Lagrange.Milky/Events/WebHookEventHandler.cs

@@ -6,6 +6,7 @@
 using System.Threading.Tasks;
 using Lagrange.Core;
 using Lagrange.Core.Events;
+using Lagrange.Core.Events.EventArgs;
 using Lagrange.Milky.Configurations;
 using Lagrange.Milky.Events.Converters;
 using Lagrange.Milky.Events.Extensions;
@@ -23,6 +24,7 @@ public class WebHookEventHandler(IServiceScopeFactory scopeFactory, ILogger<WebH
     private readonly BotContext _lagrange = lagrange;
 
     private readonly string? _token = configuration.AccessToken;
+    private readonly bool _suppressSelfMessageEvents = configuration.Event.SuppressSelfMessageEvents;
 
     private readonly string[] _targetUrls = webHookConfiguration.TargetUrls;
 
@@ -44,6 +46,8 @@ public Task StopAsync(CancellationToken ct)
 
     public async Task OnEvent<TEvent>(BotContext lagrange, TEvent @event) where TEvent : EventBase
     {
+        if (_suppressSelfMessageEvents && @event is BotMessageEvent message && message.Message.Contact.Uin == _lagrange.BotUin) return;
+        
         await using var scope = _scopeFactory.CreateAsyncScope();
         var converter = scope.ServiceProvider.GetRequiredService<IEventConverter<TEvent>>();
         byte[] bytes = Serializer.JsonSerializeToUtf8Bytes(new MilkyEvent

Lagrange.Milky/Events/WebSocketEventHandler.cs

@@ -15,6 +15,7 @@
 using Microsoft.Extensions.Logging;
 using LogLevel = Microsoft.Extensions.Logging.LogLevel;
 using Lagrange.Milky.Serialization;
+using Lagrange.Core.Events.EventArgs;
 
 namespace Lagrange.Milky.Events;
 
@@ -28,6 +29,7 @@ public sealed class WebSocketEventHandler : IHttpHandler, IGenericEventHandler,
     private readonly BotContext _lagrange;
 
     private readonly string? _token;
+    private readonly bool _suppressSelfMessageEvents;
 
     private readonly ConcurrentDictionary<Guid, (WebSocket WS, SemaphoreSlim SendLock)> _wss = new();
     private readonly CancellationTokenSource _cts = new();
@@ -162,6 +164,8 @@ private async Task SendAllAsync(byte[] payload, CancellationToken ct = default)
 
     public async Task OnEvent<TEvent>(BotContext lagrange, TEvent @event) where TEvent : EventBase
     {
+        if (_suppressSelfMessageEvents && @event is BotMessageEvent message && message.Message.Contact.Uin == _lagrange.BotUin) return;
+
         await using var scope = _scopeFactory.CreateAsyncScope();
         var converter = scope.ServiceProvider.GetRequiredService<IEventConverter<TEvent>>();
         byte[] bytes = Serializer.JsonSerializeToUtf8Bytes(new MilkyEvent

Lagrange.Milky/Resources/appsettings.json

@@ -40,6 +40,7 @@
             }
         },
         "Event": {
+            // "SuppressSelfMessageEvents": false
             "WebSocket": {
                 "Enabled": true
             }