save webView's interaction state

novasamatech · Nov 11, 2024 · ead20fc · ead20fc
1 parent f27c0ab
commit ead20fc
Show file tree

Hide file tree

Showing 5 changed files with 73 additions and 2 deletions.
diff --git a/novawallet/Common/Extension/SettingsExtension.swift b/novawallet/Common/Extension/SettingsExtension.swift
@@ -25,6 +25,7 @@ enum SettingsKey: String {
     case integrateNetworksBannerSeen
     case assetListGroupStyle
     case assetIconsAppearance
+    case webViewStates
 }
 
 extension SettingsManagerProtocol {
@@ -294,4 +295,15 @@ extension SettingsManagerProtocol {
             )
         }
     }
+
+    var webViewStates: [String: Data]? {
+        get {
+            let anyValue = anyValue(for: SettingsKey.webViewStates.rawValue)
+            return anyValue as? [String: Data]
+        }
+
+        set {
+            set(anyValue: newValue, for: SettingsKey.webViewStates.rawValue)
+        }
+    }
 }
diff --git a/novawallet/Modules/DApp/DAppBrowser/DAppBrowserPresenter.swift b/novawallet/Modules/DApp/DAppBrowser/DAppBrowserPresenter.swift
@@ -15,6 +15,8 @@ final class DAppBrowserPresenter {
     private(set) var favorites: [String: DAppFavorite]?
     private(set) var browserPage: DAppBrowserPage?
 
+    private var currentTabId: UUID?
+
     init(
         interactor: DAppBrowserInteractorInputProtocol,
         wireframe: DAppBrowserWireframeProtocol,
@@ -112,6 +114,18 @@ extension DAppBrowserPresenter: DAppBrowserPresenterProtocol {
                 return
             }
 
+            if
+                let currentTabId,
+                let currentWebView = webViewPool.getWebView(for: currentTabId),
+                let url = currentWebView.url {
+                if #available(iOS 15.0, *) {
+                    let currentWebViewState = currentWebView.interactionState
+                    tabsManager.updateStateForTab(with: url, currentWebViewState)
+                } else {
+                    // Fallback on earlier versions
+                }
+            }
+
             let webView = webViewPool.setupWebView(for: tab.uuid)
 
             let viewModel = DAppBrowserTabViewModel(
@@ -120,6 +134,8 @@ extension DAppBrowserPresenter: DAppBrowserPresenterProtocol {
                 webView: webView
             )
 
+            currentTabId = tab.uuid
+
             view?.didReceiveTab(viewModel: viewModel)
         }
     }
@@ -161,10 +177,12 @@ extension DAppBrowserPresenter: DAppBrowserInteractorOutputProtocol {
 
         let viewModel = DAppBrowserTabViewModel(
             tab: tab,
-            loadRequired: true,
+            loadRequired: tab.state == nil,
             webView: webView
         )
 
+        currentTabId = tab.uuid
+
         view?.didReceiveTab(viewModel: viewModel)
     }
 

diff --git a/novawallet/Modules/DApp/DAppBrowser/DAppBrowserViewController.swift b/novawallet/Modules/DApp/DAppBrowser/DAppBrowserViewController.swift
@@ -362,6 +362,11 @@ extension DAppBrowserViewController: DAppBrowserViewProtocol {
         transports = viewModel.tab.transports
 
         rootView.setWebView(viewModel.webView)
+        if #available(iOS 15.0, *) {
+            rootView.webView.interactionState = viewModel.tab.state
+        } else {
+            // Fallback on earlier versions
+        }
 
         configure()
 

diff --git a/novawallet/Modules/DApp/DAppBrowser/Model/DAppBrowserModel.swift b/novawallet/Modules/DApp/DAppBrowser/Model/DAppBrowserModel.swift
@@ -12,6 +12,7 @@ struct DAppBrowserTabModel {
     let url: URL
     let isDesktop: Bool
     let transports: [DAppTransportModel]
+    let state: Data?
 }
 
 struct DAppBrowserTabViewModel {

diff --git a/novawallet/Modules/DApp/DAppBrowser/TabsManager/DAppBrowserTabsManager.swift b/novawallet/Modules/DApp/DAppBrowser/TabsManager/DAppBrowserTabsManager.swift
@@ -1,26 +1,34 @@
 import Foundation
 import WebKit
+import SoraKeystore
 
 protocol DAppBrowserTabsManagerProtocol {
     func createTab(for dappBrowserModel: DAppBrowserModel) -> DAppBrowserTabModel
     func fetchTab(for id: UUID) -> DAppBrowserTabModel?
     func fetchAllTabs() -> [DAppBrowserTabModel]
+    func updateStateForTab(with url: URL, _ state: Any?)
 }
 
 class DAppBrowserTabsManager {
     private var tabs: [UUID: DAppBrowserTabModel] = [:]
 
+    private let settingsManager: SettingsManagerProtocol = SettingsManager.shared
+
     static let shared = DAppBrowserTabsManager()
 }
 
 extension DAppBrowserTabsManager: DAppBrowserTabsManagerProtocol {
     func createTab(for dappBrowserModel: DAppBrowserModel) -> DAppBrowserTabModel {
+        let state = settingsManager.webViewStates?[dappBrowserModel.url.absoluteString]
+
         let uuid = UUID()
+
         let tab = DAppBrowserTabModel(
             uuid: uuid,
             url: dappBrowserModel.url,
             isDesktop: dappBrowserModel.isDesktop,
-            transports: dappBrowserModel.transports
+            transports: dappBrowserModel.transports,
+            state: state
         )
 
         tabs[uuid] = tab
@@ -35,4 +43,31 @@ extension DAppBrowserTabsManager: DAppBrowserTabsManagerProtocol {
     func fetchAllTabs() -> [DAppBrowserTabModel] {
         Array(tabs.values)
     }
+
+    func updateStateForTab(with url: URL, _ state: Any?) {
+        guard let nsData = state as? NSData else {
+            return
+        }
+
+        let key = url.absoluteString
+
+        var dict: [String: Data] = settingsManager.webViewStates ?? [:]
+
+        let data = Data(referencing: nsData)
+        dict[key] = data
+
+        settingsManager.webViewStates = dict
+
+        if let tab = tabs.values.first(where: { $0.url == url }) {
+            let updatedTab = DAppBrowserTabModel(
+                uuid: tab.uuid,
+                url: tab.url,
+                isDesktop: tab.isDesktop,
+                transports: tab.transports,
+                state: data
+            )
+
+            tabs[tab.uuid] = updatedTab
+        }
+    }
 }