Use a ViewModel for live updates

app/src/main/java/com/lagradost/cloudstream3/MainActivity.kt

@@ -92,6 +92,8 @@ import com.lagradost.cloudstream3.syncproviders.AccountManager.Companion.appStri
 import com.lagradost.cloudstream3.syncproviders.AccountManager.Companion.inAppAuths
 import com.lagradost.cloudstream3.ui.APIRepository
 import com.lagradost.cloudstream3.ui.WatchType
+import com.lagradost.cloudstream3.ui.account.AccountHelper
+import com.lagradost.cloudstream3.ui.account.AccountViewModel
 import com.lagradost.cloudstream3.ui.download.DOWNLOAD_NAVIGATE_TO
 import com.lagradost.cloudstream3.ui.home.HomeViewModel
 import com.lagradost.cloudstream3.ui.player.BasicLink
@@ -1610,4 +1612,9 @@ class MainActivity : AppCompatActivity(), ColorPickerDialogListener {
             false
         }
     }
+
+    fun showAccountSelectLinear(context: Context) {
+        val viewModel = ViewModelProvider(this)[AccountViewModel::class.java]
+        AccountHelper.showAccountSelectLinear(context, viewModel, this)
+    }
 }

app/src/main/java/com/lagradost/cloudstream3/ui/account/AccountAdapter.kt

@@ -22,7 +22,8 @@ class AccountAdapter(
     private val accounts: List<DataStoreHelper.Account>,
     private val accountSelectCallback: (DataStoreHelper.Account) -> Unit,
     private val accountCreateCallback: (DataStoreHelper.Account) -> Unit,
-    private val accountEditCallback: (DataStoreHelper.Account) -> Unit
+    private val accountEditCallback: (DataStoreHelper.Account) -> Unit,
+    private val accountDeleteCallback: () -> Unit
 ) : RecyclerView.Adapter<AccountAdapter.AccountViewHolder>() {
 
     companion object {
@@ -62,8 +63,11 @@ class AccountAdapter(
                     } else {
                         root.setOnLongClickListener {
                             showAccountEditDialog(root.context, account, isNewAccount = false) {
+                                if (it != null) {
                                     accountEditCallback.invoke(it)
+                                } else accountDeleteCallback.invoke()
                             }
+
                             true
                         }
                     }
@@ -105,7 +109,9 @@ class AccountAdapter(
 
                     root.setOnClickListener {
                         showAccountEditDialog(root.context, account, isNewAccount = false) {
+                            if (it != null) {
                                 accountEditCallback.invoke(it)
+                            } else accountDeleteCallback.invoke()
                         }
                     }
                 }
@@ -128,6 +134,7 @@ class AccountAdapter(
                             customImage = null,
                             defaultImageIndex = image
                         ), isNewAccount = true) {
+                            if (it != null) {
                                 accountCreateCallback.invoke(it)
                             }
                         }
@@ -135,6 +142,7 @@ class AccountAdapter(
                 }
             }
         }
+    }
 
     override fun onCreateViewHolder(parent: ViewGroup, viewType: Int): AccountViewHolder =
         AccountViewHolder(

app/src/main/java/com/lagradost/cloudstream3/ui/account/AccountHelper.kt

@@ -12,6 +12,7 @@ import androidx.appcompat.app.AlertDialog
 import androidx.core.view.isGone
 import androidx.core.view.isVisible
 import androidx.core.widget.doOnTextChanged
+import androidx.lifecycle.LifecycleOwner
 import androidx.recyclerview.widget.RecyclerView
 import com.google.android.material.bottomsheet.BottomSheetDialog
 import com.lagradost.cloudstream3.AcraApplication.Companion.removeKeys
@@ -20,6 +21,7 @@ import com.lagradost.cloudstream3.databinding.AccountEditDialogBinding
 import com.lagradost.cloudstream3.databinding.AccountSelectLinearBinding
 import com.lagradost.cloudstream3.databinding.LockPinDialogBinding
 import com.lagradost.cloudstream3.mvvm.logError
+import com.lagradost.cloudstream3.mvvm.observe
 import com.lagradost.cloudstream3.ui.result.setImage
 import com.lagradost.cloudstream3.ui.result.setLinearListLayout
 import com.lagradost.cloudstream3.utils.AppUtils.setDefaultFocus
@@ -35,7 +37,7 @@ object AccountHelper {
         context: Context,
         account: DataStoreHelper.Account,
         isNewAccount: Boolean,
-        callback: (DataStoreHelper.Account) -> Unit
+        callback: (DataStoreHelper.Account?) -> Unit
     ) {
         val binding = AccountEditDialogBinding.inflate(LayoutInflater.from(context), null, false)
         val builder = AlertDialog.Builder(context, R.style.AlertDialogCustom)
@@ -65,6 +67,7 @@ object AccountHelper {
 
                         // Update UI
                         setAccount(getDefaultAccount(context), true)
+                        callback.invoke(null)
                         dialog?.dismissSafe()
                     }
 
@@ -274,7 +277,11 @@ object AccountHelper {
         }, 200)
     }
 
