From bba7333ee6581a92bbbc7479d72325e704fe7fa6 Mon Sep 17 00:00:00 2001
From: Rory& <root@rory.gay>
Date: Thu, 8 Aug 2024 02:44:16 +0200
Subject: More synapse admin apis

---
 .../Synapse/SynapseAdminApiClient.cs               | 80 ++++++++++++++++++++--
 1 file changed, 74 insertions(+), 6 deletions(-)

(limited to 'LibMatrix/Homeservers/ImplementationDetails/Synapse/SynapseAdminApiClient.cs')

diff --git a/LibMatrix/Homeservers/ImplementationDetails/Synapse/SynapseAdminApiClient.cs b/LibMatrix/Homeservers/ImplementationDetails/Synapse/SynapseAdminApiClient.cs
index ac94a7a..b3902eb 100644
--- a/LibMatrix/Homeservers/ImplementationDetails/Synapse/SynapseAdminApiClient.cs
+++ b/LibMatrix/Homeservers/ImplementationDetails/Synapse/SynapseAdminApiClient.cs
@@ -1,24 +1,32 @@
+using System.Net.Http.Json;
+using System.Text.Json.Nodes;
 using ArcaneLibs.Extensions;
 using LibMatrix.Filters;
-using LibMatrix.Responses.Admin;
+using LibMatrix.Homeservers.ImplementationDetails.Synapse.Models.Filters;
+using LibMatrix.Homeservers.ImplementationDetails.Synapse.Models.Responses;
+using LibMatrix.Responses;
 
 namespace LibMatrix.Homeservers.ImplementationDetails.Synapse;
 
 public class SynapseAdminApiClient(AuthenticatedHomeserverSynapse authenticatedHomeserver) {
-    public async IAsyncEnumerable<AdminRoomListingResult.AdminRoomListingResultRoom> SearchRoomsAsync(int limit = int.MaxValue, string orderBy = "name", string dir = "f",
-        string? searchTerm = null, LocalRoomQueryFilter? localFilter = null) {
-        AdminRoomListingResult? res = null;
+    // https://github.com/element-hq/synapse/tree/develop/docs/admin_api
+
+#region Rooms
+
+    public async IAsyncEnumerable<AdminRoomListResult.AdminRoomListResultRoom> SearchRoomsAsync(int limit = int.MaxValue, int chunkLimit = 250, string orderBy = "name",
+        string dir = "f", string? searchTerm = null, SynapseAdminLocalRoomQueryFilter? localFilter = null) {
+        AdminRoomListResult? res = null;
         var i = 0;
         int? totalRooms = null;
         do {
-            var url = $"/_synapse/admin/v1/rooms?limit={Math.Min(limit, 250)}&dir={dir}&order_by={orderBy}";
+            var url = $"/_synapse/admin/v1/rooms?limit={Math.Min(limit, chunkLimit)}&dir={dir}&order_by={orderBy}";
             if (!string.IsNullOrEmpty(searchTerm)) url += $"&search_term={searchTerm}";
 
             if (res?.NextBatch is not null) url += $"&from={res.NextBatch}";
 
             Console.WriteLine($"--- ADMIN Querying Room List with URL: {url} - Already have {i} items... ---");
 
-            res = await authenticatedHomeserver.ClientHttpClient.GetFromJsonAsync<AdminRoomListingResult>(url);
+            res = await authenticatedHomeserver.ClientHttpClient.GetFromJsonAsync<AdminRoomListResult>(url);
             totalRooms ??= res.TotalRooms;
             Console.WriteLine(res.ToJson(false));
             foreach (var room in res.Rooms) {
@@ -104,4 +112,64 @@ public class SynapseAdminApiClient(AuthenticatedHomeserverSynapse authenticatedH
             }
         } while (i < Math.Min(limit, totalRooms ?? limit));
     }
+
+#endregion
+
+#region Users
+
+    public async IAsyncEnumerable<AdminUserListResult.AdminUserListResultUser> SearchUsersAsync(int limit = int.MaxValue, int chunkLimit = 250,
+        SynapseAdminLocalUserQueryFilter? localFilter = null) {
+        // TODO: implement filters
+        string? from = null;
+        while (limit > 0) {
+            var url = new Uri("/_synapse/admin/v3/users", UriKind.Relative);
+            url = url.AddQuery("limit", Math.Min(limit, chunkLimit).ToString());
+            if (!string.IsNullOrWhiteSpace(from)) url = url.AddQuery("from", from);
+            Console.WriteLine($"--- ADMIN Querying User List with URL: {url} ---");
+            // TODO: implement URI methods in http client
+            var res = await authenticatedHomeserver.ClientHttpClient.GetFromJsonAsync<AdminUserListResult>(url.ToString());
+            foreach (var user in res.Users) {
+                limit--;
+                yield return user;
+            }
+
+            if (string.IsNullOrWhiteSpace(res.NextToken)) break;
+            from = res.NextToken;
+        }
+    }
+
+    public async Task<LoginResponse> LoginUserAsync(string userId, TimeSpan expireAfter) {
+        var url = new Uri($"/_synapse/admin/v1/users/{userId.UrlEncode()}/login", UriKind.Relative);
+        url.AddQuery("valid_until_ms", DateTimeOffset.UtcNow.Add(expireAfter).ToUnixTimeMilliseconds().ToString());
+        var resp = await authenticatedHomeserver.ClientHttpClient.PostAsJsonAsync<JsonObject>(url.ToString(), new());
+        var loginResp = await resp.Content.ReadFromJsonAsync<LoginResponse>();
+        loginResp.UserId = userId; // Synapse only returns the access token
+        return loginResp;
+    }
+
+#endregion
+
+#region Reports
+
+    public async IAsyncEnumerable<SynapseAdminEventReportListResult.SynapseAdminEventReportListResultReport> GetEventReportsAsync(int limit = int.MaxValue, int chunkLimit = 250,
+        string dir = "f", SynapseAdminLocalEventReportQueryFilter? filter = null) {
+        // TODO: implement filters
+        string? from = null;
+        while (limit > 0) {
+            var url = new Uri("/_synapse/admin/v1/event_reports", UriKind.Relative);
+            url = url.AddQuery("limit", Math.Min(limit, chunkLimit).ToString());
+            if (!string.IsNullOrWhiteSpace(from)) url = url.AddQuery("from", from);
+            Console.WriteLine($"--- ADMIN Querying Reports with URL: {url} ---");
+            var res = await authenticatedHomeserver.ClientHttpClient.GetFromJsonAsync<SynapseAdminEventReportListResult>(url.ToString());
+            foreach (var report in res.Reports) {
+                limit--;
+                yield return report;
+            }
+
+            if (string.IsNullOrWhiteSpace(res.NextToken)) break;
+            from = res.NextToken;
+        }
+    }
+
+#endregion
 }
\ No newline at end of file
-- 
cgit 1.4.1