Add init fn

aragon · Jan 31, 2017 · 7db34f7 · 7db34f7
1 parent 7d4a620
commit 7db34f7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/contracts/token/ERC20Lib.sol b/contracts/token/ERC20Lib.sol
@@ -14,6 +14,11 @@ library ERC20Lib {
   event Transfer(address indexed from, address indexed to, uint value);
   event Approval(address indexed owner, address indexed spender, uint value);
 
+  function init(TokenStorage storage self, uint _initial_supply) {
+    self.totalSupply = _initial_supply;
+    self.balances[msg.sender] = _initial_supply;
+  }
+
   function transfer(TokenStorage storage self, address _to, uint _value) internal returns (bool success) {
     self.balances[msg.sender] = self.balances[msg.sender].minus(_value);
     self.balances[_to] = self.balances[_to].plus(_value);

diff --git a/contracts/token/StandardToken.sol b/contracts/token/StandardToken.sol
@@ -22,8 +22,7 @@ import './ERC20Lib.sol';
    uint public INITIAL_SUPPLY = 10000;
 
    function StandardToken() {
-     token.totalSupply = INITIAL_SUPPLY;
-     token.balances[msg.sender] = INITIAL_SUPPLY;
+     token.init(INITIAL_SUPPLY);
    }
 
    function totalSupply() constant returns (uint) {