about summary refs log tree commit diff
path: root/Utilities/LibMatrix.DevTestBot/Bot/DevTestBot.cs
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--Utilities/LibMatrix.DevTestBot/Bot/DevTestBot.cs114
1 files changed, 114 insertions, 0 deletions
diff --git a/Utilities/LibMatrix.DevTestBot/Bot/DevTestBot.cs b/Utilities/LibMatrix.DevTestBot/Bot/DevTestBot.cs
new file mode 100644
index 0000000..e41321a
--- /dev/null
+++ b/Utilities/LibMatrix.DevTestBot/Bot/DevTestBot.cs
@@ -0,0 +1,114 @@
+using System.Diagnostics.CodeAnalysis;
+using ArcaneLibs.Extensions;
+using LibMatrix.EventTypes.Spec;
+using LibMatrix.EventTypes.Spec.State;
+using LibMatrix.ExampleBot.Bot.Interfaces;
+using LibMatrix.Helpers;
+using LibMatrix.Homeservers;
+using LibMatrix.Services;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Hosting;
+using Microsoft.Extensions.Logging;
+
+namespace LibMatrix.ExampleBot.Bot;
+
+public class DevTestBot : IHostedService {
+    private readonly HomeserverProviderService _homeserverProviderService;
+    private readonly ILogger<DevTestBot> _logger;
+    private readonly DevTestBotConfiguration _configuration;
+    private readonly IEnumerable<ICommand> _commands;
+
+    public DevTestBot(HomeserverProviderService homeserverProviderService, ILogger<DevTestBot> logger,
+        DevTestBotConfiguration configuration, IServiceProvider services) {
+        logger.LogInformation("{} instantiated!", this.GetType().Name);
+        _homeserverProviderService = homeserverProviderService;
+        _logger = logger;
+        _configuration = configuration;
+        _logger.LogInformation("Getting commands...");
+        _commands = services.GetServices<ICommand>();
+        _logger.LogInformation("Got {} commands!", _commands.Count());
+    }
+
+    /// <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;
+        }
+
+        var syncHelper = new SyncHelper(hs);
+
+        await (hs.GetRoom("!DoHEdFablOLjddKWIp:rory.gay")).JoinAsync();
+
+        // foreach (var room in await hs.GetJoinedRooms()) {
+        //     if(room.RoomId is "!OGEhHVWSdvArJzumhm:matrix.org") continue;
+        //     foreach (var stateEvent in await room.GetStateAsync<List<StateEvent>>("")) {
+        //         var _ = stateEvent.GetType;
+        //     }
+        //     _logger.LogInformation($"Got room state for {room.RoomId}!");
+        // }
+
+        syncHelper.InviteReceivedHandlers.Add(async Task (args) => {
+            var inviteEvent =
+                args.Value.InviteState.Events.FirstOrDefault(x =>
+                    x.Type == "m.room.member" && x.StateKey == hs.UserId);
+            _logger.LogInformation(
+                $"Got invite to {args.Key} by {inviteEvent.Sender} with reason: {(inviteEvent.TypedContent as RoomMemberEventContent).Reason}");
+            if (inviteEvent.Sender.EndsWith(":rory.gay") || inviteEvent.Sender == "@mxidupwitch:the-apothecary.club") {
+                try {
+                    var senderProfile = await hs.GetProfileAsync(inviteEvent.Sender);
+                    await (hs.GetRoom(args.Key)).JoinAsync(reason: $"I was invited by {senderProfile.DisplayName ?? inviteEvent.Sender}!");
+                }
+                catch (Exception e) {
+                    _logger.LogError("{}", e.ToString());
+                    await (hs.GetRoom(args.Key)).LeaveAsync(reason: "I was unable to join the room: " + e);
+                }
+            }
+        });
+        syncHelper.TimelineEventHandlers.Add(async @event => {
+            _logger.LogInformation(
+                "Got timeline event in {}: {}", @event.RoomId, @event.ToJson(indent: false, ignoreNull: true));
+
+            var room = hs.GetRoom(@event.RoomId);
+            // _logger.LogInformation(eventResponse.ToJson(indent: false));
+            if (@event is { Type: "m.room.message", TypedContent: RoomMessageEventContent message }) {
+                if (message is { MessageType: "m.text" } && message.Body.StartsWith(_configuration.Prefix)) {
+                    var command = _commands.FirstOrDefault(x => x.Name == message.Body.Split(' ')[0][_configuration.Prefix.Length..]);
+                    if (command == null) {
+                        await room.SendMessageEventAsync(
+                            new RoomMessageEventContent(messageType: "m.text", body: "Command not found!"));
+                        return;
+                    }
+
+                    var ctx = new CommandContext {
+                        Room = room,
+                        MessageEvent = @event
+                    };
+                    if (await command.CanInvoke(ctx)) {
+                        await command.Invoke(ctx);
+                    }
+                    else {
+                        await room.SendMessageEventAsync(
+                            new RoomMessageEventContent(messageType: "m.text", body: "You do not have permission to run this command!"));
+                    }
+                }
+            }
+        });
+        await syncHelper.RunSyncLoopAsync(cancellationToken: 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;
+    }
+}