# CareplanActivitiesRepository.kt

    /*
     * The MIT License
     *
     * Original work sponsored and donated by The Danish Health Data Authority (http://www.sundhedsdatastyrelsen.dk)
     *
     * Copyright (C) 2023 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

    import dk.nsp.mingraviditet.core.data.domain.careplan.SplitCareplanActivities
    import dk.nsp.mingraviditet.core.data.models.toEntity
    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 kotlinx.coroutines.CoroutineDispatcher
    import kotlinx.coroutines.flow.flatMapLatest
    import kotlinx.coroutines.flow.map
    import kotlinx.coroutines.withContext
    import java.time.LocalDate
    import javax.inject.Inject
    import javax.inject.Singleton

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

        private val dueDateMethod = pregnancyRepository.pregnancyInfo.map { it?.dueDateMethod }

        val activities = dueDateMethod.flatMapLatest { dueDate ->
            val lmp = dueDate?.lastMenstruationPeriod ?: LocalDate.now()
            val now = LocalDate.now()

            careplanActivityDao.getCareplanActivities()
                .map { initialData ->
                    initialData.ifEmpty {
                        getNetworkCareplanActivities()
                    }
                }
                .map { dbCareplanActivities -> dbCareplanActivities.map(CareplanActivityEntity::asExternalModel) }
                .map { careplanActivities ->
                    val (past, future) = careplanActivities.partition { activity ->
                        lmp.plusWeeks(activity.untilWeek.toLong()) <= now
                    }

                    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() }
    }