HomeserverEmulator work

1 files changed, 119 insertions, 0 deletions

diff --git a/Tests/LibMatrix.HomeserverEmulator/Controllers/SyncController.cs b/Tests/LibMatrix.HomeserverEmulator/Controllers/SyncController.cs
new file mode 100644
index 0000000..1653110
--- /dev/null
+++ b/Tests/LibMatrix.HomeserverEmulator/Controllers/SyncController.cs
@@ -0,0 +1,119 @@
+using System.Diagnostics;
+using System.Diagnostics.CodeAnalysis;
+using System.Security.Cryptography;
+using System.Text.Json.Nodes;
+using ArcaneLibs.Extensions;
+using LibMatrix.HomeserverEmulator.Services;
+using LibMatrix.Responses;
+using LibMatrix.Services;
+using Microsoft.AspNetCore.Mvc;
+
+namespace LibMatrix.HomeserverEmulator.Controllers;
+
+[ApiController]
+[Route("/_matrix/client/{version}/")]
+public class SyncController(ILogger<SyncController> logger, TokenService tokenService, UserStore userStore, RoomStore roomStore, HSEConfiguration cfg) : ControllerBase {
+    [HttpGet("sync")]
+    [SuppressMessage("ReSharper.DPA", "DPA0011: High execution time of MVC action", Justification = "Endpoint is expected to wait until data is available or timeout.")]
+    public async Task<SyncResponse> Sync([FromQuery] string? since = null, [FromQuery] int? timeout = 5000) {
+        var sw = Stopwatch.StartNew();
+        var token = tokenService.GetAccessToken(HttpContext);
+        if (token == null)
+            throw new MatrixException() {
+                ErrorCode = "M_MISSING_TOKEN",
+                Error = "Missing token"
+            };
+
+        var user = await userStore.GetUserByToken(token);
+        if (user == null)
+            throw new MatrixException() {
+                ErrorCode = "M_UNKNOWN_TOKEN",
+                Error = "No such user"
+            };
+        var session = user.AccessTokens[token];
+
+        if (string.IsNullOrWhiteSpace(since))
+            return InitialSync(user, session);
+
+        if (!session.SyncStates.TryGetValue(since, out var syncState))
+            if (!cfg.UnknownSyncTokenIsInitialSync)
+                throw new MatrixException() {
+                    ErrorCode = "M_UNKNOWN",
+                    Error = "Unknown sync token."
+                };
+            else
+                return InitialSync(user, session);
+
+        var response = new SyncResponse() {
+            NextBatch = Guid.NewGuid().ToString(),
+            DeviceOneTimeKeysCount = new()
+        };
+
+        session.SyncStates.Add(response.NextBatch, new() {
+            RoomPositions = syncState.RoomPositions.ToDictionary(x => x.Key, x => new UserStore.User.SessionInfo.UserSyncState.SyncRoomPosition() {
+                StatePosition = roomStore._rooms.First(y => y.RoomId == x.Key).State.Count,
+                TimelinePosition = roomStore._rooms.First(y => y.RoomId == x.Key).Timeline.Count,
+                AccountDataPosition = roomStore._rooms.First(y => y.RoomId == x.Key).AccountData[user.UserId].Count
+            })
+        });
+
+        if (!string.IsNullOrWhiteSpace(since)) {
+            while (sw.ElapsedMilliseconds < timeout && response.Rooms?.Join is not { Count: > 0 }) {
+                await Task.Delay(100);
+                var rooms = roomStore._rooms.Where(x => x.State.Any(y => y.Type == "m.room.member" && y.StateKey == user.UserId)).ToList();
+                foreach (var room in rooms) {
+                    var roomPositions = syncState.RoomPositions[room.RoomId];
+
+                    response.Rooms ??= new();
+                    response.Rooms.Join ??= new();
+                    response.Rooms.Join[room.RoomId] = new() {
+                        State = new(room.State.Skip(roomPositions.StatePosition).ToList()),
+                        Timeline = new(events: room.Timeline.Skip(roomPositions.TimelinePosition).ToList(), limited: false),
+                        AccountData = new(room.AccountData.GetOrCreate(user.UserId, _ => []).Skip(roomPositions.AccountDataPosition).ToList())
+                    };
+                    session.SyncStates[response.NextBatch].RoomPositions[room.RoomId] = new() {
+                        StatePosition = room.State.Count,
+                        TimelinePosition = room.Timeline.Count,
+                        AccountDataPosition = room.AccountData[user.UserId].Count
+                    };
+
+                    if (response.Rooms.Join[room.RoomId].State.Events.Count == 0 &&
+                        response.Rooms.Join[room.RoomId].Timeline.Events.Count == 0 &&
+                        response.Rooms.Join[room.RoomId].AccountData.Events.Count == 0
+                       )
+                        response.Rooms.Join.Remove(room.RoomId);
+                }
+            }
+        }
+
+        return response;
+    }
+
+    private SyncResponse InitialSync(UserStore.User user, UserStore.User.SessionInfo session) {
+        var response = new SyncResponse() {
+            NextBatch = Guid.NewGuid().ToString(),
+            DeviceOneTimeKeysCount = new(),
+            AccountData = new(events: user.AccountData.ToList())
+        };
+
+        session.SyncStates.Add(response.NextBatch, new());
+
+        var rooms = roomStore._rooms.Where(x => x.State.Any(y => y.Type == "m.room.member" && y.StateKey == user.UserId)).ToList();
+        foreach (var room in rooms) {
+            response.Rooms ??= new();
+            response.Rooms.Join ??= new();
+            response.Rooms.Join[room.RoomId] = new() {
+                State = new(room.State.ToList()),
+                Timeline = new(events: room.Timeline.ToList(), limited: false),
+                AccountData = new(room.AccountData.GetOrCreate(user.UserId, _ => []).ToList())
+            };
+            session.SyncStates[response.NextBatch].RoomPositions[room.RoomId] = new() {
+                StatePosition = room.State.Count,
+                TimelinePosition = room.Timeline.Count,
+                AccountDataPosition = room.AccountData[user.UserId].Count
+            };
+        }
+
+        return response;
+    }
+}
\ No newline at end of file