Dantotsu/app/src/main/java/ani/dantotsu/download/DownloadsManager.kt

package ani.dantotsu.download

import android.content.Context
import android.content.SharedPreferences
import android.os.Environment
import android.widget.Toast
import com.google.gson.Gson
import com.google.gson.reflect.TypeToken
import java.io.File
import java.io.Serializable

class DownloadsManager(private val context: Context) {
    private val prefs: SharedPreferences =
        context.getSharedPreferences("downloads_pref", Context.MODE_PRIVATE)
    private val gson = Gson()
    private val downloadsList = loadDownloads().toMutableList()

    val mangaDownloadedTypes: List<DownloadedType>
        get() = downloadsList.filter { it.type == DownloadedType.Type.MANGA }
    val animeDownloadedTypes: List<DownloadedType>
        get() = downloadsList.filter { it.type == DownloadedType.Type.ANIME }
    val novelDownloadedTypes: List<DownloadedType>
        get() = downloadsList.filter { it.type == DownloadedType.Type.NOVEL }

    private fun saveDownloads() {
        val jsonString = gson.toJson(downloadsList)
        prefs.edit().putString("downloads_key", jsonString).apply()
    }

    private fun loadDownloads(): List<DownloadedType> {
        val jsonString = prefs.getString("downloads_key", null)
        return if (jsonString != null) {
            val type = object : TypeToken<List<DownloadedType>>() {}.type
            gson.fromJson(jsonString, type)
        } else {
            emptyList()
        }
    }

    fun addDownload(downloadedType: DownloadedType) {
        downloadsList.add(downloadedType)
        saveDownloads()
    }

    fun removeDownload(downloadedType: DownloadedType) {
        downloadsList.remove(downloadedType)
        removeDirectory(downloadedType)
        saveDownloads()
    }

