From e1ae551560ad12501256fd5702dcd929baa84100 Mon Sep 17 00:00:00 2001
From: Steve Klebanoff <steve.klebanoff@gmail.com>
Date: Fri, 26 Oct 2018 15:20:04 -0700
Subject: move funct into util

---
 packages/instant/src/util/balance.ts | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 packages/instant/src/util/balance.ts

(limited to 'packages/instant/src/util')

diff --git a/packages/instant/src/util/balance.ts b/packages/instant/src/util/balance.ts
new file mode 100644
index 000000000..d442da9e0
--- /dev/null
+++ b/packages/instant/src/util/balance.ts
@@ -0,0 +1,30 @@
+import { BuyQuote } from '@0x/asset-buyer';
+import { Web3Wrapper } from '@0x/web3-wrapper';
+import { Dispatch } from 'redux';
+
+import { ZeroExInstantError } from '../types';
+
+import { errorUtil } from './error';
+
+export const balanceUtil = {
+    /**
+     * Checks to see if user has enough balance to buy assets
+     * If they do not, flash an error and return false
+     * If they do, return true
+     */
+    checkSufficientBalanceAndFlashError: async (
+        takerAddress: string,
+        buyQuote: BuyQuote,
+        web3Wrapper: Web3Wrapper,
+        dispatch: Dispatch,
+    ) => {
+        const balanceWei = await web3Wrapper.getBalanceInWeiAsync(takerAddress);
+
+        if (balanceWei < buyQuote.worstCaseQuoteInfo.totalEthAmount) {
+            const balanceError = new Error(ZeroExInstantError.InsufficientBalance);
+            errorUtil.errorFlasher.flashNewError(dispatch, balanceError);
+            return false;
+        }
+        return true;
+    },
+};
-- 
cgit