From 6512474c24ff99b997f7d5f018fc0b10662bb138 Mon Sep 17 00:00:00 2001
From: Rory& <root@rory.gay>
Date: Sat, 26 Jul 2025 16:59:18 +0200
Subject: Add support for bulk event sending, back off on server failure

---
 LibMatrix/RoomTypes/GenericRoom.cs | 44 +++++++++++++++++++++++++++++++++++---
 1 file changed, 41 insertions(+), 3 deletions(-)

(limited to 'LibMatrix/RoomTypes/GenericRoom.cs')

diff --git a/LibMatrix/RoomTypes/GenericRoom.cs b/LibMatrix/RoomTypes/GenericRoom.cs
index c34dc01..2e50e69 100644
--- a/LibMatrix/RoomTypes/GenericRoom.cs
+++ b/LibMatrix/RoomTypes/GenericRoom.cs
@@ -406,9 +406,9 @@ public class GenericRoom {
         await (await Homeserver.ClientHttpClient.PutAsJsonAsync($"/_matrix/client/v3/rooms/{RoomId}/state/{eventType}", content))
             .Content.ReadFromJsonAsync<EventIdResponse>();
 
-    public async Task<EventIdResponse?> SendStateEventAsync(string eventType, string stateKey, object content) =>
-        await (await Homeserver.ClientHttpClient.PutAsJsonAsync($"/_matrix/client/v3/rooms/{RoomId}/state/{eventType.UrlEncode()}/{stateKey.UrlEncode()}", content))
-            .Content.ReadFromJsonAsync<EventIdResponse>();
+    public async Task<EventIdResponse> SendStateEventAsync(string eventType, string stateKey, object content) =>
+        (await (await Homeserver.ClientHttpClient.PutAsJsonAsync($"/_matrix/client/v3/rooms/{RoomId}/state/{eventType.UrlEncode()}/{stateKey.UrlEncode()}", content))
+            .Content.ReadFromJsonAsync<EventIdResponse>())!;
 
     public async Task<EventIdResponse> SendTimelineEventAsync(string eventType, TimelineEventContent content) {
         var res = await Homeserver.ClientHttpClient.PutAsJsonAsync(
@@ -418,6 +418,14 @@ public class GenericRoom {
         return await res.Content.ReadFromJsonAsync<EventIdResponse>() ?? throw new Exception("Failed to send event");
     }
 
+    public async Task<EventIdResponse> SendRawTimelineEventAsync(string eventType, JsonObject content) {
+        var res = await Homeserver.ClientHttpClient.PutAsJsonAsync(
+            $"/_matrix/client/v3/rooms/{RoomId}/send/{eventType}/" + Guid.NewGuid(), content, new JsonSerializerOptions {
+                DefaultIgnoreCondition = JsonIgnoreCondition.WhenWritingNull
+            });
+        return await res.Content.ReadFromJsonAsync<EventIdResponse>() ?? throw new Exception("Failed to send event");
+    }
+
     public async Task<EventIdResponse> SendReactionAsync(string eventId, string key) =>
         await SendTimelineEventAsync("m.reaction", new RoomMessageReactionEventContent() {
             RelatesTo = new() {
@@ -615,6 +623,36 @@ public class GenericRoom {
         }
     }
 
+    public async Task BulkSendEventsAsync(IEnumerable<StateEventResponse> events) {
+        if ((await Homeserver.GetCapabilitiesAsync()).Capabilities.BulkSendEvents?.Enabled == true)
+            await Homeserver.ClientHttpClient.PostAsJsonAsync(
+                $"/_matrix/client/unstable/gay.rory.bulk_send_events/rooms/{RoomId}/bulk_send_events?_libmatrix_txn_id={Guid.NewGuid()}", events);
+        else {
+            Console.WriteLine("Homeserver does not support bulk sending events, falling back to individual sends.");
+            foreach (var evt in events)
+                await (
+                    evt.StateKey == null
+                        ? SendRawTimelineEventAsync(evt.Type, evt.RawContent!)
+                        : SendStateEventAsync(evt.Type, evt.StateKey, evt.RawContent)
+                );
+        }
+    }
+
+    public async Task BulkSendEventsAsync(IAsyncEnumerable<StateEventResponse> events) {
+        if ((await Homeserver.GetCapabilitiesAsync()).Capabilities.BulkSendEvents?.Enabled == true)
+            await Homeserver.ClientHttpClient.PostAsJsonAsync(
+                $"/_matrix/client/unstable/gay.rory.bulk_send_events/rooms/{RoomId}/bulk_send_events?_libmatrix_txn_id={Guid.NewGuid()}", events);
+        else {
+            Console.WriteLine("Homeserver does not support bulk sending events, falling back to individual sends.");
+            await foreach (var evt in events)
+                await (
+                    evt.StateKey == null
+                        ? SendRawTimelineEventAsync(evt.Type, evt.RawContent!)
+                        : SendStateEventAsync(evt.Type, evt.StateKey, evt.RawContent)
+                );
+        }
+    }
+
     public SpaceRoom AsSpace() => new SpaceRoom(Homeserver, RoomId);
     public PolicyRoom AsPolicyRoom() => new PolicyRoom(Homeserver, RoomId);
 
-- 
cgit 1.5.1