data/com.shifthackz.aisdv1.data.repository/StableDiffusionGenerationRepositoryImpl

StableDiffusionGenerationRepositoryImpl

internal class StableDiffusionGenerationRepositoryImpl(mediaStoreGateway: MediaStoreGateway, backgroundWorkObserver: BackgroundWorkObserver, localDataSource: GenerationResultDataSource.Local, val remoteDataSource: StableDiffusionGenerationDataSource.Remote, val preferenceManager: PreferenceManager, val authorizationStore: AuthorizationStore, val textToImageDemo: TextToImageDemo, val imageToImageDemo: ImageToImageDemo) : CoreGenerationRepository, StableDiffusionGenerationRepository(source)

Implements StableDiffusionGenerationRepository behavior in the SDAI data layer.

Author

Dmitriy Moroz

Constructors

StableDiffusionGenerationRepositoryImpl

constructor(mediaStoreGateway: MediaStoreGateway, backgroundWorkObserver: BackgroundWorkObserver, localDataSource: GenerationResultDataSource.Local, remoteDataSource: StableDiffusionGenerationDataSource.Remote, preferenceManager: PreferenceManager, authorizationStore: AuthorizationStore, textToImageDemo: TextToImageDemo, imageToImageDemo: ImageToImageDemo)

Properties

authorizationStore

private val authorizationStore: AuthorizationStore

Exposes the authorizationStore value used by the SDAI data layer.

backgroundWorkObserver

private val backgroundWorkObserver: BackgroundWorkObserver

Exposes the backgroundWorkObserver value used by the SDAI data layer.

imageToImageDemo

private val imageToImageDemo: ImageToImageDemo

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

remoteDataSource

private val remoteDataSource: StableDiffusionGenerationDataSource.Remote

Exposes the remoteDataSource value used by the SDAI data layer.

textToImageDemo

private val textToImageDemo: TextToImageDemo

Exposes the textToImageDemo value used by the SDAI data layer.

Functions

checkApiAvailability

open suspend override fun checkApiAvailability()

open suspend override fun checkApiAvailability(url: String)

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): List<AiGenerationResult>

generateFromText

open suspend override fun generateFromText(payload: TextToImagePayload): List<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.

interruptGeneration

open suspend override fun interruptGeneration()