about summary refs log tree commit diff
path: root/MatrixUtils.RoomUpgradeCLI/Extensions/RoomBuilderExtensions.cs
blob: 75852bcd504c3ebf99dcf21f60a92a557e6fe456 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
using LibMatrix.EventTypes.Spec.State.RoomInfo;
using LibMatrix.Helpers;
using LibMatrix.Homeservers;

namespace MatrixUtils.RoomUpgradeCLI.Extensions;

public static class RoomBuilderExtensions {
    public static async Task ApplyRoomUpgradeCLIArgs(this RoomBuilder rb, AuthenticatedHomeserverGeneric hs, string[] args, bool isNewState = false) {
        for (int i = 0; i < args.Length; i++) {
            // Console.WriteLine($"Parsing arg {i}: {args[i]}");
            switch (args[i]) {
                case "--alias":
                    rb.AliasLocalPart = args[++i];
                    break;
                case "--avatar-url":
                    rb.Avatar!.Url = args[++i];
                    break;
                case "--copy-avatar": {
                    var room = hs.GetRoom(args[++i]);
                    rb.Avatar = await room.GetAvatarUrlAsync() ?? throw new ArgumentException($"Room {room.RoomId} does not have an avatar");
                    break;
                }
                case "--copy-powerlevels": {
                    var room = hs.GetRoom(args[++i]);
                    rb.PowerLevels = await room.GetPowerLevelsAsync() ?? throw new ArgumentException($"Room {room.RoomId} does not have power levels???");
                    break;
                }
                case "--invite-admin":
                    var inviteAdmin = args[++i];
                    if (!inviteAdmin.StartsWith('@')) {
                        throw new ArgumentException("Invalid user reference: " + inviteAdmin);
                    }

                    rb.Invites.Add(inviteAdmin, "Marked explicitly as admin to be invited");
                    break;
                case "--invite":
                    var inviteUser = args[++i];
                    if (!inviteUser.StartsWith('@')) {
                        throw new ArgumentException("Invalid user reference: " + inviteUser);
                    }

                    rb.Invites.Add(inviteUser, "Marked explicitly to be invited");
                    break;
                case "--name":
                    var nameEvt = rb.Name = new() { Name = "" };
                    while (i + 1 < args.Length && !args[i + 1].StartsWith("--")) {
                        nameEvt.Name += (nameEvt.Name.Length > 0 ? " " : "") + args[++i];
                    }

                    break;
                case "--topic":
                    var topicEvt = rb.Topic = new() { Topic = "" };
                    while (i + 1 < args.Length && !args[i + 1].StartsWith("--")) {
                        topicEvt.Topic += (topicEvt.Topic.Length > 0 ? " " : "") + args[++i];
                    }

                    break;
                case "--federate":
                    rb.IsFederatable = GetBoolArg(args, ref i, true);
                    break;
                case "--public":
                case "--invite-only":
                case "--knock":
                case "--restricted":
                case "--knock_restricted":
                case "--private":
                    rb.JoinRules.JoinRule = args[i].Replace("--", "").ToLowerInvariant() switch {
                        "public" => RoomJoinRulesEventContent.JoinRules.Public,
                        "invite-only" => RoomJoinRulesEventContent.JoinRules.Invite,
                        "knock" => RoomJoinRulesEventContent.JoinRules.Knock,
                        "restricted" => RoomJoinRulesEventContent.JoinRules.Restricted,
                        "knock_restricted" => RoomJoinRulesEventContent.JoinRules.KnockRestricted,
                        "private" => RoomJoinRulesEventContent.JoinRules.Private,
                        _ => throw new ArgumentException("Unknown join rule: " + args[i])
                    };
                    break;
                case "--join-rule":
                    if (i + 1 >= args.Length || !args[i + 1].StartsWith("--")) {
                        throw new ArgumentException("Expected join rule after --join-rule");
                    }

                    rb.JoinRules.JoinRule = args[++i].ToLowerInvariant() switch {
                        "public" => RoomJoinRulesEventContent.JoinRules.Public,
                        "invite" => RoomJoinRulesEventContent.JoinRules.Invite,
                        "knock" => RoomJoinRulesEventContent.JoinRules.Knock,
                        "restricted" => RoomJoinRulesEventContent.JoinRules.Restricted,
                        "knock_restricted" => RoomJoinRulesEventContent.JoinRules.KnockRestricted,
                        "private" => RoomJoinRulesEventContent.JoinRules.Private,
                        _ => throw new ArgumentException("Unknown join rule: " + args[i])
                    };
                    break;
                case "--history-visibility":
                    rb.HistoryVisibility = new RoomHistoryVisibilityEventContent {
                        HistoryVisibility = args[++i].ToLowerInvariant() switch {
                            "shared" => RoomHistoryVisibilityEventContent.HistoryVisibilityTypes.Shared,
                            "invited" => RoomHistoryVisibilityEventContent.HistoryVisibilityTypes.Invited,
                            "joined" => RoomHistoryVisibilityEventContent.HistoryVisibilityTypes.Joined,
                            "world_readable" => RoomHistoryVisibilityEventContent.HistoryVisibilityTypes.WorldReadable,
                            _ => throw new ArgumentException("Unknown history visibility: " + args[i])
                        }
                    };
                    break;
                case "--type":
                    rb.Type = args[++i];
                    break;
                case "--version":
                    rb.Version = args[++i];
                    // if (!RoomBuilder.V12PlusRoomVersions.Contains(rb.Version)) {
                    // logger.LogWarning("Using room version {Version} which is not v12 or higher, this may cause issues with some features.", rb.Version);
                    // }
                    break;
                case "--encryption":
                    if (args[i + 1].StartsWith("--")) {
                        rb.Encryption.Algorithm = "m.megolm.v1.aes-sha2";
                    }
                    else {
                        rb.Encryption.Algorithm = args[++i];
                        if (rb.Encryption.Algorithm == "null")
                            rb.Encryption.Algorithm = null; // disable encryption
                    }

                    break;
                // upgrade options
                case "--invite-members":
                    if (rb is not RoomUpgradeBuilder upgradeBuilder) {
                        throw new InvalidOperationException("Invite members can only be used with room upgrades");
                    }

                    upgradeBuilder.UpgradeOptions.InviteMembers = GetBoolArg(args, ref i, true);
                    break;
                case "--invite-powerlevel-users":
                case "--invite-power-level-users":
                    if (rb is not RoomUpgradeBuilder upgradeBuilderInvite) {
                        throw new InvalidOperationException("Invite powerlevel users can only be used with room upgrades");
                    }

                    upgradeBuilderInvite.UpgradeOptions.InvitePowerlevelUsers = GetBoolArg(args, ref i, true);
                    break;
                case "--synapse-admin-join-local-users":
                    rb.SynapseAdminAutoAcceptLocalInvites = GetBoolArg(args, ref i, true);
                    break;
                case "--migrate-bans":
                    if (rb is not RoomUpgradeBuilder upgradeBuilderBan) {
                        throw new InvalidOperationException("Migrate bans can only be used with room upgrades");
                    }

                    upgradeBuilderBan.UpgradeOptions.MigrateBans = GetBoolArg(args, ref i, true);
                    break;
                case "--migrate-empty-state-events":
                    if (rb is not RoomUpgradeBuilder upgradeBuilderEmpty) {
                        throw new InvalidOperationException("Migrate empty state events can only be used with room upgrades");
                    }

                    upgradeBuilderEmpty.UpgradeOptions.MigrateEmptyStateEvents = GetBoolArg(args, ref i, true);
                    break;
                case "--upgrade-unstable-values":
                    if (rb is not RoomUpgradeBuilder upgradeBuilderUnstable) {
                        throw new InvalidOperationException("Update unstable values can only be used with room upgrades");
                    }

                    upgradeBuilderUnstable.UpgradeOptions.UpgradeUnstableValues = GetBoolArg(args, ref i, true);
                    break;
                case "--msc4321-policy-list-upgrade":
                    if (rb is not RoomUpgradeBuilder upgradeBuilderPolicy) {
                        throw new InvalidOperationException("MSC4321 policy list upgrade can only be used with room upgrades");
                    }

                    upgradeBuilderPolicy.UpgradeOptions.Msc4321PolicyListUpgradeOptions.Enable = true;
                    upgradeBuilderPolicy.UpgradeOptions.Msc4321PolicyListUpgradeOptions.UpgradeType = args[++i].ToLowerInvariant() switch {
                        "move" => RoomUpgradeBuilder.Msc4321PolicyListUpgradeOptions.Msc4321PolicyListUpgradeType.Move,
                        "transition" => RoomUpgradeBuilder.Msc4321PolicyListUpgradeOptions.Msc4321PolicyListUpgradeType.Transition,
                        _ => throw new ArgumentException("Unknown MSC4321 policy list upgrade type: " + args[i])
                    };
                    break;
                case "--force-upgrade":
                    if (rb is not RoomUpgradeBuilder upgradeBuilderForce) {
                        throw new InvalidOperationException("Force upgrade can only be used with room upgrades");
                    }

                    upgradeBuilderForce.UpgradeOptions.ForceUpgrade = GetBoolArg(args, ref i, true);
                    break;
                case "--noop-upgrade":
                    if (rb is not RoomUpgradeBuilder upgradeBuilderNoop) {
                        throw new InvalidOperationException("No-op upgrade can only be used with room upgrades");
                    }

                    upgradeBuilderNoop.UpgradeOptions.NoopUpgrade = GetBoolArg(args, ref i, true);
                    break;
                case "--upgrade":
                    if (rb is not RoomUpgradeBuilder upgradeBuilderUpgrade) {
                        throw new InvalidOperationException("Upgrade can only be used with room upgrades");
                    }

                    if (isNewState) {
                        upgradeBuilderUpgrade.OldRoomId = args[++i];
                        Console.WriteLine($"Popping arg for --upgrade(isNewState={isNewState}): " + upgradeBuilderUpgrade.OldRoomId);
                    }

                    break;
                case "--help":
                    PrintHelpAndExit();
                    return;
                default:
                    throw new ArgumentException("Unknown argument: " + args[i]);
            }
        }
    }

