From 7963342a148771e4bb4aae86be5ebf2b9cb180bb Mon Sep 17 00:00:00 2001
From: Mustafa Merza <mustafa.merza95@gmail.com>
Date: Mon, 12 Aug 2024 13:53:36 +0300
Subject: [PATCH] - Added updating folders when file manager screen appears.

---
 .../Modules/Documents/DocumentsTableViewController.swift | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/MiniScanner/Modules/Documents/DocumentsTableViewController.swift b/MiniScanner/Modules/Documents/DocumentsTableViewController.swift
index 34a6a5a..117b381 100644
--- a/MiniScanner/Modules/Documents/DocumentsTableViewController.swift
+++ b/MiniScanner/Modules/Documents/DocumentsTableViewController.swift
@@ -108,8 +108,6 @@ extension DocumentsTableViewController {
     
     private func setupUI() {
         
-        fetchFolders()
-        
         setupToolbar()
         
         setupTableView()
@@ -197,6 +195,8 @@ extension DocumentsTableViewController {
         
         self.tabBarController?.tabBar.isHidden = false
         
+        fetchFolders()
+        
         fetchScanSessions()
         
         handleBackIcon()
@@ -234,9 +234,12 @@ extension DocumentsTableViewController {
     
     private func fetchFolders() {
         
+        let index = folders.firstIndex { $0.id == selectedFolder?.id } ?? 0
+        
         var folders = getFoldersUseCase.execute()
         folders.insert(.allDocuments, at: 0)
-        folders[0].isSelected = true
+        
+        folders[index].isSelected = true
         
         self.folders = folders
         
-- 
GitLab