1 files changed, 83 insertions, 0 deletions
diff --git a/Tests/LibMatrix.HomeserverEmulator/Services/RoomStore.cs b/Tests/LibMatrix.HomeserverEmulator/Services/RoomStore.cs
new file mode 100644
index 0000000..b4624ab
--- /dev/null
+++ b/Tests/LibMatrix.HomeserverEmulator/Services/RoomStore.cs
@@ -0,0 +1,83 @@
+using System.Collections.Concurrent;
+using System.Security.Cryptography;
+using LibMatrix.EventTypes;
+using LibMatrix.EventTypes.Spec.State;
+using LibMatrix.Responses;
+
+namespace LibMatrix.HomeserverEmulator.Services;
+
+public class RoomStore {
+    public ConcurrentBag<Room> _rooms = new();
+    private Dictionary<string, Room> _roomsById = new();
+
+    private void RebuildIndexes() {
+        _roomsById = _rooms.ToDictionary(u => u.RoomId);
+    }
+
+    public Room? GetRoomById(string roomId, bool createIfNotExists = false) {
+        if (_roomsById.TryGetValue(roomId, out var user)) {
+            return user;
+        }
+
+        if (!createIfNotExists)
+            return null;
+
+        return CreateRoom(new() { });
+    }
+
+    public Room CreateRoom(CreateRoomRequest request) {
+        var room = new Room {
+            RoomId = $"!{Guid.NewGuid().ToString()}"
+        };
+        if (!string.IsNullOrWhiteSpace(request.Name))
+            room.SetStateInternal(new StateEvent() {
+                Type = RoomNameEventContent.EventId,
+                TypedContent = new RoomNameEventContent() {
+                    Name = request.Name
+                }
+            });
+
+        if (!string.IsNullOrWhiteSpace(request.RoomAliasName))
+            room.SetStateInternal(new StateEvent() {
+                Type = RoomCanonicalAliasEventContent.EventId,
+                TypedContent = new RoomCanonicalAliasEventContent() {
+                    Alias = $"#{request.RoomAliasName}:localhost"
+                }
+            });
+
+        foreach (var stateEvent in request.InitialState ?? []) {
+            room.SetStateInternal(stateEvent);
+        }
+
+        _rooms.Add(room);
+        RebuildIndexes();
+        return room;
+    }
+
+    public class Room {
+        public string RoomId { get; set; }
+        public List<StateEventResponse> State { get; set; } = new();
+        public Dictionary<string, EventContent> Timeline { get; set; } = new();
+
+        internal StateEventResponse SetStateInternal(StateEvent request) {
+            var state = new StateEventResponse() {
+                Type = request.Type,
+                StateKey = request.StateKey,
+                RawContent = request.RawContent,
+                EventId = Guid.NewGuid().ToString()
+            };
+            State.Add(state);
+            return state;
+        }
+
+        public StateEventResponse AddUser(string userId) {
+            return SetStateInternal(new() {
+                Type = RoomMemberEventContent.EventId,
+                StateKey = userId,
+                TypedContent = new RoomMemberEventContent() {
+                    Membership = "join"
+                }
+            });
+        }
+    }
+}
\ No newline at end of file