From 8864be681443f7b847297fae8aef7dc8e692c6d9 Mon Sep 17 00:00:00 2001
From: rhunk <101876869+rhunk@users.noreply.github.com>
Date: Sat, 4 May 2024 19:00:16 +0200
Subject: [PATCH] fix(download_processor): handle duplicates - checks if the
 file already exists and if it does, compares its contents with the input
 file, if contents differ, deletes existing file

---
 .../snapenhance/download/DownloadProcessor.kt | 29 ++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

diff --git a/app/src/main/kotlin/me/rhunk/snapenhance/download/DownloadProcessor.kt b/app/src/main/kotlin/me/rhunk/snapenhance/download/DownloadProcessor.kt
index 49545dc8..4e083d91 100644
--- a/app/src/main/kotlin/me/rhunk/snapenhance/download/DownloadProcessor.kt
+++ b/app/src/main/kotlin/me/rhunk/snapenhance/download/DownloadProcessor.kt
@@ -89,7 +89,6 @@ class DownloadProcessor (
 
     private fun newFFMpegProcessor(pendingTask: PendingTask) = FFMpegProcessor.newFFMpegProcessor(remoteSideContext, pendingTask)
 
-    @SuppressLint("UnspecifiedRegisterReceiverFlag")
     suspend fun saveMediaToGallery(pendingTask: PendingTask, inputFile: File, metadata: DownloadMetadata) {
         if (coroutineContext.job.isCancelled) return
 
@@ -129,6 +128,34 @@ class DownloadProcessor (
                 }
             }
 
+            // checks if the file already exists and if it does, compares its contents with the input file, if contents differ, deletes existing file.
+            outputFileFolder.findFile(fileName)?.let { existingFile ->
+                pendingTask.updateProgress("Comparing existing media")
+                remoteSideContext.androidContext.contentResolver.openInputStream(existingFile.uri)?.use { existingInputStream ->
+                    val buffer1 = ByteArray(1024 * 1024)
+                    val buffer2 = ByteArray(1024 * 1024)
+                    var read1: Int
+                    var read2: Int
+
+                    inputFile.inputStream().use { inputStream ->
+                        while (true) {
+                            read1 = inputStream.read(buffer1)
+                            read2 = existingInputStream.read(buffer2)
+                            if (read1 != read2 || !buffer1.contentEquals(buffer2)) {
+                                existingFile.delete()
+                                return@let
+                            }
+                            if (read1 == -1) break
+                        }
+                    }
+                }
+
+                pendingTask.task.extra = existingFile.uri.toString()
+                pendingTask.success()
+                callbackOnFailure(translation["already_downloaded_toast"])
+                return
+            }
+
             val outputFile = outputFileFolder.createFile(fileType.mimeType, fileName)!!
 
             pendingTask.updateProgress("Saving media to gallery")