Fix zip closing

app/src/main/kotlin/org/koitharu/kotatsu/core/zip/ZipOutput.kt

@@ -103,10 +103,13 @@ class ZipOutput(
 				}
 				val zipEntry = ZipEntry(name)
 				putNextEntry(zipEntry)
+				try {
 					fis.copyTo(this)
+				} finally {
 					closeEntry()
 				}
 			}
+		}
 		return true
 	}
 
@@ -117,8 +120,11 @@ class ZipOutput(
 		}
 		val zipEntry = ZipEntry(name)
 		putNextEntry(zipEntry)
+		try {
 			content.byteInputStream().copyTo(this)
+		} finally {
 			closeEntry()
+		}
 		return true
 	}
 }

app/src/main/kotlin/org/koitharu/kotatsu/local/data/output/LocalMangaDirOutput.kt

@@ -6,6 +6,7 @@ import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.runInterruptible
 import kotlinx.coroutines.sync.Mutex
 import kotlinx.coroutines.sync.withLock
+import okhttp3.internal.closeQuietly
 import org.koitharu.kotatsu.core.model.isLocal
 import org.koitharu.kotatsu.core.util.ext.deleteAwait
 import org.koitharu.kotatsu.core.util.ext.takeIfReadable
@@ -90,7 +91,7 @@ class LocalMangaDirOutput(
 
 	override fun close() {
 		for (output in chaptersOutput.values) {
-			output.close()
+			output.closeQuietly()
 		}
 	}
 
@@ -119,10 +120,21 @@ class LocalMangaDirOutput(
 	}
 
 	private suspend fun ZipOutput.flushAndFinish() = runInterruptible(Dispatchers.IO) {
+		val e: Throwable? = try {
 			finish()
+			null
+		} catch (e: Throwable) {
+			e
+		} finally {
 			close()
+		}
+		if (e == null) {
 			val resFile = File(file.absolutePath.removeSuffix(SUFFIX_TMP))
 			file.renameTo(resFile)
+		} else {
+			file.delete()
+			throw e
+		}
 	}
 
 	private fun chapterFileName(chapter: IndexedValue<MangaChapter>): String {

app/src/main/kotlin/org/koitharu/kotatsu/local/data/output/LocalMangaUtil.kt

@@ -2,8 +2,6 @@ package org.koitharu.kotatsu.local.data.output
 
 import androidx.core.net.toFile
 import androidx.core.net.toUri
-import kotlinx.coroutines.Dispatchers
-import kotlinx.coroutines.runInterruptible
 import org.koitharu.kotatsu.core.model.isLocal
 import org.koitharu.kotatsu.parsers.model.Manga
 
@@ -16,26 +14,14 @@ class LocalMangaUtil(
 	}
 
 	suspend fun deleteChapters(ids: Set<Long>) {
-		newOutput().use { output ->
-			when (output) {
-				is LocalMangaZipOutput -> runInterruptible(Dispatchers.IO) {
-					LocalMangaZipOutput.filterChapters(output, ids)
-				}
-
-				is LocalMangaDirOutput -> {
+		val file = manga.url.toUri().toFile()
+		if (file.isDirectory) {
+			LocalMangaDirOutput(file, manga).use { output ->
 				output.deleteChapters(ids)
 				output.finish()
 			}
-			}
-		}
-	}
-
-	private suspend fun newOutput(): LocalMangaOutput = runInterruptible(Dispatchers.IO) {
-		val file = manga.url.toUri().toFile()
-		if (file.isDirectory) {
-			LocalMangaDirOutput(file, manga)
 		} else {
-			LocalMangaZipOutput(file, manga)
+			LocalMangaZipOutput.filterChapters(file, manga, ids)
 		}
 	}
 }

app/src/main/kotlin/org/koitharu/kotatsu/local/data/output/LocalMangaZipOutput.kt

@@ -5,6 +5,7 @@ import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.runInterruptible
 import kotlinx.coroutines.sync.Mutex
 import kotlinx.coroutines.sync.withLock
+import okhttp3.internal.closeQuietly
 import org.koitharu.kotatsu.core.model.isLocal
 import org.koitharu.kotatsu.core.util.ext.deleteAwait
 import org.koitharu.kotatsu.core.util.ext.readText
@@ -52,7 +53,8 @@ class LocalMangaZipOutput(
 		index.setCoverEntry(name)
 	}
 
-	override suspend fun addPage(chapter: IndexedValue<MangaChapter>, file: File, pageNumber: Int, ext: String) = mutex.withLock {
+	override suspend fun addPage(chapter: IndexedValue<MangaChapter>, file: File, pageNumber: Int, ext: String) =
+		mutex.withLock {
 			val name = buildString {
 				append(FILENAME_PATTERN.format(chapter.value.branch.hashCode(), chapter.index + 1, pageNumber))
 				if (ext.isNotEmpty() && ext.length <= 4) {
@@ -70,9 +72,10 @@ class LocalMangaZipOutput(
 
 	override suspend fun finish() = mutex.withLock {
 		runInterruptible(Dispatchers.IO) {
+			output.use { output ->
 				output.put(ENTRY_NAME_INDEX, index.toString())
 				output.finish()
-			output.close()
+			}
 		}
 		rootFile.deleteAwait()
 		output.file.renameTo(rootFile)
@@ -115,8 +118,10 @@ class LocalMangaZipOutput(
 
 		private const val FILENAME_PATTERN = "%08d_%03d%03d"
 
-		@WorkerThread
-		fun filterChapters(subject: LocalMangaZipOutput, idsToRemove: Set<Long>) {
+		suspend fun filterChapters(file: File, manga: Manga, idsToRemove: Set<Long>) =
+			runInterruptible(Dispatchers.IO) {
+				val subject = LocalMangaZipOutput(file, manga)
+				try {
 					ZipFile(subject.rootFile).use { zip ->
 						val index = MangaIndex(zip.readText(zip.getEntry(ENTRY_NAME_INDEX)))
 						idsToRemove.forEach { id -> index.removeChapter(id) }
@@ -151,6 +156,15 @@ class LocalMangaZipOutput(
 						subject.rootFile.delete()
 						subject.output.file.renameTo(subject.rootFile)
 					}
+				} catch (e: Throwable) {
+					subject.closeQuietly()
+					try {
+						subject.output.file.delete()
+					} catch (e2: Throwable) {
+						e.addSuppressed(e2)
+					}
+					throw e
+				}
 			}
 	}
 }