package me.eternal.purrfect.download

import android.content.ContentUris
import android.content.ContentValues
import android.content.Intent
import android.graphics.Bitmap
import android.graphics.BitmapFactory
import android.net.Uri
import android.os.Build
import android.os.Environment
import android.provider.MediaStore
import android.widget.Toast
import androidx.documentfile.provider.DocumentFile
import com.google.gson.GsonBuilder
import kotlinx.coroutines.*
import kotlinx.coroutines.sync.Semaphore
import kotlinx.coroutines.sync.withPermit
import me.eternal.purrfect.RemoteSideContext
import me.eternal.purrfect.bridge.DownloadCallback
import me.eternal.purrfect.common.Constants
import me.eternal.purrfect.common.ReceiversConfig
import me.eternal.purrfect.common.data.FileType
import me.eternal.purrfect.common.data.download.DownloadMediaType
import me.eternal.purrfect.common.data.download.DownloadMetadata
import me.eternal.purrfect.common.data.download.DownloadRequest
import me.eternal.purrfect.common.data.download.InputMedia
import me.eternal.purrfect.common.data.download.SplitMediaAssetType
import me.eternal.purrfect.common.util.snap.MediaDownloaderHelper
import me.eternal.purrfect.common.util.snap.RemoteMediaResolver
import me.eternal.purrfect.core.features.impl.downloader.decoder.AttachmentType
import me.eternal.purrfect.task.PendingTask
import me.eternal.purrfect.task.PendingTaskListener
import me.eternal.purrfect.task.Task
import me.eternal.purrfect.task.TaskStatus
import me.eternal.purrfect.task.TaskType
import java.io.File
import java.io.FileOutputStream
import java.io.InputStream
import java.net.HttpURLConnection
import java.net.URL
import java.util.concurrent.ConcurrentHashMap
import javax.xml.parsers.DocumentBuilderFactory
import javax.xml.transform.TransformerFactory
import javax.xml.transform.dom.DOMSource
import javax.xml.transform.stream.StreamResult
import kotlin.coroutines.coroutineContext
import kotlin.io.encoding.Base64
import kotlin.io.encoding.ExperimentalEncodingApi

data class DownloadedFile(
    val file: File,
    val fileType: FileType
)

/**
 * DownloadProcessor handles the download requests of the user
 */
