data/com.shifthackz.aisdv1.data.repository/ArliAiGenerationRepositoryImpl

ArliAiGenerationRepositoryImpl

internal class ArliAiGenerationRepositoryImpl(mediaStoreGateway: MediaStoreGateway, localDataSource: GenerationResultDataSource.Local, backgroundWorkObserver: BackgroundWorkObserver, val preferenceManager: PreferenceManager, val remoteDataSource: ArliAiGenerationDataSource.Remote) : CoreGenerationRepository, ArliAiGenerationRepository(source)

Generates images through ArliAI and persists returned gallery records.

Each generation uses a payload model override when present, otherwise it falls back to the selected ArliAI model saved in preferences.

Author

Dmitriy Moroz

Constructors

ArliAiGenerationRepositoryImpl

constructor(mediaStoreGateway: MediaStoreGateway, localDataSource: GenerationResultDataSource.Local, backgroundWorkObserver: BackgroundWorkObserver, preferenceManager: PreferenceManager, remoteDataSource: ArliAiGenerationDataSource.Remote)

Properties

backgroundWorkObserver

private val backgroundWorkObserver: BackgroundWorkObserver

Exposes the backgroundWorkObserver 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

private val preferenceManager: PreferenceManager

Exposes the preferenceManager value used by the SDAI data layer.

private val preferenceManager: PreferenceManager

remoteDataSource

private val remoteDataSource: ArliAiGenerationDataSource.Remote

Functions

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>

Generates image-to-image results and saves each returned image to local history.

generateFromText

open suspend override fun generateFromText(payload: TextToImagePayload): List<AiGenerationResult>

Generates text-to-image results and saves each returned image to local history.

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.

open suspend override fun validateApiKey(): Boolean

Validates the currently saved ArliAI API key.