From ea9d801d0884e9aca50ba378ab1f3b99f32a3e3f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kuba=20Szczodrzy=C5=84ski?= <kuba@szczodrzynski.pl>
Date: Wed, 8 Sep 2021 22:48:35 +0200
Subject: [PATCH] [DB] Workaround missing event types after profile archiving.
 (#60)

---
 .../edziennik/ui/modules/agenda/AgendaFragment.kt | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/app/src/main/java/pl/szczodrzynski/edziennik/ui/modules/agenda/AgendaFragment.kt b/app/src/main/java/pl/szczodrzynski/edziennik/ui/modules/agenda/AgendaFragment.kt
index 8d228dcc..8607217a 100644
--- a/app/src/main/java/pl/szczodrzynski/edziennik/ui/modules/agenda/AgendaFragment.kt
+++ b/app/src/main/java/pl/szczodrzynski/edziennik/ui/modules/agenda/AgendaFragment.kt
@@ -21,6 +21,7 @@ import kotlinx.coroutines.*
 import pl.szczodrzynski.edziennik.App
 import pl.szczodrzynski.edziennik.MainActivity
 import pl.szczodrzynski.edziennik.R
+import pl.szczodrzynski.edziennik.data.db.entity.EventType
 import pl.szczodrzynski.edziennik.data.db.entity.Metadata
 import pl.szczodrzynski.edziennik.data.db.entity.Profile
 import pl.szczodrzynski.edziennik.databinding.FragmentAgendaCalendarBinding
@@ -136,9 +137,22 @@ class AgendaFragment : Fragment(), CoroutineScope {
         }
     }
 
+    private suspend fun checkEventTypes() {
+        withContext(Dispatchers.Default) {
+            val eventTypes = app.db.eventTypeDao().getAllNow(app.profileId).map {
+                it.id
+            }
+            val defaultEventTypes = EventType.getTypeColorMap().keys
+            if (!eventTypes.containsAll(defaultEventTypes)) {
+                app.db.eventTypeDao().addDefaultTypes(activity, app.profileId)
+            }
+        }
+    }
+
     private fun createDefaultAgendaView() { (b as? FragmentAgendaDefaultBinding)?.let { b -> launch {
         if (!isAdded)
             return@launch
+        checkEventTypes()
         delay(500)
 
         agendaDefault = AgendaFragmentDefault(activity, app, b)
@@ -146,6 +160,7 @@ class AgendaFragment : Fragment(), CoroutineScope {
     }}}
 
     private fun createCalendarAgendaView() { (b as? FragmentAgendaCalendarBinding)?.let { b -> launch {
+        checkEventTypes()
         delay(300)
 
         val dayList = mutableListOf<EventDay>()