From 44aad384649b032d557164d29b4c494e84f00882 Mon Sep 17 00:00:00 2001
From: Yuriy Glukhov <yuriy.glukhov@gmail.com>
Date: Fri, 10 Aug 2018 18:12:18 +0300
Subject: [PATCH] Use stint in wrc20 challenge

---
 eth_contracts.nimble |  2 +-
 examples/test.nim    | 14 ++++++++++----
 2 files changed, 11 insertions(+), 5 deletions(-)

diff --git a/eth_contracts.nimble b/eth_contracts.nimble
index 2ecd876..fab7ab2 100644
--- a/eth_contracts.nimble
+++ b/eth_contracts.nimble
@@ -6,7 +6,7 @@ skipDirs      = @["examples"]
 
 # Dependencies
 
-requires "nim >= 0.18.1"
+requires "nim >= 0.18.1", "stint"
 
 proc buildExample(name: string) =
   exec "nim c --out:examples/" & name & ".wasm examples/" & name
diff --git a/examples/test.nim b/examples/test.nim
index 8ecb16e..c0cb2a6 100644
--- a/examples/test.nim
+++ b/examples/test.nim
@@ -1,7 +1,7 @@
 ## ewasm “WRC20” token contract coding challenge
 ## https://gist.github.com/axic/16158c5c88fbc7b1d09dfa8c658bc363
 
-import ../eth_contracts
+import ../eth_contracts, stint
 
 proc do_balance() =
   if getCallDataSize() != 24:
@@ -30,15 +30,21 @@ proc do_transfer() =
   var recipientBalance: array[32, byte]
   storageLoad(recipient, addr recipientBalance)
 
+  var sb = readUintBE[256](senderBalance)
+  var rb = readUintBE[256](recipientBalance)
+  let v = value.u256
+
   block:
     # This is a quick stub to avoid using stint for now. Works only with
     # single byte values.
-    if sender_balance[31].uint64 < value:
+    if sb < v:
       revert(nil, 0)
 
-    sender_balance[31] -= value.byte
-    recipient_balance[31] += value.byte
+    sb -= v
+    rb += v
 
+  senderBalance = sb.toByteArrayBE()
+  recipientBalance = rb.toByteArrayBE()
   storageStore(sender, addr senderBalance)
   storageStore(recipient, addr recipientBalance)