优化

app/src/main/java/io/legado/app/ui/book/local/rule/TxtTocRuleActivity.kt

@@ -1,10 +1,21 @@
 package io.legado.app.ui.book.local.rule
 
+import android.annotation.SuppressLint
 import android.os.Bundle
+import android.view.Menu
+import android.view.MenuItem
 import androidx.activity.viewModels
+import com.google.android.material.snackbar.Snackbar
+import io.legado.app.R
 import io.legado.app.base.VMBaseActivity
 import io.legado.app.data.entities.TxtTocRule
 import io.legado.app.databinding.ActivityTxtTocRuleBinding
+import io.legado.app.databinding.DialogEditTextBinding
+import io.legado.app.databinding.DialogTocRegexEditBinding
+import io.legado.app.lib.dialogs.alert
+import io.legado.app.utils.ACache
+import io.legado.app.utils.snackbar
+import io.legado.app.utils.splitNotBlank
 import io.legado.app.utils.viewbindingdelegate.viewBinding
 
 class TxtTocRuleActivity : VMBaseActivity<ActivityTxtTocRuleBinding, TxtTocRuleViewModel>(),
@@ -15,6 +26,7 @@ class TxtTocRuleActivity : VMBaseActivity<ActivityTxtTocRuleBinding, TxtTocRuleV
     private val adapter: TxtTocRuleAdapter by lazy {
         TxtTocRuleAdapter(this, this)
     }
+    private val importTocRuleKey = "tocRuleUrl"
 
     override fun onActivityCreated(savedInstanceState: Bundle?) {
         initView()
@@ -25,16 +37,45 @@ class TxtTocRuleActivity : VMBaseActivity<ActivityTxtTocRuleBinding, TxtTocRuleV
 
     }
 
+    override fun onCompatCreateOptionsMenu(menu: Menu): Boolean {
+        menuInflater.inflate(R.menu.txt_toc_regex, menu)
+        return super.onCompatCreateOptionsMenu(menu)
+    }
+
+    override fun onCompatOptionsItemSelected(item: MenuItem): Boolean {
+        when (item.itemId) {
+            R.id.menu_add -> edit(TxtTocRule())
+            R.id.menu_default -> viewModel.importDefault()
+            R.id.menu_import -> showImportDialog()
+        }
+        return super.onCompatOptionsItemSelected(item)
+    }
+
     override fun del(source: TxtTocRule) {
         viewModel.del(source)
     }
 
     override fun edit(source: TxtTocRule) {
-
+        alert(titleResource = R.string.txt_toc_regex) {
+            val alertBinding = DialogTocRegexEditBinding.inflate(layoutInflater)
+            alertBinding.apply {
+                tvRuleName.setText(source.name)
+                tvRuleRegex.setText(source.rule)
+            }
+            customView { alertBinding.root }
+            okButton {
+                alertBinding.apply {
+                    source.name = tvRuleName.text.toString()
+                    source.rule = tvRuleRegex.text.toString()
+                    viewModel.save(source)
+                }
+            }
+            cancelButton()
+        }
     }
 
     override fun update(vararg source: TxtTocRule) {
-
+        viewModel.update(*source)
     }
 
     override fun toTop(source: TxtTocRule) {
@@ -50,7 +91,48 @@ class TxtTocRuleActivity : VMBaseActivity<ActivityTxtTocRuleBinding, TxtTocRuleV
     }
 
     override fun upCountView() {
+        binding.selectActionBar
+            .upCountView(adapter.selection.size, adapter.itemCount)
+    }
 
