From 182a821fc5668072898e456eedadb9f424f4a960 Mon Sep 17 00:00:00 2001
From: nyne <me@nyne.dev>
Date: Sun, 15 Jun 2025 14:45:13 +0800
Subject: [PATCH] Fixed the issue where the download task would stop after
 exiting the reader. Close #387

---
 lib/network/download.dart | 2 +-
 lib/network/images.dart   | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/lib/network/download.dart b/lib/network/download.dart
index bb8c6d3..7ab6777 100644
--- a/lib/network/download.dart
+++ b/lib/network/download.dart
@@ -552,7 +552,7 @@ class _ImageDownloadWrapper {
   void start() async {
     int lastBytes = 0;
     try {
-      await for (var p in ImageDownloader.loadComicImage(
+      await for (var p in ImageDownloader.loadComicImageUnwrapped(
           image, task.source.key, task.comicId, chapter)) {
         if (isCancelled) {
           return;
diff --git a/lib/network/images.dart b/lib/network/images.dart
index 26189f7..74749ac 100644
--- a/lib/network/images.dart
+++ b/lib/network/images.dart
@@ -111,6 +111,11 @@ abstract class ImageDownloader {
     return stream.stream;
   }
 
+  static Stream<ImageDownloadProgress> loadComicImageUnwrapped(
+      String imageKey, String? sourceKey, String cid, String eid) {
+    return _loadComicImage(imageKey, sourceKey, cid, eid);
+  }
+
   static Stream<ImageDownloadProgress> _loadComicImage(
       String imageKey, String? sourceKey, String cid, String eid) async* {
     final cacheKey = "$imageKey@$sourceKey@$cid@$eid";