diff options
author | TheArcaneBrony <myrainbowdash949@gmail.com> | 2023-08-14 04:26:27 +0200 |
---|---|---|
committer | TheArcaneBrony <myrainbowdash949@gmail.com> | 2023-08-14 04:26:27 +0200 |
commit | 8fdc48e5b21b1eea61534b181585858727500f34 (patch) | |
tree | 71cc2655ca1e2b5556c2c7c1ad550c5df6b499ef /LibMatrix.ExampleBot | |
parent | Split LibMatrix into separate repo (diff) | |
download | LibMatrix-8fdc48e5b21b1eea61534b181585858727500f34.tar.xz |
Move more projects over
Diffstat (limited to 'LibMatrix.ExampleBot')
-rw-r--r-- | LibMatrix.ExampleBot/Bot/Commands/CmdCommand.cs | 46 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/Bot/Commands/HelpCommand.cs | 28 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/Bot/Commands/PingCommand.cs | 17 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/Bot/FileStorageProvider.cs | 35 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/Bot/Interfaces/CommandContext.cs | 12 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/Bot/Interfaces/ICommand.cs | 12 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/Bot/MRUBot.cs | 114 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/Bot/MRUBotConfiguration.cs | 12 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/LibMatrix.ExampleBot.csproj | 32 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/Program.cs | 28 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/Properties/launchSettings.json | 26 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/appsettings.Development.json | 9 | ||||
-rw-r--r-- | LibMatrix.ExampleBot/appsettings.json | 13 |
13 files changed, 384 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 diff --git a/LibMatrix.ExampleBot/LibMatrix.ExampleBot.csproj b/LibMatrix.ExampleBot/LibMatrix.ExampleBot.csproj new file mode 100644 index 0000000..03a3f0b --- /dev/null +++ b/LibMatrix.ExampleBot/LibMatrix.ExampleBot.csproj @@ -0,0 +1,32 @@ +<Project Sdk="Microsoft.NET.Sdk"> + + <PropertyGroup> + <OutputType>Exe</OutputType> + <TargetFramework>net8.0</TargetFramework> + <LangVersion>preview</LangVersion> + <ImplicitUsings>enable</ImplicitUsings> + <Nullable>enable</Nullable> + <PublishAot>false</PublishAot> + <InvariantGlobalization>true</InvariantGlobalization> + <PublishTrimmed>true</PublishTrimmed> + <PublishReadyToRun>true</PublishReadyToRun> + <PublishSingleFile>true</PublishSingleFile> + <PublishReadyToRunShowWarnings>true</PublishReadyToRunShowWarnings> + <PublishTrimmedShowLinkerSizeComparison>true</PublishTrimmedShowLinkerSizeComparison> + <PublishTrimmedShowLinkerSizeComparisonWarnings>true</PublishTrimmedShowLinkerSizeComparisonWarnings> + </PropertyGroup> + + <ItemGroup> + <ProjectReference Include="..\LibMatrix\LibMatrix.csproj" /> + </ItemGroup> + + <ItemGroup> + <PackageReference Include="ArcaneLibs" Version="1.0.0-preview3020494760.012ed3f" /> + <PackageReference Include="Microsoft.Extensions.Hosting" Version="8.0.0-preview.5.23280.8" /> + </ItemGroup> + <ItemGroup> + <Content Include="appsettings*.json"> + <CopyToOutputDirectory>Always</CopyToOutputDirectory> + </Content> + </ItemGroup> +</Project> diff --git a/LibMatrix.ExampleBot/Program.cs b/LibMatrix.ExampleBot/Program.cs new file mode 100644 index 0000000..93a5f27 --- /dev/null +++ b/LibMatrix.ExampleBot/Program.cs @@ -0,0 +1,28 @@ +// See https://aka.ms/new-console-template for more information + +using LibMatrix.ExampleBot.Bot; +using LibMatrix.ExampleBot.Bot.Interfaces; +using LibMatrix.Extensions; +using LibMatrix.Services; +using Microsoft.Extensions.DependencyInjection; +using Microsoft.Extensions.Hosting; + +Console.WriteLine("Hello, World!"); + +var host = Host.CreateDefaultBuilder(args).ConfigureServices((_, services) => { + services.AddScoped<TieredStorageService>(x => + new( + cacheStorageProvider: new FileStorageProvider("bot_data/cache/"), + dataStorageProvider: new FileStorageProvider("bot_data/data/") + ) + ); + services.AddScoped<MRUBotConfiguration>(); + services.AddRoryLibMatrixServices(); + foreach (var commandClass in new ClassCollector<ICommand>().ResolveFromAllAccessibleAssemblies()) { + Console.WriteLine($"Adding command {commandClass.Name}"); + services.AddScoped(typeof(ICommand), commandClass); + } + services.AddHostedService<MRUBot>(); +}).UseConsoleLifetime().Build(); + +await host.RunAsync(); diff --git a/LibMatrix.ExampleBot/Properties/launchSettings.json b/LibMatrix.ExampleBot/Properties/launchSettings.json new file mode 100644 index 0000000..997e294 --- /dev/null +++ b/LibMatrix.ExampleBot/Properties/launchSettings.json @@ -0,0 +1,26 @@ +{ + "$schema": "http://json.schemastore.org/launchsettings.json", + "profiles": { + "Default": { + "commandName": "Project", + "dotnetRunMessages": true, + "environmentVariables": { + + } + }, + "Development": { + "commandName": "Project", + "dotnetRunMessages": true, + "environmentVariables": { + "DOTNET_ENVIRONMENT": "Development" + } + }, + "Local config": { + "commandName": "Project", + "dotnetRunMessages": true, + "environmentVariables": { + "DOTNET_ENVIRONMENT": "Local" + } + } + } +} diff --git a/LibMatrix.ExampleBot/appsettings.Development.json b/LibMatrix.ExampleBot/appsettings.Development.json new file mode 100644 index 0000000..27bbd50 --- /dev/null +++ b/LibMatrix.ExampleBot/appsettings.Development.json @@ -0,0 +1,9 @@ +{ + "Logging": { + "LogLevel": { + "Default": "Debug", + "System": "Information", + "Microsoft": "Information" + } + } +} \ No newline at end of file diff --git a/LibMatrix.ExampleBot/appsettings.json b/LibMatrix.ExampleBot/appsettings.json new file mode 100644 index 0000000..5668b53 --- /dev/null +++ b/LibMatrix.ExampleBot/appsettings.json @@ -0,0 +1,13 @@ +{ + "Logging": { + "LogLevel": { + "Default": "Debug", + "System": "Information", + "Microsoft": "Information" + } + }, + "Bot": { + "Homeserver": "rory.gay", + "AccessToken": "syt_xxxxxxxxxxxxxxxxx" + } +} \ No newline at end of file |