From 359e1b42c8c3bc4f58d33dbceb40afa161e7ede9 Mon Sep 17 00:00:00 2001
From: Nicolas Buquet <nbuquet@buquet-net.com>
Date: Wed, 29 Nov 2023 18:08:56 +0100
Subject: [PATCH] =?UTF-8?q?Nouvelle=20impl=C3=A9mentation=20de=20l'activat?=
 =?UTF-8?q?ion=20automatique=20du=20cross-signing=20ne=20n=C3=A9cessitant?=
 =?UTF-8?q?=20pas=20de=20modification=20dans=20Matrix-ios-SDK.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Riot/Modules/Home/HomeViewController.m | 68 +++++++++++++++++++-------
 1 file changed, 51 insertions(+), 17 deletions(-)

diff --git a/Riot/Modules/Home/HomeViewController.m b/Riot/Modules/Home/HomeViewController.m
index 8a8df9f2e..20b01ff6c 100644
--- a/Riot/Modules/Home/HomeViewController.m
+++ b/Riot/Modules/Home/HomeViewController.m
@@ -40,7 +40,7 @@ @interface HomeViewController () <SecureBackupSetupCoordinatorBridgePresenterDel
     CGFloat selectedCollectionViewContentOffset;
     
     // Tchap: initial sync done listener
-    id tchapInitialSyncDoneListener;
+    id tchapSyncDoneListener;
 }
 
 @property (nonatomic, strong) SecureBackupSetupCoordinatorBridgePresenter *secureBackupSetupCoordinatorBridgePresenter;
@@ -104,20 +104,55 @@ - (void)viewDidLoad
 
     // Change the table data source. It must be the home view controller itself.
     self.recentsTableView.dataSource = self;
-    
-    // Tchap: listen to Tchap InitialSyncDone notification added by Tchap in Matrix SDK.
-    if (tchapInitialSyncDoneListener == nil)
+}
+
+// Tchap: set-up cross-signing auto-activation
+- (void)tchapSetupCrossSigningAutoActivation
+{
+    // Tchap: listen to Tchap syncDone notification to check the status of the cross-signing after account info are updated.
+    if (tchapSyncDoneListener == nil)
     {
-        tchapInitialSyncDoneListener = [NSNotificationCenter.defaultCenter addObserverForName:kTchapMXSessionInitialSyncDone
-                                                                                       object:nil
-                                                                                        queue:NSOperationQueue.mainQueue
-                                                                                   usingBlock:^(NSNotification * _Nonnull notification) {
-            // Try to force activate cross-signing after initial sync.
-            [self showCrossSigningSetup];
+        __weak typeof(self) weakSelf = self;
+        
+        tchapSyncDoneListener = [NSNotificationCenter.defaultCenter addObserverForName:kMXSessionDidSyncNotification
+                                                                                object:nil
+                                                                                 queue:NSOperationQueue.mainQueue
+                                                                            usingBlock:^(NSNotification * _Nonnull notification) {
+            
+            __strong __typeof(weakSelf)strongSelf = weakSelf;
+            MXSession *session = strongSelf.mainSession;
+            
+            // Remove listener after first invocation.
+            [self tchapUnsetupCrossSigningAutoActivation];
+            
+            // Try to force activate cross-signing after sync if:
+            //   - device is the only one listed on the account (no risk of conflict with other devices)
+            //   - Cross-signing is not enabled for this account.
+            //   - or Cross-signing exists on the account but is not trusted by this device. The account password can be asked to activate cross-signing in this case.
+            NSString *currentUserId = session.myUserId;
+            MXCrossSigningState state = session.crypto.crossSigning.state;
+            if (currentUserId != nil
+                && [session.crypto devicesForUser:currentUserId].count < 2
+                && (session.crypto.crossSigning.state == MXCrossSigningStateNotBootstrapped || session.crypto.crossSigning.state == MXCrossSigningStateCrossSigningExists) )
+            {
+                [strongSelf showCrossSigningSetup];
+            }
         }];
     }
 }
 
+// Tchap: remove intial sync done listener
+- (void)tchapUnsetupCrossSigningAutoActivation
+{
+    if (tchapSyncDoneListener)
+    {
+        [NSNotificationCenter.defaultCenter removeObserver:tchapSyncDoneListener
+                                                      name:kMXSessionDidSyncNotification
+                                                    object:nil];
+        tchapSyncDoneListener = nil;
+    }
+}
+
 - (void)viewWillAppear:(BOOL)animated
 {
     [super viewWillAppear:animated];
@@ -138,6 +173,10 @@ - (void)viewWillAppear:(BOOL)animated
         [recentsDataSource searchWithPatterns:nil];
         [self.recentsSearchBar setText:nil];
     }
+    
+    // Tchap: set-up cross-signing auto-activation in viewWillAppear rather than in viewDidLoad
+    // because this controller is only created once and never released!
+    [self tchapSetupCrossSigningAutoActivation];
 }
 
 - (void)viewWillTransitionToSize:(CGSize)size withTransitionCoordinator:(id <UIViewControllerTransitionCoordinator>)coordinator
@@ -154,13 +193,8 @@ - (void)viewWillTransitionToSize:(CGSize)size withTransitionCoordinator:(id <UIV
 - (void)destroy
 {
     // Tchap: remove intial sync done listener
-    if (tchapInitialSyncDoneListener)
-    {
-        [NSNotificationCenter.defaultCenter removeObserver:tchapInitialSyncDoneListener
-                                                      name:kTchapMXSessionInitialSyncDone
-                                                    object:nil];
-        tchapInitialSyncDoneListener = nil;
-    }
+    [self tchapUnsetupCrossSigningAutoActivation];
+    
     [super destroy];
 }