Add FileSelectionManager to storage lib
This new manager will be responsible for handling file selection by user prior to restore.
This commit is contained in:
Torsten Grote
parent
adbc412d20
commit
2b07b8417c
1 changed files with 218 additions and 0 deletions
|
|
@ -0,0 +1,218 @@
|
|||
/*
|
||||
* SPDX-FileCopyrightText: 2024 The Calyx Institute
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*/
|
||||
|
||||
package org.calyxos.backup.storage.ui.restore
|
||||
|
||||
import androidx.annotation.UiThread
|
||||
import kotlinx.coroutines.flow.MutableStateFlow
|
||||
import kotlinx.coroutines.flow.StateFlow
|
||||
import kotlinx.coroutines.flow.asStateFlow
|
||||
import org.calyxos.backup.storage.backup.BackupSnapshot
|
||||
import org.calyxos.backup.storage.restore.RestorableFile
|
||||
|
||||
public sealed interface FilesItem {
|
||||
public val name: String
|
||||
public val dir: String
|
||||
public val level: Int
|
||||
public val selected: Boolean
|
||||
public val size: Long
|
||||
public val lastModified: Long?
|
||||
}
|
||||
|
||||
public data class FileItem internal constructor(
|
||||
internal val file: RestorableFile,
|
||||
override val level: Int,
|
||||
override val selected: Boolean,
|
||||
) : FilesItem {
|
||||
override val name: String get() = file.name
|
||||
override val dir: String get() = file.dir
|
||||
override val size: Long get() = file.size
|
||||
override val lastModified: Long? get() = file.lastModified
|
||||
}
|
||||
|
||||
public data class FolderItem(
|
||||
override val dir: String,
|
||||
override val name: String,
|
||||
override val level: Int,
|
||||
val numFiles: Int,
|
||||
override val size: Long,
|
||||
override val lastModified: Long?,
|
||||
override val selected: Boolean,
|
||||
val partiallySelected: Boolean,
|
||||
val expanded: Boolean,
|
||||
) : FilesItem {
|
||||
init {
|
||||
check(selected || !partiallySelected) {
|
||||
"$dir was not selected, but partially selected"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public class FileSelectionManager {
|
||||
|
||||
private val allFolders = HashMap<String, FolderItem>()
|
||||
private val allFiles = HashMap<String, MutableList<FileItem>>()
|
||||
private var expandedFolder: String? = null
|
||||
|
||||
private val mFiles = MutableStateFlow<List<FilesItem>>(emptyList())
|
||||
public val files: StateFlow<List<FilesItem>> = mFiles.asStateFlow()
|
||||
|
||||
@UiThread
|
||||
public fun onSnapshotChosen(snapshot: BackupSnapshot) {
|
||||
snapshot.mediaFilesList.forEach { mediaFile ->
|
||||
cacheFileItem(RestorableFile(mediaFile))
|
||||
}
|
||||
snapshot.documentFilesList.forEach { documentFile ->
|
||||
cacheFileItem(RestorableFile(documentFile))
|
||||
}
|
||||
// figure out indentation level and display names for folders
|
||||
val sortedFolders = allFiles.keys.sorted()
|
||||
val levels = calculateFolderIndentationLevels(sortedFolders)
|
||||
val list = mutableListOf<FilesItem>()
|
||||
sortedFolders.forEach { folder ->
|
||||
// get size and lastModified from files in that folder
|
||||
val fileItems = allFiles[folder] ?: error("$folder not in allFiles")
|
||||
val size = fileItems.sumOf { it.file.size }
|
||||
val lastModified = fileItems.maxOf { it.file.lastModified ?: -1 }
|
||||
|
||||
val level = levels[folder] ?: error("No level for $folder")
|
||||
val folderItem = FolderItem(
|
||||
dir = folder,
|
||||
name = level.second,
|
||||
level = level.first,
|
||||
numFiles = fileItems.size,
|
||||
size = size,
|
||||
lastModified = if (lastModified == -1L) null else lastModified,
|
||||
selected = true,
|
||||
partiallySelected = false,
|
||||
expanded = false,
|
||||
)
|
||||
allFolders[folder] = folderItem
|
||||
list.add(folderItem)
|
||||
allFiles[folder] = fileItems.sortedBy { it.name }.map {
|
||||
it.copy(level = level.first + 1)
|
||||
}.toMutableList()
|
||||
}
|
||||
mFiles.value = list
|
||||
}
|
||||
|
||||
@UiThread
|
||||
internal fun onExpandClicked(clickedFolderItem: FolderItem) {
|
||||
// un-expand previously expanded folder, if any
|
||||
expandedFolder?.let { folder ->
|
||||
allFolders[folder] = allFolders[folder]?.copy(expanded = false)
|
||||
?: error("Expanded folder $folder not in allFolders")
|
||||
}
|
||||
|
||||
// update clickedFolderItem's expanded state in cache
|
||||
val newFolderItem = clickedFolderItem.copy(expanded = !clickedFolderItem.expanded)
|
||||
allFolders[clickedFolderItem.dir] = newFolderItem
|
||||
if (newFolderItem.expanded) expandedFolder = clickedFolderItem.dir
|
||||
|
||||
// re-build file tree for UI
|
||||
mFiles.value = rebuildListFromCache()
|
||||
}
|
||||
|
||||
@UiThread
|
||||
internal fun onCheckedChanged(toggledFilesItem: FilesItem) {
|
||||
if (toggledFilesItem is FileItem) {
|
||||
onFileItemCheckedChanged(toggledFilesItem)
|
||||
} else if (toggledFilesItem is FolderItem) {
|
||||
onFolderItemCheckedChanged(toggledFilesItem)
|
||||
}
|
||||
// re-build list from cache, so selection state gets updated there
|
||||
mFiles.value = rebuildListFromCache()
|
||||
}
|
||||
|
||||
private fun cacheFileItem(restorableFile: RestorableFile) {
|
||||
val fileItem = FileItem(restorableFile, 0, true)
|
||||
allFiles.getOrPut(restorableFile.dir) {
|
||||
mutableListOf()
|
||||
}.add(fileItem)
|
||||
}
|
||||
|
||||
private fun calculateFolderIndentationLevels(
|
||||
sortedFolders: List<String>,
|
||||
): Map<String, Pair<Int, String>> {
|
||||
val levels = mutableMapOf<String, Pair<Int, String>>()
|
||||
sortedFolders.forEach { folder ->
|
||||
val parts = folder.split('/')
|
||||
for (i in parts.size - 1 downTo 0) {
|
||||
val subPath = parts.subList(0, i).joinToString("/")
|
||||
if (subPath.isBlank()) continue
|
||||
val subPathLevel = levels[subPath]
|
||||
if (subPathLevel != null) {
|
||||
val name = if (i >= parts.size - 1) {
|
||||
parts[i]
|
||||
} else {
|
||||
parts.subList(i, parts.size).joinToString { "/" }
|
||||
}
|
||||
levels[folder] = Pair(subPathLevel.first + 1, name)
|
||||
return@forEach
|
||||
}
|
||||
}
|
||||
levels[folder] = Pair(0, folder.ifEmpty { "/" })
|
||||
}
|
||||
return levels
|
||||
}
|
||||
|
||||
@UiThread
|
||||
private fun rebuildListFromCache(): MutableList<FilesItem> {
|
||||
val list = mutableListOf<FilesItem>()
|
||||
allFolders.keys.sorted().forEach { folder ->
|
||||
val folderItem = allFolders[folder] ?: error("No item for $folder")
|
||||
list.add(folderItem)
|
||||
val fileItems = allFiles[folder] ?: error("$folder not in allFiles")
|
||||
if (folderItem.expanded) {
|
||||
list.addAll(fileItems)
|
||||
}
|
||||
}
|
||||
return list
|
||||
}
|
||||
|
||||
@UiThread
|
||||
private fun onFileItemCheckedChanged(fileItem: FileItem) {
|
||||
// get all file items from this dir and update only the changed one
|
||||
val fileItems = allFiles[fileItem.dir]
|
||||
?: error("no files for ${fileItem.dir}")
|
||||
fileItems.replaceAll {
|
||||
if (it.file == fileItem.file) it.copy(selected = !it.selected)
|
||||
else it
|
||||
}
|
||||
// figure out how to update parent folder
|
||||
var allSelected = true
|
||||
var noneSelected = true
|
||||
fileItems.forEach { item ->
|
||||
if (item.selected) noneSelected = false
|
||||
else allSelected = false
|
||||
}
|
||||
// update parent folder
|
||||
val folderItem = allFolders[fileItem.dir]
|
||||
?: error("no folder for ${fileItem.dir}")
|
||||
allFolders[fileItem.dir] = folderItem.copy(
|
||||
selected = allSelected || !noneSelected,
|
||||
partiallySelected = !allSelected && !noneSelected,
|
||||
)
|
||||
}
|
||||
|
||||
@UiThread
|
||||
private fun onFolderItemCheckedChanged(folderItem: FolderItem) {
|
||||
val newSelected = if (!folderItem.selected) {
|
||||
true // was not selected, so now it should be
|
||||
} else if (folderItem.partiallySelected) {
|
||||
true // was only partially selected, so now select all
|
||||
} else {
|
||||
false // was fully selected, so now deselect
|
||||
}
|
||||
allFiles[folderItem.dir]?.replaceAll {
|
||||
it.copy(selected = newSelected)
|
||||
}
|
||||
allFolders[folderItem.dir] = folderItem.copy(
|
||||
selected = newSelected,
|
||||
partiallySelected = false,
|
||||
)
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in a new issue