presentation/com.shifthackz.aisdv1.presentation.screen.storageusage/StorageUsageViewModel

StorageUsageViewModel

class StorageUsageViewModel(dispatchersProvider: DispatchersProvider, val getAllGalleryUseCase: GetAllGalleryUseCase, val deleteAllGalleryUseCase: DeleteAllGalleryUseCase, val getLocalOnnxModelsUseCase: GetLocalOnnxModelsUseCase, val getLocalMediaPipeModelsUseCase: GetLocalMediaPipeModelsUseCase, val getLocalSdxlModelsUseCase: GetLocalSdxlModelsUseCase, val getLocalCoreMlModelsUseCase: GetLocalCoreMlModelsUseCase, val getLocalBonsaiModelsUseCase: GetLocalBonsaiModelsUseCase, val deleteModelUseCase: DeleteModelUseCase, val storageUsageObserver: StorageUsageObserver, val buildInfoProvider: BuildInfoProvider, val router: StorageUsageRouter, val platformActions: StorageUsagePlatformActions, val onError: (Throwable) -> Unit = {}) : BaseMviViewModel<StorageUsageState, StorageUsageIntent, EmptyEffect> (source)

ViewModel for the standalone storage usage screen.

Storage usage is assembled from platform cache bytes, generated gallery payloads, and downloaded model directories. The ViewModel listens to StorageUsageObserver instead of lifecycle callbacks so Settings summaries and the standalone screen refresh from the same invalidation signal.

Author

Dmitriy Moroz

Parameters

dispatchersProvider

App coroutine dispatchers used by the MVI base class and IO work.

getAllGalleryUseCase

Reads generated gallery records whose payload strings are counted.

deleteAllGalleryUseCase

Deletes gallery records after explicit user confirmation.

getLocalOnnxModelsUseCase

Reads locally configured ONNX models.

getLocalMediaPipeModelsUseCase

Reads locally configured MediaPipe models.

getLocalSdxlModelsUseCase

Reads locally configured SDXL models.

getLocalCoreMlModelsUseCase

Reads locally configured Core ML models on supported builds.

deleteModelUseCase

Deletes downloaded local model metadata and files.

storageUsageObserver

Shared invalidation stream used by Settings and this screen.

buildInfoProvider

Build metadata used to filter platform-supported model categories.

router

Standalone route navigation contract.

platformActions

Filesystem bridge for cache and downloaded model byte accounting.

onError

Error callback forwarded to the app-level error handling pipeline.

Constructors

StorageUsageViewModel

constructor(dispatchersProvider: DispatchersProvider, getAllGalleryUseCase: GetAllGalleryUseCase, deleteAllGalleryUseCase: DeleteAllGalleryUseCase, getLocalOnnxModelsUseCase: GetLocalOnnxModelsUseCase, getLocalMediaPipeModelsUseCase: GetLocalMediaPipeModelsUseCase, getLocalSdxlModelsUseCase: GetLocalSdxlModelsUseCase, getLocalCoreMlModelsUseCase: GetLocalCoreMlModelsUseCase, getLocalBonsaiModelsUseCase: GetLocalBonsaiModelsUseCase, deleteModelUseCase: DeleteModelUseCase, storageUsageObserver: StorageUsageObserver, buildInfoProvider: BuildInfoProvider, router: StorageUsageRouter, platformActions: StorageUsagePlatformActions, onError: (Throwable) -> Unit = {})

Properties

private val _effect: Channel<E>

private val _state: MutableStateFlow<S>

buildInfoProvider

private val buildInfoProvider: BuildInfoProvider

protected val currentState: StorageUsageState

deleteAllGalleryUseCase

private val deleteAllGalleryUseCase: DeleteAllGalleryUseCase

deleteModelUseCase

private val deleteModelUseCase: DeleteModelUseCase

open override val effect: Flow<EmptyEffect>

effectDispatcher

private val effectDispatcher: CoroutineDispatcher

getAllGalleryUseCase

private val getAllGalleryUseCase: GetAllGalleryUseCase

getLocalBonsaiModelsUseCase

private val getLocalBonsaiModelsUseCase: GetLocalBonsaiModelsUseCase

getLocalCoreMlModelsUseCase

private val getLocalCoreMlModelsUseCase: GetLocalCoreMlModelsUseCase

getLocalMediaPipeModelsUseCase

private val getLocalMediaPipeModelsUseCase: GetLocalMediaPipeModelsUseCase

getLocalOnnxModelsUseCase

private val getLocalOnnxModelsUseCase: GetLocalOnnxModelsUseCase

getLocalSdxlModelsUseCase

private val getLocalSdxlModelsUseCase: GetLocalSdxlModelsUseCase

private val ioDispatcher: CoroutineDispatcher

private val onError: (Throwable) -> Unit

platformActions

private val platformActions: StorageUsagePlatformActions

private val router: StorageUsageRouter

open override val state: StateFlow<StorageUsageState>

storageUsageObserver

private val storageUsageObserver: StorageUsageObserver

Functions

expect open fun addCloseable(closeable: AutoCloseable)

expect fun addCloseable(key: String, closeable: AutoCloseable)

internal expect fun clear()

private fun clearAll(categories: Set<UsageCategory>)

private fun clearCategory(category: UsageCategory)

clearCategoryInternal

private suspend fun clearCategoryInternal(category: UsageCategory)

protected fun emitEffect(effect: EmptyEffect)

protected fun emitState(state: StorageUsageState)

expect fun <T : AutoCloseable> getCloseable(key: String): T?

getCoreMlModelBytes

private suspend fun getCoreMlModelBytes(allowedModes: List<ServerSource>, modelIds: List<String>): Long

Reads Core ML model bytes from the entire platform model store on iOS-style targets.

getDownloadedModels

private suspend fun getDownloadedModels(getModels: suspend () -> List<LocalAiModel>): List<LocalAiModel>

protected fun launch(dispatcher: CoroutineDispatcher, start: CoroutineStart, block: suspend CoroutineScope.() -> Unit): Job

loadStorageUsage

private suspend fun loadStorageUsage(selectedCategory: UsageCategory?): UsageState

observeStorageUsage

private fun observeStorageUsage()

protected expect open fun onCleared()

open override fun processIntent(intent: StorageUsageIntent)

requestClearAll

private fun requestClearAll()

requestClearCategory

private fun requestClearCategory(category: UsageCategory)

private fun selectCategory(category: UsageCategory)

shouldUseCoreMlModelStoreFallback

private fun shouldUseCoreMlModelStoreFallback(): Boolean

Returns true when Core ML is the only local model provider exposed by this platform.

protected fun updateState(mutation: (StorageUsageState) -> StorageUsageState)