From 7f371a95320a1d8aae29f8ca15f7fd972367b60e Mon Sep 17 00:00:00 2001
From: Allan Wang <me@allanwang.ca>
Date: Mon, 22 Apr 2019 03:21:30 -0400
Subject: Update parsers and add widget items

---
 app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt     | 12 ++++++++++--
 .../main/kotlin/com/pitchedapps/frost/db/NotificationDb.kt   |  9 ++++++---
 2 files changed, 16 insertions(+), 5 deletions(-)

(limited to 'app/src/main/kotlin/com/pitchedapps/frost/db')

diff --git a/app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt b/app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt
index e1b1d4c4..67372e23 100644
--- a/app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt
+++ b/app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt
@@ -20,6 +20,7 @@ import android.content.Context
 import androidx.room.Database
 import androidx.room.Room
 import androidx.room.RoomDatabase
+import com.pitchedapps.frost.BuildConfig
 import org.koin.dsl.module.module
 import org.koin.standalone.StandAloneContext
 
@@ -69,15 +70,22 @@ class FrostDatabase(private val privateDb: FrostPrivateDatabase, private val pub
     }
 
     companion object {
+
+        private fun <T : RoomDatabase> RoomDatabase.Builder<T>.frostBuild() = if (BuildConfig.DEBUG) {
+            fallbackToDestructiveMigration().build()
+        } else {
+            build()
+        }
+
         fun create(context: Context): FrostDatabase {
             val privateDb = Room.databaseBuilder(
                 context, FrostPrivateDatabase::class.java,
                 FrostPrivateDatabase.DATABASE_NAME
-            ).build()
+            ).frostBuild()
             val publicDb = Room.databaseBuilder(
                 context, FrostPublicDatabase::class.java,
                 FrostPublicDatabase.DATABASE_NAME
-            ).build()
+            ).frostBuild()
             return FrostDatabase(privateDb, publicDb)
         }
 
diff --git a/app/src/main/kotlin/com/pitchedapps/frost/db/NotificationDb.kt b/app/src/main/kotlin/com/pitchedapps/frost/db/NotificationDb.kt
index 8936d682..532bb435 100644
--- a/app/src/main/kotlin/com/pitchedapps/frost/db/NotificationDb.kt
+++ b/app/src/main/kotlin/com/pitchedapps/frost/db/NotificationDb.kt
@@ -64,7 +64,8 @@ data class NotificationEntity(
     val timestamp: Long,
     val profileUrl: String?,
     // Type essentially refers to channel
-    val type: String
+    val type: String,
+    val unread: Boolean
 ) {
     constructor(
         type: String,
@@ -77,7 +78,8 @@ data class NotificationEntity(
         content.text,
         content.timestamp,
         content.profileUrl,
-        type
+        type,
+        content.unread
     )
 }
 
@@ -94,7 +96,8 @@ data class NotificationContentEntity(
         title = notif.title,
         text = notif.text,
         timestamp = notif.timestamp,
-        profileUrl = notif.profileUrl
+        profileUrl = notif.profileUrl,
+        unread = notif.unread
     )
 }
 
-- 
cgit v1.2.3