data/com.shifthackz.aisdv1.data.repository/BonsaiGenerationRepositoryImpl

BonsaiGenerationRepositoryImpl

internal class BonsaiGenerationRepositoryImpl(mediaStoreGateway: MediaStoreGateway, localDataSource: GenerationResultDataSource.Local, backgroundWorkObserver: BackgroundWorkObserver, val preferenceManager: PreferenceManager, val bonsaiDiffusion: BonsaiDiffusion, val downloadableLocalDataSource: DownloadableModelDataSource.Local, val fileProviderDescriptor: FileProviderDescriptor) : CoreGenerationRepository, BonsaiGenerationRepository(source)

Implements BonsaiGenerationRepository behavior in the SDAI data layer.

Author

Dmitriy Moroz

Constructors

BonsaiGenerationRepositoryImpl

constructor(mediaStoreGateway: MediaStoreGateway, localDataSource: GenerationResultDataSource.Local, backgroundWorkObserver: BackgroundWorkObserver, preferenceManager: PreferenceManager, bonsaiDiffusion: BonsaiDiffusion, downloadableLocalDataSource: DownloadableModelDataSource.Local, fileProviderDescriptor: FileProviderDescriptor)

Properties

backgroundWorkObserver

private val backgroundWorkObserver: BackgroundWorkObserver

Exposes the backgroundWorkObserver value used by the SDAI data layer.

bonsaiDiffusion

private val bonsaiDiffusion: BonsaiDiffusion

Exposes the bonsaiDiffusion value used by the SDAI data layer.

downloadableLocalDataSource

private val downloadableLocalDataSource: DownloadableModelDataSource.Local

Exposes the downloadableLocalDataSource value used by the SDAI data layer.

fileProviderDescriptor

private val fileProviderDescriptor: FileProviderDescriptor

Exposes the fileProviderDescriptor value used by the SDAI data layer.

localDataSource

private val localDataSource: GenerationResultDataSource.Local

Exposes the localDataSource value used by the SDAI data layer.

mediaStoreGateway

private val mediaStoreGateway: MediaStoreGateway

Exposes the mediaStoreGateway value used by the SDAI data layer.

preferenceManager

private val preferenceManager: PreferenceManager

Exposes the preferenceManager value used by the SDAI data layer.

Functions

exportAsync

private fun exportAsync(result: AiGenerationResult)

Executes the exportAsync step in the SDAI data layer.

exportToMediaStoreAsync

protected suspend fun exportToMediaStoreAsync(result: AiGenerationResult)

Executes the exportToMediaStoreAsync step in the SDAI data layer.

generate

private suspend fun generate(payload: TextToImagePayload, modelPath: String): AiGenerationResult

generateFromText

open suspend override fun generateFromText(payload: TextToImagePayload): AiGenerationResult

getInfoAsync

protected suspend fun getInfoAsync(): MediaStoreInfo

Loads SDAI data through getInfoAsync.

getSelectedSupportedModel

private suspend fun getSelectedSupportedModel(): LocalAiModel

insertGenerationResult

protected suspend fun insertGenerationResult(ai: AiGenerationResult): AiGenerationResult

Performs the SDAI side effect handled by insertGenerationResult.

interruptGeneration

open suspend override fun interruptGeneration()

observeStatus

open override fun observeStatus(): Flow<LocalDiffusionStatus>

resolveModelPath

private fun LocalAiModel.resolveModelPath(): String