Add account balance calculation (#298)

near · May 12, 2020 · 8651f7a · 8651f7a
1 parent 1ff1a50
commit 8651f7a
Show file tree

Hide file tree

Showing 3 changed files with 65 additions and 0 deletions.
diff --git a/lib/account.d.ts b/lib/account.d.ts
diff --git a/lib/account.js b/lib/account.js
diff --git a/src/account.ts b/src/account.ts
@@ -36,6 +36,16 @@ export interface AccountState {
     amount: string;
     staked: string;
     code_hash: string;
+    storage_usage: number;
+    locked: string;
+    balance: AccountBalance;
+}
+
+export interface AccountBalance {
+    total: string;
+    stateStaked: string;
+    staked: string;
+    available: string;
 }
 
 /**
@@ -70,6 +80,7 @@ export class Account {
      */
     async state(): Promise<AccountState> {
         await this.ready;
+        this._state.balance = await this.getAccountBalance();
         return this._state;
     }
 
@@ -324,4 +335,25 @@ export class Account {
         });
         return result;
     }
+
+    /**
+     * Returns calculated account balance
+     * @returns {Promise<AccountBalance>}
+     */
+    private async getAccountBalance(): Promise<AccountBalance> {
+        const genesisConfig = await this.connection.provider.genesisConfig();
+
+        const costPerByte = new BN(genesisConfig.runtime_config.storage_amount_per_byte);
+        const stateStaked = new BN(this._state.storage_usage).mul(costPerByte);
+        const staked = new BN(this._state.locked);
+        const totalBalance = new BN(this._state.amount).add(staked);
+        const availableBalance = totalBalance.sub(staked).sub(stateStaked);
+
+        return {
+            total: totalBalance.toString(),
+            stateStaked: stateStaked.toString(),
+            staked: staked.toString(),
+            available: availableBalance.toString()
+        };
+    }
 }