From 63227a1f1fce2523d811fb2ed9e930f8b570ab2a Mon Sep 17 00:00:00 2001
From: Alexander Capehart <alex@oxycblt.org>
Date: Mon, 17 Mar 2025 06:37:36 -0600
Subject: [PATCH] musikr: fix incorrect cache cleanups

---
 .../org/oxycblt/musikr/pipeline/ExtractStep.kt     | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/musikr/src/main/java/org/oxycblt/musikr/pipeline/ExtractStep.kt b/musikr/src/main/java/org/oxycblt/musikr/pipeline/ExtractStep.kt
index d6a2cb214..8ca2fa5ad 100644
--- a/musikr/src/main/java/org/oxycblt/musikr/pipeline/ExtractStep.kt
+++ b/musikr/src/main/java/org/oxycblt/musikr/pipeline/ExtractStep.kt
@@ -99,12 +99,22 @@ private class ExtractStepImpl(
                                 it.song.cover?.id,
                                 it.song.addedMs)
                         cache.write(cachedSong)
-                        exclude.add(cachedSong)
                         Finalized(it.song)
                     }
                 }
             }
-            .map { it.extracted }
+            .map {
+                if (it.extracted is RawSong) {
+                    exclude.add(
+                        CachedSong(
+                            it.extracted.file,
+                            it.extracted.properties,
+                            it.extracted.tags,
+                            it.extracted.cover?.id,
+                            it.extracted.addedMs))
+                }
+                it.extracted
+            }
             .flowOn(Dispatchers.IO)
             .buffer(Channel.UNLIMITED)
             .onCompletion { cache.cleanup(exclude) }