[Model] WalletModel P2CS transactions connected.

furszy · furszy · commit 65888607b87c · 2019-11-08T18:24:33.000-03:00
diff --git a/src/qt/walletmodel.cpp b/src/qt/walletmodel.cpp
@@ -351,7 +351,30 @@ WalletModel::SendCoinsReturn WalletModel::prepareTransaction(WalletModelTransact
             setAddress.insert(rcp.address);
             ++nAddresses;
 
-            CScript scriptPubKey = GetScriptForDestination(CBitcoinAddress(rcp.address.toStdString()).Get());
+            CScript scriptPubKey;
+            CBitcoinAddress out = CBitcoinAddress(rcp.address.toStdString());
+
+            if (rcp.isP2CS) {
+                CBitcoinAddress ownerAdd;
+                if (rcp.ownerAddress.isEmpty()) {
+                    // Create new internal owner address
+                    if (!getNewAddress(ownerAdd).result)
+                        return CannotCreateInternalAddress;
+                } else {
+                    ownerAdd = CBitcoinAddress(rcp.ownerAddress.toStdString());
+                }
+
+                CKeyID stakerId;
+                CKeyID ownerId;
+                if(!out.GetKeyID(stakerId) || !ownerAdd.GetKeyID(ownerId)) {
+                    return InvalidAddress;
+                }
+
+                scriptPubKey = GetScriptForStakeDelegation(stakerId, ownerId);
+            } else {
+                // Regular P2PK or P2PKH
+                scriptPubKey = GetScriptForDestination(out.Get());
+            }
             vecSend.push_back(std::pair<CScript, CAmount>(scriptPubKey, rcp.amount));
 
             total += rcp.amount;
@@ -418,9 +441,11 @@ WalletModel::SendCoinsReturn WalletModel::sendCoins(WalletModelTransaction& tran
         return AnonymizeOnlyUnlocked;
     }
 
+    bool fColdStakingActive = sporkManager.IsSporkActive(SPORK_17_COLDSTAKING_ENFORCEMENT);
+
     // Double check tx before do anything
     CValidationState state;
-    if(!CheckTransaction(*transaction.getTransaction(), true, true, state, true)){
+    if(!CheckTransaction(*transaction.getTransaction(), true, true, state, true, fColdStakingActive)){
         return TransactionCommitFailed;
     }
 
diff --git a/src/qt/walletmodel.h b/src/qt/walletmodel.h
@@ -54,6 +54,12 @@ class SendCoinsRecipient
     QString label;
     AvailableCoinsType inputType;
     bool useSwiftTX = false;
+
+    // Cold staking.
+    bool isP2CS = false;
+    QString ownerAddress;
+
+    // Amount
     CAmount amount;
     // If from a payment request, this is used for storing the memo
     QString message;
@@ -119,7 +125,8 @@ class WalletModel : public QObject
         TransactionCreationFailed, // Error returned when wallet is still locked
         TransactionCommitFailed,
         AnonymizeOnlyUnlocked,
-        InsaneFee
+        InsaneFee,
+        CannotCreateInternalAddress
     };
 
     enum EncryptionStatus {
@@ -210,6 +217,18 @@ class WalletModel : public QObject
             CZerocoinSpendReceipt &receipt
     );
 
+    // ###################
+    // Cold Staking
+    // ###################
+
+
+
+    // ###################
+    // End Cold Staking
+    // ###################
+
+
+
     // Wallet encryption
     bool setWalletEncrypted(bool encrypted, const SecureString& passphrase);
     // Passphrase only needed when unlocking
