Skip to content

Commit

Permalink
Extract file importation logic from view model to manger
Browse files Browse the repository at this point in the history
  • Loading branch information
@LunarX
LunarX committed Oct 21, 2024
1 parent a57de00 commit 3232c99
Show file tree
Hide file tree
Showing 2 changed files with 202 additions and 181 deletions.
183 changes: 12 additions & 171 deletions app/src/main/java/com/infomaniak/swisstransfer/ui/screen/newtransfer/NewTransferViewModel.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,63 +17,37 @@
*/
package com.infomaniak.swisstransfer.ui.screen.newtransfer

import android.content.Context
import android.net.Uri
import android.util.Log
import androidx.core.net.toUri
import androidx.lifecycle.SavedStateHandle
import androidx.lifecycle.ViewModel
import androidx.lifecycle.viewModelScope
import com.infomaniak.swisstransfer.ui.components.FileUi
import com.infomaniak.swisstransfer.ui.screen.newtransfer.TransferFilesManager.PickedFile
import dagger.hilt.android.lifecycle.HiltViewModel
import dagger.hilt.android.qualifiers.ApplicationContext
import io.sentry.Sentry
import io.sentry.SentryLevel
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.FlowPreview
import kotlinx.coroutines.channels.Channel
import kotlinx.coroutines.flow.*
import kotlinx.coroutines.flow.SharingStarted
import kotlinx.coroutines.flow.debounce
import kotlinx.coroutines.flow.stateIn
import kotlinx.coroutines.launch
import kotlinx.coroutines.sync.Mutex
import kotlinx.coroutines.sync.withLock
import java.io.File
import javax.inject.Inject

@HiltViewModel
class NewTransferViewModel @Inject constructor(
@ApplicationContext private val appContext: Context,
private val transferFilesManager: TransferFilesManager,
private val savedStateHandle: SavedStateHandle,
) : ViewModel() {

// Importing a file locally can take up time. We can't base the list of already used names on _files because a new import with
// the same name could occur while the file is not finished importing yet.
// This list needs to mark a name a "taken" as soon as the file is queued to be imported and until the file is removed from
// the list of already imported files we listen to in the LazyRow.
private val alreadyUsedFileNames = AlreadyUsedFileNamesSet()
private val _files = MutableStateFlow<List<FileUi>>(emptyList())
private val files: StateFlow<List<FileUi>> = _files
@OptIn(FlowPreview::class)
val filesDebounced = files
val filesDebounced = transferFilesManager.importedFiles
.debounce(50)
.stateIn(
scope = viewModelScope,
started = SharingStarted.Lazily,
initialValue = emptyList(),
)

private val _failedFiles = MutableSharedFlow<PickedFile>()
val failedFiles: SharedFlow<PickedFile> = _failedFiles

private val filesToImport: TransferCountChannel = TransferCountChannel()
val filesToImportCount: StateFlow<Int> = filesToImport.count
val currentSessionTotalUploadedFiles: StateFlow<Int> = filesToImport.currentSessionTotalUploadedFiles

private val filesMutationMutex = Mutex()

private val localCopyFolderFile by lazy { File(appContext.cacheDir, LOCAL_COPY_FOLDER) }
private val localCopyFolder get() = localCopyFolderFile.apply { if (!exists()) mkdirs() }
val failedFiles = transferFilesManager.failedFiles
val filesToImportCount = transferFilesManager.filesToImportCount
val currentSessionTotalUploadedFiles = transferFilesManager.currentSessionTotalUploadedFiles

private var isFirstViewModelCreation: Boolean
get() = savedStateHandle.get<Boolean>(IS_VIEW_MODEL_RESTORED_KEY) ?: true
Expand All @@ -87,160 +61,27 @@ class NewTransferViewModel @Inject constructor(
isFirstViewModelCreation = false
// Remove old imported files in case it would've crashed or similar to start with a clean slate. This is required for
// already imported files restoration to not pick up old files in some extreme cases.
removeLocalCopyFolder()
transferFilesManager.removeLocalCopyFolder()
} else {
restoreAlreadyImportedFiles()
transferFilesManager.restoreAlreadyImportedFiles()
}

observeFilesToImport()
transferFilesManager.copyPickedFilesToLocalStorage()
}
}

fun addFiles(uris: List<Uri>) {
viewModelScope.launch(Dispatchers.IO) {
val newFiles = transferFilesManager.getFiles(uris, isAlreadyUsed = { alreadyUsedFileNames.contains(it) })

alreadyUsedFileNames.addAll(newFiles.map { it.fileName })

newFiles.forEach { filesToImport.send(it) }
transferFilesManager.addFiles(uris)
}
}

