From 7201a74aeaa6b646e97d7c9c2c4cc979ace5f4f7 Mon Sep 17 00:00:00 2001
From: Amir Bandeali <abandeali1@gmail.com>
Date: Thu, 8 Feb 2018 17:09:34 -0800
Subject: Move utils dir into src

---
 packages/contracts/src/utils/balances.ts | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 packages/contracts/src/utils/balances.ts

(limited to 'packages/contracts/src/utils/balances.ts')

diff --git a/packages/contracts/src/utils/balances.ts b/packages/contracts/src/utils/balances.ts
new file mode 100644
index 000000000..2ebc10313
--- /dev/null
+++ b/packages/contracts/src/utils/balances.ts
@@ -0,0 +1,30 @@
+import { BigNumber } from '@0xproject/utils';
+import * as _ from 'lodash';
+import * as Web3 from 'web3';
+
+import { DummyTokenContract } from '../contract_wrappers/generated/dummy_token';
+
+import { BalancesByOwner } from './types';
+
+export class Balances {
+    private _tokenContractInstances: DummyTokenContract[];
+    private _ownerAddresses: string[];
+    constructor(tokenContractInstances: DummyTokenContract[], ownerAddresses: string[]) {
+        this._tokenContractInstances = tokenContractInstances;
+        this._ownerAddresses = ownerAddresses;
+    }
+    public async getAsync(): Promise<BalancesByOwner> {
+        const balancesByOwner: BalancesByOwner = {};
+        for (const tokenContractInstance of this._tokenContractInstances) {
+            for (const ownerAddress of this._ownerAddresses) {
+                let balance = await tokenContractInstance.balanceOf.callAsync(ownerAddress);
+                balance = new BigNumber(balance);
+                if (_.isUndefined(balancesByOwner[ownerAddress])) {
+                    balancesByOwner[ownerAddress] = {};
+                }
+                balancesByOwner[ownerAddress][tokenContractInstance.address] = balance;
+            }
+        }
+        return balancesByOwner;
+    }
+}
-- 
cgit v1.2.3