about summary refs log tree commit diff
path: root/LibMatrix.ExampleBot/Bot
diff options
context:
space:
mode:
authorTheArcaneBrony <myrainbowdash949@gmail.com>2023-09-04 06:29:00 +0200
committerTheArcaneBrony <myrainbowdash949@gmail.com>2023-09-04 06:29:00 +0200
commit9dcce18cda5317ea1150eed06d6589b6285577e6 (patch)
tree1b36a9ddffa312e58daab075c43fb482c2bae905 /LibMatrix.ExampleBot/Bot
parentToo many changes to name... (diff)
downloadLibMatrix-9dcce18cda5317ea1150eed06d6589b6285577e6.tar.xz
Add start of Media Moderator PoC bot
Diffstat (limited to 'LibMatrix.ExampleBot/Bot')
-rw-r--r--LibMatrix.ExampleBot/Bot/Commands/CmdCommand.cs46
-rw-r--r--LibMatrix.ExampleBot/Bot/Commands/HelpCommand.cs24
-rw-r--r--LibMatrix.ExampleBot/Bot/Commands/PingCommand.cs15
-rw-r--r--LibMatrix.ExampleBot/Bot/FileStorageProvider.cs39
-rw-r--r--LibMatrix.ExampleBot/Bot/Interfaces/CommandContext.cs12
-rw-r--r--LibMatrix.ExampleBot/Bot/Interfaces/ICommand.cs12
-rw-r--r--LibMatrix.ExampleBot/Bot/MRUBot.cs117
-rw-r--r--LibMatrix.ExampleBot/Bot/MRUBotConfiguration.cs12
-rw-r--r--LibMatrix.ExampleBot/Bot/StartupTasks/ServerRoomSizeCalulator.cs72
9 files changed, 0 insertions, 349 deletions
diff --git a/LibMatrix.ExampleBot/Bot/Commands/CmdCommand.cs b/LibMatrix.ExampleBot/Bot/Commands/CmdCommand.cs
deleted file mode 100644
index ca10326..0000000
--- a/LibMatrix.ExampleBot/Bot/Commands/CmdCommand.cs
+++ /dev/null
@@ -1,46 +0,0 @@
-using LibMatrix.ExampleBot.Bot.Interfaces;
-using LibMatrix.StateEventTypes.Spec;
-
-namespace LibMatrix.ExampleBot.Bot.Commands;
-
-public class CmdCommand : ICommand {
-    public string Name => "cmd";
-    public string Description => "Runs a command on the host system";
-
-    public Task<bool> CanInvoke(CommandContext ctx) {
-        return Task.FromResult(ctx.MessageEvent.Sender.EndsWith(":rory.gay"));
-    }
-
-    public async Task Invoke(CommandContext ctx) {
-        var cmd = ctx.Args.Aggregate("\"", (current, arg) => current + arg + " ");
-
-        cmd = cmd.Trim();
-        cmd += "\"";
-
-        await ctx.Room.SendMessageEventAsync("m.room.message", new RoomMessageEventData {
-            Body = $"Command being executed: `{cmd}`"
-        });
-
-        var output = ArcaneLibs.Util.GetCommandOutputSync(
-                Environment.OSVersion.Platform == PlatformID.Unix ? "/bin/sh" : "cmd.exe",
-                (Environment.OSVersion.Platform == PlatformID.Unix ? "-c " : "/c ") + cmd)
-            .Replace("`", "\\`")
-            .Split("\n").ToList();
-        foreach (var _out in output) Console.WriteLine($"{_out.Length:0000} {_out}");
-
-        var msg = "";
-        while (output.Count > 0) {
-            Console.WriteLine("Adding: " + output[0]);
-            msg += output[0] + "\n";
-            output.RemoveAt(0);
-            if ((output.Count > 0 && (msg + output[0]).Length > 64000) || output.Count == 0) {
-                await ctx.Room.SendMessageEventAsync("m.room.message", new RoomMessageEventData {
-                    FormattedBody = $"```ansi\n{msg}\n```",
-                    // Body = Markdig.Markdown.ToHtml(msg),
-                    Format = "org.matrix.custom.html"
-                });
-                msg = "";
-            }
-        }
-    }
-}
diff --git a/LibMatrix.ExampleBot/Bot/Commands/HelpCommand.cs b/LibMatrix.ExampleBot/Bot/Commands/HelpCommand.cs
deleted file mode 100644
index 69766d1..0000000
--- a/LibMatrix.ExampleBot/Bot/Commands/HelpCommand.cs
+++ /dev/null
@@ -1,24 +0,0 @@
-using System.Text;
-using LibMatrix.ExampleBot.Bot.Interfaces;
-using LibMatrix.StateEventTypes.Spec;
-using Microsoft.Extensions.DependencyInjection;
-
-namespace LibMatrix.ExampleBot.Bot.Commands;
-
-public class HelpCommand(IServiceProvider services) : ICommand {
-    public string Name { get; } = "help";
-    public string Description { get; } = "Displays this help message";
-
-    public async Task Invoke(CommandContext ctx) {
-        var sb = new StringBuilder();
-        sb.AppendLine("Available commands:");
-        var commands = services.GetServices<ICommand>().ToList();
-        foreach (var command in commands) {
-            sb.AppendLine($"- {command.Name}: {command.Description}");
-        }
-
-        await ctx.Room.SendMessageEventAsync("m.room.message", new RoomMessageEventData {
-            Body = sb.ToString()
-        });
-    }
-}
diff --git a/LibMatrix.ExampleBot/Bot/Commands/PingCommand.cs b/LibMatrix.ExampleBot/Bot/Commands/PingCommand.cs
deleted file mode 100644
index a7c65b5..0000000
--- a/LibMatrix.ExampleBot/Bot/Commands/PingCommand.cs
+++ /dev/null
@@ -1,15 +0,0 @@
-using LibMatrix.ExampleBot.Bot.Interfaces;
-using LibMatrix.StateEventTypes.Spec;
-
-namespace LibMatrix.ExampleBot.Bot.Commands;
-
-public class PingCommand : ICommand {
-    public string Name { get; } = "ping";
-    public string Description { get; } = "Pong!";
-
-    public async Task Invoke(CommandContext ctx) {
-        await ctx.Room.SendMessageEventAsync("m.room.message", new RoomMessageEventData {
-            Body = "pong!"
-        });
-    }
-}
diff --git a/LibMatrix.ExampleBot/Bot/FileStorageProvider.cs b/LibMatrix.ExampleBot/Bot/FileStorageProvider.cs
deleted file mode 100644
index 2dfcee5..0000000
--- a/LibMatrix.ExampleBot/Bot/FileStorageProvider.cs
+++ /dev/null
@@ -1,39 +0,0 @@
-using System.Text.Json;
-using ArcaneLibs.Extensions;
-using LibMatrix.Extensions;
-using LibMatrix.Interfaces.Services;
-using Microsoft.Extensions.Logging;
-
-namespace LibMatrix.ExampleBot.Bot;
-
-public class FileStorageProvider : IStorageProvider {
-    private readonly ILogger<FileStorageProvider> _logger;
-
-    public string TargetPath { get; }
-
-    /// <summary>
-    /// Creates a new instance of <see cref="FileStorageProvider" />.
-    /// </summary>
-    /// <param name="targetPath"></param>
-    public FileStorageProvider(string targetPath) {
-        new Logger<FileStorageProvider>(new LoggerFactory()).LogInformation("test");
-        Console.WriteLine($"Initialised FileStorageProvider with path {targetPath}");
-        TargetPath = targetPath;
-        if(!Directory.Exists(targetPath)) {
-            Directory.CreateDirectory(targetPath);
-        }
-    }
-
-    public async Task SaveObjectAsync<T>(string key, T value) => await File.WriteAllTextAsync(Path.Join(TargetPath, key), value?.ToJson());
-
-    public async Task<T?> LoadObjectAsync<T>(string key) => JsonSerializer.Deserialize<T>(await File.ReadAllTextAsync(Path.Join(TargetPath, key)));
-
-    public Task<bool> ObjectExistsAsync(string key) => Task.FromResult(File.Exists(Path.Join(TargetPath, key)));
-
-    public Task<List<string>> GetAllKeysAsync() => Task.FromResult(Directory.GetFiles(TargetPath).Select(Path.GetFileName).ToList());
-
-    public Task DeleteObjectAsync(string key) {
-        File.Delete(Path.Join(TargetPath, key));
-        return Task.CompletedTask;
-    }
-}
diff --git a/LibMatrix.ExampleBot/Bot/Interfaces/CommandContext.cs b/LibMatrix.ExampleBot/Bot/Interfaces/CommandContext.cs
deleted file mode 100644
index ec61a1e..0000000
--- a/LibMatrix.ExampleBot/Bot/Interfaces/CommandContext.cs
+++ /dev/null
@@ -1,12 +0,0 @@
-using LibMatrix.Responses;
-using LibMatrix.RoomTypes;
-using LibMatrix.StateEventTypes.Spec;
-
-namespace LibMatrix.ExampleBot.Bot.Interfaces;
-
-public class CommandContext {
-    public GenericRoom Room { get; set; }
-    public StateEventResponse MessageEvent { get; set; }
-    public string CommandName => (MessageEvent.TypedContent as RoomMessageEventData).Body.Split(' ')[0][1..];
-    public string[] Args => (MessageEvent.TypedContent as RoomMessageEventData).Body.Split(' ')[1..];
-}
diff --git a/LibMatrix.ExampleBot/Bot/Interfaces/ICommand.cs b/LibMatrix.ExampleBot/Bot/Interfaces/ICommand.cs
deleted file mode 100644
index 393ddbb..0000000
--- a/LibMatrix.ExampleBot/Bot/Interfaces/ICommand.cs
+++ /dev/null
@@ -1,12 +0,0 @@
-namespace LibMatrix.ExampleBot.Bot.Interfaces; 
-
-public interface ICommand {
-    public string Name { get; }
-    public string Description { get; }
-
-    public Task<bool> CanInvoke(CommandContext ctx) {
-        return Task.FromResult(true);
-    }
-    
-    public Task Invoke(CommandContext ctx);
-}
\ No newline at end of file
diff --git a/LibMatrix.ExampleBot/Bot/MRUBot.cs b/LibMatrix.ExampleBot/Bot/MRUBot.cs
deleted file mode 100644
index 4f9b173..0000000
--- a/LibMatrix.ExampleBot/Bot/MRUBot.cs
+++ /dev/null
@@ -1,117 +0,0 @@
-using System.Diagnostics.CodeAnalysis;
-using ArcaneLibs.Extensions;
-using LibMatrix.ExampleBot.Bot.Interfaces;
-using LibMatrix.Extensions;
-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;
-
-public class MRUBot : IHostedService {
-    private readonly HomeserverProviderService _homeserverProviderService;
-    private readonly ILogger<MRUBot> _logger;
-    private readonly MRUBotConfiguration _configuration;
-    private readonly IEnumerable<ICommand> _commands;
-
-    public MRUBot(HomeserverProviderService homeserverProviderService, ILogger<MRUBot> logger,
-        MRUBotConfiguration configuration, IServiceProvider services) {
-        logger.LogInformation("MRUBot hosted service instantiated!");
-        _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;
-        }
-
-        await (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}!");
-        // }
-
-        hs.SyncHelper.InviteReceivedHandlers.Add(async Task (args) => {
-            var inviteEvent =
-                args.Value.InviteState.Events.FirstOrDefault(x =>
-                    x.Type == "m.room.member" && x.StateKey == hs.WhoAmI.UserId);
-            _logger.LogInformation(
-                $"Got invite to {args.Key} by {inviteEvent.Sender} with reason: {(inviteEvent.TypedContent as RoomMemberEventData).Reason}");
-            if (inviteEvent.Sender.EndsWith(":rory.gay") || inviteEvent.Sender == "@mxidupwitch:the-apothecary.club") {
-                try {
-                    var senderProfile = await hs.GetProfile(inviteEvent.Sender);
-                    await (await hs.GetRoom(args.Key)).JoinAsync(reason: $"I was invited by {senderProfile.DisplayName ?? inviteEvent.Sender}!");
-                }
-                catch (Exception e) {
-                    _logger.LogError("{}", e.ToString());
-                    await (await hs.GetRoom(args.Key)).LeaveAsync(reason: "I was unable to join the room: " + e);
-                }
-            }
-        });
-        hs.SyncHelper.TimelineEventHandlers.Add(async @event => {
-            _logger.LogInformation(
-                "Got timeline event in {}: {}", @event.RoomId, @event.ToJson(indent: false, ignoreNull: true));
-
-            var room = await hs.GetRoom(@event.RoomId);
-            // _logger.LogInformation(eventResponse.ToJson(indent: false));
-            if (@event is { Type: "m.room.message", TypedContent: RoomMessageEventData 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("m.room.message",
-                            new RoomMessageEventData {
-                                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("m.room.message",
-                            new RoomMessageEventData {
-                                MessageType = "m.text",
-                                Body = "You do not have permission to run this command!"
-                            });
-                    }
-                }
-            }
-        });
-        await hs.SyncHelper.RunSyncLoop(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;
-    }
-}
diff --git a/LibMatrix.ExampleBot/Bot/MRUBotConfiguration.cs b/LibMatrix.ExampleBot/Bot/MRUBotConfiguration.cs
deleted file mode 100644
index c7620df..0000000
--- a/LibMatrix.ExampleBot/Bot/MRUBotConfiguration.cs
+++ /dev/null
@@ -1,12 +0,0 @@
-using Microsoft.Extensions.Configuration;
-
-namespace LibMatrix.ExampleBot.Bot; 
-
-public class MRUBotConfiguration {
-    public MRUBotConfiguration(IConfiguration config) {
-        config.GetRequiredSection("Bot").Bind(this);
-    }
-    public string Homeserver { get; set; } = "";
-    public string AccessToken { get; set; } = "";
-    public string Prefix { get; set; }
-}
\ No newline at end of file
diff --git a/LibMatrix.ExampleBot/Bot/StartupTasks/ServerRoomSizeCalulator.cs b/LibMatrix.ExampleBot/Bot/StartupTasks/ServerRoomSizeCalulator.cs
deleted file mode 100644
index 4785192..0000000
--- a/LibMatrix.ExampleBot/Bot/StartupTasks/ServerRoomSizeCalulator.cs
+++ /dev/null
@@ -1,72 +0,0 @@
-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;
-    }
-}