about summary refs log tree commit diff
path: root/LibMatrix.ExampleBot/Bot
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--LibMatrix.ExampleBot/Bot/Commands/CmdCommand.cs46
-rw-r--r--LibMatrix.ExampleBot/Bot/Commands/HelpCommand.cs28
-rw-r--r--LibMatrix.ExampleBot/Bot/Commands/PingCommand.cs17
-rw-r--r--LibMatrix.ExampleBot/Bot/FileStorageProvider.cs35
-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.cs114
-rw-r--r--LibMatrix.ExampleBot/Bot/MRUBotConfiguration.cs12
8 files changed, 276 insertions, 0 deletions
diff --git a/LibMatrix.ExampleBot/Bot/Commands/CmdCommand.cs b/LibMatrix.ExampleBot/Bot/Commands/CmdCommand.cs
new file mode 100644
index 0000000..7b54b0c
--- /dev/null
+++ b/LibMatrix.ExampleBot/Bot/Commands/CmdCommand.cs
@@ -0,0 +1,46 @@
+using LibMatrix.ExampleBot.Bot.Interfaces;
+
+namespace LibMatrix.ExampleBot.Bot.Commands;
+
+public class CmdCommand : ICommand {
+    public string Name { get; } = "cmd";
+    public string Description { get; } = "Runs a command on the host system";
+
+    public async Task<bool> CanInvoke(CommandContext ctx) {
+        return ctx.MessageEvent.Sender.EndsWith(":rory.gay");
+    }
+
+    public async Task Invoke(CommandContext ctx) {
+        var cmd = "\"";
+        foreach (var arg in ctx.Args) cmd += arg + " ";
+
+        cmd = cmd.Trim();
+        cmd += "\"";
+
+        await ctx.Room.SendMessageEventAsync("m.room.message", new() {
+            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() {
+                    FormattedBody = $"```ansi\n{msg}\n```",
+                    // Body = Markdig.Markdown.ToHtml(msg),
+                    Format = "org.matrix.custom.html"
+                });
+                msg = "";
+            }
+        }
+    }
+}
\ No newline at end of file
diff --git a/LibMatrix.ExampleBot/Bot/Commands/HelpCommand.cs b/LibMatrix.ExampleBot/Bot/Commands/HelpCommand.cs
new file mode 100644
index 0000000..a259b3e
--- /dev/null
+++ b/LibMatrix.ExampleBot/Bot/Commands/HelpCommand.cs
@@ -0,0 +1,28 @@
+using System.Text;
+using LibMatrix.ExampleBot.Bot.Interfaces;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace LibMatrix.ExampleBot.Bot.Commands; 
+
+public class HelpCommand : ICommand {
+    private readonly IServiceProvider _services;
+    public HelpCommand(IServiceProvider services) {
+        _services = services;
+    }
+
+    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() {
+            Body = sb.ToString(),
+        });
+    }
+}
\ No newline at end of file
diff --git a/LibMatrix.ExampleBot/Bot/Commands/PingCommand.cs b/LibMatrix.ExampleBot/Bot/Commands/PingCommand.cs
new file mode 100644
index 0000000..664dc53
--- /dev/null
+++ b/LibMatrix.ExampleBot/Bot/Commands/PingCommand.cs
@@ -0,0 +1,17 @@
+using LibMatrix.ExampleBot.Bot.Interfaces;
+
+namespace LibMatrix.ExampleBot.Bot.Commands; 
+
+public class PingCommand : ICommand {
+    public PingCommand() {
+    }
+
+    public string Name { get; } = "ping";
+    public string Description { get; } = "Pong!";
+
+    public async Task Invoke(CommandContext ctx) {
+        await ctx.Room.SendMessageEventAsync("m.room.message", new() {
+            Body = "pong!"
+        });
+    }
+}
\ No newline at end of file
diff --git a/LibMatrix.ExampleBot/Bot/FileStorageProvider.cs b/LibMatrix.ExampleBot/Bot/FileStorageProvider.cs
new file mode 100644
index 0000000..249aba3
--- /dev/null
+++ b/LibMatrix.ExampleBot/Bot/FileStorageProvider.cs
@@ -0,0 +1,35 @@
+using System.Text.Json;
+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), ObjectExtensions.ToJson(value));
+
+    public async Task<T?> LoadObjectAsync<T>(string key) => JsonSerializer.Deserialize<T>(await File.ReadAllTextAsync(Path.Join(TargetPath, key)));
+
+    public async Task<bool> ObjectExistsAsync(string key) => File.Exists(Path.Join(TargetPath, key));
+
+    public async Task<List<string>> GetAllKeysAsync() => Directory.GetFiles(TargetPath).Select(Path.GetFileName).ToList();
+
+    public async Task DeleteObjectAsync(string key) => File.Delete(Path.Join(TargetPath, key));
+}
diff --git a/LibMatrix.ExampleBot/Bot/Interfaces/CommandContext.cs b/LibMatrix.ExampleBot/Bot/Interfaces/CommandContext.cs
new file mode 100644
index 0000000..ec61a1e
--- /dev/null
+++ b/LibMatrix.ExampleBot/Bot/Interfaces/CommandContext.cs
@@ -0,0 +1,12 @@
+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
new file mode 100644
index 0000000..393ddbb
--- /dev/null
+++ b/LibMatrix.ExampleBot/Bot/Interfaces/ICommand.cs
@@ -0,0 +1,12 @@
+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
new file mode 100644
index 0000000..eecab84
--- /dev/null
+++ b/LibMatrix.ExampleBot/Bot/MRUBot.cs
@@ -0,0 +1,114 @@
+using System.Diagnostics.CodeAnalysis;
+using LibMatrix.ExampleBot.Bot.Interfaces;
+using LibMatrix.Extensions;
+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.Count()} commands!");
+    }
+
+    /// <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);
+        AuthenticatedHomeServer 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 async Task StopAsync(CancellationToken cancellationToken) {
+        _logger.LogInformation("Shutting down bot!");
+    }
+}
diff --git a/LibMatrix.ExampleBot/Bot/MRUBotConfiguration.cs b/LibMatrix.ExampleBot/Bot/MRUBotConfiguration.cs
new file mode 100644
index 0000000..c7620df
--- /dev/null
+++ b/LibMatrix.ExampleBot/Bot/MRUBotConfiguration.cs
@@ -0,0 +1,12 @@
+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