+    @SuppressLint("InflateParams")
+    private fun showImportDialog() {
+        val aCache = ACache.get(this, cacheDir = false)
+        val defaultUrl = "https://gitee.com/fisher52/YueDuJson/raw/master/myTxtChapterRule.json"
+        val cacheUrls: MutableList<String> = aCache
+            .getAsString(importTocRuleKey)
+            ?.splitNotBlank(",")
+            ?.toMutableList()
+            ?: mutableListOf()
+        if (!cacheUrls.contains(defaultUrl)) {
+            cacheUrls.add(0, defaultUrl)
+        }
+        alert(titleResource = R.string.import_on_line) {
+            val alertBinding = DialogEditTextBinding.inflate(layoutInflater).apply {
+                editView.hint = "url"
+                editView.setFilterValues(cacheUrls)
+                editView.delCallBack = {
+                    cacheUrls.remove(it)
+                    aCache.put(importTocRuleKey, cacheUrls.joinToString(","))
+                }
+            }
+            customView { alertBinding.root }
+            okButton {
+                val text = alertBinding.editView.text?.toString()
+                text?.let {
+                    if (!cacheUrls.contains(it)) {
+                        cacheUrls.add(0, it)
+                        aCache.put(importTocRuleKey, cacheUrls.joinToString(","))
+                    }
+                    Snackbar.make(binding.root, R.string.importing, Snackbar.LENGTH_INDEFINITE)
+                        .show()
+                    viewModel.importOnLine(it) { msg ->
+                        binding.root.snackbar(msg)
+                    }
+                }
+            }
+            cancelButton()
+        }
     }
 
 }

app/src/main/java/io/legado/app/ui/book/local/rule/TxtTocRuleAdapter.kt

@@ -12,6 +12,17 @@ class TxtTocRuleAdapter(context: Context, private val callback: Callback) :
 
     private val selected = linkedSetOf<TxtTocRule>()
 
+    val selection: List<TxtTocRule>
+        get() {
+            val selection = arrayListOf<TxtTocRule>()
+            getItems().map {
+                if (selected.contains(it)) {
+                    selection.add(it)
+                }
+            }
+            return selection.sortedBy { it.serialNumber }
+        }
+
     override fun getViewBinding(parent: ViewGroup): ItemTxtTocRuleBinding {
         return ItemTxtTocRuleBinding.inflate(inflater, parent, false)
     }

app/src/main/java/io/legado/app/ui/book/local/rule/TxtTocRuleViewModel.kt

@@ -4,18 +4,52 @@ import android.app.Application
 import io.legado.app.base.BaseViewModel
 import io.legado.app.data.appDb
 import io.legado.app.data.entities.TxtTocRule
+import io.legado.app.help.DefaultData
+import io.legado.app.help.http.newCallResponseBody
+import io.legado.app.help.http.okHttpClient
+import io.legado.app.help.http.text
+import io.legado.app.utils.GSON
+import io.legado.app.utils.fromJsonArray
 
 class TxtTocRuleViewModel(app: Application) : BaseViewModel(app) {
 
+    fun save(txtTocRule: TxtTocRule) {
+        execute {
+            appDb.txtTocRuleDao.insert(txtTocRule)
+        }
+    }
+
     fun del(txtTocRule: TxtTocRule) {
         execute {
             appDb.txtTocRuleDao.delete(txtTocRule)
         }
     }
 
-    fun update(txtTocRule: TxtTocRule) {
+    fun update(vararg txtTocRule: TxtTocRule) {
+        execute {
+            appDb.txtTocRuleDao.update(*txtTocRule)
+        }
+    }
+
+    fun importDefault() {
+        execute {
+            DefaultData.importDefaultTocRules()
+        }
+    }
+
+    fun importOnLine(url: String, finally: (msg: String) -> Unit) {
         execute {
-            appDb.txtTocRuleDao.update(txtTocRule)
+            okHttpClient.newCallResponseBody {
+                url(url)
+            }.text("utf-8").let { json ->
+                GSON.fromJsonArray<TxtTocRule>(json)?.let {
+                    appDb.txtTocRuleDao.insert(*it.toTypedArray())
+                }
+            }
+        }.onSuccess {
+            finally("导入成功")
+        }.onError {
+            finally("导入失败")
         }
     }