up

app/src/main/java/io/legado/app/ui/book/info/BookInfoActivity.kt

@@ -232,13 +232,11 @@ class BookInfoActivity : VMBaseActivity<BookInfoViewModel>(R.layout.activity_boo
         }
     }
 
-    override fun curOrigin(): String? {
+    override val curOrigin: String?
-        return viewModel.bookData.value?.origin
+        get() = viewModel.bookData.value?.origin
-    }
 
-    override fun oldBook(): Book? {
+    override val oldBook: Book?
-        return viewModel.bookData.value
+        get() = viewModel.bookData.value
-    }
 
     override fun changeTo(book: Book) {
         upLoading(true)

app/src/main/java/io/legado/app/ui/book/read/ReadBookActivity.kt

@@ -330,13 +330,11 @@ class ReadBookActivity : VMBaseActivity<ReadBookViewModel>(R.layout.activity_rea
         return viewModel.chapterSize
     }
 
-    override fun curOrigin(): String? {
+    override val curOrigin: String?
-        return viewModel.bookData.value?.origin
+        get() = viewModel.bookData.value?.origin
-    }
 
-    override fun oldBook(): Book? {
+    override val oldBook: Book?
-        return viewModel.bookData.value
+        get() = viewModel.bookData.value
-    }
 
     override fun changeTo(book: Book) {
         viewModel.changeTo(book)

app/src/main/java/io/legado/app/ui/changesource/ChangeSourceDialog.kt

@@ -35,6 +35,7 @@ class ChangeSourceDialog : DialogFragment(),
         }
     }
 
+    private var callBack: CallBack? = null
     private lateinit var viewModel: ChangeSourceViewModel
     private lateinit var changeSourceAdapter: ChangeSourceAdapter
 
@@ -49,6 +50,7 @@ class ChangeSourceDialog : DialogFragment(),
 
     override fun onViewCreated(view: View, savedInstanceState: Bundle?) {
         super.onViewCreated(view, savedInstanceState)
+        callBack = activity as? CallBack
         viewModel.searchStateData.observe(viewLifecycleOwner, Observer {
             refresh_progress_bar.isAutoLoading = it
         })
@@ -114,31 +116,24 @@ class ChangeSourceDialog : DialogFragment(),
     }
 
     override fun changeTo(searchBook: SearchBook) {
-        val activity = activity
+        val book = searchBook.toBook()
-        if (activity is CallBack) {
+        callBack?.oldBook?.let { oldBook ->
-            val book = searchBook.toBook()
+            book.durChapterIndex = oldBook.durChapterIndex
-            activity.oldBook()?.let { oldBook ->
+            book.durChapterPos = oldBook.durChapterPos
-                book.durChapterIndex = oldBook.durChapterIndex
+            book.durChapterTitle = oldBook.durChapterTitle
-                book.durChapterPos = oldBook.durChapterPos
+            book.customCoverUrl = oldBook.customCoverUrl
-                book.durChapterTitle = oldBook.durChapterTitle
+            book.customIntro = oldBook.customIntro
-                book.customCoverUrl = oldBook.customCoverUrl
+            book.order = oldBook.order
-                book.customIntro = oldBook.customIntro
+            if (book.coverUrl.isNullOrEmpty()) {
-                book.order = oldBook.order
+                book.coverUrl = oldBook.getDisplayCover()
-                if (book.coverUrl.isNullOrEmpty()) {
-                    book.coverUrl = oldBook.getDisplayCover()
-                }
-                activity.changeTo(book)
             }
+            callBack?.changeTo(book)
         }
         dismiss()
     }
 
     override fun curOrigin(): String {
-        val activity = activity
+        return callBack?.curOrigin ?: ""
-        if (activity is CallBack) {
-            return activity.curOrigin() ?: ""
-        }
-        return ""
     }
 
     override fun adapter(): ChangeSourceAdapter {
@@ -146,8 +141,8 @@ class ChangeSourceDialog : DialogFragment(),
     }
 
     interface CallBack {
-        fun curOrigin(): String?
+        val curOrigin: String?
-        fun oldBook(): Book?
+        val oldBook: Book?
         fun changeTo(book: Book)
     }