Replace new database.

Author: zhaobozhen

build.gradle

@@ -6,7 +6,7 @@ buildscript {
   }
   dependencies {
     classpath "com.android.tools.build:gradle:7.3.1"
-    classpath "org.jetbrains.kotlin:kotlin-gradle-plugin:1.8.0"
+    classpath "org.jetbrains.kotlin:kotlin-gradle-plugin:1.7.0"
 
     // NOTE: Do not place your application dependencies here; they belong
     // in the individual module build.gradle files

library/build.gradle

@@ -36,11 +36,10 @@ android {
 }
 
 dependencies {
-  api 'androidx.core:core-ktx:1.9.0'
-  api 'androidx.appcompat:appcompat:1.6.0'
-  api 'org.jetbrains.kotlinx:kotlinx-coroutines-android:1.6.4'
+  implementation 'androidx.core:core-ktx:1.9.0'
+  implementation 'org.jetbrains.kotlinx:kotlinx-coroutines-android:1.6.4'
 
-  api 'androidx.room:room-runtime:2.5.0'
-  api 'androidx.room:room-ktx:2.5.0'
-  api 'javax.annotation:javax.annotation-api:1.3.2'
+  implementation 'androidx.room:room-runtime:2.5.0'
+  implementation 'androidx.room:room-ktx:2.5.0'
+  compileOnly 'javax.annotation:javax.annotation-api:1.3.2'
 }

library/src/main/java/com/absinthe/rulesbundle/LCRules.kt

@@ -14,6 +14,7 @@ object LCRules : IAPI {
     private var ruleRepo: RuleRepository? = null
 
     fun init(context: Context) {
+        Repositories.checkRulesDatabase(context)
         contextRef = WeakReference(context)
         ruleRepo = RuleRepository(RuleDatabase.getDatabase(context).ruleDao())
     }

library/src/main/java/com/absinthe/rulesbundle/Repositories.kt

@@ -0,0 +1,45 @@
+package com.absinthe.rulesbundle
+
+import android.content.Context
+import com.absinthe.rulesbundle.utils.FileUtils
+import com.absinthe.rulesbundle.utils.HashUtils
+import java.io.File
+
+object Repositories {
+
+    const val RULES_DATABASE_NAME = "rules_database"
+
+    fun checkRulesDatabase(context: Context) {
+        if (!checkMd5(context) && !checkVersion(context)) {
+            deleteRulesDatabase(context)
+        }
+    }
+
+    private fun deleteRulesDatabase(context: Context) {
+        val databaseDir = context.getDatabasePath(RULES_DATABASE_NAME).parent
+        FileUtils.delete(File(databaseDir, RULES_DATABASE_NAME))
+        FileUtils.delete(File(databaseDir, "${RULES_DATABASE_NAME}-shm"))
+        FileUtils.delete(File(databaseDir, "${RULES_DATABASE_NAME}-wal"))
+    }
+
+    private fun checkMd5(context: Context): Boolean {
+        val databaseDir = context.getDatabasePath(RULES_DATABASE_NAME).parent
+        val localDB = File(databaseDir, RULES_DATABASE_NAME)
+        if (localDB.exists().not()) return false
+        val localDBMd5 = HashUtils.md5(localDB.readBytes())
+        val assetDBMd5 = context.assets.open(LCRules.getRulesAssetPath()).use {
+            HashUtils.md5(it.readBytes())
+        }
+        return localDBMd5 == assetDBMd5
+    }
+
+    private fun checkVersion(context: Context): Boolean {
+        val versionFile = File(context.filesDir, "lcrules/version")
+        if (versionFile.exists().not()) return false
+        if (versionFile.isDirectory.not()) return false
+        versionFile.listFiles()?.takeIf { it.isNotEmpty() }?.let {
+            val version = runCatching { it[0].name.toInt() }.getOrDefault(0)
+            return version >= LCRules.getVersion()
+        } ?: return false
+    }
+}

library/src/main/java/com/absinthe/rulesbundle/RuleDatabase.kt

@@ -26,7 +26,7 @@ abstract class RuleDatabase : RoomDatabase() {
             return Room.databaseBuilder(
                 context.applicationContext,
                 RuleDatabase::class.java,
-                "rules_database"
+                Repositories.RULES_DATABASE_NAME
             ).fallbackToDestructiveMigration()
                 .createFromAsset(LCRules.getRulesAssetPath())
                 .build()

library/src/main/java/com/absinthe/rulesbundle/utils/FileUtils.kt

@@ -0,0 +1,38 @@
+package com.absinthe.rulesbundle.utils
+
+import java.io.File
+
+object FileUtils {
+
+    fun delete(file: File?): Boolean {
+        if (file == null) return false
+        return if (file.isDirectory) {
+            deleteDir(file)
+        } else {
+            deleteFile(file)
+        }
+    }
+
+    private fun deleteDir(dir: File?): Boolean {
+        if (dir == null) return false
+        // dir doesn't exist then return true
+        if (!dir.exists()) return true
+        // dir isn't a directory then return false
+        if (!dir.isDirectory) return false
+        val files = dir.listFiles()
+        if (files != null && files.isNotEmpty()) {
+            for (file in files) {
+                if (file.isFile) {
+                    if (!file.delete()) return false
+                } else if (file.isDirectory) {
+                    if (!deleteDir(file)) return false
+                }
+            }
+        }
+        return dir.delete()
+    }
+
+    private fun deleteFile(file: File?): Boolean {
+        return file != null && (!file.exists() || file.isFile && file.delete())
+    }
+}

library/src/main/java/com/absinthe/rulesbundle/utils/HashUtils.kt

@@ -0,0 +1,25 @@
+package com.absinthe.rulesbundle.utils
+
+import java.security.MessageDigest
+
+object HashUtils {
+    fun md5(byteArray: ByteArray): String = digest(byteArray, "MD5")
+
+    fun sha1(byteArray: ByteArray): String = digest(byteArray, "SHA-1")
+
+    fun sha256(byteArray: ByteArray): String = digest(byteArray, "SHA-256")
+
+    fun sha512(byteArray: ByteArray): String = digest(byteArray, "SHA-512")
+
+    private fun digest(byteArray: ByteArray, algorithm: String): String {
+        val messageDigest = MessageDigest.getInstance(algorithm).apply {
+            reset()
+        }
+        messageDigest.update(byteArray)
+        return messageDigest.digest().toHexString()
+    }
+
+    private fun ByteArray.toHexString() : String {
+        return this.joinToString("") { it.toString(16) }
+    }
+}