update badge when click notify

ekibun · Nov 13, 2019 · 542e2eb · 542e2eb
1 parent 8206d6e
commit 542e2eb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/app/src/main/java/soko/ekibun/bangumi/ui/main/MainActivity.kt b/app/src/main/java/soko/ekibun/bangumi/ui/main/MainActivity.kt
@@ -69,10 +69,12 @@ class MainActivity : BaseActivity() {
                 when(it.itemId){
                     R.id.notify_type_inbox -> {
                         mainPresenter.user?.notify = Pair(0, mainPresenter.user?.notify?.second ?: 0)
+                        notifyMenu?.badge = mainPresenter.user?.notify?.let { it.first + it.second } ?: 0
                         WebActivity.launchUrl(this, "${Bangumi.SERVER}/pm")
                     }
                     R.id.notify_type_notify -> {
                         mainPresenter.user?.notify = Pair(mainPresenter.user?.notify?.first ?: 0, 0)
+                        notifyMenu?.badge = mainPresenter.user?.notify?.let { it.first + it.second } ?: 0
                         WebActivity.launchUrl(this, "${Bangumi.SERVER}/notify" + if (notify == 0) "/all" else "")
                     }
                 }