Actually lock mn collateral (#513)

Author: Duddino

scripts/composables/use_masternode.js

@@ -40,8 +40,10 @@ export const useMasternode = defineStore('masternode', () => {
         await database.addMasternode(toRaw(masternode.value));
     });
 
-    fetchProposalsFromDatabase().then(() => {});
-    fetchMasternodeFromDatabase().then(() => {});
+    getEventEmitter().on('wallet-import', () => {
+        fetchProposalsFromDatabase().then(() => {});
+        fetchMasternodeFromDatabase().then(() => {});
+    });
     getEventEmitter().on('toggle-network', () => {
         fetchProposalsFromDatabase().then(() => {});
         fetchMasternodeFromDatabase().then(() => {});

scripts/composables/use_wallet.js

@@ -137,6 +137,8 @@ export const useWallet = defineStore('wallet', () => {
     const isCreatingTransaction = () => createAndSendTransaction.isLocked();
     const getMasternodeUTXOs = () => wallet.getMasternodeUTXOs();
     const getPath = (script) => wallet.getPath(script);
+    const lockCoin = (out) => wallet.lockCoin(out);
+    const unlockCoin = (out) => wallet.unlockCoin(out);
 
     getEventEmitter().on('toggle-network', async () => {
         isEncrypted.value = await hasEncryptedWallet();
@@ -198,5 +200,7 @@ export const useWallet = defineStore('wallet', () => {
         getMasternodeUTXOs,
         getPath,
         blockCount,
+        lockCoin,
+        unlockCoin,
     };
 });

scripts/masternode/Masternode.vue

@@ -13,6 +13,7 @@ import { getNetwork } from '../network/network_manager.js';
 import { translation, ALERTS } from '../i18n.js';
 import { generateMasternodePrivkey, parseIpAddress } from '../misc';
 import { useAlerts } from '../composables/use_alerts.js';
+import { COutpoint } from '../transaction.js';
 
 const { createAlert } = useAlerts();
 
@@ -28,6 +29,22 @@ const masternodePrivKey = ref('');
 // Array of possible masternode UTXOs
 const possibleUTXOs = ref(wallet.getMasternodeUTXOs());
 
+watch(masternode, (masternode, oldValue) => {
+    if (oldValue?.collateralTxId) {
+        wallet.unlockCoin(
+            new COutpoint({ txid: oldValue.collateralTxId, n: oldValue.outidx })
+        );
+    }
+
+    if (masternode?.collateralTxId) {
+        wallet.lockCoin(
+            new COutpoint({
+                txid: masternode.collateralTxId,
+                n: masternode.outidx,
+            })
+        );
+    }
+});
 function updatePossibleUTXOs() {
     possibleUTXOs.value = wallet.getMasternodeUTXOs();
 }