Room upgrade CLI changes

1 files changed, 35 insertions, 0 deletions

diff --git a/MatrixUtils.RoomUpgradeCLI/Commands/ImportUpgradeStateCommand.cs b/MatrixUtils.RoomUpgradeCLI/Commands/ImportUpgradeStateCommand.cs
new file mode 100644

index 0000000..960905b
--- /dev/null
+++ b/MatrixUtils.RoomUpgradeCLI/Commands/ImportUpgradeStateCommand.cs
@@ -0,0 +1,35 @@
+using System.Text.Json;
+using ArcaneLibs.Extensions;
+using LibMatrix.Helpers;
+using LibMatrix.Homeservers;
+
+namespace MatrixUtils.RoomUpgradeCLI.Commands;
+
+public class ImportUpgradeStateCommand(ILogger<ImportUpgradeStateCommand> logger, IHost host, RuntimeContext ctx, AuthenticatedHomeserverGeneric hs) : IHostedService {
+    public async Task StartAsync(CancellationToken cancellationToken) {
+        if (ctx.Args.Length <= 1) {
+            await PrintHelp();
+            return;
+        }
+        var filename = ctx.Args[1];
+        if (filename.StartsWith("--")) {
+            Console.WriteLine("Filename cannot start with --, please provide a valid filename.");
+            await PrintHelp();
+        }
+
+        var rb = await JsonSerializer.DeserializeAsync<RoomUpgradeBuilder>(File.OpenRead(filename));
+        await rb!.ImportAsync(hs.GetRoom(rb.OldRoomId));
+        await File.WriteAllTextAsync(filename, rb.ToJson(), cancellationToken);
+
+        await host.StopAsync(cancellationToken);
+    }
+
+    public async Task StopAsync(CancellationToken cancellationToken) { }
+
+    private async Task PrintHelp() {
+        Console.WriteLine("Usage: import-upgrade-state [filename]");
+        Console.WriteLine("Options:");
+        Console.WriteLine("  --help     Show this help message");
+        await host.StopAsync();
+    }
+}
\ No newline at end of file