From c0970f3906c2e798a6a60763524fa39ce796debc Mon Sep 17 00:00:00 2001
From: apanizo <adolfo.panizo@gmail.com>
Date: Wed, 22 Aug 2018 13:43:43 +0200
Subject: [PATCH] WA-521 Logic for handling txs via provider signatures

---
 src/logic/safe/safeBlockchainOperations.js | 45 ++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/src/logic/safe/safeBlockchainOperations.js b/src/logic/safe/safeBlockchainOperations.js
index 650775c9..efa2d531 100644
--- a/src/logic/safe/safeBlockchainOperations.js
+++ b/src/logic/safe/safeBlockchainOperations.js
@@ -3,6 +3,9 @@ import { calculateGasOf, checkReceiptStatus, calculateGasPrice } from '~/logic/w
 import { type Operation, submitOperation } from '~/logic/safe/safeTxHistory'
 import { getDailyLimitModuleFrom } from '~/logic/contracts/dailyLimitContracts'
 import { getSafeEthereumInstance } from '~/logic/safe/safeFrontendOperations'
+import { generateMetamaskSignature, generateTxGasEstimateFrom, estimateDataGas } from '~/logic/safe/safeTxSigner'
+import { storeSignature, getSignaturesFrom } from '~/utils/localStorage/signatures'
+import { signaturesViaMetamask } from '~/config'
 
 export const approveTransaction = async (
   safeAddress: string,
@@ -15,6 +18,14 @@ export const approveTransaction = async (
 ) => {
   const gasPrice = await calculateGasPrice()
 
+  if (signaturesViaMetamask()) {
+    const safe = await getSafeEthereumInstance(safeAddress)
+    const txGasEstimate = await generateTxGasEstimateFrom(safe, safeAddress, data, to, valueInWei, operation)
+    const signature =
+      await generateMetamaskSignature(safe, safeAddress, sender, to, valueInWei, nonce, data, operation, txGasEstimate)
+    storeSignature(safeAddress, nonce, signature)
+  }
+
   const gnosisSafe = await getSafeEthereumInstance(safeAddress)
   const txData = gnosisSafe.contract.approveTransactionWithParameters.getData(to, valueInWei, data, operation, nonce)
   const gas = await calculateGasOf(txData, sender, safeAddress)
@@ -39,6 +50,40 @@ export const executeTransaction = async (
 ) => {
   const gasPrice = await calculateGasPrice()
 
+  if (signaturesViaMetamask()) {
+    const safe = await getSafeEthereumInstance(safeAddress)
+    const txGasEstimate = await generateTxGasEstimateFrom(safe, safeAddress, data, to, valueInWei, operation)
+    const signature =
+      await generateMetamaskSignature(safe, safeAddress, sender, to, valueInWei, nonce, data, operation, txGasEstimate)
+    storeSignature(safeAddress, nonce, signature)
+
+    const sigs = getSignaturesFrom(safeAddress, nonce)
+
+    const threshold = await safe.getThreshold()
+    const gas = await estimateDataGas(safe, to, valueInWei, data, operation, txGasEstimate, 0, nonce, Number(threshold))
+    const numOwners = await safe.getOwners()
+    const gasIncludingRemovingStoreUpfront = gas + txGasEstimate + (numOwners.length * 15000)
+
+    const txReceipt = await safe.execTransactionAndPaySubmitter(
+      to,
+      valueInWei,
+      data,
+      operation,
+      txGasEstimate,
+      0,
+      0,
+      0,
+      sigs,
+      { from: sender, gas: gasIncludingRemovingStoreUpfront, gasPrice },
+    )
+
+    const txHash = txReceipt.tx
+    await checkReceiptStatus(txHash)
+    // await submitOperation(safeAddress, to, valueInWei, data, operation, nonce, txHash, sender, 'execution')
+
+    return txHash
+  }
+
   const gnosisSafe = await getSafeEthereumInstance(safeAddress)
   const txConfirmationData =
     gnosisSafe.contract.execTransactionIfApproved.getData(to, valueInWei, data, operation, nonce)