Revert "Remove CalcMissingRewardTempFix"

This reverts commit db227f7.
DeFiCh · Mar 3, 2023 · 4c15585 · 4c15585
1 parent db227f7
commit 4c15585
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/src/masternodes/masternodes.cpp b/src/masternodes/masternodes.cpp
@@ -1325,3 +1325,24 @@ CAmount CCustomCSView::GetFeeBurnPctFromAttributes() const {
     return attributes->GetValue(feeBurnPctKey, Params().GetConsensus().props.feeBurnPct);
 }
 
+void CalcMissingRewardTempFix(CCustomCSView &mnview, const uint32_t targetHeight, const CWallet &wallet) {
+    mnview.ForEachMasternode([&](const uint256 &id, const CMasternode &node) {
+        if (node.rewardAddressType) {
+            const CScript rewardAddress = GetScriptForDestination(node.rewardAddressType == PKHashType ?
+                                                                  CTxDestination(PKHash(node.rewardAddress)) :
+                                                                  CTxDestination(WitnessV0KeyHash(node.rewardAddress)));
+            if (IsMineCached(wallet, rewardAddress) == ISMINE_SPENDABLE) {
+                mnview.CalculateOwnerRewards(rewardAddress, targetHeight);
+            }
+        }
+
+        const CScript rewardAddress = GetScriptForDestination(node.ownerType == PKHashType ?
+                                                              CTxDestination(PKHash(node.ownerAuthAddress)) :
+                                                              CTxDestination(WitnessV0KeyHash(node.ownerAuthAddress)));
+        if (IsMineCached(wallet, rewardAddress) == ISMINE_SPENDABLE) {
+            mnview.CalculateOwnerRewards(rewardAddress, targetHeight);
+        }
+
+        return true;
+    });
+}
diff --git a/src/masternodes/masternodes.h b/src/masternodes/masternodes.h
@@ -46,6 +46,10 @@ CAmount GetTokenCreationFee(int height);
 CAmount GetMnCollateralAmount(int height);
 CAmount GetProposalCreationFee(int height, const CCustomCSView &view, const CCreateProposalMessage &msg);
 
+// Update owner rewards for MNs missing call to CalculateOwnerRewards after voter fee distributions.
+// Missing call fixed in: https://github.com/DeFiCh/ain/pull/1766
+void CalcMissingRewardTempFix(CCustomCSView &mnview, const uint32_t targetHeight, const CWallet &wallet);
+
 enum class UpdateMasternodeType : uint8_t {
     None             = 0x00,
     OwnerAddress     = 0x01,