Configured retry button in FilesFragment

itszechs · Jun 21, 2022 · 950422c · 950422c
1 parent 65fad07
commit 950422c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/app/src/main/java/zechs/drive/stream/ui/files/FilesFragment.kt b/app/src/main/java/zechs/drive/stream/ui/files/FilesFragment.kt
@@ -132,9 +132,14 @@ class FilesFragment : BaseFragment() {
             }
             is Resource.Loading -> {
                 isLoading = true
-                if (!viewModel.hasLoaded) {
+                if (!viewModel.hasLoaded || viewModel.hasFailed) {
                     isLoading(true)
                 }
+                binding.error.root.apply {
+                    if (isVisible) {
+                        isGone = true
+                    }
+                }
             }
         }
     }
@@ -186,6 +191,12 @@ class FilesFragment : BaseFragment() {
             error.apply {
                 root.isVisible = true
                 errorTxt.text = msg ?: getString(R.string.something_went_wrong)
+                btnRetry.apply {
+                    isVisible = true
+                    setOnClickListener {
+                        viewModel.queryFiles(args.query)
+                    }
+                }
             }
         }
         isLoading = false

diff --git a/app/src/main/java/zechs/drive/stream/ui/files/FilesViewModel.kt b/app/src/main/java/zechs/drive/stream/ui/files/FilesViewModel.kt
@@ -38,6 +38,9 @@ class FilesViewModel @Inject constructor(
 
     var hasLoaded = false
 
+    var hasFailed = false
+        private set
+
     var isLastPage = nextPageToken == null
         private set
 
@@ -57,6 +60,7 @@ class FilesViewModel @Inject constructor(
                 } else {
                     getTeamDrives(drive)
                 }
+                hasFailed = false
             } else {
                 _filesList.postValue(Resource.Error("Unable to build Drive Service"))
                 Log.d(TAG, "Unable to build Drive Service")
@@ -78,6 +82,7 @@ class FilesViewModel @Inject constructor(
             _filesList.postValue(Resource.Error(e.message ?: "Something went wrong"))
             Log.e(TAG, "Something went wrong", e)
         }
+        hasFailed = true
     }
 
     private suspend fun getQuery(drive: Drive, query: String) {