# CareplanActivitiesRepository.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.careplanActivities

    import dk.nsp.mingraviditet.core.data.domain.careplan.SplitCareplanActivities
    import dk.nsp.mingraviditet.core.data.domain.pregnancyInfo.PregnancyInfoDomainModel
    import dk.nsp.mingraviditet.core.data.models.toEntity
    import dk.nsp.mingraviditet.core.data.repository.overview.PregnancyRepository
    import dk.nsp.mingraviditet.core.data.util.weekOrNull
    import dk.nsp.mingraviditet.core.database.dao.CareplanActivityDao
    import dk.nsp.mingraviditet.core.database.model.CareplanActivityEntity
    import dk.nsp.mingraviditet.core.database.model.asExternalModel
    import dk.nsp.mingraviditet.core.model.data.CareplanActivity
    import dk.nsp.mingraviditet.core.network.appointments.AppointmentsOfflineDataSource
    import dk.nsp.mingraviditet.core.network.appointments.model.NetworkCareplanActivity
    import dk.nsp.mingraviditet.core.shared.network.Dispatcher
    import dk.nsp.mingraviditet.core.shared.network.MGDispatchers
    import dk.nsp.mingraviditet.core.shared.network.ResultWrapper
    import kotlinx.coroutines.CoroutineDispatcher
    import kotlinx.coroutines.flow.Flow
    import kotlinx.coroutines.flow.flatMapLatest
    import kotlinx.coroutines.flow.map
    import kotlinx.coroutines.withContext
    import javax.inject.Inject
    import javax.inject.Singleton
    import kotlin.collections.map

    @Singleton
    class CareplanActivitiesRepository @Inject constructor(
        pregnancyRepository: PregnancyRepository,
        private val appointmentsDataSource: AppointmentsOfflineDataSource,
        @Dispatcher(MGDispatchers.IO) private val ioDispatcher: CoroutineDispatcher,
        private val careplanActivityDao: CareplanActivityDao,
    ) {

        private val weekNumber: Flow<Int> =
            pregnancyRepository.pregnancyInfo.map { pregnancyResult: ResultWrapper<PregnancyInfoDomainModel> ->
                pregnancyResult.weekOrNull() ?: 0
            }
        val activities = weekNumber.flatMapLatest { weekNumber ->
            careplanActivityDao.getCareplanActivities()
                .map { initialData ->
                    initialData.ifEmpty {
                        getNetworkCareplanActivities()
                    }
                }
                .map { dbCareplanActivities -> dbCareplanActivities.map(CareplanActivityEntity::asExternalModel) }
                .map { careplanActivities ->
                    careplanActivities.sortedBy { it.untilWeek }
                }
                .map { careplanActivities ->
                    val (past, future) = careplanActivities.partition { activity ->
                        activity.untilWeek < weekNumber
                    }

                    SplitCareplanActivities(past = past, future = future)
                }
        }

        private suspend fun getNetworkCareplanActivities(): List<CareplanActivityEntity> =
            withContext(ioDispatcher) {
                val networkCareplanActivities = appointmentsDataSource.getCareplanTemplate()

                val entities =
                    networkCareplanActivities.activities.map(NetworkCareplanActivity::toEntity)
                careplanActivityDao.insertCareplanActivities(entities)

                entities
            }

        suspend fun getCareplanActivity(id: Int): CareplanActivity =
            withContext(ioDispatcher) { careplanActivityDao.getCareplanActivity(id).asExternalModel() }
    }