path: root/app/src/main/java/com/zeapo/pwdstore/SearchableRepositoryViewModel.kt

/*
 * Copyright © 2014-2020 The Android Password Store Authors. All Rights Reserved.
 * SPDX-License-Identifier: GPL-3.0-only
 */
package com.zeapo.pwdstore

import android.app.Application
import android.view.LayoutInflater
import android.view.View
import android.view.ViewGroup
import androidx.lifecycle.AndroidViewModel
import androidx.lifecycle.MutableLiveData
import androidx.lifecycle.asFlow
import androidx.lifecycle.asLiveData
import androidx.preference.PreferenceManager
import androidx.recyclerview.widget.DiffUtil
import androidx.recyclerview.widget.ListAdapter
import androidx.recyclerview.widget.RecyclerView
import com.zeapo.pwdstore.autofill.oreo.AutofillPreferences
import com.zeapo.pwdstore.autofill.oreo.DirectoryStructure
import com.zeapo.pwdstore.utils.PasswordItem
import com.zeapo.pwdstore.utils.PasswordRepository
import java.io.File
import java.text.Collator
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.ExperimentalCoroutinesApi
import kotlinx.coroutines.FlowPreview
import kotlinx.coroutines.flow.Flow
import kotlinx.coroutines.flow.asFlow
import kotlinx.coroutines.flow.distinctUntilChanged
import kotlinx.coroutines.flow.emptyFlow
import kotlinx.coroutines.flow.filter
import kotlinx.coroutines.flow.map
import kotlinx.coroutines.flow.mapLatest
import kotlinx.coroutines.flow.toList
import kotlinx.coroutines.yield

private fun File.toPasswordItem(root: File) = if (isFile)
    PasswordItem.newPassword(name, this, root)
else
    PasswordItem.newCategory(name, this, root)

private fun PasswordItem.fuzzyMatch(filter: String): Int {
    var i = 0
    var j = 0
    var score = 0
    var bonus = 0
    var bonusIncrement = 0

    val toMatch = longName

    while (i < filter.length && j < toMatch.length) {
        when {
            filter[i].isWhitespace() -> i++
            filter[i].toLowerCase() == toMatch[j].toLowerCase() -> {
                i++
                bonusIncrement += 1
                bonus += bonusIncrement
                score += bonus
            }
            else -> {
                bonus = 0
                bonusIncrement = 0
            }
        }
        j++
    }
    return if (i == filter.length) score else 0
}

private val CaseInsensitiveComparator = Collator.getInstance().apply {
    strength = Collator.PRIMARY
}

private fun PasswordItem.Companion.makeComparator(
    typeSortOrder: PasswordRepository.PasswordSortOrder,
    directoryStructure: DirectoryStructure
): Comparator<PasswordItem> {
    return when (typeSortOrder) {
        PasswordRepository.PasswordSortOrder.FOLDER_FIRST -> compareBy { it.type }
        PasswordRepository.PasswordSortOrder.INDEPENDENT -> compareBy<PasswordItem>()
        PasswordRepository.PasswordSortOrder.FILE_FIRST -> compareByDescending { it.type }
    }
        .then(compareBy(nullsLast(CaseInsensitiveComparator)) {
            directoryStructure.getIdentifierFor(
                it.file
            )
        })
        .then(compareBy(CaseInsensitiveComparator) { directoryStructure.getUsernameFor(it.file) })
}

enum class FilterMode {
    ListOnly,
    StrictDomain,
    Fuzzy
}

enum class SearchMode {
    RecursivelyInSubdirectories,
    InCurrentDirectoryOnly
}

private data class SearchAction(
    val currentDir: File,
    val filter: String,
    val filterMode: FilterMode,
    val searchMode: SearchMode,
    val listFilesOnly: Boolean
)

@ExperimentalCoroutinesApi
@FlowPreview
class SearchableRepositoryViewModel(application: Application) : AndroidViewModel(application) {
    private val root = PasswordRepository.getRepositoryDirectory(application)
    private val settings = PreferenceManager.getDefaultSharedPreferences(getApplication())
    private val showHiddenDirs = settings.getBoolean("show_hidden_folders", false)
    private val searchFromRoot = settings.getBoolean("search_from_root", false)
    private val defaultSearchMode = if (settings.getBoolean("filter_recursively", true)) {
        SearchMode.RecursivelyInSubdirectories
    } else {
        SearchMode.InCurrentDirectoryOnly
    }

    private val typeSortOrder = PasswordRepository.PasswordSortOrder.getSortOrder(settings)
    private val directoryStructure = AutofillPreferences.directoryStructure(application)
    private val itemComparator = PasswordItem.makeComparator(typeSortOrder, directoryStructure)

    private val searchAction = MutableLiveData(
        SearchAction(
            currentDir = root,
            filter = "",
            filterMode = FilterMode.ListOnly,
            searchMode = SearchMode.InCurrentDirectoryOnly,
            listFilesOnly = true
        )
    )
    private val searchActionFlow = searchAction.asFlow()
        .map { it.copy(filter = it.filter.trim()) }
        .distinctUntilChanged()

