1 files changed, 98 insertions, 0 deletions
diff --git a/Utilities/LibMatrix.Utilities.Bot/BotServiceInstaller.cs b/Utilities/LibMatrix.Utilities.Bot/BotServiceInstaller.cs
new file mode 100644
index 0000000..ff0bc9e
--- /dev/null
+++ b/Utilities/LibMatrix.Utilities.Bot/BotServiceInstaller.cs
@@ -0,0 +1,98 @@
+using ArcaneLibs;
+using LibMatrix.Homeservers;
+using LibMatrix.Services;
+using LibMatrix.Utilities.Bot.AppServices;
+using LibMatrix.Utilities.Bot.Interfaces;
+using LibMatrix.Utilities.Bot.Services;
+using Microsoft.Extensions.DependencyInjection;
+
+namespace LibMatrix.Utilities.Bot;
+
+public static class BotServiceInstallerExtensions {
+    public static BotServiceInstaller AddMatrixBot(this IServiceCollection services) {
+        return new BotServiceInstaller(services).AddMatrixBot();
+    }
+}
+
+public class BotServiceInstaller(IServiceCollection services) {
+    public BotServiceInstaller AddMatrixBot() {
+        services.AddSingleton<LibMatrixBotConfiguration>();
+
+        services.AddSingleton<AuthenticatedHomeserverGeneric>(x => {
+            var config = x.GetService<LibMatrixBotConfiguration>() ?? throw new Exception("No configuration found!");
+            var hsProvider = x.GetService<HomeserverProviderService>() ?? throw new Exception("No homeserver provider found!");
+
+            if (x.GetService<AppServiceConfiguration>() is AppServiceConfiguration appsvcConfig)
+                config.AccessToken = appsvcConfig.AppserviceToken;
+            else if (Environment.GetEnvironmentVariable("LIBMATRIX_ACCESS_TOKEN_PATH") is string path)
+                config.AccessTokenPath = path;
+
+            if (string.IsNullOrWhiteSpace(config.AccessToken) && string.IsNullOrWhiteSpace(config.AccessTokenPath))
+                throw new Exception("Unable to add bot service without an access token or access token path!");
+
+            if (!string.IsNullOrWhiteSpace(config.AccessTokenPath)) {
+                var token = File.ReadAllText(config.AccessTokenPath);
+                config.AccessToken = token.Trim();
+            }
+
+            var hs = hsProvider.GetAuthenticatedWithToken(config.Homeserver, config.AccessToken).Result;
+
+            return hs;
+        });
+
+        return this;
+    }
+
+    public BotServiceInstaller AddCommandHandler() {
+        Console.WriteLine("Adding command handler...");
+        services.AddSingleton(s => s.GetRequiredService<LibMatrixBotConfiguration>().CommandListener
+                                   ?? throw new Exception("Command handling is enabled, but configuration is missing the LibMatrixBot:CommandListener configuration section!")
+        );
+        services.AddHostedService<CommandListenerHostedService>();
+        return this;
+    }
+
+    public BotServiceInstaller DiscoverAllCommands() {
+        foreach (var commandClass in ClassCollector<ICommand>.ResolveFromAllAccessibleAssemblies()) {
+            Console.WriteLine($"Adding command {commandClass.Name}");
+            services.AddScoped(typeof(ICommand), commandClass);
+        }
+
+        return this;
+    }
+
+    public BotServiceInstaller AddCommands(IEnumerable<Type> commandClasses) {
+        foreach (var commandClass in commandClasses) {
+            if (!commandClass.IsAssignableTo(typeof(ICommand)))
+                throw new Exception($"Type {commandClass.Name} is not assignable to ICommand!");
+            Console.WriteLine($"Adding command {commandClass.Name}");
+            services.AddScoped(typeof(ICommand), commandClass);
+        }
+
+        return this;
+    }
+
+    public BotServiceInstaller WithCommandResultHandler(Func<CommandResult, Task> commandResultHandler) {
+        services.AddSingleton(commandResultHandler);
+        return this;
+    }
+
+    public BotServiceInstaller WithInviteHandler(Func<RoomInviteContext, Task> inviteHandler) {
+        services.AddSingleton(inviteHandler);
+        services.AddSingleton(s => s.GetRequiredService<LibMatrixBotConfiguration>().InviteListener
+                                   ?? throw new Exception("Invite handling is enabled, but configuration is missing the LibMatrixBot:InviteListener configuration section!")
+        );
+        services.AddHostedService<InviteHandlerHostedService>();
+        return this;
+    }
+
+    public BotServiceInstaller WithInviteHandler<T>() where T : class, IRoomInviteHandler {
+        services.AddSingleton<T>();
+        services.AddSingleton<Func<RoomInviteContext, Task>>(sp => sp.GetRequiredService<T>().HandleInviteAsync);
+        services.AddSingleton(s => s.GetRequiredService<LibMatrixBotConfiguration>().InviteListener
+                                   ?? throw new Exception("Invite handling is enabled, but configuration is missing the LibMatrixBot:InviteListener configuration section!")
+        );
+        services.AddHostedService<InviteHandlerHostedService>();
+        return this;
+    }
+}
\ No newline at end of file