[Model] * TransactionTableModel, update cachedDelegations when new p2cs output is received.

furszy · furszy · commit 51019026bc04 · 2019-11-08T18:24:38.000-03:00
* WalletModel, isDelegatedToMe method created.
diff --git a/src/qt/transactiontablemodel.cpp b/src/qt/transactiontablemodel.cpp
@@ -152,16 +152,7 @@ class TransactionTablePriv
 
                     // Check for delegations
                     if (record.type == TransactionRecord::P2CSDelegation) {
-                        CSDelegation delegation(false, record.address);
-                        delegation.cachedTotalAmount += record.credit;
-                        int index = cachedDelegations.indexOf(delegation);
-                        if (index == -1) {
-                            cachedDelegations.append(delegation);
-                        } else {
-                            CSDelegation del = cachedDelegations[index];
-                            del.delegatedUtxo.append(record.getTxID());
-                            del.cachedTotalAmount += record.credit;
-                        }
+                        checkForDelegations(record, cachedDelegations);
                     }
                 }
 
@@ -175,6 +166,20 @@ class TransactionTablePriv
         return std::make_pair(cachedWallet, cachedDelegations);
     }
 
+    static void checkForDelegations(const TransactionRecord& record, QList<CSDelegation>& cachedDelegations) {
+        CSDelegation delegation(false, record.address);
+        delegation.cachedTotalAmount += record.credit;
+        int index = cachedDelegations.indexOf(delegation);
+        if (index == -1) {
+            cachedDelegations.append(delegation);
+        } else {
+            CSDelegation del = cachedDelegations[index];
+            del.delegatedUtxo.append(record.getTxID());
+            del.cachedTotalAmount += record.credit;
+        }
+
+    }
+
     static bool HasZcTxesIfNeeded(const TransactionRecord& record) {
         return (record.type == TransactionRecord::ZerocoinMint ||
                 record.type == TransactionRecord::ZerocoinSpend ||
@@ -235,6 +240,12 @@ class TransactionTablePriv
                         for (const TransactionRecord& rec : toInsert) {
                             cachedWallet.insert(insert_idx, rec);
                             if (!hasZcTxes) hasZcTxes = HasZcTxesIfNeeded(rec);
+
+                            // Check for delegations
+                            if (rec.type == TransactionRecord::P2CSDelegation) {
+                                checkForDelegations(rec, cachedDelegations);
+                            }
+
                             insert_idx += 1;
                             // Return record
                             ret = rec;
diff --git a/src/qt/walletmodel.cpp b/src/qt/walletmodel.cpp
@@ -506,6 +506,20 @@ const CWalletTx* WalletModel::getTx(uint256 id){
     return wallet->GetWalletTx(id);
 }
 
+bool WalletModel::isDelegatedToMe(QString id) {
+    uint256 hashTx;
+    hashTx.SetHex(id.toStdString());
+    const CWalletTx* tx = getTx(hashTx);
+    if (tx && tx->HasP2CSOutputs()) {
+        for (auto out : tx->vout) {
+            if (wallet->IsMine(out) & ISMINE_COLD) {
+                return true;
+            }
+        }
+    }
+    return false;
+}
+
 bool WalletModel::mintCoins(CAmount value, CCoinControl* coinControl ,std::string &strError){
     CWalletTx wtx;
     std::vector<CDeterministicMint> vMints;
diff --git a/src/qt/walletmodel.h b/src/qt/walletmodel.h
@@ -277,6 +277,11 @@ class WalletModel : public QObject
      */
     PairResult getNewStakingAddress(CBitcoinAddress& ret, std::string label = "") const;
 
+    /**
+     * If the wallet has the tx and it was a stakes delegation to this wallet
+     */
+    bool isDelegatedToMe(QString txId);
+
     bool whitelistAddressFromColdStaking(const QString &addressStr);
     bool blacklistAddressFromColdStaking(const QString &address);
     bool updateAddressBookPurpose(const QString &addressStr, const std::string& purpose);