@OptIn(ExperimentalEncodingApi::class)
class DownloadProcessor (
    private val remoteSideContext: RemoteSideContext,
    private val callback: DownloadCallback
) {
    companion object {
        private val downloadSemaphore = Semaphore(3)
    }

    private data class GallerySaveResult(
        val uri: Uri,
        val alreadyDownloaded: Boolean = false
    )

    private val translation by lazy {
        remoteSideContext.translation.getCategory("download_processor")
    }

    private val gson by lazy { GsonBuilder().setPrettyPrinting().create() }

    private fun fallbackToast(message: Any) {
        android.os.Handler(remoteSideContext.androidContext.mainLooper).post {
            Toast.makeText(remoteSideContext.androidContext, message.toString(), Toast.LENGTH_SHORT).show()
        }
    }

    private fun callbackOnSuccess(path: String) = runCatching {
        callback.onSuccess(path)
    }.onFailure {
        fallbackToast(it)
    }

    private fun callbackOnFailure(message: String, throwable: String? = null) = runCatching {
        callback.onFailure(message, throwable)
    }.onFailure {
        fallbackToast("$message\n$throwable")
    }

    private fun callbackOnProgress(message: String) = runCatching {
        callback.onProgress(message)
    }.onFailure {
        fallbackToast(it)
    }

    private fun newFFMpegProcessor(pendingTask: PendingTask) = FFMpegProcessor.newFFMpegProcessor(remoteSideContext, pendingTask)

    suspend fun saveMediaToGallery(pendingTask: PendingTask, inputFile: File, metadata: DownloadMetadata) {
        if (coroutineContext.job.isCancelled) return

        runCatching {
            var fileType = FileType.fromFile(inputFile)

            if (fileType.isImage) {
                remoteSideContext.config.root.downloader.forceImageFormat.getNullable()?.let { format ->
                    val bitmap = BitmapFactory.decodeFile(inputFile.absolutePath) ?: throw Exception("Failed to decode bitmap")
                    @Suppress("DEPRECATION") val compressFormat = when (format) {
                        "png" -> Bitmap.CompressFormat.PNG
                        "jpg" -> Bitmap.CompressFormat.JPEG
                        "webp" -> Bitmap.CompressFormat.WEBP
                        else -> throw Exception("Invalid image format")
                    }

                    pendingTask.updateProgress("Converting image to $format")
                    inputFile.outputStream().use {
                        bitmap.compress(compressFormat, 100, it)
                    }
                    bitmap.recycle()
                    fileType = FileType.fromFile(inputFile)
                }
            }

            val fileName = buildOutputFileName(metadata.outputPath, fileType)
            val configuredFolder = remoteSideContext.config.root.downloader.saveFolder.get().orEmpty().trim()
            val saveResult = if (configuredFolder.isBlank()) {
                saveToSystemDefault(fileName, fileType, inputFile, metadata)
            } else {
                runCatching {
                    saveToConfiguredFolder(
                        configuredFolder = configuredFolder,
                        fileName = fileName,
                        fileType = fileType,
                        inputFile = inputFile,
                        metadata = metadata,
                        pendingTask = pendingTask
                    )
                }.onFailure {
                    remoteSideContext.log.error("Failed to save to configured folder, falling back to system default", it)
                }.getOrNull() ?: saveToSystemDefault(fileName, fileType, inputFile, metadata)
            } ?: throw Exception("Failed to save media (no output uri)")

            pendingTask.task.extra = saveResult.uri.toString()
            pendingTask.success()

            if (saveResult.alreadyDownloaded) {
                callbackOnFailure(translation["already_downloaded_toast"])
                return
            }

            runCatching {
                remoteSideContext.androidContext.sendBroadcast(Intent("android.intent.action.MEDIA_SCANNER_SCAN_FILE").apply {
                    data = saveResult.uri
                })
            }.onFailure {
                remoteSideContext.log.error("Failed to scan media file", it)
                callbackOnFailure(translation.format("failed_gallery_toast", "error" to it.toString()), it.message)
            }

            remoteSideContext.log.verbose("download complete")
            callbackOnSuccess(fileName)
        }.onFailure { exception ->
            remoteSideContext.log.error("Failed to save media to gallery", exception)
            callbackOnFailure(translation.format("failed_gallery_toast", "error" to exception.toString()), exception.message)
            pendingTask.fail("Failed to save media to gallery")
        }
    }

    private fun streamsMatch(stream1: InputStream, stream2: InputStream): Boolean {
        stream1.use { s1 ->
            stream2.use { s2 ->
                val buffer1 = ByteArray(1024 * 1024)
                val buffer2 = ByteArray(1024 * 1024)
                while (true) {
                    val read1 = s1.read(buffer1)
                    val read2 = s2.read(buffer2)
                    if (read1 != read2) return false
                    if (read1 == -1) return true
                    for (i in 0 until read1) {
                        if (buffer1[i] != buffer2[i]) return false
                    }
                }
            }
        }
    }

    private fun findExistingMediaUri(collection: Uri, fileName: String, relativePath: String): Uri? {
        val projection = arrayOf(MediaStore.MediaColumns._ID)
        val selection = "${MediaStore.MediaColumns.DISPLAY_NAME} = ? AND ${MediaStore.MediaColumns.RELATIVE_PATH} = ?"
        val selectionArgs = arrayOf(fileName, relativePath)
        return remoteSideContext.androidContext.contentResolver.query(
            collection,
            projection,
            selection,
            selectionArgs,
            null
        )?.use { cursor ->
            if (!cursor.moveToFirst()) return@use null
            val id = cursor.getLong(cursor.getColumnIndexOrThrow(MediaStore.MediaColumns._ID))
            ContentUris.withAppendedId(collection, id)
        }
    }

    private fun contentMatches(uri: Uri, inputFile: File): Boolean {
        val existingInputStream = remoteSideContext.androidContext.contentResolver.openInputStream(uri) ?: return false
        return streamsMatch(existingInputStream, inputFile.inputStream())
    }

    private fun buildOutputFileName(outputPath: String, fileType: FileType): String {
        val rawName = outputPath.trimEnd('/').substringAfterLast("/").ifBlank { "media" }
        val targetExtension = fileType.fileExtension?.lowercase() ?: "dat"
        
        val baseName = if (rawName.lowercase().endsWith(".$targetExtension")) {
            rawName.substringBeforeLast(".")
        } else {
            rawName
        }
        
        return "${sanitizeFileName(baseName)}.$targetExtension"
    }

    private fun sanitizeFileName(name: String): String {
        return name
            .replace(Regex("[\\\\/:*?\"<>|]"), "_")
            .replace(Regex("\\p{Cntrl}"), "")
            .replace(Regex("\\s+"), " ")
            .trim()
            .trim('.')
            .ifBlank { "media" }
    }

    private fun sanitizeRelativePath(path: String): String {
        return path.trimEnd('/').split("/")
            .mapNotNull { segment ->
                segment.trim()
                    .takeIf { it.isNotBlank() }
                    ?.let(::sanitizeFileName)
            }
            .joinToString("/")
    }

    private fun appendNameSuffix(fileName: String, index: Int): String {
        val extension = fileName.substringAfterLast('.', "")
        val baseName = fileName.substringBeforeLast(".", fileName)
        return if (extension.isBlank()) {
            "$baseName ($index)"
        } else {
            "$baseName ($index).$extension"
        }
    }

    private fun saveToConfiguredFolder(
        configuredFolder: String,
        fileName: String,
        fileType: FileType,
        inputFile: File,
        metadata: DownloadMetadata,
        pendingTask: PendingTask,
    ): GallerySaveResult {
        val outputFolder = DocumentFile.fromTreeUri(remoteSideContext.androidContext, Uri.parse(configuredFolder))
            ?: throw Exception("Failed to open output folder")

        val outputFileFolder = metadata.outputPath.let {
            if (it.contains("/")) {
                it.substringBeforeLast("/").split("/").fold(outputFolder) { folder, name ->
                    folder.findFile(name)
                        ?: folder.createDirectory(name)
                        ?: throw Exception("Failed to create output directory $name")
                }
            } else {
                outputFolder
            }
        }

        var finalFileName = fileName
        var collisionCount = 0
        
        while (true) {
            val existingFile = outputFileFolder.findFile(finalFileName) ?: break
            
            if (existingFile.length() == inputFile.length() && !remoteSideContext.config.root.downloader.allowDuplicate.get()) {
                val existingInputStream = remoteSideContext.androidContext.contentResolver.openInputStream(existingFile.uri)
                if (existingInputStream != null && streamsMatch(existingInputStream, inputFile.inputStream())) {
                    return GallerySaveResult(existingFile.uri, alreadyDownloaded = true)
                }
            }
            
            collisionCount++
            finalFileName = appendNameSuffix(fileName, collisionCount)
        }

        val outputFile = outputFileFolder.createFile(fileType.mimeType, finalFileName)
            ?: throw Exception("Failed to create output file $finalFileName")

        pendingTask.updateProgress("Saving media to gallery")
        val outputStream = remoteSideContext.androidContext.contentResolver.openOutputStream(outputFile.uri)
            ?: throw Exception("Failed to open output stream for $finalFileName")

        outputStream.use { currentOutputStream ->
            inputFile.inputStream().use { inputStream ->
                inputStream.copyTo(currentOutputStream)
            }
        }

        return GallerySaveResult(outputFile.uri)
    }

    private fun saveToSystemDefault(
        fileName: String,
        fileType: FileType,
        inputFile: File,
        metadata: DownloadMetadata,
    ): GallerySaveResult? {
        val subPath = sanitizeRelativePath(
            metadata.outputPath.substringBeforeLast("/", missingDelimiterValue = "")
                .replace("\\", "/")
        )
        val baseRelative = when {
            fileType.isImage -> Environment.DIRECTORY_PICTURES
            fileType.isVideo -> Environment.DIRECTORY_MOVIES
            else -> Environment.DIRECTORY_DOWNLOADS
        }
        val relativePath = listOfNotNull(baseRelative, "Purrfect", subPath.takeIf { it.isNotBlank() })
            .joinToString("/") + "/"

        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.Q) {
            val collection = when {
                fileType.isImage -> MediaStore.Images.Media.getContentUri(MediaStore.VOLUME_EXTERNAL_PRIMARY)
                fileType.isVideo -> MediaStore.Video.Media.getContentUri(MediaStore.VOLUME_EXTERNAL_PRIMARY)
                else -> MediaStore.Downloads.getContentUri(MediaStore.VOLUME_EXTERNAL_PRIMARY)
            }
            val resolver = remoteSideContext.androidContext.contentResolver
            
            for (attempt in 0..100) {
                val candidateName = if (attempt == 0) fileName else appendNameSuffix(fileName, attempt)
                
                findExistingMediaUri(collection, candidateName, relativePath)?.let { existingUri ->
                    if (contentMatches(existingUri, inputFile)) {
                        return GallerySaveResult(existingUri, alreadyDownloaded = true)
                    }
                    return@let
                } ?: run {
                    val values = ContentValues().apply {
                        put(MediaStore.MediaColumns.DISPLAY_NAME, candidateName)
                        put(MediaStore.MediaColumns.MIME_TYPE, fileType.mimeType)
                        put(MediaStore.MediaColumns.RELATIVE_PATH, relativePath)
                        put(MediaStore.MediaColumns.IS_PENDING, 1)
                    }

                    val uri = runCatching { resolver.insert(collection, values) }.getOrNull() ?: return@run

                    runCatching {
                        resolver.openOutputStream(uri)?.use { out ->
                            inputFile.inputStream().use { it.copyTo(out) }
                        } ?: throw IllegalStateException("Failed to open output stream")

                        ContentValues().apply { put(MediaStore.MediaColumns.IS_PENDING, 0) }.also { resolver.update(uri, it, null, null) }
                        return GallerySaveResult(uri)
                    }.onFailure {
                        runCatching { resolver.delete(uri, null, null) }
                    }
                }
            }
            throw IllegalStateException("Failed to allocate unique filename")
        } else {
            @Suppress("DEPRECATION")
            val baseDir = Environment.getExternalStoragePublicDirectory(baseRelative)
            val destDir = File(baseDir, "Purrfect" + (if (subPath.isNotBlank()) "/$subPath" else ""))
            destDir.mkdirs()
            
            var destFile = File(destDir, fileName)
            var suffix = 1
            while (destFile.exists()) {
                if (destFile.length() == inputFile.length() && !remoteSideContext.config.root.downloader.allowDuplicate.get() && streamsMatch(destFile.inputStream(), inputFile.inputStream())) {
                    return GallerySaveResult(Uri.fromFile(destFile), alreadyDownloaded = true)
                }
                destFile = File(destDir, appendNameSuffix(fileName, suffix++))
            }
            
            FileOutputStream(destFile).use { out -> inputFile.inputStream().use { it.copyTo(out) } }
            runCatching {
                remoteSideContext.androidContext.sendBroadcast(Intent("android.intent.action.MEDIA_SCANNER_SCAN_FILE").apply { data = Uri.fromFile(destFile) })
            }
            return GallerySaveResult(Uri.fromFile(destFile))
        }
    }

    private fun createMediaTempFile(): File {
        return File.createTempFile("media", ".tmp")
    }

    private suspend fun downloadInputMedias(pendingTask: PendingTask, downloadRequest: DownloadRequest): Map<InputMedia, File> {
        val downloadedMedias = mutableMapOf<InputMedia, File>()
        var totalSize = 1L
        val inputMediaDownloadedBytes = mutableMapOf<InputMedia, Long>()
        val inputMediaProgress = ConcurrentHashMap<InputMedia, String>()

        fun updateDownloadProgress() {
            pendingTask.updateProgress(
                inputMediaProgress.values.joinToString("\n"),
                progress = (inputMediaDownloadedBytes.values.sum() * 100 / totalSize.coerceAtLeast(1)).toInt().coerceIn(0, 100)
            )
        }

        coroutineScope {
            downloadRequest.inputMedias.forEach { inputMedia ->
                fun setProgress(progress: String) {
                    inputMediaProgress[inputMedia] = progress
                    updateDownloadProgress()
                }

                fun handleInputStream(inputStream: InputStream, estimatedSize: Long = 0L) {
                    createMediaTempFile().apply {
                        val decryptedInputStream = (inputMedia.encryption?.decryptInputStream(inputStream) ?: inputStream).buffered()
                        val buffer = ByteArray(1024 * 1024 * 2) // 2MB
                        var read: Int
                        var totalRead = 0L

                        outputStream().use { outputStream ->
                            while (decryptedInputStream.read(buffer).also { read = it } != -1) {
                                outputStream.write(buffer, 0, read)
                                totalRead += read
                                inputMediaDownloadedBytes[inputMedia] = totalRead
                                setProgress("${totalRead / 1024}KB/${estimatedSize / 1024}KB")
                            }
                        }

                        if (totalRead == 0L) {
                            throw java.io.IOException("Download failed: Received 0 bytes")
                        }

                        if (estimatedSize > 0 && totalRead < (estimatedSize * 0.9)) { // 90% threshold for bolt media variability
                             throw java.io.IOException("Incomplete download: $totalRead/$estimatedSize bytes")
                        }
                    }.also { downloadedMedias[inputMedia] = it }
                }

                launch {
                    when (inputMedia.type) {
                        DownloadMediaType.PROTO_MEDIA -> {
                            RemoteMediaResolver.downloadBoltMedia(Base64.UrlSafe.decode(inputMedia.content), decryptionCallback = { it }, resultCallback = { inputStream, length ->
                                totalSize += length
                                inputStream.use {
                                    handleInputStream(it, estimatedSize = length)
                                }
                            })
                        }
                        DownloadMediaType.REMOTE_MEDIA -> {
                            with(URL(inputMedia.content).openConnection() as HttpURLConnection) {
                                requestMethod = "GET"
                                setRequestProperty("User-Agent", Constants.USER_AGENT)
                                connect()
                                totalSize += contentLength.toLong()
                                inputStream.use {
                                    handleInputStream(it, estimatedSize = contentLength.toLong())
                                }
                            }
                        }
                        DownloadMediaType.DIRECT_MEDIA -> {
                            val decoded = Base64.UrlSafe.decode(inputMedia.content)
                            totalSize += decoded.size.toLong()
                            handleInputStream(decoded.inputStream(), estimatedSize = decoded.size.toLong())
                        }
                        else -> {
                            File(inputMedia.content).inputStream().use {
                                totalSize += it.available().toLong()
                                handleInputStream(it, estimatedSize = it.available().toLong())
                            }
                        }
                    }
                }
            }
        }

        return downloadedMedias
    }

    private suspend fun downloadRemoteMedia(pendingTask: PendingTask, metadata: DownloadMetadata, downloadedMedias: Map<InputMedia, File>, downloadRequest: DownloadRequest) {
        downloadRequest.inputMedias.first().let { inputMedia ->
            val mediaType = inputMedia.type
            val media = downloadedMedias[inputMedia]!!

            if (!downloadRequest.isDashPlaylist) {
                if (inputMedia.attachmentType == AttachmentType.NOTE.key) {
                    remoteSideContext.config.root.downloader.forceVoiceNoteFormat.getNullable()?.let { format ->
                        val outputFile = File.createTempFile("voice_note", ".$format")
                        newFFMpegProcessor(pendingTask).execute(FFMpegProcessor.Request(
                            action = FFMpegProcessor.Action.CONVERSION,
                            inputs = listOf(media.absolutePath),
                            output = outputFile
                        ))
                        media.delete()
                        saveMediaToGallery(pendingTask, outputFile, metadata)
                        outputFile.delete()
                        return
                    }
                }

                saveMediaToGallery(pendingTask, media, metadata)
                media.delete()
                return
            }

            assert(mediaType == DownloadMediaType.REMOTE_MEDIA)

            val playlistXml = DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(media)
            val baseUrlNodeList = playlistXml.getElementsByTagName("BaseURL")
            for (i in 0 until baseUrlNodeList.length) {
                val baseUrlNode = baseUrlNodeList.item(i)
                val baseUrl = baseUrlNode.textContent
                // FIX: Only add prefix if it's not already a full URL
                if (!baseUrl.startsWith("http://") && !baseUrl.startsWith("https://")) {
                    baseUrlNode.textContent = "${RemoteMediaResolver.CF_ST_CDN_D}$baseUrl"
                }
            }

            val dashOptions = downloadRequest.dashOptions!!

            val dashPlaylistFile = renameFromFileType(media, FileType.MPD)
            dashPlaylistFile.outputStream().use {
                TransformerFactory.newInstance().newTransformer().transform(DOMSource(playlistXml), StreamResult(it))
            }

            callbackOnProgress(translation.format("download_toast", "path" to dashPlaylistFile.nameWithoutExtension))
            val outputFile = File.createTempFile("dash", ".mp4")
            runCatching {
                newFFMpegProcessor(pendingTask).execute(FFMpegProcessor.Request(
                    action = FFMpegProcessor.Action.DOWNLOAD_DASH,
                    inputs = listOf(dashPlaylistFile.absolutePath),
                    output = outputFile,
                    startTime = dashOptions.offsetTime,
                    duration = dashOptions.duration
                ))
                saveMediaToGallery(pendingTask, outputFile, metadata)
            }.onFailure { exception ->
                if (coroutineContext.job.isCancelled) return@onFailure
                remoteSideContext.log.error("Failed to download dash media", exception)
                callbackOnFailure(translation.format("failed_processing_toast", "error" to exception.toString()), exception.message)
                pendingTask.fail("Failed to download dash media")
            }

            dashPlaylistFile.delete()
            outputFile.delete()
            media.delete()
        }
    }

    private fun renameFromFileType(file: File, fileType: FileType): File {
        val newFile = File(file.parentFile, file.nameWithoutExtension + "." + fileType.fileExtension)
        file.renameTo(newFile)
        return newFile
    }

    fun enqueue(downloadRequest: DownloadRequest, downloadMetadata: DownloadMetadata) {
        remoteSideContext.coroutineScope.launch {
            remoteSideContext.taskManager.getTaskByHash(downloadMetadata.mediaIdentifier)?.let { task ->
                remoteSideContext.log.debug("already queued or downloaded")

                if (task.status.isFinalStage()) {
                    if (task.status != TaskStatus.SUCCESS) return@let
                    // check if the media file has been deleted
                    if (task.type == TaskType.DOWNLOAD) {
                        val outputFile = runCatching {
                            DocumentFile.fromSingleUri(remoteSideContext.androidContext, Uri.parse(task.extra))
                        }.getOrNull()

                        if (outputFile != null && !outputFile.exists()) {
                            return@let
                        }
                    }
                    callbackOnFailure(translation["already_downloaded_toast"])
                    return@launch
                } else {
                    callbackOnFailure(translation["already_queued_toast"], null)
                }
                return@launch
            }

            downloadSemaphore.withPermit {
                callbackOnProgress(translation["download_started_toast"])
                remoteSideContext.log.debug("downloading media")
                val pendingTask = remoteSideContext.taskManager.createPendingTask(
                    Task(
                        type = TaskType.DOWNLOAD,
                        title = downloadMetadata.downloadSource,
                        author = downloadMetadata.mediaAuthor,
                        hash = downloadMetadata.mediaIdentifier
                    )
                ).apply {
                    status = TaskStatus.RUNNING
                    addListener(PendingTaskListener(onCancel = {
                        coroutineContext.job.cancel()
                    }))
                    updateProgress("Downloading...")
                }

                runCatching {
                    if (downloadRequest.isAudioStream) {
                        val streamUrl = downloadRequest.inputMedias.first().content
                        val outputFile = File.createTempFile("audio_stream", ".mp3")

                        callbackOnProgress("Downloading audio stream")
                        pendingTask.updateProgress("Downloading audio stream")
                        newFFMpegProcessor(pendingTask).execute(FFMpegProcessor.Request(
                            action = FFMpegProcessor.Action.DOWNLOAD_AUDIO_STREAM,
                            inputs = listOf(streamUrl),
                            output = outputFile,
                            audioStreamFormat = downloadRequest.audioStreamFormat
                        ))
                        saveMediaToGallery(pendingTask, outputFile, downloadMetadata)
                        return@launch
                    }

                    //first download all input medias into cache
                    val downloadedMedias = downloadInputMedias(pendingTask, downloadRequest).map {
                        it.key to it.value
                    }.toMap().toMutableMap()
                    remoteSideContext.log.verbose("downloaded ${downloadedMedias.size} medias")

                    var shouldMergeOverlay = downloadRequest.shouldMergeOverlay

                    //if there is a zip file, extract it and replace the downloaded media with the extracted ones
                    downloadedMedias.values.find { FileType.fromFile(it) == FileType.ZIP }?.let { zipFile ->
                        val oldDownloadedMedias = downloadedMedias.toMap()
                        downloadedMedias.clear()

                        zipFile.inputStream().use { zipFileInputStream ->
                            MediaDownloaderHelper.getSplitElements(zipFileInputStream) { type, inputStream ->
                                createMediaTempFile().apply {
                                    outputStream().use {
                                        inputStream.copyTo(it)
                                    }
                                }.also {
                                    downloadedMedias[InputMedia(
                                        type = DownloadMediaType.LOCAL_MEDIA,
                                        content = it.absolutePath,
                                        isOverlay = type == SplitMediaAssetType.OVERLAY
                                    )] = it
                                }
                            }
                        }

                        oldDownloadedMedias.forEach { (_, value) ->
                            value.delete()
                        }

                        shouldMergeOverlay = true
                    }

                    if (shouldMergeOverlay) {
                        assert(downloadedMedias.size == 2)
                        val media = downloadedMedias.entries.first { !it.key.isOverlay }.value
                        val overlayMedia = downloadedMedias.entries.first { it.key.isOverlay }.value

                        val mediaFileType = FileType.fromFile(media)
                        val overlayFileType = FileType.fromFile(overlayMedia)

                        val renamedMedia = renameFromFileType(media, mediaFileType)
                        val renamedOverlayMedia = renameFromFileType(overlayMedia, overlayFileType)

                        if (mediaFileType.isImage && overlayFileType.isImage) {
                            runCatching {
                                callbackOnProgress(translation.format("processing_toast", "path" to media.nameWithoutExtension))
                                val originalBitmap = BitmapFactory.decodeFile(renamedMedia.absolutePath) ?: throw Exception("Failed to decode original image")
                                val overlayBitmap = BitmapFactory.decodeFile(renamedOverlayMedia.absolutePath) ?: throw Exception("Failed to decode overlay image")

                                val mergedBitmap = me.eternal.purrfect.core.util.media.PreviewUtils.mergeBitmapOverlay(originalBitmap, overlayBitmap)
                                val mergedImage: File = File.createTempFile("merged", "." + (mediaFileType.fileExtension ?: "jpg"))

                                val compressFormat = when (mediaFileType) {
                                    FileType.PNG -> Bitmap.CompressFormat.PNG
                                    FileType.WEBP -> if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.R) Bitmap.CompressFormat.WEBP_LOSSLESS else Bitmap.CompressFormat.WEBP
                                    else -> Bitmap.CompressFormat.JPEG
                                }

                                mergedImage.outputStream().use {
                                    mergedBitmap.compress(compressFormat, 100, it)
                                }

                                originalBitmap.recycle()
                                overlayBitmap.recycle()
                                mergedBitmap.recycle()

                                saveMediaToGallery(pendingTask, mergedImage, downloadMetadata)
                                mergedImage.delete()
                                renamedOverlayMedia.delete()
                                renamedMedia.delete()
                                return@launch
                            }.onFailure {
                                remoteSideContext.log.error("Failed to merge image overlay using Bitmap, falling back to FFmpeg", it)
                            }
                        }

                        val mergedOverlay: File = File.createTempFile("merged", ".mp4")
                        runCatching {
                            callbackOnProgress(translation.format("processing_toast", "path" to media.nameWithoutExtension))

                            newFFMpegProcessor(pendingTask).execute(FFMpegProcessor.Request(
                                action = FFMpegProcessor.Action.MERGE_OVERLAY,
                                inputs = listOf(renamedMedia.absolutePath),
                                output = mergedOverlay,
                                overlay = renamedOverlayMedia
                            ))

                            saveMediaToGallery(pendingTask, mergedOverlay, downloadMetadata)
                        }.onFailure { exception ->
                            if (coroutineContext.job.isCancelled) return@onFailure
                            remoteSideContext.log.error("Failed to merge overlay", exception)
                            callbackOnFailure(translation.format("failed_processing_toast", "error" to exception.toString()), exception.message)
                            pendingTask.fail("Failed to merge overlay")
                        }

                        mergedOverlay.delete()
                        renamedOverlayMedia.delete()
                        renamedMedia.delete()
                        return@launch
                    }

                    downloadRemoteMedia(pendingTask, downloadMetadata, downloadedMedias, downloadRequest)
                }.onFailure { exception ->
                    pendingTask.fail("Failed to download media")
                    remoteSideContext.log.error("Failed to download media", exception)
                    callbackOnFailure(translation["failed_generic_toast"], exception.message)
                }
            }
        }
    }

    fun onReceive(intent: Intent) {
        val downloadMetadata = gson.fromJson(intent.getStringExtra(ReceiversConfig.DOWNLOAD_METADATA_EXTRA)!!, DownloadMetadata::class.java)
        val downloadRequest = gson.fromJson(intent.getStringExtra(ReceiversConfig.DOWNLOAD_REQUEST_EXTRA)!!, DownloadRequest::class.java)

        enqueue(downloadRequest, downloadMetadata)
    }
}