fix folder-download/upload notification success message

0xNaimulHasanTaky · Feb 11, 2024 · 2b1bfb5 · 2b1bfb5
1 parent a8b5f36
commit 2b1bfb5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/...n/java/de/felixnuesse/extract/notifications/implementations/DownloadWorkerNotification.kt b/...n/java/de/felixnuesse/extract/notifications/implementations/DownloadWorkerNotification.kt
@@ -33,8 +33,8 @@ class DownloadWorkerNotification(var context: Context) : WorkerNotification(cont
     override fun generateSuccessMessage(statusObject: StatusObject, fileItem: FileItem): String {
 
         val transfers = statusObject.getTotalTransfers()
-        val message = if (transfers <= 1) {
-            statusObject.getTotalTransfers().toString()
+        val message = if (transfers <= 1 && !fileItem.isDir) {
+            statusObject.getTransfers().toString()
         } else {
             fileItem.name
         }

diff --git a/...ain/java/de/felixnuesse/extract/notifications/implementations/UploadWorkerNotification.kt b/...ain/java/de/felixnuesse/extract/notifications/implementations/UploadWorkerNotification.kt
@@ -30,8 +30,8 @@ class UploadWorkerNotification(var context: Context) : WorkerNotification(contex
 
     override fun generateSuccessMessage(statusObject: StatusObject, fileItem: FileItem): String {
         val transfers = statusObject.getTotalTransfers()
-        val message = if (transfers <= 1) {
-            statusObject.getTotalTransfers().toString()
+        val message = if (transfers <= 1 && !fileItem.isDir) {
+            statusObject.getTransfers().toString()
         } else {
             fileItem.name
         }