fun removeFileByUid(uid: String) {
viewModelScope.launch(Dispatchers.IO) {
var fileName: String? = null

filesMutationMutex.withLock {
val files = _files.value.toMutableList()

val index = files.indexOfFirst { it.uid == uid }.takeIf { it != -1 } ?: return@withLock
val fileToRemove = files.removeAt(index)
fileName = fileToRemove.fileName

runCatching { File(fileToRemove.uri).delete() }

_files.value = files
}

fileName?.let {
alreadyUsedFileNames.remove(it)
}
transferFilesManager.removeFileByUid(uid)
}
}

private suspend fun restoreAlreadyImportedFiles() {
if (!localCopyFolderFile.exists()) return

val alreadyCopiedFiles = localCopyFolderFile.listFiles() ?: return
val restoredFileData = transferFilesManager.getRestoredFileData(alreadyCopiedFiles)

if (alreadyCopiedFiles.size != restoredFileData.size) {
Sentry.withScope { scope ->
scope.level = SentryLevel.ERROR
Sentry.captureMessage("Failure of the restoration of the already imported files after a process kill")
}
}

alreadyUsedFileNames.addAll(restoredFileData.map { it.fileName })

filesMutationMutex.withLock { _files.value += restoredFileData }
}

private fun removeLocalCopyFolder() {
if (localCopyFolderFile.exists()) runCatching { localCopyFolderFile.deleteRecursively() }
}

private suspend fun observeFilesToImport() {
filesToImport.consume { fileToImport ->
Log.i(TAG, "Importing ${fileToImport.uri}")
val copiedFile = copyFileLocally(fileToImport.uri, fileToImport.fileName)

if (copiedFile == null) {
reportFailedImportation(fileToImport)
return@consume
}

Log.i(TAG, "Successfully imported ${fileToImport.uri}")

filesMutationMutex.withLock {
_files.value += FileUi(
uid = fileToImport.fileName,
fileName = fileToImport.fileName,
fileSizeInBytes = fileToImport.fileSizeInBytes,
mimeType = null,
uri = copiedFile.toUri().toString(),
)
}
}
}

private fun copyFileLocally(uri: Uri, fileName: String): File? {
val file = File(localCopyFolder, fileName).apply {
if (exists()) delete()
runCatching { createNewFile() }.onFailure { return null }

runCatching {
val inputStream = appContext.contentResolver.openInputStream(uri) ?: return null

inputStream.use { inputStream ->
outputStream().use { outputStream ->
inputStream.copyTo(outputStream)
}
}
}.onFailure {
return null
}
}

return file
}

private suspend fun reportFailedImportation(file: PickedFile) {
Log.w(TAG, "Failed importation of ${file.uri}");
_failedFiles.emit(file)
}

private class TransferCountChannel {
private val channel = Channel<PickedFile>(capacity = Channel.UNLIMITED)

private val _count = MutableStateFlow(0)
val count: StateFlow<Int> = _count

// Session resets when reaching 0 files in the queue
private val _currentSessionTotalUploadedFiles = MutableStateFlow(0)
val currentSessionTotalUploadedFiles: StateFlow<Int> = _currentSessionTotalUploadedFiles

private val countMutex = Mutex()

suspend fun send(element: PickedFile) {
countMutex.withLock {
_count.value += 1
_currentSessionTotalUploadedFiles.value += 1
}
channel.send(element)
}

suspend fun consume(process: suspend (PickedFile) -> Unit) {
for (element in channel) {
process(element)
countMutex.withLock {
val newValue = _count.value - 1
_count.value = newValue
if (newValue == 0) _currentSessionTotalUploadedFiles.value = 0
}
}
}
}

class AlreadyUsedFileNamesSet {
private val alreadyUsedFileNames = mutableSetOf<String>()
private val mutex = Mutex()

suspend fun contains(fileName: String): Boolean = mutex.withLock { alreadyUsedFileNames.contains(fileName) }
suspend fun addAll(filesNames: List<String>): Boolean = mutex.withLock { alreadyUsedFileNames.addAll(filesNames) }
suspend fun remove(filesName: String): Boolean = mutex.withLock { alreadyUsedFileNames.remove(filesName) }
}

companion object {
private const val TAG = "File importation"
const val LOCAL_COPY_FOLDER = "local_copy_folder"
Expand Down
Loading

0 comments on commit 3232c99

Please sign in to comment.