fix moonriver

Author: Denis Lyazgin

feature-staking-impl/src/main/java/jp/co/soramitsu/feature_staking_impl/presentation/confirm/ConfirmStakingViewModel.kt

@@ -227,7 +227,12 @@ class ConfirmStakingViewModel(
                     is SetupStakingProcess.ReadyToSubmit.Parachain -> {
                         val collator = currentProcessState.payload.blockProducers.first()
                         val amount = bondPayload?.amount ?: error("Amount cant be null")
-                        val delegationCount = (scenarioInteractor.stakingStateFlow.first() as StakingState.Parachain.Delegator).delegations.size
+                        val delegationCount = when (val state = scenarioInteractor.stakingStateFlow.first()) {
+                            is StakingState.Parachain.Delegator -> state.delegations.size
+                            is StakingState.Parachain.Collator -> 0 // todo add collators support
+                            is StakingState.Parachain.None -> 0
+                            else -> 0
+                        }
                         setupStakingInteractor.estimateFinalParachainFee(collator, it.planksFromAmount(amount), delegationCount)
                     }
                 }
@@ -287,7 +292,12 @@ class ConfirmStakingViewModel(
                 val token = controllerAssetFlow.first().token
                 val collator = currentProcessState.payload.blockProducers.first()
                 val amount = bondPayload?.amount ?: error("Amount cant be null")
-                val delegationCount = (scenarioInteractor.stakingStateFlow.first() as StakingState.Parachain.Delegator).delegations.size
+                val delegationCount = when (val state = scenarioInteractor.stakingStateFlow.first()) {
+                    is StakingState.Parachain.Delegator -> state.delegations.size
+                    is StakingState.Parachain.Collator -> 0 // todo add collators support
+                    is StakingState.Parachain.None -> 0
+                    else -> 0
+                }
                 val accountAddress = (scenarioInteractor.stakingStateFlow.first() as StakingState.Parachain).accountAddress
                 setupStakingInteractor.setupStaking(collator, token.planksFromAmount(amount), delegationCount, accountAddress)
             }

feature-staking-impl/src/main/java/jp/co/soramitsu/feature_staking_impl/presentation/staking/main/StakingViewState.kt

@@ -115,10 +115,6 @@ sealed class StakeViewState<S>(
     private val availableManageActions: Set<ManageStakeAction>
 ) : StakingViewState() {
 
-    init {
-        syncStakingRewards()
-    }
-
     val manageStakingActionsButtonVisible = availableManageActions.isNotEmpty()
 
     private val _showManageActionsEvent = MutableLiveData<Event<ManageStakingBottomSheet.Payload>>()
@@ -156,7 +152,7 @@ sealed class StakeViewState<S>(
         _showStatusAlertEvent.value = Event(titleAndMessage)
     }
 
-    private fun syncStakingRewards() {
+    protected fun syncStakingRewards() {
         scope.launch {
             val syncResult = stakingInteractor.syncStakingRewards(stakeState.chain.id, stakeState.rewardsAddress)
 
@@ -207,7 +203,11 @@ class ValidatorViewState(
     summaryFlowProvider = { relayChainScenarioInteractor.observeValidatorSummary(validatorState).shareIn(scope, SharingStarted.Eagerly, replay = 1) },
     statusMessageProvider = { getValidatorStatusTitleAndMessage(resourceManager, it) },
     availableManageActions = ManageStakeAction.values().toSet() - ManageStakeAction.VALIDATORS
-)
+) {
+    init {
+        syncStakingRewards()
+    }
+}
 
 private fun getValidatorStatusTitleAndMessage(
     resourceManager: ResourceManager,
@@ -237,7 +237,11 @@ class StashNoneViewState(
     summaryFlowProvider = { relayChainScenarioInteractor.observeStashSummary(stashState).shareIn(scope, SharingStarted.Eagerly, replay = 1) },
     statusMessageProvider = { getStashStatusTitleAndMessage(resourceManager, it) },
     availableManageActions = ManageStakeAction.values().toSet() - ManageStakeAction.PAYOUTS
-)
+) {
+    init {
+        syncStakingRewards()
+    }
+}
 
 private fun getStashStatusTitleAndMessage(
     resourceManager: ResourceManager,
@@ -265,7 +269,11 @@ class NominatorViewState(
     summaryFlowProvider = { relayChainScenarioInteractor.observeNominatorSummary(nominatorState).shareIn(scope, SharingStarted.Eagerly, replay = 1) },
     statusMessageProvider = { getNominatorStatusTitleAndMessage(resourceManager, it) },
     availableManageActions = ManageStakeAction.values().toSet()
-)
+) {
+    init {
+        syncStakingRewards()
+    }
+}
 
 private fun getNominatorStatusTitleAndMessage(
     resourceManager: ResourceManager,