token.js throw error on invalid balance call

ArweaveTeam · Jun 24, 2020 · 413a689 · 413a689
1 parent 61ef3b4
commit 413a689
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/examples/token.js b/examples/token.js
@@ -39,6 +39,12 @@ export function handle(state, action) {
   }
 
   if (input.function == 'balance') {
+    if (typeof input.target !== 'string') {
+      throw new ContractError(`Must specific target to get balance for`);
+    }
+    if (typeof balances[target] !== 'number') {
+      throw new ContractError(`Cannnot get balance, target does not exist`);
+    }
     let target = input.target;
     let ticker = state.ticker;
     let divisibility = state.divisibility;