1 files changed, 106 insertions, 0 deletions
diff --git a/app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt b/app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt
new file mode 100644
index 00000000..67372e23
--- /dev/null
+++ b/app/src/main/kotlin/com/pitchedapps/frost/db/Database.kt
@@ -0,0 +1,106 @@
+/*
+ * Copyright 2019 Allan Wang
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package com.pitchedapps.frost.db
+
+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
+
+interface FrostPrivateDao {
+    fun cookieDao(): CookieDao
+    fun notifDao(): NotificationDao
+    fun cacheDao(): CacheDao
+}
+
+@Database(
+    entities = [CookieEntity::class, NotificationEntity::class, CacheEntity::class],
+    version = 1,
+    exportSchema = true
+)
+abstract class FrostPrivateDatabase : RoomDatabase(), FrostPrivateDao {
+    companion object {
+        const val DATABASE_NAME = "frost-priv-db"
+    }
+}
+
+interface FrostPublicDao {
+    fun genericDao(): GenericDao
+}
+
+@Database(entities = [GenericEntity::class], version = 1, exportSchema = true)
+abstract class FrostPublicDatabase : RoomDatabase(), FrostPublicDao {
+    companion object {
+        const val DATABASE_NAME = "frost-db"
+    }
+}
+
+interface FrostDao : FrostPrivateDao, FrostPublicDao {
+    fun close()
+}
+
+/**
+ * Composition of all database interfaces
+ */
+class FrostDatabase(private val privateDb: FrostPrivateDatabase, private val publicDb: FrostPublicDatabase) :
+    FrostDao,
+    FrostPrivateDao by privateDb,
+    FrostPublicDao by publicDb {
+
+    override fun close() {
+        privateDb.close()
+        publicDb.close()
+    }
+
+    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
+            ).frostBuild()
+            val publicDb = Room.databaseBuilder(
+                context, FrostPublicDatabase::class.java,
+                FrostPublicDatabase.DATABASE_NAME
+            ).frostBuild()
+            return FrostDatabase(privateDb, publicDb)
+        }
+
+        fun module(context: Context) = module {
+            single { create(context) }
+            single { get<FrostDatabase>().cookieDao() }
+            single { get<FrostDatabase>().cacheDao() }
+            single { get<FrostDatabase>().notifDao() }
+            single { get<FrostDatabase>().genericDao() }
+        }
+
+        /**
+         * Get from koin
+         * For the most part, you can retrieve directly from other koin components
+         */
+        fun get(): FrostDatabase = StandAloneContext.getKoin().koinContext.get()
+    }
+}