Improve data saver handling

2021-12-12 12:16:07 -05:00
parent df0083e2e3
commit 6e8ee38238
3 changed files with 18 additions and 15 deletions
@@ -2,7 +2,11 @@ package exh.util

 import com.tfcporciuncula.flow.Preference
 import eu.kanade.tachiyomi.data.preference.PreferencesHelper
-import tachiyomi.source.Source
+import eu.kanade.tachiyomi.source.Source
+import eu.kanade.tachiyomi.source.model.Page
+import eu.kanade.tachiyomi.source.online.HttpSource
+import okhttp3.Response
+import rx.Observable

 interface DataSaver {

@@ -14,6 +18,15 @@ interface DataSaver {
                return imageUrl
            }
        }
+
+        fun HttpSource.fetchImage(page: Page, dataSaver: DataSaver): Observable<Response> {
+            val imageUrl = page.imageUrl ?: return fetchImage(page)
+            page.imageUrl = dataSaver.compress(imageUrl)
+            return fetchImage(page)
+                .doOnNext {
+                    page.imageUrl = imageUrl
+                }
+        }
    }
 }