add restore wallet mints

Author: hakob91

src/info/sigma/zcoinEngineExtension.js

@@ -20,6 +20,7 @@ import type { TxOptions } from '../../utils/coinUtils.js'
 import { logger } from '../../utils/logger'
 import { getMintsToSpend } from './coinOperations'
 import {
+  createPrivateCoin,
   createSpendTX,
   getMintCommitmentsForValue,
   parseJsonTransactionForSpend,
@@ -28,7 +29,7 @@ import {
 } from './coinUtils'
 import type { SpendCoin } from './coinUtils.js'
 import type { PrivateCoin } from './flowTypes'
-import { denominations, OP_SIGMA_MINT } from './flowTypes'
+import { denominations, OP_SIGMA_MINT, RESTORE_FILE } from './flowTypes'
 import { ZcoinStateExtension } from './zcoinStateExtension'
 
 const MILLI_TO_SEC = 1000
@@ -65,6 +66,17 @@ export class ZcoinEngineExtension implements CurrencyEngineExtension {
     this.runLooperIfNeed()
   }
 
+  async resyncBlockchain() {
+    window.setTimeout(() => {
+      this.forceReloadSpendTransactions()
+      this.zcoinStateExtensions.wakeUpConnections()
+    }, 5000)
+  }
+
+  async killEngine() {
+    this.cancelAllLooperMethods()
+  }
+
   onTxFetched(txid: string) {
     logger.info(`zcoinEngineExtension -> onTxFetched called txId ${txid}`)
     if (txid in this.savedSpendTransactionValues) {
@@ -112,6 +124,11 @@ export class ZcoinEngineExtension implements CurrencyEngineExtension {
   async loop() {
     logger.info('zcoinEngineExtension -> loop called')
 
+    const restored = await this.restore()
+    if (!restored) {
+      return
+    }
+
     // TODO: can move into newTx
     const utxos = this.engineState.getUTXOs()
     const needToMint = sumUtxos(utxos)
@@ -135,7 +152,6 @@ export class ZcoinEngineExtension implements CurrencyEngineExtension {
           }
         ]
       }
-      // mint and getMintMetadataLoop must not work paralelly in order to avoid problems related to modifying mint.json parallely
       await this.mint(edgeInfo)
     }
 
@@ -144,6 +160,86 @@ export class ZcoinEngineExtension implements CurrencyEngineExtension {
     this.onBalanceChanged()
   }
 
+  async restore(): Promise<boolean> {
+    logger.info('zcoinEngineExtension -> restore')
+
+    try {
+      const restoreJsonStr = await this.walletLocalEncryptedDisklet.getText(
+        RESTORE_FILE
+      )
+      if (restoreJsonStr && JSON.parse(restoreJsonStr).restored) {
+        return true
+      }
+    } catch (e) {
+      logger.error('zcoinEngineExtension -> something went wrong', e)
+    }
+
+    const mintData: PrivateCoin[] = []
+
+    let usedSerialNumbers = []
+    const usedCoins = await this.zcoinStateExtensions.retrieveUsedCoinSerials()
+    usedSerialNumbers = usedCoins.serials
+
+    const latestCoinIds = await this.zcoinStateExtensions.retrieveLatestCoinIds()
+    let commitmentCount = 0
+    for (const coinInfo of latestCoinIds) {
+      coinInfo.anonymitySet = []
+      for (let i = 0; i < coinInfo.id; i++) {
+        const anonimitySet = await this.zcoinStateExtensions.retrieveAnonymitySet(
+          coinInfo.denom,
+          i + 1
+        )
+        coinInfo.anonymitySet = coinInfo.anonymitySet.concat(
+          anonimitySet.serializedCoins
+        )
+        commitmentCount += anonimitySet.serializedCoins.length
+      }
+    }
+
+    let counter = 0
+    let index = -1
+    while (counter++ < 100 && index++ < commitmentCount) {
+      // commitment is only dependant to private key and index, that's why coin value is hardcoded
+      const coin = await createPrivateCoin(
+        100000000,
+        this.currencyEngine.walletInfo.keys.dataKey,
+        index,
+        this.io
+      )
+      logger.info('zcoinEngineExtension -> restore coin ', coin)
+      const isSpend = usedSerialNumbers.includes(coin.serialNumber)
+      for (const coinInfo of latestCoinIds) {
+        if (coinInfo.anonymitySet.includes(coin.commitment)) {
+          mintData.push({
+            value: coinInfo.denom,
+            index: index,
+            commitment: coin.commitment,
+            serialNumber: '',
+            groupId: coinInfo.id,
+            isSpend: isSpend,
+            spendTxId: ''
+          })
+          counter = 0
+          break
+        }
+      }
+    }
+
+    try {
+      logger.info('zcoinEngineExtension -> restore try save ', mintData)
+      await this.zcoinStateExtensions.writeMintedCoins(mintData)
+      await this.walletLocalEncryptedDisklet.setText(
+        RESTORE_FILE,
+        JSON.stringify({ restored: true })
+      )
+      logger.info('zcoinEngineExtension -> restored')
+    } catch (e) {
+      return false
+    }
+
+    return true
+  }
+
   runLooperIfNeed() {
     this.canRunLoop = true
     this.addLooperMethodToLoop('loop', 60000)
@@ -657,4 +753,12 @@ export class ZcoinEngineExtension implements CurrencyEngineExtension {
 
     return spendTransactionValues
   }
+
+  forceReloadSpendTransactions() {
+    this.zcoinStateExtensions.mintedCoins.forEach(item => {
+      if (item.spendTxId) {
+        this.zcoinStateExtensions.handleNewTxid(item.spendTxId, true)
+      }
+    })
+  }
 }

src/info/sigma/zcoinStateExtension.js

@@ -89,6 +89,12 @@ export class ZcoinStateExtension implements EngineStateExtension {
     })
   }
 
+  dumpData(): any {
+    return {
+      'engineState.missingTxsVerbose': this.missingTxsVerbose
+    }
+  }
+
   pickNextTask(uri: string, stratumVersion: string): StratumTask | void {
     logger.info('zcoinEngineExtension -> zcoinStateExtension -> pickNextTask')
     const serverState = this.engineState.serverStates[uri]