    private static bool GetBoolArg(string[] args, ref int i, bool defaultValue) {
        if (i + 1 < args.Length && bool.TryParse(args[i + 1], out var result)) {
            i++;
            return result;
        }

        return defaultValue;
    }

    private static void PrintHelpAndExit() {
        Console.WriteLine("""
                            --help                                             Show this help message
                            --version <version>                                Set the room version (e.g. 9, 10, 11, 12)
                          -- New room options --
                            --federate [True|false]                            Set whether the room is federatable [WARNING: Cannot be updated later!]
                            --type <type>                                      Set the room type (e.g. m.space, m.room, support.feline.policy.list.msc.v1 etc.) [WARNING: Cannot be updated later!]
                            --alias <alias>                                    Set the room alias (local part)
                            --avatar-url <url>                                 Set the room avatar URL
                            --copy-avatar <roomId>                             Copy the avatar from an existing room
                            --copy-powerlevels <roomId>                        Copy power levels from an existing room
                            --invite <userId>                                  Invite a user (userId must start with '@')
                            --invite-admin <userId>                            Invite a user as an admin (userId must start with '@')
                            --synapse-admin-join-local-users [True|false]      Automatically accept local user invites during room creation (Synapse only, requires synapse admin access)
                            --name <name>                                      Set the room name (can be multiple words)
                            --topic <topic>                                    Set the room topic (can be multiple words)
                            --join-rule <rule>                                 Set the room join rule (public, invite, knock, restricted, knock_restricted, private)
                                                                               Aliases: --public, --invite, --knock, --restricted, --knock_restricted, --private
                            --history-visibility <visibility>                  Set the room history visibility (shared, invited, joined, world_readable)
                          -- Upgrade options --
                            --upgrade <roomId>                                 Create a room upgrade file instead of a new room file - WARNING: incompatible with non-upgrade options
                            --invite-members [True|false]                      Invite members during room upgrade
                            --invite-local-users [True|false]                  Invite local users during room upgrade (also see --synapse-admin-join-local-users)
                            --invite-powerlevel-users [True|false]             Invite users with power levels during room upgrade
                            --migrate-bans [True|false]                        Migrate bans during room upgrade
                            --migrate-empty-state-events [True|false]          Migrate empty state events during room upgrade
                            --upgrade-unstable-values [True|false]             Upgrade unstable values during room upgrade
                            --msc4321-policy-list-upgrade <move|transition>    Upgrade MSC4321 policy list
                            --force-upgrade [True|false]                       Force upgrade even if you don't have the required permissions
                            --noop-upgrade [True|false]                        Perform the upgrade, but do not tombstone the old room
                          WARNING: The --upgrade option is incompatible with options listed under "New room", please use the equivalent options in the `modify` command instead.
                          """);
        Environment.Exit(0);
    }
}