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
|
@page "/Tools/Info/KnownHomeserverList"
@using ArcaneLibs.Extensions
@using LibMatrix.RoomTypes
@using SpawnDev.BlazorJS.WebWorkers
@inject WebWorkerService workerService
<h3>Known Homeserver List</h3>
<hr/>
@if (!IsFinished) {
<p>
<b>Loading... @RoomCount rooms remaining to process...</b>
</p>
}
@{
var shownCounts = counts.OrderByDescending(x => x.Value).AsEnumerable();
if (!IsFinished && counts.Count > 500) {
shownCounts = shownCounts.Where(x => x.Value > 5);
}
}
@foreach (var (homeserver, members) in shownCounts.ToList()) {
<p>@homeserver - @members users</p>
}
<hr/>
@code {
Dictionary<string, List<string>> homeservers { get; set; } = new();
Dictionary<string, int> counts { get; set; } = new();
// List<HomeserverInfo> Homeservers = new();
bool IsFinished { get; set; }
// HomeserverInfoQueryProgress QueryProgress { get; set; } = new();
AuthenticatedHomeserverGeneric? hs { get; set; }
int RoomCount { get; set; } = 0;
protected override async Task OnInitializedAsync() {
hs = await sessionStore.GetCurrentHomeserver(navigateOnFailure: true);
if (hs is null) return;
var ss = new SemaphoreSlim(32, 32);
var rooms = await hs.GetJoinedRooms();
RoomCount = rooms.Count;
var fetchTasks = rooms.Select(roomId => workerService.TaskPool.Invoke(() => InternalGetMembersByHomeserver(hs.WellKnownUris.Client, hs.AccessToken, roomId.RoomId))).ToList().ToAsyncEnumerable();
// var fetchTasks = rooms.Select(async x => {
// await ss.WaitAsync();
// var res = await x.GetMembersByHomeserverAsync();
// ss.Release();
// return res;
// }).ToAsyncEnumerable();
await foreach (var result in fetchTasks) {
foreach (var (resHomeserver, resMembers) in result) {
if (!homeservers.TryAdd(resHomeserver, resMembers)) {
homeservers[resHomeserver].AddRange(resMembers);
}
counts[resHomeserver] = homeservers[resHomeserver].Count;
}
RoomCount--;
StateHasChanged();
// await Task.Delay(250);
await Task.Yield();
}
foreach (var resHomeserver in homeservers.Keys) {
homeservers[resHomeserver] = homeservers[resHomeserver].Distinct().ToList();
counts[resHomeserver] = homeservers[resHomeserver].Count;
StateHasChanged();
await Task.Yield();
}
IsFinished = true;
StateHasChanged();
Console.WriteLine("Rerendered!");
await base.OnInitializedAsync();
}
private static async Task<Dictionary<string, List<string>>> InternalGetMembersByHomeserver(string homeserverBaseUrl, string accessToken, string roomId) {
var hs = new AuthenticatedHomeserverGeneric(homeserverBaseUrl, new() { Client = homeserverBaseUrl }, null, accessToken);
var room = hs.GetRoom(roomId);
return await room.GetMembersByHomeserverAsync();
}
}
|
