From aaf490c35145c29019cb1bb3702a68440e932c69 Mon Sep 17 00:00:00 2001
From: Doug <6060466+pixlwave@users.noreply.github.com>
Date: Thu, 12 Sep 2024 10:38:07 +0100
Subject: [PATCH] Make sure we reset the feature flag when upgrading to SSS.
 (#3272)

---
 ElementX/Sources/Screens/HomeScreen/HomeScreenViewModel.swift | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ElementX/Sources/Screens/HomeScreen/HomeScreenViewModel.swift b/ElementX/Sources/Screens/HomeScreen/HomeScreenViewModel.swift
index 36228d23c..e55fd0259 100644
--- a/ElementX/Sources/Screens/HomeScreen/HomeScreenViewModel.swift
+++ b/ElementX/Sources/Screens/HomeScreen/HomeScreenViewModel.swift
@@ -143,6 +143,7 @@ class HomeScreenViewModel: HomeScreenViewModelType, HomeScreenViewModelProtocol
         case .skipRecoveryKeyConfirmation:
             state.securityBannerMode = .dismissed
         case .confirmSlidingSyncUpgrade:
+            appSettings.slidingSyncDiscovery = .native
             actionsSubject.send(.logout)
         case .skipSlidingSyncUpgrade:
             state.slidingSyncMigrationBannerMode = .dismissed
@@ -326,6 +327,7 @@ class HomeScreenViewModel: HomeScreenViewModelType, HomeScreenViewModelProtocol
                                                               title: L10n.bannerMigrateToNativeSlidingSyncForceLogoutTitle,
                                                               primaryButton: .init(title: L10n.bannerMigrateToNativeSlidingSyncAction,
                                                                                    action: { [weak self] in
+                                                                                       self?.appSettings.slidingSyncDiscovery = .native
                                                                                        self?.actionsSubject.send(.logoutWithoutConfirmation)
                                                                                    }))
                 }