fix: check if secrets are cached instead of checking sync

basically checked everywhere if the accountData update would have been used, if not just removed the checks

lib/encryption/ssss.dart

@@ -251,13 +251,9 @@ class SSSS {
 
     final accountDataType = EventTypes.secretStorageKey(keyId);
     // noooow we set the account data
-    final waitForAccountData = client.onSync.stream.firstWhere((syncUpdate) =>
-        syncUpdate.accountData != null &&
-        syncUpdate.accountData!
-            .any((accountData) => accountData.type == accountDataType));
+
     await client.setAccountData(
         client.userID!, accountDataType, content.toJson());
-    await waitForAccountData;
 
     final key = open(keyId);
     await key.setPrivateKey(privateKey);
@@ -747,6 +743,11 @@ class OpenSSSS {
       throw Exception('SSSS not unlocked');
     }
     await ssss.store(type, secret, keyId, privateKey, add: add);
+
+    while (!(await ssss.encryption.keyManager.isCached())) {
+      Logs().v('Wait for secret to come down sync');
+      await ssss.client.oneShotSync(timeout: Duration(seconds: 3));
+    }
   }
 
   Future<void> validateAndStripOtherKeys(String type, String secret) async {

lib/encryption/utils/bootstrap.dart

@@ -27,6 +27,7 @@ import 'package:matrix/encryption/key_manager.dart';
 import 'package:matrix/encryption/ssss.dart';
 import 'package:matrix/encryption/utils/base64_unpadded.dart';
 import 'package:matrix/matrix.dart';
+import 'package:matrix/src/utils/map_copy_extension.dart';
 
 enum BootstrapState {
   /// Is loading.
@@ -292,12 +293,15 @@ class Bootstrap {
         }
         // alright, we re-encrypted all the secrets. We delete the dead weight only *after* we set our key to the default key
       }
-      final updatedAccountData = client.onSync.stream.firstWhere((syncUpdate) =>
-          syncUpdate.accountData != null &&
-          syncUpdate.accountData!.any((accountData) =>
-              accountData.type == EventTypes.SecretStorageDefaultKey));
       await encryption.ssss.setDefaultKeyId(newSsssKey!.keyId);
-      await updatedAccountData;
+      // how useful is this? wouldn't a old ssss be fine here anyway?
+      while (client.accountData
+          .where((_, BasicEvent event) =>
+              event.type == EventTypes.SecretStorageDefaultKey)
+          .isNotEmpty) {
+        Logs().v('Waiting accountData to have m.secret_storage.default_key');
+        await client.oneShotSync(timeout: Duration(seconds: 3));
+      }
       if (oldSsssKeys != null) {
         for (final entry in secretMap!.entries) {
           Logs().v('Validate and stripe other keys ${entry.key}...');
@@ -479,33 +483,20 @@ class Bootstrap {
               ));
       Logs().v('Device signing keys have been uploaded.');
       // aaaand set the SSSS secrets
-      final futures = <Future<void>>[];
       if (masterKey != null) {
-        futures.add(
-          client.onSync.stream
-              .firstWhere((syncUpdate) =>
-                  masterKey?.publicKey != null &&
-                  client.userDeviceKeys[client.userID]?.masterKey?.ed25519Key ==
-                      masterKey?.publicKey)
-              .then((_) => Logs().v('New Master Key was created')),
-        );
+        while (!(masterKey.publicKey != null &&
+            client.userDeviceKeys[client.userID]?.masterKey?.ed25519Key ==
+                masterKey.publicKey)) {
+          Logs().v('Waiting for master to be created');
+          await client.oneShotSync(timeout: Duration(seconds: 3));
+        }
       }
       for (final entry in secretsToStore.entries) {
-        futures.add(
-          client.onSync.stream
-              .firstWhere((syncUpdate) =>
-                  syncUpdate.accountData != null &&
-                  syncUpdate.accountData!
-                      .any((accountData) => accountData.type == entry.key))
-              .then((_) =>
-                  Logs().v('New Key with type ${entry.key} was created')),
-        );
         Logs().v('Store new SSSS key ${entry.key}...');
         await newSsssKey?.store(entry.key, entry.value);
       }
       Logs().v(
           'Wait for MasterKey and ${secretsToStore.entries.length} keys to be created');
-      await Future.wait<void>(futures);
       final keysToSign = <SignableKey>[];
       if (masterKey != null) {
         if (client.userDeviceKeys[client.userID]?.masterKey?.ed25519Key !=
@@ -581,14 +572,6 @@ class Bootstrap {
       );
       Logs().v('Store the secret...');
       await newSsssKey?.store(megolmKey, base64.encode(privKey));
-      Logs().v('Wait for secret to come down sync');
-
-      if (!await encryption.keyManager.isCached()) {
-        await client.onSync.stream.firstWhere((syncUpdate) =>
-            syncUpdate.accountData != null &&
-            syncUpdate.accountData!
-                .any((accountData) => accountData.type == megolmKey));
-      }
 
       Logs().v(
           'And finally set all megolm keys as needing to be uploaded again...');

lib/src/client.dart

@@ -1641,8 +1641,8 @@ class Client extends MatrixApi {
 
   /// Immediately start a sync and wait for completion.
   /// If there is an active sync already, wait for the active sync instead.
-  Future<void> oneShotSync() {
-    return _sync();
+  Future<void> oneShotSync({Duration? timeout}) {
+    return _sync(timeout: timeout);
   }
 
   /// Pass a timeout to set how long the server waits before sending an empty response.

lib/src/utils/map_copy_extension.dart

@@ -16,7 +16,7 @@
  *   along with this program.  If not, see <https://www.gnu.org/licenses/>.
  */
 
-extension MapCopyExtension on Map<String, dynamic> {
+extension MapExtension<K, V> on Map<K, V> {
   /// Deep-copies a given json map
   Map<String, dynamic> copy() {
     final copy = Map<String, dynamic>.from(this);
@@ -30,4 +30,14 @@ extension MapCopyExtension on Map<String, dynamic> {
     }
     return copy;
   }
+
+  Map<K, V> where(bool Function(K, V) condition) {
+    final Map<K, V> result = {};
+    for (final element in entries) {
+      if (condition(element.key, element.value)) {
+        result[element.key] = element.value;
+      }
+    }
+    return result;
+  }
 }

test/matrix_api/map_copy_extension_test.dart

@@ -21,8 +21,8 @@ import 'package:test/test.dart';
 import 'package:matrix/src/utils/map_copy_extension.dart';
 
 void main() {
-  group('Map-copy-extension', () {
-    test('it should work', () {
+  group('map extension', () {
+    test('copy-extension should work', () {
       final original = <String, dynamic>{
         'attr': 'fox',
         'child': <String, dynamic>{
@@ -37,4 +37,12 @@ void main() {
       expect(copy['child']['list'], [1, 2]);
     });
   });
+  test('where-extension should work', () {
+    final original = <String, dynamic>{
+      'attr': 'fox',
+      'attr2': 'fox2',
+    };
+    expect(original.where((k, v) => k == 'attr').length, 1);
+    expect(original.where((k, v) => k == 'attr'), {'attr': 'fox'});
+  });
 }