data/com.shifthackz.aisdv1.data.repository/StabilityAiGenerationRepositoryImpl

StabilityAiGenerationRepositoryImpl

internal class StabilityAiGenerationRepositoryImpl(mediaStoreGateway: MediaStoreGateway, backgroundWorkObserver: BackgroundWorkObserver, localDataSource: GenerationResultDataSource.Local, val preferenceManager: PreferenceManager, val generationRds: StabilityAiGenerationDataSource.Remote, val creditsRds: StabilityAiCreditsRemoteDataSource, val creditsLds: StabilityAiCreditsDataSource.Local) : CoreGenerationRepository, StabilityAiGenerationRepository(source)

Implements StabilityAiGenerationRepository behavior in the SDAI data layer.

Author

Dmitriy Moroz

Constructors

StabilityAiGenerationRepositoryImpl

constructor(mediaStoreGateway: MediaStoreGateway, backgroundWorkObserver: BackgroundWorkObserver, localDataSource: GenerationResultDataSource.Local, preferenceManager: PreferenceManager, generationRds: StabilityAiGenerationDataSource.Remote, creditsRds: StabilityAiCreditsRemoteDataSource, creditsLds: StabilityAiCreditsDataSource.Local)

Properties

backgroundWorkObserver

private val backgroundWorkObserver: BackgroundWorkObserver

Exposes the backgroundWorkObserver value used by the SDAI data layer.

creditsLds

private val creditsLds: StabilityAiCreditsDataSource.Local

Exposes the creditsLds value used by the SDAI data layer.

creditsRds

private val creditsRds: StabilityAiCreditsRemoteDataSource

Exposes the creditsRds value used by the SDAI data layer.

generationRds

private val generationRds: StabilityAiGenerationDataSource.Remote

Exposes the generationRds 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.

generateFromImage

open suspend override fun generateFromImage(payload: ImageToImagePayload): AiGenerationResult

generateFromText

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

getInfoAsync

protected suspend fun getInfoAsync(): MediaStoreInfo

Loads SDAI data through getInfoAsync.

insertGenerationResult

protected suspend fun insertGenerationResult(ai: AiGenerationResult): AiGenerationResult

Performs the SDAI side effect handled by insertGenerationResult.

refreshCredits

private suspend fun refreshCredits(ai: AiGenerationResult): AiGenerationResult

validateApiKey

open suspend override fun validateApiKey(): Boolean