presentation/com.shifthackz.aisdv1.presentation.screen.setup/ServerSetupViewModel

ServerSetupViewModel

class ServerSetupViewModel(val launchSource: LaunchSource = LaunchSource.SPLASH, val dispatchersProvider: DispatchersProvider, val buildInfoProvider: BuildInfoProvider, val getConfigurationUseCase: GetConfigurationUseCase, val getLocalOnnxModelsUseCase: GetLocalOnnxModelsUseCase, val getLocalMediaPipeModelsUseCase: GetLocalMediaPipeModelsUseCase, val getLocalSdxlModelsUseCase: GetLocalSdxlModelsUseCase, val getLocalCoreMlModelsUseCase: GetLocalCoreMlModelsUseCase, val getLocalBonsaiModelsUseCase: GetLocalBonsaiModelsUseCase, val fetchHuggingFaceModelsUseCase: FetchHuggingFaceModelsUseCase, val urlValidator: UrlValidator, val stringValidator: CommonStringValidator, val filePathValidator: FilePathValidator, val connectToA1111UseCase: ConnectToA1111UseCase, val connectToSwarmUiUseCase: ConnectToSwarmUiUseCase, val connectToHordeUseCase: ConnectToHordeUseCase, val connectToHuggingFaceUseCase: ConnectToHuggingFaceUseCase, val connectToLocalDiffusionUseCase: ConnectToLocalDiffusionUseCase, val connectToMediaPipeUseCase: ConnectToMediaPipeUseCase, val connectToSdxlUseCase: ConnectToSdxlUseCase, val connectToCoreMlUseCase: ConnectToCoreMlUseCase, val connectToBonsaiUseCase: ConnectToBonsaiUseCase, val connectToOpenAiUseCase: ConnectToOpenAiUseCase, val connectToStabilityAiUseCase: ConnectToStabilityAiUseCase, val connectToFalAiUseCase: ConnectToFalAiUseCase, val connectToArliAiUseCase: ConnectToArliAiUseCase, val downloadModelUseCase: DownloadModelUseCase, val deleteModelUseCase: DeleteModelUseCase, val downloadGuard: ServerSetupDownloadGuard, val storageUsageObserver: StorageUsageObserver, val linksProvider: LinksProvider, val preferenceManager: PreferenceManager, val router: ServerSetupRouter, val onError: (Throwable) -> Unit = {}) : BaseMviViewModel<ServerSetupState, ServerSetupIntent, ServerSetupEffect> (source)

Owns provider setup state, validation, local model downloads, and final connection side effects.

Author

Dmitriy Moroz

Parameters

launchSource

Screen that opened setup, used to decide back navigation and completion route.

dispatchersProvider

App coroutine dispatchers used by setup work.

buildInfoProvider

Build metadata used to filter platform-supported local providers.

getConfigurationUseCase

Reads the current provider configuration.

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.

fetchHuggingFaceModelsUseCase

Fetches Hugging Face model list for remote setup.

urlValidator

Validates provider URL input.

stringValidator

Validates text settings entered by the user.

filePathValidator

Validates local file path settings.

connectToA1111UseCase

Applies Automatic1111 configuration.

connectToSwarmUiUseCase

Applies SwarmUI configuration.

connectToHordeUseCase

Applies Horde configuration.

connectToHuggingFaceUseCase

Applies Hugging Face configuration.

connectToLocalDiffusionUseCase

Applies ONNX local diffusion configuration.

connectToMediaPipeUseCase

Applies MediaPipe local diffusion configuration.

connectToSdxlUseCase

Applies SDXL local diffusion configuration.

connectToCoreMlUseCase

Applies Core ML local diffusion configuration.

connectToOpenAiUseCase

Applies OpenAI configuration.

connectToStabilityAiUseCase

Applies Stability AI configuration.

connectToFalAiUseCase

Applies Fal AI configuration.

downloadModelUseCase

Downloads a selected local model.

deleteModelUseCase

Deletes a selected local model.

downloadGuard

Platform guard for model download prerequisites.

storageUsageObserver

Invalidates Settings storage summaries after model changes.

linksProvider

External links and demo mode URL used by setup.

preferenceManager

Persists setup preferences.

router

Setup navigation contract.

onError

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

Constructors

ServerSetupViewModel

constructor(launchSource: LaunchSource = LaunchSource.SPLASH, dispatchersProvider: DispatchersProvider, buildInfoProvider: BuildInfoProvider, getConfigurationUseCase: GetConfigurationUseCase, getLocalOnnxModelsUseCase: GetLocalOnnxModelsUseCase, getLocalMediaPipeModelsUseCase: GetLocalMediaPipeModelsUseCase, getLocalSdxlModelsUseCase: GetLocalSdxlModelsUseCase, getLocalCoreMlModelsUseCase: GetLocalCoreMlModelsUseCase, getLocalBonsaiModelsUseCase: GetLocalBonsaiModelsUseCase, fetchHuggingFaceModelsUseCase: FetchHuggingFaceModelsUseCase, urlValidator: UrlValidator, stringValidator: CommonStringValidator, filePathValidator: FilePathValidator, connectToA1111UseCase: ConnectToA1111UseCase, connectToSwarmUiUseCase: ConnectToSwarmUiUseCase, connectToHordeUseCase: ConnectToHordeUseCase, connectToHuggingFaceUseCase: ConnectToHuggingFaceUseCase, connectToLocalDiffusionUseCase: ConnectToLocalDiffusionUseCase, connectToMediaPipeUseCase: ConnectToMediaPipeUseCase, connectToSdxlUseCase: ConnectToSdxlUseCase, connectToCoreMlUseCase: ConnectToCoreMlUseCase, connectToBonsaiUseCase: ConnectToBonsaiUseCase, connectToOpenAiUseCase: ConnectToOpenAiUseCase, connectToStabilityAiUseCase: ConnectToStabilityAiUseCase, connectToFalAiUseCase: ConnectToFalAiUseCase, connectToArliAiUseCase: ConnectToArliAiUseCase, downloadModelUseCase: DownloadModelUseCase, deleteModelUseCase: DeleteModelUseCase, downloadGuard: ServerSetupDownloadGuard, storageUsageObserver: StorageUsageObserver, linksProvider: LinksProvider, preferenceManager: PreferenceManager, router: ServerSetupRouter, onError: (Throwable) -> Unit = {})

