mirror of https://github.com/tasks/tasks
Begin refactoring FilterSettingsActivity
Alex Baker
12 months ago
parent
e3edd4797a
commit
be5a4dfc01
@ -0,0 +1,157 @@
|
|||||||
|
package org.tasks.activities
|
||||||
|
|
||||||
|
import androidx.lifecycle.SavedStateHandle
|
||||||
|
import androidx.lifecycle.ViewModel
|
||||||
|
import androidx.lifecycle.viewModelScope
|
||||||
|
import com.todoroo.astrid.api.PermaSql
|
||||||
|
import com.todoroo.astrid.core.CriterionInstance
|
||||||
|
import dagger.hilt.android.lifecycle.HiltViewModel
|
||||||
|
import kotlinx.collections.immutable.ImmutableList
|
||||||
|
import kotlinx.collections.immutable.persistentListOf
|
||||||
|
import kotlinx.collections.immutable.toImmutableList
|
||||||
|
import kotlinx.coroutines.flow.MutableStateFlow
|
||||||
|
import kotlinx.coroutines.flow.StateFlow
|
||||||
|
import kotlinx.coroutines.flow.update
|
||||||
|
import kotlinx.coroutines.launch
|
||||||
|
import org.tasks.activities.FilterSettingsActivity.Companion.EXTRA_CRITERIA
|
||||||
|
import org.tasks.activities.FilterSettingsActivity.Companion.TOKEN_FILTER
|
||||||
|
import org.tasks.data.dao.TaskDao.TaskCriteria.activeAndVisible
|
||||||
|
import org.tasks.data.db.Database
|
||||||
|
import org.tasks.data.entity.Task
|
||||||
|
import org.tasks.data.rawQuery
|
||||||
|
import org.tasks.data.sql.Field
|
||||||
|
import org.tasks.data.sql.Query
|
||||||
|
import org.tasks.data.sql.UnaryCriterion
|
||||||
|
import org.tasks.db.QueryUtils
|
||||||
|
import org.tasks.filters.CustomFilter
|
||||||
|
import org.tasks.filters.FilterCriteriaProvider
|
||||||
|
import javax.inject.Inject
|
||||||
|
import kotlin.math.max
|
||||||
|
|
||||||
|
@HiltViewModel
|
||||||
|
class FilterSettingsViewModel @Inject constructor(
|
||||||
|
savedStateHandle: SavedStateHandle,
|
||||||
|
private val filterCriteriaProvider: FilterCriteriaProvider,
|
||||||
|
private val database: Database
|
||||||
|
) : ViewModel() {
|
||||||
|
data class ViewState(
|
||||||
|
val filter: CustomFilter? = null,
|
||||||
|
val fabExtended: Boolean = false,
|
||||||
|
val criteria: ImmutableList<CriterionInstance> = persistentListOf(),
|
||||||
|
)
|
||||||
|
|
||||||
|
private val _viewState = MutableStateFlow(
|
||||||
|
ViewState(
|
||||||
|
filter = savedStateHandle.get<CustomFilter>(TOKEN_FILTER),
|
||||||
|
)
|
||||||
|
)
|
||||||
|
val viewState: StateFlow<ViewState> = _viewState
|
||||||
|
|
||||||
|
init {
|
||||||
|
_viewState.value.filter
|
||||||
|
?.let { filter ->
|
||||||
|
viewModelScope.launch {
|
||||||
|
setCriteria(filterCriteriaProvider.fromString(filter.criterion))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
?: savedStateHandle.get<String>(EXTRA_CRITERIA)?.let { criteria ->
|
||||||
|
viewModelScope.launch {
|
||||||
|
setCriteria(filterCriteriaProvider.fromString(criteria))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
?: setCriteria(emptyList())
|
||||||
|
}
|
||||||
|
|
||||||
|
fun setFabExtended(extended: Boolean) {
|
||||||
|
_viewState.update { it.copy(fabExtended = extended) }
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun setCriteria(criteria: List<CriterionInstance>) {
|
||||||
|
_viewState.update {
|
||||||
|
it.copy(
|
||||||
|
criteria = criteria.ifEmpty { universe() }.toImmutableList(),
|
||||||
|
fabExtended = it.filter == null || criteria.size <= 1,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
viewModelScope.launch {
|
||||||
|
_viewState.update {
|
||||||
|
it.copy(criteria = updateCounts(it.criteria).toImmutableList())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fun removeAt(index: Int) {
|
||||||
|
setCriteria(_viewState.value.criteria.toMutableList().apply { removeAt(index) })
|
||||||
|
}
|
||||||
|
|
||||||
|
fun move(from: Int, to: Int) {
|
||||||
|
setCriteria(
|
||||||
|
_viewState.value.criteria
|
||||||
|
.toMutableList()
|
||||||
|
.apply {
|
||||||
|
val criterion = removeAt(from)
|
||||||
|
add(to, criterion)
|
||||||
|
}
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
fun addCriteria(instance: CriterionInstance) {
|
||||||
|
setCriteria(_viewState.value.criteria.toMutableList().apply { add(instance) })
|
||||||
|
}
|
||||||
|
|
||||||
|
private suspend fun updateCounts(criteria: List<CriterionInstance>): List<CriterionInstance> {
|
||||||
|
val newList = mutableListOf<CriterionInstance>()
|
||||||
|
var max = 0
|
||||||
|
var last = -1
|
||||||
|
val sql = StringBuilder(Query.select(Field.COUNT).from(Task.TABLE).toString())
|
||||||
|
.append(" WHERE ")
|
||||||
|
|
||||||
|
for (instance in criteria.map { CriterionInstance(it) }) {
|
||||||
|
when (instance.type) {
|
||||||
|
CriterionInstance.TYPE_ADD -> sql.append("OR ")
|
||||||
|
CriterionInstance.TYPE_SUBTRACT -> sql.append("AND NOT ")
|
||||||
|
CriterionInstance.TYPE_INTERSECT -> sql.append("AND ")
|
||||||
|
}
|
||||||
|
|
||||||
|
// special code for all tasks universe
|
||||||
|
if (instance.type == CriterionInstance.TYPE_UNIVERSE || instance.criterion.sql == null) {
|
||||||
|
sql.append(activeAndVisible()).append(' ')
|
||||||
|
} else {
|
||||||
|
var subSql: String = instance.criterion.sql.replace(
|
||||||
|
"?",
|
||||||
|
UnaryCriterion.sanitize(instance.valueFromCriterion!!)
|
||||||
|
)
|
||||||
|
subSql = PermaSql.replacePlaceholdersForQuery(subSql)
|
||||||
|
sql.append(Task.ID).append(" IN (").append(subSql).append(")")
|
||||||
|
}
|
||||||
|
val sqlString = QueryUtils.showHiddenAndCompleted(sql.toString())
|
||||||
|
database.rawQuery(sqlString) { cursor ->
|
||||||
|
cursor.step()
|
||||||
|
instance.start = if (last == -1) cursor.getInt(0) else last
|
||||||
|
instance.end = cursor.getInt(0)
|
||||||
|
last = instance.end
|
||||||
|
max = max(max, last)
|
||||||
|
}
|
||||||
|
newList.add(instance)
|
||||||
|
}
|
||||||
|
for (instance in newList) {
|
||||||
|
instance.max = max
|
||||||
|
}
|
||||||
|
return newList
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun universe() = listOf(
|
||||||
|
CriterionInstance().apply {
|
||||||
|
criterion = filterCriteriaProvider.startingUniverse
|
||||||
|
type = CriterionInstance.TYPE_UNIVERSE
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
fun setCriterion(index: Int, criterionInstance: CriterionInstance) {
|
||||||
|
setCriteria(
|
||||||
|
_viewState.value.criteria.toMutableList().apply {
|
||||||
|
set(index, criterionInstance)
|
||||||
|
}
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue