LibreTube/app/src/main/java/com/github/libretube/util/DownloadHelper.kt
2022-11-11 18:39:56 +01:00

58 lines
1.6 KiB
Kotlin

package com.github.libretube.util
import android.content.Context
import android.os.Build
import com.github.libretube.obj.DownloadedFile
import java.io.File
object DownloadHelper {
const val VIDEO_DIR = "video"
const val AUDIO_DIR = "audio"
const val METADATA_DIR = "metadata"
const val THUMBNAIL_DIR = "thumbnail"
private fun getOfflineStorageDir(context: Context): File {
if (Build.VERSION.SDK_INT < Build.VERSION_CODES.M) return context.filesDir
return try {
context.getExternalFilesDir(null)!!
} catch (e: Exception) {
context.filesDir
}
}
fun getDownloadDir(context: Context, path: String): File {
return File(
getOfflineStorageDir(context),
path
).apply {
if (!this.exists()) this.mkdirs()
}
}
private fun File.toDownloadedFile(): DownloadedFile {
return DownloadedFile(
name = this.name,
size = this.length()
)
}
fun getDownloadedFiles(context: Context): MutableList<DownloadedFile> {
val videoFiles = getDownloadDir(context, VIDEO_DIR).listFiles().orEmpty()
val audioFiles = getDownloadDir(context, AUDIO_DIR).listFiles().orEmpty().toMutableList()
val files = mutableListOf<DownloadedFile>()
videoFiles.forEach {
audioFiles.removeIf { audioFile -> audioFile.name == it.name }
files.add(it.toDownloadedFile())
}
audioFiles.forEach {
files.add(it.toDownloadedFile())
}
return files
}
}