Properties

_effect

private val _effect: Channel<E>

_state

private val _state: MutableStateFlow<S>

buildInfoProvider

private val buildInfoProvider: BuildInfoProvider

connectToA1111UseCase

private val connectToA1111UseCase: ConnectToA1111UseCase

connectToArliAiUseCase

private val connectToArliAiUseCase: ConnectToArliAiUseCase

connectToBonsaiUseCase

private val connectToBonsaiUseCase: ConnectToBonsaiUseCase

connectToCoreMlUseCase

private val connectToCoreMlUseCase: ConnectToCoreMlUseCase

connectToFalAiUseCase

private val connectToFalAiUseCase: ConnectToFalAiUseCase

connectToHordeUseCase

private val connectToHordeUseCase: ConnectToHordeUseCase

connectToHuggingFaceUseCase

private val connectToHuggingFaceUseCase: ConnectToHuggingFaceUseCase

connectToLocalDiffusionUseCase

private val connectToLocalDiffusionUseCase: ConnectToLocalDiffusionUseCase

connectToMediaPipeUseCase

private val connectToMediaPipeUseCase: ConnectToMediaPipeUseCase

connectToOpenAiUseCase

private val connectToOpenAiUseCase: ConnectToOpenAiUseCase

connectToSdxlUseCase

private val connectToSdxlUseCase: ConnectToSdxlUseCase

connectToStabilityAiUseCase

private val connectToStabilityAiUseCase: ConnectToStabilityAiUseCase

connectToSwarmUiUseCase

private val connectToSwarmUiUseCase: ConnectToSwarmUiUseCase

currentState

protected val currentState: ServerSetupState

deleteModelUseCase

private val deleteModelUseCase: DeleteModelUseCase

dispatchersProvider

private val dispatchersProvider: DispatchersProvider

downloadGuard

private val downloadGuard: ServerSetupDownloadGuard

downloadJobs

private val downloadJobs: MutableMap<String, Job>

downloadModelUseCase

private val downloadModelUseCase: DownloadModelUseCase

effect

open override val effect: Flow<ServerSetupEffect>

effectDispatcher

private val effectDispatcher: CoroutineDispatcher

fetchHuggingFaceModelsUseCase

private val fetchHuggingFaceModelsUseCase: FetchHuggingFaceModelsUseCase

filePathValidator

private val filePathValidator: FilePathValidator

getConfigurationUseCase

private val getConfigurationUseCase: GetConfigurationUseCase

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

intentProcessor

private val intentProcessor: ServerSetupIntentProcessor

launchSource

private val launchSource: LaunchSource

linksProvider

private val linksProvider: LinksProvider

onError

private val onError: (Throwable) -> Unit

preferenceManager

private val preferenceManager: PreferenceManager

router

private val router: ServerSetupRouter

state

open override val state: StateFlow<ServerSetupState>

storageUsageObserver

private val storageUsageObserver: StorageUsageObserver

stringValidator

private val stringValidator: CommonStringValidator

urlValidator

private val urlValidator: UrlValidator

Functions

addCloseable

expect open fun addCloseable(closeable: AutoCloseable)

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

clear

@MainThread

internal expect fun clear()

connectToArliAi

private suspend fun connectToArliAi(): Result<Unit>

connectToAutomaticInstance

private suspend fun connectToAutomaticInstance(): Result<Unit>

connectToBonsai

private suspend fun connectToBonsai(): Result<Unit>

connectToCoreMl

private suspend fun connectToCoreMl(): Result<Unit>

connectToFalAi

private suspend fun connectToFalAi(): Result<Unit>

connectToHorde

private suspend fun connectToHorde(): Result<Unit>

connectToHuggingFace

private suspend fun connectToHuggingFace(): Result<Unit>

connectToLocalDiffusion

private suspend fun connectToLocalDiffusion(): Result<Unit>

connectToMediaPipe

private suspend fun connectToMediaPipe(): Result<Unit>

connectToOpenAi

private suspend fun connectToOpenAi(): Result<Unit>

connectToSdxl

private suspend fun connectToSdxl(): Result<Unit>

connectToServer

private fun connectToServer()

connectToStabilityAi

private suspend fun connectToStabilityAi(): Result<Unit>

connectToSwarmUi

private suspend fun connectToSwarmUi(): Result<Unit>

deleteLocalModel

private fun deleteLocalModel(id: String)

download

private fun download(modelId: String, url: String)

emitEffect

protected fun emitEffect(effect: ServerSetupEffect)

emitState

protected fun emitState(state: ServerSetupState)

getCloseable

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

launch

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

localModelDownloadClickReducer

private fun localModelDownloadClickReducer(value: ServerSetupState.LocalModel)

onCleared

protected expect open fun onCleared()

onDownloadFailure

private fun onDownloadFailure(localModel: ServerSetupState.LocalModel, t: Throwable)

processIntent

open override fun processIntent(intent: ServerSetupIntent)

scopeLaunchDownload

private fun scopeLaunchDownload(localModel: ServerSetupState.LocalModel, url: String): Job

updateState

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

validate

private fun validate(): Boolean

validateAndConnectToServer

private fun validateAndConnectToServer()