From c190753e1df67dd0a124f31d2bcbffa375b41447 Mon Sep 17 00:00:00 2001
From: gnuxie <Gnuxie@protonmail.com>
Date: Fri, 16 Feb 2024 17:19:47 +0000
Subject: [PATCH] Enable MPS's Member/ServerBanSynchronisationProtections by
 default.

---
 .../DefaultEnabledProtectionsMigration.ts     | 37 ++++++++++++++++++-
 1 file changed, 35 insertions(+), 2 deletions(-)

diff --git a/src/protections/DefaultEnabledProtectionsMigration.ts b/src/protections/DefaultEnabledProtectionsMigration.ts
index b01f8c26..c8a9ee09 100644
--- a/src/protections/DefaultEnabledProtectionsMigration.ts
+++ b/src/protections/DefaultEnabledProtectionsMigration.ts
@@ -3,7 +3,7 @@
  * All rights reserved.
  */
 
-import { ActionError,DRAUPNIR_SCHEMA_VERSION_KEY, MjolnirEnabledProtectionsEvent, MjolnirEnabledProtectionsEventType, Ok, SchemedDataManager, Value } from "matrix-protection-suite";
+import { ActionError,ActionException,ActionExceptionKind,DRAUPNIR_SCHEMA_VERSION_KEY, MjolnirEnabledProtectionsEvent, MjolnirEnabledProtectionsEventType, Ok, SchemedDataManager, Value, findProtection } from "matrix-protection-suite";
 
 export const DefaultEnabledProtectionsMigration = new SchemedDataManager<MjolnirEnabledProtectionsEvent>([
   async function enableBanPropagationByDefault(input) {
@@ -12,11 +12,44 @@ export const DefaultEnabledProtectionsMigration = new SchemedDataManager<Mjolnir
         `The data for ${MjolnirEnabledProtectionsEventType} is corrupted.`
       );
     }
+    const banPropagationProtection = findProtection('BanPropagationProtection');
+    if (banPropagationProtection === undefined) {
+        const message = `Cannot find the BanPropagationProtection`;
+        return ActionException.Result(message, {
+            exception: new TypeError(message),
+            exceptionKind: ActionExceptionKind.Unknown,
+        })
+    }
     const enabled = new Set(input.enabled);
-    enabled.add('BanPropagationProtection');
+    enabled.add(banPropagationProtection.name);
     return Ok({
       enabled: [...enabled],
       [DRAUPNIR_SCHEMA_VERSION_KEY]: 1,
     });
   },
+  async function enableMemberAndServerSynchronisationByDefault(input) {
+    if (!Value.Check(MjolnirEnabledProtectionsEvent, input)) {
+        return ActionError.Result(
+          `The data for ${MjolnirEnabledProtectionsEventType} is corrupted.`
+        );
+    }
+    const enabledProtections = new Set(input.enabled);
+    // we go through the process of finding them just so we can be sure that we spell their names correctly.
+    const memberBanSynchronisationProtection = findProtection('MemberBanSynchronisationProtection');
+    const serverBanSynchronisationProtection = findProtection('ServerBanSynchronisationProtection');
+    if (memberBanSynchronisationProtection === undefined || serverBanSynchronisationProtection === undefined) {
+        const message = `Cannot find the member ban or server ban synchronisation protections`;
+        return ActionException.Result(message, {
+            exception: new TypeError(message),
+            exceptionKind: ActionExceptionKind.Unknown
+        });
+    }
+    for (const protection of [memberBanSynchronisationProtection, serverBanSynchronisationProtection]) {
+        enabledProtections.add(protection.name);
+    }
+    return Ok({
+        enabled: [...enabledProtections],
+        [DRAUPNIR_SCHEMA_VERSION_KEY]: 2,
+    });
+  }
 ]);