    fun removeMedia(title: String, type: DownloadedType.Type) {
        val subDirectory = if (type == DownloadedType.Type.MANGA) {
            "Manga"
        } else if (type == DownloadedType.Type.ANIME) {
            "Anime"
        } else {
            "Novel"
        }
        val directory = File(
            context.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS),
            "Dantotsu/$subDirectory/$title"
        )
        if (directory.exists()) {
            val deleted = directory.deleteRecursively()
            if (deleted) {
                Toast.makeText(context, "Successfully deleted", Toast.LENGTH_SHORT).show()
            } else {
                Toast.makeText(context, "Failed to delete directory", Toast.LENGTH_SHORT).show()
            }
        } else {
            Toast.makeText(context, "Directory does not exist", Toast.LENGTH_SHORT).show()
            cleanDownloads()
        }
        downloadsList.removeAll { it.title == title }
        saveDownloads()
    }

    private fun cleanDownloads() {
        cleanDownload(DownloadedType.Type.MANGA)
        cleanDownload(DownloadedType.Type.ANIME)
        cleanDownload(DownloadedType.Type.NOVEL)
    }

    private fun cleanDownload(type: DownloadedType.Type) {
        // remove all folders that are not in the downloads list
        val subDirectory = if (type == DownloadedType.Type.MANGA) {
            "Manga"
        } else if (type == DownloadedType.Type.ANIME) {
            "Anime"
        } else {
            "Novel"
        }
        val directory = File(
            context.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS),
            "Dantotsu/$subDirectory"
        )
        val downloadsSubLists = if (type == DownloadedType.Type.MANGA) {
            mangaDownloadedTypes
        } else if (type == DownloadedType.Type.ANIME) {
            animeDownloadedTypes
        } else {
            novelDownloadedTypes
        }
        if (directory.exists()) {
            val files = directory.listFiles()
            if (files != null) {
                for (file in files) {
                    if (!downloadsSubLists.any { it.title == file.name }) {
                        val deleted = file.deleteRecursively()
                    }
                }
            }
        }
        //now remove all downloads that do not have a folder
        val iterator = downloadsList.iterator()
        while (iterator.hasNext()) {
            val download = iterator.next()
            val downloadDir = File(directory, download.title)
            if ((!downloadDir.exists() && download.type == type) || download.title.isBlank()) {
                iterator.remove()
            }
        }
    }

    fun saveDownloadsListToJSONFileInDownloadsFolder(downloadsList: List<DownloadedType>)  //for debugging
    {
        val jsonString = gson.toJson(downloadsList)
        val file = File(
            Environment.getExternalStoragePublicDirectory(Environment.DIRECTORY_DOWNLOADS),
            "Dantotsu/downloads.json"
        )
        if (file.parentFile?.exists() == false) {
            file.parentFile?.mkdirs()
        }
        if (!file.exists()) {
            file.createNewFile()
        }
        file.writeText(jsonString)
    }

    fun queryDownload(downloadedType: DownloadedType): Boolean {
        return downloadsList.contains(downloadedType)
    }

    fun queryDownload(title: String, chapter: String, type: DownloadedType.Type? = null): Boolean {
        return if (type == null) {
            downloadsList.any { it.title == title && it.chapter == chapter }
        } else {
            downloadsList.any { it.title == title && it.chapter == chapter && it.type == type }
        }
    }

    private fun removeDirectory(downloadedType: DownloadedType) {
        val directory = if (downloadedType.type == DownloadedType.Type.MANGA) {
            File(
                context.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS),
                "Dantotsu/Manga/${downloadedType.title}/${downloadedType.chapter}"
            )
        } else if (downloadedType.type == DownloadedType.Type.ANIME) {
            File(
                context.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS),
                "Dantotsu/Anime/${downloadedType.title}/${downloadedType.chapter}"
            )
        } else {
            File(
                context.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS),
                "Dantotsu/Novel/${downloadedType.title}/${downloadedType.chapter}"
            )
        }

        // Check if the directory exists and delete it recursively
        if (directory.exists()) {
            val deleted = directory.deleteRecursively()
            if (deleted) {
                Toast.makeText(context, "Successfully deleted", Toast.LENGTH_SHORT).show()
            } else {
                Toast.makeText(context, "Failed to delete directory", Toast.LENGTH_SHORT).show()
            }
        } else {
            Toast.makeText(context, "Directory does not exist", Toast.LENGTH_SHORT).show()
        }
    }

    fun exportDownloads(downloadedType: DownloadedType) { //copies to the downloads folder available to the user
        val directory = if (downloadedType.type == DownloadedType.Type.MANGA) {
            File(
                context.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS),
                "Dantotsu/Manga/${downloadedType.title}/${downloadedType.chapter}"
            )
        } else if (downloadedType.type == DownloadedType.Type.ANIME) {
            File(
                context.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS),
                "Dantotsu/Anime/${downloadedType.title}/${downloadedType.chapter}"
            )
        } else {
            File(
                context.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS),
                "Dantotsu/Novel/${downloadedType.title}/${downloadedType.chapter}"
            )
        }
        val destination = File(
            Environment.getExternalStoragePublicDirectory(Environment.DIRECTORY_DOWNLOADS),
            "Dantotsu/${downloadedType.title}/${downloadedType.chapter}"
        )
        if (directory.exists()) {
            val copied = directory.copyRecursively(destination, true)
            if (copied) {
                Toast.makeText(context, "Successfully copied", Toast.LENGTH_SHORT).show()
            } else {
                Toast.makeText(context, "Failed to copy directory", Toast.LENGTH_SHORT).show()
            }
        } else {
            Toast.makeText(context, "Directory does not exist", Toast.LENGTH_SHORT).show()
        }
    }

    fun purgeDownloads(type: DownloadedType.Type) {
        val directory = if (type == DownloadedType.Type.MANGA) {
            File(context.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS), "Dantotsu/Manga")
        } else if (type == DownloadedType.Type.ANIME) {
            File(context.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS), "Dantotsu/Anime")
        } else {
            File(context.getExternalFilesDir(Environment.DIRECTORY_DOWNLOADS), "Dantotsu/Novel")
        }
        if (directory.exists()) {
            val deleted = directory.deleteRecursively()
            if (deleted) {
                Toast.makeText(context, "Successfully deleted", Toast.LENGTH_SHORT).show()
            } else {
                Toast.makeText(context, "Failed to delete directory", Toast.LENGTH_SHORT).show()
            }
        } else {
            Toast.makeText(context, "Directory does not exist", Toast.LENGTH_SHORT).show()
        }

        downloadsList.removeAll { it.type == type }
        saveDownloads()
    }

    companion object {
        const val novelLocation = "Dantotsu/Novel"
        const val mangaLocation = "Dantotsu/Manga"
        const val animeLocation = "Dantotsu/Anime"
    }

}

data class DownloadedType(val title: String, val chapter: String, val type: Type) : Serializable {
    enum class Type {
        MANGA,
        ANIME,
        NOVEL
    }
}