From ef7e0ecf0dd8aa1fc0827d0bd419be1f9e6093ab Mon Sep 17 00:00:00 2001
From: Blatzar <46196380+Blatzar@users.noreply.github.com>
Date: Sat, 28 Jan 2023 00:54:47 +0100
Subject: [PATCH] Fixed rotation and loading delay

---
 .../ui/library/LibraryFragment.kt             |  1 +
 .../ui/library/LibraryViewModel.kt            |  1 -
 .../ui/library/ViewpagerAdapter.kt            | 20 ++++++++++++++-----
 3 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/app/src/main/java/com/lagradost/cloudstream3/ui/library/LibraryFragment.kt b/app/src/main/java/com/lagradost/cloudstream3/ui/library/LibraryFragment.kt
index 97fc4b72..5e2e5c45 100644
--- a/app/src/main/java/com/lagradost/cloudstream3/ui/library/LibraryFragment.kt
+++ b/app/src/main/java/com/lagradost/cloudstream3/ui/library/LibraryFragment.kt
@@ -360,6 +360,7 @@ class LibraryFragment : Fragment() {
     }
 
     override fun onConfigurationChanged(newConfig: Configuration) {
+        (viewpager.adapter as? ViewpagerAdapter)?.rebind()
         super.onConfigurationChanged(newConfig)
     }
 }
diff --git a/app/src/main/java/com/lagradost/cloudstream3/ui/library/LibraryViewModel.kt b/app/src/main/java/com/lagradost/cloudstream3/ui/library/LibraryViewModel.kt
index 07f88fbf..5f64880c 100644
--- a/app/src/main/java/com/lagradost/cloudstream3/ui/library/LibraryViewModel.kt
+++ b/app/src/main/java/com/lagradost/cloudstream3/ui/library/LibraryViewModel.kt
@@ -97,7 +97,6 @@ class LibraryViewModel : ViewModel() {
                     )
                 }
 
-                delay(5000)
                 _pages.postValue(Resource.Success(pages))
             }
         }
diff --git a/app/src/main/java/com/lagradost/cloudstream3/ui/library/ViewpagerAdapter.kt b/app/src/main/java/com/lagradost/cloudstream3/ui/library/ViewpagerAdapter.kt
index 3e31e875..33a40386 100644
--- a/app/src/main/java/com/lagradost/cloudstream3/ui/library/ViewpagerAdapter.kt
+++ b/app/src/main/java/com/lagradost/cloudstream3/ui/library/ViewpagerAdapter.kt
@@ -28,18 +28,28 @@ class ViewpagerAdapter(
     override fun onBindViewHolder(holder: RecyclerView.ViewHolder, position: Int) {
         when (holder) {
             is PageViewHolder -> {
-                holder.bind(pages[position])
+                holder.bind(pages[position], unbound.remove(position))
             }
         }
     }
 
+    private val unbound = mutableSetOf<Int>()
+    /**
+     * Used to mark all pages for re-binding and forces all items to be refreshed
+     * Without this the pages will still use the same adapters
+     **/
+    fun rebind() {
+        unbound.addAll(0..pages.size)
+        this.notifyItemRangeChanged(0, pages.size)
+    }
+
     inner class PageViewHolder(private val itemViewTest: View) :
         RecyclerView.ViewHolder(itemViewTest) {
-        fun bind(page: SyncAPI.Page) {
-            if (itemViewTest.page_recyclerview?.adapter == null) {
-                itemView.page_recyclerview?.spanCount =
-                    this@PageViewHolder.itemView.context.getSpanCount() ?: 3
+        fun bind(page: SyncAPI.Page, rebind: Boolean) {
+            itemView.page_recyclerview?.spanCount =
+                this@PageViewHolder.itemView.context.getSpanCount() ?: 3
 
+            if (itemViewTest.page_recyclerview?.adapter == null || rebind) {
                 // Only add the items after it has been attached since the items rely on ItemWidth
                 // Which is only determined after the recyclerview is attached.
                 // If this fails then item height becomes 0 when there is only one item