-    fun showAccountSelectLinear(context: Context) {
+    fun showAccountSelectLinear(
+        context: Context,
+        viewModel: AccountViewModel,
+        lifecycleOwner: LifecycleOwner
+    ) {
         fun onAccountUpdated(account: DataStoreHelper.Account) {
             val currentAccounts = DataStoreHelper.accounts.toMutableList()
 
@@ -318,8 +325,10 @@ object AccountHelper {
         recyclerView.addItemDecoration(AccountSelectLinearItemDecoration(itemWidth, itemHeight))
 
         recyclerView.setLinearListLayout(isHorizontal = true)
+
+        lifecycleOwner.observe(viewModel.accountsLiveData) { liveAccounts ->
             recyclerView.adapter = AccountAdapter(
-            getAccounts(context),
+                liveAccounts,
                 accountSelectCallback = { account ->
                     // Check if the selected account has a lock PIN set
                     if (account.lockPin != null) {
@@ -338,10 +347,16 @@ object AccountHelper {
                 },
                 accountCreateCallback = {
                     onAccountUpdated(it)
-                builder.dismissSafe()
+                    viewModel.updateAccounts(getAccounts(context))
                 },
-            // Editing is done using AccountSelectActivity
-            accountEditCallback = {}
+                accountEditCallback = {
+                    onAccountUpdated(it)
+                    viewModel.updateAccounts(getAccounts(context))
+                },
+                accountDeleteCallback = {
+                    viewModel.updateAccounts(getAccounts(context))
+                }
             )
         }
     }
+}

app/src/main/java/com/lagradost/cloudstream3/ui/account/AccountSelectActivity.kt

@@ -4,6 +4,7 @@ import android.content.Intent
 import android.os.Bundle
 import android.widget.ImageView
 import androidx.appcompat.app.AppCompatActivity
+import androidx.lifecycle.ViewModelProvider
 import androidx.recyclerview.widget.GridLayoutManager
 import androidx.recyclerview.widget.RecyclerView
 import com.lagradost.cloudstream3.CommonActivity
@@ -11,6 +12,7 @@ import com.lagradost.cloudstream3.CommonActivity.loadThemes
 import com.lagradost.cloudstream3.MainActivity
 import com.lagradost.cloudstream3.R
 import com.lagradost.cloudstream3.databinding.ActivityAccountSelectBinding
+import com.lagradost.cloudstream3.mvvm.observe
 import com.lagradost.cloudstream3.ui.account.AccountAdapter.Companion.VIEW_TYPE_EDIT_ACCOUNT
 import com.lagradost.cloudstream3.ui.account.AccountAdapter.Companion.VIEW_TYPE_SELECT_ACCOUNT
 import com.lagradost.cloudstream3.ui.account.AccountHelper.showPinInputDialog
@@ -23,11 +25,18 @@ class AccountSelectActivity : AppCompatActivity() {
 
     override fun onCreate(savedInstanceState: Bundle?) {
         super.onCreate(savedInstanceState)
+
+        // Are we editing and coming from MainActivity?
+        val isEditingFromMainActivity = intent.getBooleanExtra(
+            "isEditingFromMainActivity",
+            false
+        )
+
         val accounts = getAccounts(this@AccountSelectActivity)
         
         // Don't show account selection if there is only
         // one account that exists
-        if (accounts.count() <= 1) {
+        if (!isEditingFromMainActivity && accounts.count() <= 1) {
             navigateToMainActivity()
             return
         }
@@ -42,26 +51,41 @@ class AccountSelectActivity : AppCompatActivity() {
 
         val recyclerView: RecyclerView = binding.accountRecyclerView
 
-        // Are we editing and coming from MainActivity?
-        val isEditingFromMainActivity = intent.getBooleanExtra(
-            "isEditingFromMainActivity",
-            false
-        )
+        val viewModel = ViewModelProvider(this)[AccountViewModel::class.java]
 
+        observe(viewModel.accountsLiveData) { liveAccounts ->
             val adapter = AccountAdapter(
-            accounts,
+                liveAccounts,
                 // Handle the selected account
                 accountSelectCallback = { onAccountSelected(it) },
-            accountCreateCallback = { onAccountUpdated(it) },
+                accountCreateCallback = {
+                    onAccountUpdated(it)
+
+                    viewModel.updateAccounts(
+                        getAccounts(binding.root.context)
+                    )
+                },
                 accountEditCallback = {
                     onAccountUpdated(it)
 
+                    viewModel.updateAccounts(
+                        getAccounts(binding.root.context)
+                    )
+
                     // We came from MainActivity, return there
                     // and switch to the edited account
                     if (isEditingFromMainActivity) {
-                    DataStoreHelper.setAccount(it, it.keyIndex != DataStoreHelper.selectedKeyIndex)
+                        DataStoreHelper.setAccount(
+                            it,
+                            it.keyIndex != DataStoreHelper.selectedKeyIndex
+                        )
                         navigateToMainActivity()
                     }
+                },
+                accountDeleteCallback = {
+                    viewModel.updateAccounts(
+                        getAccounts(binding.root.context)
+                    )
                 }
             )
 
@@ -98,6 +122,7 @@ class AccountSelectActivity : AppCompatActivity() {
 
                 adapter.notifyDataSetChanged()
             }
+        }
 
         if (isTvSettings()) {
             val spanSize = if (accounts.count() <= 6) {
@@ -137,8 +162,6 @@ class AccountSelectActivity : AppCompatActivity() {
 
         DataStoreHelper.accounts = currentAccounts.toTypedArray()
         DataStoreHelper.currentHomePage = currentHomePage
-
-        this.recreate()
     }
 
     private fun setAccount(account: DataStoreHelper.Account) {

app/src/main/java/com/lagradost/cloudstream3/ui/account/AccountViewModel.kt

@@ -0,0 +1,21 @@
+package com.lagradost.cloudstream3.ui.account
+
+import androidx.lifecycle.LiveData
+import androidx.lifecycle.MutableLiveData
+import androidx.lifecycle.ViewModel
+import com.lagradost.cloudstream3.utils.DataStoreHelper
+
+class AccountViewModel : ViewModel() {
+    private val _accountsLiveData = MutableLiveData<List<DataStoreHelper.Account>>()
+    val accountsLiveData: LiveData<List<DataStoreHelper.Account>> get() = _accountsLiveData
+
+    init {
+        // Default to using the getAccounts function to retrieve initial data
+        val initialAccounts = DataStoreHelper.accounts.toList()
+        _accountsLiveData.value = initialAccounts
+    }
+
+    fun updateAccounts(newAccounts: List<DataStoreHelper.Account>) {
+        _accountsLiveData.value = newAccounts
+    }
+}

app/src/main/java/com/lagradost/cloudstream3/ui/home/HomeFragment.kt

@@ -38,7 +38,6 @@ import com.lagradost.cloudstream3.mvvm.observe
 import com.lagradost.cloudstream3.mvvm.observeNullable
 import com.lagradost.cloudstream3.ui.APIRepository.Companion.noneApi
 import com.lagradost.cloudstream3.ui.APIRepository.Companion.randomApi
-import com.lagradost.cloudstream3.ui.account.AccountHelper.showAccountSelectLinear
 import com.lagradost.cloudstream3.ui.result.txt
 import com.lagradost.cloudstream3.ui.search.*
 import com.lagradost.cloudstream3.ui.search.SearchHelper.handleSearchClickCallback
@@ -497,8 +496,9 @@ class HomeFragment : Fragment() {
             homeApiFab.setOnClickListener(apiChangeClickListener)
             homeChangeApi.setOnClickListener(apiChangeClickListener)
             homeSwitchAccount.setOnClickListener { v ->
-                showAccountSelectLinear(v?.context ?: return@setOnClickListener)
+                (CommonActivity.activity as MainActivity).showAccountSelectLinear(v?.context ?: return@setOnClickListener)
             }
+
             homeRandom.setOnClickListener {
                 if (listHomepageItems.isNotEmpty()) {
                     activity.loadSearchResult(listHomepageItems.random())

app/src/main/java/com/lagradost/cloudstream3/ui/home/HomeParentItemAdapterPreview.kt

@@ -27,7 +27,6 @@ import com.lagradost.cloudstream3.mvvm.Resource
 import com.lagradost.cloudstream3.mvvm.debugException
 import com.lagradost.cloudstream3.mvvm.observe
 import com.lagradost.cloudstream3.ui.WatchType
-import com.lagradost.cloudstream3.ui.account.AccountHelper.showAccountSelectLinear
 import com.lagradost.cloudstream3.ui.home.HomeFragment.Companion.selectHomepage
 import com.lagradost.cloudstream3.ui.result.FOCUS_SELF
 import com.lagradost.cloudstream3.ui.result.ResultViewModel2
@@ -479,7 +478,7 @@ class HomeParentItemAdapterPreview(
             }
 
             homeAccount?.setOnClickListener { v ->
-                showAccountSelectLinear(v?.context ?: return@setOnClickListener)
+                (activity as MainActivity).showAccountSelectLinear(v?.context ?: return@setOnClickListener)
             }
 
             (binding as? FragmentHomeHeadTvBinding)?.apply {