    private val passwordItemsFlow = searchActionFlow
        .mapLatest { searchAction ->
            val dirToSearch =
                if (searchFromRoot && searchAction.filterMode != FilterMode.ListOnly) root else searchAction.currentDir
            val listResultFlow = when (searchAction.searchMode) {
                SearchMode.RecursivelyInSubdirectories -> listFilesRecursively(dirToSearch)
                SearchMode.InCurrentDirectoryOnly -> listFiles(dirToSearch)
            }
            val prefilteredResultFlow =
                if (searchAction.listFilesOnly) listResultFlow.filter { it.isFile } else listResultFlow
            val filterModeToUse =
                if (searchAction.filter == "") FilterMode.ListOnly else searchAction.filterMode
            when (filterModeToUse) {
                FilterMode.ListOnly -> {
                    prefilteredResultFlow
                        .map { it.toPasswordItem(root) }
                        .toList()
                        .sortedWith(itemComparator)
                }
                FilterMode.StrictDomain -> {
                    check(searchAction.listFilesOnly) { "Searches with StrictDomain search mode can only list files" }
                    prefilteredResultFlow
                        .filter { file ->
                            val toMatch =
                                directoryStructure.getIdentifierFor(file) ?: return@filter false
                            // In strict domain mode, we match
                            // * the search term exactly,
                            // * subdomains of the search term,
                            // * or the search term plus an arbitrary protocol.
                            toMatch == searchAction.filter ||
                                toMatch.endsWith(".${searchAction.filter}") ||
                                toMatch.endsWith("://${searchAction.filter}")
                        }
                        .map { it.toPasswordItem(root) }
                        .toList()
                        .sortedWith(itemComparator)
                }
                FilterMode.Fuzzy -> {
                    prefilteredResultFlow
                        .map {
                            val item = it.toPasswordItem(root)
                            Pair(item.fuzzyMatch(searchAction.filter), item)
                        }
                        .filter { it.first > 0 }
                        .toList()
                        .sortedWith(
                            compareByDescending<Pair<Int, PasswordItem>> { it.first }.thenBy(
                                itemComparator
                            ) { it.second })
                        .map { it.second }
                }
            }
        }

    val passwordItemsList = passwordItemsFlow.asLiveData(Dispatchers.IO)

    fun list(currentDir: File) {
        require(currentDir.isDirectory) { "Can only list files in a directory" }
        searchAction.postValue(
            SearchAction(
                filter = "",
                currentDir = currentDir,
                filterMode = FilterMode.ListOnly,
                searchMode = SearchMode.InCurrentDirectoryOnly,
                listFilesOnly = false
            )
        )
    }

    fun search(
        filter: String,
        currentDir: File? = null,
        filterMode: FilterMode = FilterMode.Fuzzy,
        searchMode: SearchMode? = null,
        listFilesOnly: Boolean = false
    ) {
        require(currentDir?.isDirectory != false) { "Can only search in a directory" }
        val action = SearchAction(
            filter = filter.trim(),
            currentDir = currentDir ?: searchAction.value!!.currentDir,
            filterMode = filterMode,
            searchMode = searchMode ?: defaultSearchMode,
            listFilesOnly = listFilesOnly
        )
        searchAction.postValue(action)
    }

    private fun shouldTake(file: File) = with(file) {
        if (isDirectory) {
            !isHidden || showHiddenDirs
        } else {
            !isHidden && file.extension == "gpg"
        }
    }

    private fun listFiles(dir: File): Flow<File> {
        return dir.listFiles { file -> shouldTake(file) }?.asFlow() ?: emptyFlow()
    }

    private fun listFilesRecursively(dir: File): Flow<File> {
        return dir
            .walkTopDown().onEnter { file -> shouldTake(file) }
            .asFlow()
            .map {
                yield()
                it
            }
            .filter { file -> shouldTake(file) }
    }
}

private object PasswordItemDiffCallback : DiffUtil.ItemCallback<PasswordItem>() {
    override fun areItemsTheSame(oldItem: PasswordItem, newItem: PasswordItem) =
        oldItem.file.absolutePath == newItem.file.absolutePath

    override fun areContentsTheSame(oldItem: PasswordItem, newItem: PasswordItem) = oldItem == newItem
}

class DelegatedSearchableRepositoryAdapter<T : RecyclerView.ViewHolder>(
    private val layoutRes: Int,
    private val viewHolderCreator: (view: View) -> T,
    private val viewHolderBinder: T.(item: PasswordItem) -> Unit
) : ListAdapter<PasswordItem, T>(PasswordItemDiffCallback) {

    override fun onCreateViewHolder(parent: ViewGroup, viewType: Int): T {
        val view = LayoutInflater.from(parent.context)
            .inflate(layoutRes, parent, false)
        return viewHolderCreator(view)
    }

    override fun onBindViewHolder(holder: T, position: Int) {
        viewHolderBinder.invoke(holder, getItem(position))
    }
}