1 files changed, 120 insertions, 0 deletions
diff --git a/ModAS.Server/Controllers/Debug/DebugController.cs b/ModAS.Server/Controllers/Debug/DebugController.cs
new file mode 100644
index 0000000..7bec3e5
--- /dev/null
+++ b/ModAS.Server/Controllers/Debug/DebugController.cs
@@ -0,0 +1,120 @@
+using System.Collections.Frozen;
+using ArcaneLibs.Extensions;
+using Elastic.Apm;
+using Elastic.Apm.Api;
+using LibMatrix.Homeservers;
+using Microsoft.AspNetCore.Authorization;
+using Microsoft.AspNetCore.Mvc;
+using ModAS.Server.Attributes;
+using ModAS.Server.Services;
+using MxApiExtensions.Services;
+
+namespace ModAS.Server.Controllers.Debug;
+
+/// <summary>
+///   Provides debugging endpoints.
+/// </summary>
+/// <param name="config"><inheritdoc cref="ModASConfiguration"/></param>
+/// <param name="authHsProvider"><inheritdoc cref="UserProviderService"/></param>
+/// <param name="roomContextService"><inheritdoc cref="RoomContextService"/></param>
+[ApiController]
+[UserAuth(AnyRoles = AuthRoles.Developer | AuthRoles.Administrator)]
+public class DebugController(ModASConfiguration config, UserProviderService authHsProvider, RoomContextService roomContextService) : ControllerBase {
+    /// <summary>
+    ///  Returns a JSON object containing the request and response headers.
+    /// </summary>
+    /// <returns>JSON object with request and partial response headers.</returns>
+    [HttpGet("/_matrix/_modas/debug")]
+    public IActionResult Index() {
+        return Ok(new {
+            Request = Request.Headers,
+            Response = Response.Headers
+        });
+    }
+
+    /// <summary>
+    ///  Returns a JSON object containing the configuration.
+    /// </summary>
+    /// <returns></returns>
+    [HttpGet("/_matrix/_modas/debug/config")]
+    public IActionResult Config() {
+        return Ok(config);
+    }
+
+    [HttpGet("/_matrix/_modas/debug/known_users")]
+    public IActionResult KnownUsers() {
+        return Ok(authHsProvider.KnownUsers.Keys);
+    }
+
+    [HttpGet("/_matrix/_modas/debug/test_locate_users")]
+    public async IAsyncEnumerable<string> TestLocateUsers([FromQuery] string startUser) {
+        List<AuthenticatedHomeserverGeneric> foundUsers = (await authHsProvider.GetValidUsers()).Select(x => x.Value).ToList();
+        if (!foundUsers.Any(x => x.WhoAmI.UserId == startUser)) {
+            foundUsers.Add(await authHsProvider.GetImpersonatedHomeserver(startUser));
+        }
+
+        List<string> processedRooms = [], processedUsers = [];
+        var foundNew = true;
+        while (foundNew) {
+            var span1 = currentTransaction.StartSpan("iterateUsers", ApiConstants.TypeApp);
+            foundNew = false;
+            var usersToProcess = foundUsers.Where(x => !processedUsers.Any(y => x.WhoAmI.UserId == y)).ToFrozenSet();
+            Console.WriteLine($"Got {usersToProcess.Count} users: {string.Join(", ", usersToProcess)}");
+
+            var rooms = usersToProcess.Select(async x => await x.GetJoinedRooms());
+            var roomLists = rooms.ToAsyncEnumerable();
+            await foreach (var roomList in roomLists) {
+                if (roomList is null) continue;
+                foreach (var room in roomList) {
+                    if (processedRooms.Contains(room.RoomId)) continue;
+                    processedRooms.Add(room.RoomId);
+                    var roomMembers = await room.GetMembersListAsync(false);
+                    foreach (var roomMember in roomMembers) {
+                        if (roomMember.StateKey.EndsWith(':' + config.ServerName) && !foundUsers.Any(x => x.WhoAmI.UserId == roomMember.StateKey)) {
+                            foundUsers.Add(await authHsProvider.GetImpersonatedHomeserver(roomMember.StateKey));
+                            foundNew = true;
+                            yield return roomMember.StateKey;
+                        }
+                    }
+                }
+            }
+
+            // await foreach (var task in tasks) {
+            //     if (task is null) continue;
+            //     foreach (var user in task) {
+            //         if (foundUsers.Contains(user)) continue;
+            //         foundUsers.Add(user);
+            //         foundNew = true;
+            //         yield return user;
+            //     }
+            // }
+
+            span1.End();
+        }
+    }
+
+    [HttpGet("/_matrix/_modas/debug/room_contexts")]
+    public IActionResult RoomContexts() {
+        return Ok(roomContextService.RoomContexts.Values);
+    }
+
+    [HttpGet("/_matrix/_modas/debug/room_contexts/{roomId}")]
+    public async Task<IActionResult> RoomContext(string roomId) {
+        var roomContext = await roomContextService.GetRoomContext(roomId);
+        if (roomContext is null) return NotFound("Room not found");
+        return Ok(roomContext);
+    }
+
+    [HttpGet("/_matrix/_modas/debug/room_contexts/by_user/{userId}")]
+    public async IAsyncEnumerable<RoomContextService.RoomContext> RoomContextByUser(string userId) {
+        var user = await authHsProvider.GetImpersonatedHomeserver(userId);
+        var rooms = await user.GetJoinedRooms();
+        var contexts = rooms.Select(x => roomContextService.GetRoomContext(x.RoomId)).ToAsyncEnumerable();
+        await foreach (var context in contexts) {
+            if (context is null) continue;
+            yield return context;
+        }
+    }
+
+    private static ITransaction currentTransaction => Agent.Tracer.CurrentTransaction;
+}
\ No newline at end of file