From dae1a25664606415e054f3e3b20bbbfabdbb0e99 Mon Sep 17 00:00:00 2001
From: Rory& <root@rory.gay>
Date: Tue, 13 May 2025 08:16:46 +0200
Subject: Allow specifying expected membership when fetching member list, add

... helpers to get member IDs, add sending reactions, add selfbot-only
command mode in bot utils
---
 LibMatrix/RoomTypes/GenericRoom.cs | 39 +++++++++++++++++++++++++++++++-------
 LibMatrix/RoomTypes/SpaceRoom.cs   |  2 +-
 2 files changed, 33 insertions(+), 8 deletions(-)

(limited to 'LibMatrix')

diff --git a/LibMatrix/RoomTypes/GenericRoom.cs b/LibMatrix/RoomTypes/GenericRoom.cs
index 93736a3..bc1bc90 100644
--- a/LibMatrix/RoomTypes/GenericRoom.cs
+++ b/LibMatrix/RoomTypes/GenericRoom.cs
@@ -232,8 +232,11 @@ public class GenericRoom {
         return await res.Content.ReadFromJsonAsync<RoomIdResponse>() ?? throw new Exception("Failed to join room?");
     }
 
-    public async IAsyncEnumerable<StateEventResponse> GetMembersEnumerableAsync(bool joinedOnly = true) {
-        var res = await Homeserver.ClientHttpClient.GetAsync($"/_matrix/client/v3/rooms/{RoomId}/members");
+    public async IAsyncEnumerable<StateEventResponse> GetMembersEnumerableAsync(string? membership = null) {
+        var url = $"/_matrix/client/v3/rooms/{RoomId}/members";
+        var isMembershipSet = !string.IsNullOrWhiteSpace(membership);
+        if (isMembershipSet) url += $"?membership={membership}";
+        var res = await Homeserver.ClientHttpClient.GetAsync(url);
         var result = await JsonSerializer.DeserializeAsync<ChunkedStateEventResponse>(await res.Content.ReadAsStreamAsync(), new JsonSerializerOptions() {
             TypeInfoResolver = ChunkedStateEventResponseSerializerContext.Default
         });
@@ -242,13 +245,16 @@ public class GenericRoom {
 
         foreach (var resp in result.Chunk ?? []) {
             if (resp.Type != "m.room.member") continue;
-            if (joinedOnly && resp.RawContent?["membership"]?.GetValue<string>() != "join") continue;
+            if (isMembershipSet && resp.RawContent?["membership"]?.GetValue<string>() != membership) continue;
             yield return resp;
         }
     }
 
-    public async Task<FrozenSet<StateEventResponse>> GetMembersListAsync(bool joinedOnly = true) {
-        var res = await Homeserver.ClientHttpClient.GetAsync($"/_matrix/client/v3/rooms/{RoomId}/members");
+    public async Task<FrozenSet<StateEventResponse>> GetMembersListAsync(string? membership = null) {
+        var url = $"/_matrix/client/v3/rooms/{RoomId}/members";
+        var isMembershipSet = !string.IsNullOrWhiteSpace(membership);
+        if (isMembershipSet) url += $"?membership={membership}";
+        var res = await Homeserver.ClientHttpClient.GetAsync(url);
         var result = await JsonSerializer.DeserializeAsync<ChunkedStateEventResponse>(await res.Content.ReadAsStreamAsync(), new JsonSerializerOptions() {
             TypeInfoResolver = ChunkedStateEventResponseSerializerContext.Default
         });
@@ -258,13 +264,23 @@ public class GenericRoom {
         var members = new List<StateEventResponse>();
         foreach (var resp in result.Chunk ?? []) {
             if (resp.Type != "m.room.member") continue;
-            if (joinedOnly && resp.RawContent?["membership"]?.GetValue<string>() != "join") continue;
+            if (isMembershipSet && resp.RawContent?["membership"]?.GetValue<string>() != membership) continue;
             members.Add(resp);
         }
 
         return members.ToFrozenSet();
     }
 
+    public async IAsyncEnumerable<string> GetMemberIdsEnumerableAsync(string? membership = null) {
+        await foreach (var evt in GetMembersEnumerableAsync(membership))
+            yield return evt.StateKey!;
+    }
+    
+    public async Task<FrozenSet<string>> GetMemberIdsListAsync(string? membership = null) {
+        var members = await GetMembersListAsync(membership);
+        return members.Select(x => x.StateKey!).ToFrozenSet();
+    }
+
 #region Utility shortcuts
 
     public Task<EventIdResponse> SendMessageEventAsync(RoomMessageEventContent content) =>
@@ -393,6 +409,15 @@ public class GenericRoom {
         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() {
+                RelationType = "m.annotation",
+                EventId = eventId,
+                Key = key
+            }
+        });
+
     public async Task<EventIdResponse?> SendFileAsync(string fileName, Stream fileStream, string messageType = "m.file", string contentType = "application/octet-stream") {
         var url = await Homeserver.UploadFile(fileName, fileStream);
         var content = new RoomMessageEventContent() {
@@ -586,4 +611,4 @@ public class GenericRoom {
 public class RoomIdResponse {
     [JsonPropertyName("room_id")]
     public string RoomId { get; set; }
-}
+}
\ No newline at end of file
diff --git a/LibMatrix/RoomTypes/SpaceRoom.cs b/LibMatrix/RoomTypes/SpaceRoom.cs
index 4563ed3..0c74be5 100644
--- a/LibMatrix/RoomTypes/SpaceRoom.cs
+++ b/LibMatrix/RoomTypes/SpaceRoom.cs
@@ -17,7 +17,7 @@ public class SpaceRoom(AuthenticatedHomeserverGeneric homeserver, string roomId)
     }
 
     public async Task<EventIdResponse> AddChildAsync(GenericRoom room) {
-        var members = room.GetMembersEnumerableAsync(true);
+        var members = room.GetMembersEnumerableAsync("join");
         Dictionary<string, int> memberCountByHs = new();
         await foreach (var member in members) {
             var server = member.StateKey.Split(':')[1];
-- 
cgit 1.5.1