# PregnancyRepository.kt

    /*
     * The MIT License
     *
     * Original work sponsored and donated by The Danish Health Data Authority (http://www.sundhedsdatastyrelsen.dk)
     *
     * Copyright (C) 2025 The Danish Health Data Authority (http://www.sundhedsdatastyrelsen.dk)
     *
     *
     * Permission is hereby granted, free of charge, to any person obtaining a copy of
     * this software and associated documentation files (the "Software"), to deal in
     * the Software without restriction, including without limitation the rights to
     * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
     * of the Software, and to permit persons to whom the Software is furnished to do
     * so, subject to the following conditions:
     *
     *
     * The above copyright notice and this permission notice shall be included in all
     * copies or substantial portions of the Software.
     *
     *
     * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
     * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
     * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
     * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
     * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
     * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
     * SOFTWARE.
     */
    package dk.nsp.mingraviditet.core.data.repository.overview

    import dk.nsp.mingraviditet.core.data.domain.settings.AppPreferences
    import dk.nsp.mingraviditet.core.data.repository.UserPreferencesRepository
    import dk.nsp.mingraviditet.core.datastore.PregnancyPreferencesDataSource
    import dk.nsp.mingraviditet.core.shared.network.Dispatcher
    import dk.nsp.mingraviditet.core.shared.network.Loading
    import dk.nsp.mingraviditet.core.shared.network.MGDispatchers
    import kotlinx.coroutines.CoroutineDispatcher
    import kotlinx.coroutines.CoroutineScope
    import kotlinx.coroutines.async
    import kotlinx.coroutines.flow.SharingStarted
    import kotlinx.coroutines.flow.emitAll
    import kotlinx.coroutines.flow.flowOn
    import kotlinx.coroutines.flow.stateIn
    import kotlinx.coroutines.flow.transformLatest
    import kotlinx.coroutines.withContext
    import java.time.LocalDate
    import javax.inject.Inject
    import javax.inject.Singleton

    @Singleton
    class PregnancyRepository @Inject constructor(
        private val pregnancyPreferencesDataSource: PregnancyPreferencesDataSource,
        userPreferencesRepository: UserPreferencesRepository,
        authenticatedPregnancyDataProvider: AuthenticatedPregnancyDataProvider,
        anonymousPregnancyDataProvider: AnonymousPregnancyDataProvider,
        @Dispatcher(MGDispatchers.IO) private val ioDispatcher: CoroutineDispatcher,
    ) {

        val dataProviders = listOf(
            authenticatedPregnancyDataProvider,
            anonymousPregnancyDataProvider,
        )

        val pregnancyInfo =
            userPreferencesRepository.preferences.transformLatest { userPrefs: AppPreferences ->
                val dataProvider = dataProviders.firstOrNull {
                    it.canHandle(userPrefs)
                }
                if (dataProvider == null) {
                    emit(Loading)
                    return@transformLatest
                }
                emitAll(dataProvider.pregnancyInfo)
            }
                .flowOn(ioDispatcher)
                .stateIn(
                    scope = CoroutineScope(ioDispatcher),
                    started = SharingStarted.Lazily,
                    initialValue = Loading,
                )

        fun CoroutineScope.refreshPregnancyInfo() = async(ioDispatcher) {
            dataProviders.forEach { it.refreshPregnancyInfo() }
        }

        suspend fun setExpectedDueDate(date: LocalDate?) = withContext(ioDispatcher) {
            pregnancyPreferencesDataSource.setExpectedDueDate(date)
        }

        suspend fun setDateOfBirth(date: LocalDate?) = withContext(ioDispatcher) {
            pregnancyPreferencesDataSource.setDateOfBirth(date)
        }

        suspend fun setLastMenstruationPeriod(date: LocalDate?) = withContext(ioDispatcher) {
            pregnancyPreferencesDataSource.setLastMenstruationPeriod(date)
        }

        fun CoroutineScope.setDateOfTermination(date: LocalDate?) = async(ioDispatcher) {
            pregnancyPreferencesDataSource.setDateOfTermination(date)
        }

        fun CoroutineScope.clearPregnancy() = async(ioDispatcher) {
            dataProviders.forEach {
                it.run {
                    clearPregnancy()
                }
            }
        }
    }