From 2ddf83e9c4392ed50ae89838f0320a085c30d445 Mon Sep 17 00:00:00 2001
From: kunfei <gekunfei@live.com>
Date: Sun, 1 Mar 2020 14:23:18 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BC=98=E5=8C=96=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../source/edit/BookSourceEditActivity.kt     | 23 ++++++++++++++-----
 1 file changed, 17 insertions(+), 6 deletions(-)

diff --git a/app/src/main/java/io/legado/app/ui/book/source/edit/BookSourceEditActivity.kt b/app/src/main/java/io/legado/app/ui/book/source/edit/BookSourceEditActivity.kt
index c0eab1b32..46f3eb49f 100644
--- a/app/src/main/java/io/legado/app/ui/book/source/edit/BookSourceEditActivity.kt
+++ b/app/src/main/java/io/legado/app/ui/book/source/edit/BookSourceEditActivity.kt
@@ -24,6 +24,7 @@ import io.legado.app.data.entities.rule.*
 import io.legado.app.lib.dialogs.alert
 import io.legado.app.lib.theme.ATH
 import io.legado.app.ui.book.source.debug.BookSourceDebugActivity
+import io.legado.app.ui.login.SourceLogin
 import io.legado.app.ui.widget.KeyboardToolPop
 import io.legado.app.utils.GSON
 import io.legado.app.utils.applyTint
@@ -67,22 +68,20 @@ class BookSourceEditActivity :
 
     override fun onCompatOptionsItemSelected(item: MenuItem): Boolean {
         when (item.itemId) {
-            R.id.menu_save -> {
-                val source = getSource()
+            R.id.menu_save -> getSource().let { source ->
                 if (checkSource(source)) {
                     viewModel.save(source) { setResult(Activity.RESULT_OK); finish() }
                 }
             }
-            R.id.menu_debug_source -> {
-                val source = getSource()
+            R.id.menu_debug_source -> getSource().let { source ->
                 if (checkSource(source)) {
                     viewModel.save(source) {
                         startActivity<BookSourceDebugActivity>(Pair("key", source.bookSourceUrl))
                     }
                 }
             }
-            R.id.menu_copy_source -> {
-                GSON.toJson(getSource())?.let { sourceStr ->
+            R.id.menu_copy_source -> getSource().let { source ->
+                GSON.toJson(source)?.let { sourceStr ->
                     val clipboard = getSystemService(Context.CLIPBOARD_SERVICE) as? ClipboardManager
                     clipboard?.setPrimaryClip(ClipData.newPlainText(null, sourceStr))
                 }
@@ -101,6 +100,18 @@ class BookSourceEditActivity :
                     toast(R.string.can_not_open)
                 }
             }
+            R.id.menu_login -> getSource().let {
+                if (checkSource(it)) {
+                    if (it.loginUrl.isNullOrEmpty()) {
+                        toast(R.string.source_no_login)
+                    } else {
+                        startActivity<SourceLogin>(
+                            Pair("sourceUrl", it.bookSourceUrl),
+                            Pair("loginUrl", it.loginUrl)
+                        )
+                    }
+                }
+            }
         }
         return super.onCompatOptionsItemSelected(item)
     }