Rename

app/src/main/java/io/legato/kazusa/help/RuleBigDataHelp.kt

@@ -0,0 +1,146 @@
+package io.legato.kazusa.help
+
+import io.legato.kazusa.data.appDb
+import io.legato.kazusa.utils.FileUtils
+import io.legato.kazusa.utils.MD5Utils
+import io.legato.kazusa.utils.externalFiles
+import io.legato.kazusa.utils.getFile
+import kotlinx.coroutines.Dispatchers.IO
+import kotlinx.coroutines.withContext
+import splitties.init.appCtx
+import java.io.File
+
+object RuleBigDataHelp {
+
+    private val ruleDataDir = FileUtils.createFolderIfNotExist(appCtx.externalFiles, "ruleData")
+    private val bookData = FileUtils.createFolderIfNotExist(ruleDataDir, "book")
+    private val rssData = FileUtils.createFolderIfNotExist(ruleDataDir, "rss")
+
+    suspend fun clearInvalid() {
+        withContext(IO) {
+            bookData.listFiles()?.forEach {
+                if (it.isFile) {
+                    FileUtils.delete(it)
+                } else {
+                    val bookUrlFile = it.getFile("bookUrl.txt")
+                    if (!bookUrlFile.exists()) {
+                        FileUtils.delete(it)
+                    } else {
+                        val bookUrl = bookUrlFile.readText()
+                        if (!appDb.bookDao.has(bookUrl)) {
+                            FileUtils.delete(it)
+                        }
+                    }
+                }
+            }
+            rssData.listFiles()?.forEach {
+                if (it.isFile) {
+                    FileUtils.delete(it)
+                } else {
+                    val originFile = it.getFile("origin.txt")
+                    if (!originFile.exists()) {
+                        FileUtils.delete(it)
+                    } else {
+                        val origin = originFile.readText()
+                        if (!appDb.rssSourceDao.has(origin)) {
+                            FileUtils.delete(it)
+                        }
+                    }
+                }
+            }
+        }
+    }
+
+    fun putBookVariable(bookUrl: String, key: String, value: String?) {
+        val md5BookUrl = MD5Utils.md5Encode(bookUrl)
+        val md5Key = MD5Utils.md5Encode(key)
+        if (value == null) {
+            FileUtils.delete(FileUtils.getPath(bookData, md5BookUrl, "$md5Key.txt"), true)
+        } else {
+            val valueFile = FileUtils.createFileIfNotExist(bookData, md5BookUrl, "$md5Key.txt")
+            valueFile.writeText(value)
+            val bookUrlFile = File(FileUtils.getPath(bookData, md5BookUrl, "bookUrl.txt"))
+            if (!bookUrlFile.exists()) {
+                bookUrlFile.writeText(bookUrl)
+            }
+        }
+    }
+
+    fun getBookVariable(bookUrl: String, key: String?): String? {
+        val md5BookUrl = MD5Utils.md5Encode(bookUrl)
+        val md5Key = MD5Utils.md5Encode(key)
+        val file = File(FileUtils.getPath(bookData, md5BookUrl, "$md5Key.txt"))
+        if (file.exists()) {
+            return file.readText()
+        }
+        return null
+    }
+
+    fun hasBookVariable(bookUrl: String, key: String): Boolean {
+        val md5BookUrl = MD5Utils.md5Encode(bookUrl)
+        val md5Key = MD5Utils.md5Encode(key)
+        val file = File(FileUtils.getPath(bookData, md5BookUrl, "$md5Key.txt"))
+        return file.exists()
+    }
+
+    fun putChapterVariable(bookUrl: String, chapterUrl: String, key: String, value: String?) {
+        val md5BookUrl = MD5Utils.md5Encode(bookUrl)
+        val md5ChapterUrl = MD5Utils.md5Encode(chapterUrl)
+        val md5Key = MD5Utils.md5Encode(key)
+        if (value == null) {
+            FileUtils.delete(FileUtils.getPath(bookData, md5BookUrl, md5ChapterUrl, "$md5Key.txt"))
+        } else {
+            val valueFile =
+                FileUtils.createFileIfNotExist(bookData, md5BookUrl, md5ChapterUrl, "$md5Key.txt")
+            valueFile.writeText(value)
+            val bookUrlFile = File(FileUtils.getPath(bookData, md5BookUrl, "bookUrl.txt"))
+            if (!bookUrlFile.exists()) {
+                bookUrlFile.writeText(bookUrl)
+            }
+        }
+    }
+
+    fun getChapterVariable(bookUrl: String, chapterUrl: String, key: String): String? {
+        val md5BookUrl = MD5Utils.md5Encode(bookUrl)
+        val md5ChapterUrl = MD5Utils.md5Encode(chapterUrl)
+        val md5Key = MD5Utils.md5Encode(key)
+        val file = File(FileUtils.getPath(bookData, md5BookUrl, md5ChapterUrl, "$md5Key.txt"))
+        if (file.exists()) {
+            return file.readText()
+        }
+        return null
+    }
+
+    fun putRssVariable(origin: String, link: String, key: String, value: String?) {
+        val md5Origin = MD5Utils.md5Encode(origin)
+        val md5Link = MD5Utils.md5Encode(link)
+        val md5Key = MD5Utils.md5Encode(key)
+        val filePath = FileUtils.getPath(rssData, md5Origin, md5Link, "$md5Key.txt")
+        if (value == null) {
+            FileUtils.delete(filePath)
+        } else {
+            val valueFile = FileUtils.createFileIfNotExist(filePath)
+            valueFile.writeText(value)
+            val originFile = File(FileUtils.getPath(rssData, md5Origin, "origin.txt"))
+            if (!originFile.exists()) {
+                originFile.writeText(origin)
+            }
+            val linFile = File(FileUtils.getPath(rssData, md5Origin, md5Link, "origin.txt"))
+            if (!linFile.exists()) {
+                linFile.writeText(link)
+            }
+        }
+    }
+
+    fun getRssVariable(origin: String, link: String, key: String): String? {
+        val md5Origin = MD5Utils.md5Encode(origin)
+        val md5Link = MD5Utils.md5Encode(link)
+        val md5Key = MD5Utils.md5Encode(key)
+        val filePath = FileUtils.getPath(rssData, md5Origin, md5Link, "$md5Key.txt")
+        val file = File(filePath)
+        if (file.exists()) {
+            return file.readText()
+        }
+        return null
+    }
+}