1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
|
using System.Collections.Concurrent;
using System.Collections.Frozen;
using System.Collections.Immutable;
using System.Collections.ObjectModel;
using System.Collections.Specialized;
using System.Security.Cryptography;
using System.Text.Json;
using System.Text.Json.Nodes;
using System.Text.Json.Serialization;
using ArcaneLibs;
using ArcaneLibs.Collections;
using ArcaneLibs.Extensions;
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();
public RoomStore(HSEConfiguration config) {
if (config.StoreData) {
var path = Path.Combine(config.DataStoragePath, "rooms");
if (!Directory.Exists(path)) Directory.CreateDirectory(path);
foreach (var file in Directory.GetFiles(path)) {
var room = JsonSerializer.Deserialize<Room>(File.ReadAllText(file));
if (room is not null) _rooms.Add(room);
}
}
else
Console.WriteLine("Data storage is disabled, not loading rooms from disk");
RebuildIndexes();
}
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 : NotifyPropertyChanged {
private CancellationTokenSource _debounceCts = new();
private ObservableCollection<StateEventResponse> _timeline;
private ObservableDictionary<string, List<StateEventResponse>> _accountData;
public Room(string roomId) {
if (string.IsNullOrWhiteSpace(roomId)) throw new ArgumentException("Value cannot be null or whitespace.", nameof(roomId));
if (roomId[0] != '!') throw new ArgumentException("Room ID must start with !", nameof(roomId));
RoomId = roomId;
State = FrozenSet<StateEventResponse>.Empty;
Timeline = new();
AccountData = new();
}
public string RoomId { get; set; }
public FrozenSet<StateEventResponse> State { get; private set; }
public ObservableCollection<StateEventResponse> Timeline {
get => _timeline;
set {
if (Equals(value, _timeline)) return;
_timeline = new(value);
_timeline.CollectionChanged += (sender, args) => {
if (args.Action == NotifyCollectionChangedAction.Add) {
foreach (StateEventResponse state in args.NewItems) {
if (state.StateKey is not null)
// we want state to be deduplicated by type and key, and we want the latest state to be the one that is returned
RebuildState();
}
}
SaveDebounced();
};
RebuildState();
OnPropertyChanged();
}
}
public ObservableDictionary<string, List<StateEventResponse>> AccountData {
get => _accountData;
set {
if (Equals(value, _accountData)) return;
_accountData = new(value);
_accountData.CollectionChanged += (sender, args) => SaveDebounced();
OnPropertyChanged();
}
}
public ImmutableList<StateEventResponse> JoinedMembers =>
State.Where(s => s is { Type: RoomMemberEventContent.EventId, TypedContent: RoomMemberEventContent { Membership: "join" } }).ToImmutableList();
internal StateEventResponse SetStateInternal(StateEvent request) {
var state = new StateEventResponse() {
Type = request.Type,
StateKey = request.StateKey,
EventId = Guid.NewGuid().ToString(),
RoomId = RoomId,
OriginServerTs = DateTimeOffset.Now.ToUnixTimeMilliseconds(),
Sender = "",
RawContent = request.RawContent ?? (request.TypedContent is not null
? new JsonObject()
: JsonSerializer.Deserialize<JsonObject>(JsonSerializer.Serialize(request.TypedContent)))
};
Timeline.Add(state);
if (state.StateKey is not null)
// we want state to be deduplicated by type and key, and we want the latest state to be the one that is returned
State = Timeline.Where(s => s.StateKey != null)
.OrderByDescending(s => s.OriginServerTs)
.DistinctBy(x => (x.Type, x.StateKey))
.ToFrozenSet();
return state;
}
public StateEventResponse AddUser(string userId) {
var state = SetStateInternal(new() {
Type = RoomMemberEventContent.EventId,
StateKey = userId,
TypedContent = new RoomMemberEventContent() {
Membership = "join"
},
});
state.Sender = userId;
return state;
}
public async Task SaveDebounced() {
if (!HSEConfiguration.Current.StoreData) return;
await _debounceCts.CancelAsync();
_debounceCts = new CancellationTokenSource();
try {
await Task.Delay(250, _debounceCts.Token);
// Ensure all state events are in the timeline
State.Where(s => !Timeline.Contains(s)).ToList().ForEach(s => Timeline.Add(s));
var path = Path.Combine(HSEConfiguration.Current.DataStoragePath, "rooms", $"{RoomId}.json");
Console.WriteLine($"Saving room {RoomId} to {path}!");
await File.WriteAllTextAsync(path, this.ToJson(ignoreNull: true));
}
catch (TaskCanceledException) { }
}
private void RebuildState() {
State = Timeline //.Where(s => s.Type == state.Type && s.StateKey == state.StateKey)
.Where(x => x.StateKey != null)
.OrderByDescending(s => s.OriginServerTs)
.DistinctBy(x => (x.Type, x.StateKey))
.ToFrozenSet();
}
}
}
|
