about summary refs log tree commit diff
path: root/MatrixUtils.RoomUpgradeCLI/Commands/NewFromRoomDirCommand.cs
diff options
context:
space:
mode:
Diffstat (limited to 'MatrixUtils.RoomUpgradeCLI/Commands/NewFromRoomDirCommand.cs')
-rw-r--r--MatrixUtils.RoomUpgradeCLI/Commands/NewFromRoomDirCommand.cs115
1 files changed, 115 insertions, 0 deletions
diff --git a/MatrixUtils.RoomUpgradeCLI/Commands/NewFromRoomDirCommand.cs b/MatrixUtils.RoomUpgradeCLI/Commands/NewFromRoomDirCommand.cs
new file mode 100644

index 0000000..40ab791 --- /dev/null +++ b/MatrixUtils.RoomUpgradeCLI/Commands/NewFromRoomDirCommand.cs
@@ -0,0 +1,115 @@ +using ArcaneLibs.Extensions; +using LibMatrix.EventTypes.Spec.State.RoomInfo; +using LibMatrix.Helpers; +using LibMatrix.Homeservers; +using MatrixUtils.RoomUpgradeCLI.Extensions; + +namespace MatrixUtils.RoomUpgradeCLI.Commands; + +public class NewFromRoomDirCommand(ILogger<NewFromRoomDirCommand> logger, IHost host, RuntimeContext ctx, AuthenticatedHomeserverGeneric hs) : IHostedService { + public async Task StartAsync(CancellationToken cancellationToken) { + if (ctx.Args.Length <= 1) { + await PrintHelp(); + return; + } + + var dirName = ctx.Args[1]; + if (dirName.StartsWith("--")) { + Console.WriteLine("Directory name cannot start with --, please provide a valid directory name."); + await PrintHelp(); + } + + if (Directory.Exists(dirName)) + Directory.Delete(dirName, true); + Directory.CreateDirectory(dirName); + List<Task> tasks = []; + await foreach (var rooms in hs.EnumeratePublicRoomsAsync().WithCancellation(cancellationToken)) { + // foreach (var room in rooms.Chunk) { } + tasks.AddRange(rooms.Chunk.Select(x=> ProcessRoom(dirName, x))); + } + await Task.WhenAll(tasks); + + // var rb = ctx.Args.Contains("--upgrade") ? new RoomUpgradeBuilder() : new RoomBuilder(); + // + // // check for room membership! + // if (rb is RoomUpgradeBuilder rub) { + + // } + await host.StopAsync(cancellationToken); + } + + private async Task ProcessRoom(string dirName, PublicRoomDirectoryResult.PublicRoomListItem roomListItem) { + Console.WriteLine(roomListItem.Name ?? roomListItem.RoomId); + var room = hs.GetRoom(roomListItem.RoomId); + var rb = new RoomUpgradeBuilder() { + OldRoomId = roomListItem.RoomId + }; + + await rb.ApplyRoomUpgradeCLIArgs(hs, ctx.Args[2..], isNewState: true); + try { + var membership = await room.GetStateAsync<RoomMemberEventContent>(RoomMemberEventContent.EventId, hs.UserId); + } + catch (Exception e) { + Console.WriteLine("Error checking room membership: " + e.Message); + Console.WriteLine("Please ensure you are a member of the room you are trying to upgrade. -- ABORTING --"); + await host.StopAsync(); + return; + } + + await rb.ImportAsync(hs.GetRoom(roomListItem.RoomId)); + + var validFileNameChars = (roomListItem.Name ?? roomListItem.CanonicalAlias ?? roomListItem.RoomId) + // .Replace('&', '_') + // .Replace(':', '_') + // .Replace('\'', '_') + // .Replace(' ', '_') + .ToList(); + validFileNameChars.RemoveAll(Path.GetInvalidFileNameChars().Contains); + var filename = string.Join("", validFileNameChars); + while (File.Exists(filename)) + filename += "_"; + + await File.WriteAllTextAsync(dirName + "/" + filename + ".json", rb.ToJson()); + } + + public async Task StopAsync(CancellationToken cancellationToken) { } + + private async Task PrintHelp() { + Console.WriteLine("Usage: new [filename] [options]"); + Console.WriteLine("Options:"); + Console.WriteLine(" --help Show this help message"); + Console.WriteLine(" --version <version> Set the room version (e.g. 9, 10, 11, 12)"); + Console.WriteLine("-- New room options --"); + Console.WriteLine(" --alias <alias> Set the room alias (local part)"); + Console.WriteLine(" --avatar-url <url> Set the room avatar URL"); + Console.WriteLine(" --copy-avatar <roomId> Copy the avatar from an existing room"); + Console.WriteLine(" --copy-powerlevels <roomId> Copy power levels from an existing room"); + Console.WriteLine(" --invite-admin <userId> Invite a user as an admin (userId must start with '@')"); + Console.WriteLine(" --invite <userId> Invite a user (userId must start with '@')"); + Console.WriteLine(" --name <name> Set the room name (can be multiple words)"); + Console.WriteLine(" --topic <topic> Set the room topic (can be multiple words)"); + Console.WriteLine(" --federate <true|false> Set whether the room is federatable"); + Console.WriteLine(" --public Set the room join rule to public"); + Console.WriteLine(" --invite-only Set the room join rule to invite-only"); + Console.WriteLine(" --knock Set the room join rule to knock"); + Console.WriteLine(" --restricted Set the room join rule to restricted"); + Console.WriteLine(" --knock_restricted Set the room join rule to knock_restricted"); + Console.WriteLine(" --private Set the room join rule to private"); + Console.WriteLine(" --join-rule <rule> Set the room join rule (public, invite, knock, restricted, knock_restricted, private)"); + Console.WriteLine(" --history-visibility <visibility> Set the room history visibility (shared, invited, joined, world_readable)"); + Console.WriteLine(" --type <type> Set the room type (e.g. m.space, m.room, support.feline.policy.list.msc.v1 etc.)"); + // upgrade opts + Console.WriteLine("-- Upgrade options --"); + Console.WriteLine( + " --upgrade <roomId> Create a room upgrade file instead of a new room file - WARNING: incompatible with non-upgrade options"); + Console.WriteLine(" --invite-members Invite members during room upgrade"); + Console.WriteLine(" --invite-powerlevel-users Invite users with power levels during room upgrade"); + Console.WriteLine(" --migrate-bans Migrate bans during room upgrade"); + Console.WriteLine(" --migrate-empty-state-events Migrate empty state events during room upgrade"); + Console.WriteLine(" --upgrade-unstable-values Upgrade unstable values during room upgrade"); + Console.WriteLine(" --msc4321-policy-list-upgrade <move|transition> Upgrade MSC4321 policy list"); + Console.WriteLine( + "WARNING: The --upgrade option is incompatible with options listed under \"New room\", please use the equivalent options in the `modify` command instead."); + await host.StopAsync(); + } +} \ No newline at end of file