add shortcut for vestedAmount after vesting end

00f323d1 · Francisco Giordano · Martín Triay · 80e591f4 · 00f323d1
Commit 00f323d1 authored Aug 28, 2017 by Francisco Giordano Committed by Martín Triay Oct 04, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

TokenVesting.sol contracts/token/TokenVesting.sol +2 -0

No files found.
--- a/contracts/token/TokenVesting.sol
+++ b/contracts/token/TokenVesting.sol
@@ -70,6 +70,8 @@ contract TokenVesting is Ownable {
  function vestedAmount(ERC20Basic token) constant returns (uint256) {
    if (now < cliff) {
      return 0;
+    } else if (now >= end) {
+      return token.balanceOf(this);
    } else {
      uint256 currentBalance = token.balanceOf(this);
      uint256 totalBalance = currentBalance.add(released[token]);