using System.Diagnostics.CodeAnalysis; using LibMatrix.ExampleBot.Bot.Interfaces; using LibMatrix.Homeservers; using LibMatrix.Services; using Microsoft.Extensions.Hosting; using Microsoft.Extensions.Logging; namespace LibMatrix.ExampleBot.Bot.StartupTasks; public class ServerRoomSizeCalulator : IHostedService { private readonly HomeserverProviderService _homeserverProviderService; private readonly ILogger _logger; private readonly DevTestBotConfiguration _configuration; private readonly IEnumerable _commands; public ServerRoomSizeCalulator(HomeserverProviderService homeserverProviderService, ILogger logger, DevTestBotConfiguration configuration, IServiceProvider services) { logger.LogInformation("Server room size calculator hosted service instantiated!"); _homeserverProviderService = homeserverProviderService; _logger = logger; _configuration = configuration; } /// Triggered when the application host is ready to start the service. /// Indicates that the start process has been aborted. [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 hs.GetRoom("!DoHEdFablOLjddKWIp:rory.gay").JoinAsync(); Dictionary 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); } /// Triggered when the application host is performing a graceful shutdown. /// Indicates that the shutdown process should no longer be graceful. public Task StopAsync(CancellationToken cancellationToken) { _logger.LogInformation("Shutting down bot!"); return Task.CompletedTask; } }