about summary refs log tree commit diff
path: root/ExampleBots/LibMatrix.ExampleBot/Bot/StartupTasks/ServerRoomSizeCalulator.cs
blob: 47851922a6acf001ac129386277174996247a075 (plain) (blame)
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
using System.Diagnostics.CodeAnalysis;
using ArcaneLibs.Extensions;
using LibMatrix.ExampleBot.Bot.Interfaces;
using LibMatrix.Homeservers;
using LibMatrix.Services;
using LibMatrix.StateEventTypes.Spec;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Logging;

namespace LibMatrix.ExampleBot.Bot.StartupTasks;

public class ServerRoomSizeCalulator : IHostedService {
    private readonly HomeserverProviderService _homeserverProviderService;
    private readonly ILogger<ServerRoomSizeCalulator> _logger;
    private readonly MRUBotConfiguration _configuration;
    private readonly IEnumerable<ICommand> _commands;

    public ServerRoomSizeCalulator(HomeserverProviderService homeserverProviderService, ILogger<ServerRoomSizeCalulator> logger,
        MRUBotConfiguration configuration, IServiceProvider services) {
        logger.LogInformation("Server room size calculator hosted service instantiated!");
        _homeserverProviderService = homeserverProviderService;
        _logger = logger;
        _configuration = configuration;
    }

    /// <summary>Triggered when the application host is ready to start the service.</summary>
    /// <param name="cancellationToken">Indicates that the start process has been aborted.</param>
    [SuppressMessage("ReSharper", "FunctionNeverReturns")]
    public async Task StartAsync(CancellationToken cancellationToken) {
        Directory.GetFiles("bot_data/cache").ToList().ForEach(File.Delete);
        AuthenticatedHomeserverGeneric hs;
        try {
            hs = await _homeserverProviderService.GetAuthenticatedWithToken(_configuration.Homeserver,
                _configuration.AccessToken);
        }
        catch (Exception e) {
            _logger.LogError("{}", e.Message);
            throw;
        }

        await (await hs.GetRoom("!DoHEdFablOLjddKWIp:rory.gay")).JoinAsync();

        Dictionary<string, int> totalRoomSize = new();
        foreach (var room in await hs.GetJoinedRooms()) {
            var stateList = room.GetFullStateAsync().ToBlockingEnumerable().ToList();
            var roomSize = stateList.Count;
            if (roomSize > 10000) {
                await File.AppendAllLinesAsync("large_rooms.txt", new[] { $"{{ \"{room.RoomId}\", {roomSize} }}," }, cancellationToken);
            }

            var roomHs = room.RoomId.Split(":")[1];
            if (totalRoomSize.ContainsKey(roomHs)) {
                totalRoomSize[roomHs] += roomSize;
            }
            else {
                totalRoomSize.Add(roomHs, roomSize);
            }

            _logger.LogInformation($"Got room state for {room.RoomId}!");
        }

        await File.WriteAllTextAsync("server_size.txt", string.Join('\n', totalRoomSize.Select(x => $"{{ \"{x.Key}\", {x.Value} }},")), cancellationToken);
    }

    /// <summary>Triggered when the application host is performing a graceful shutdown.</summary>
    /// <param name="cancellationToken">Indicates that the shutdown process should no longer be graceful.</param>
    public Task StopAsync(CancellationToken cancellationToken) {
        _logger.LogInformation("Shutting down bot!");
        return Task.CompletedTask;
    }
}