// Vulnerability label: TD1
pragma solidity 0.5.5;
/**
 * @title OMNIS-BIT ICO CONTRACT
 * @dev ERC-20 Token Standard Compliant
 * Official OMNIS-BIT SmartContract
 * website: www.cryptohunters.club
 */

// ----------------------------------------------------------------------------
// Safe maths
// ----------------------------------------------------------------------------
library SafeMath {
    function add(uint a, uint b) internal pure returns(uint c) {
        c = a + b;
        require(c >= a);
    }

    function sub(uint a, uint b) internal pure returns(uint c) {
        require(b <= a);
        c = a - b;
    }

    function mul(uint a, uint b) internal pure returns(uint c) {
        c = a * b;
        require(a == 0 || c / a == b);
    }

    function div(uint a, uint b) internal pure returns(uint c) {
        require(b > 0);
        c = a / b;
    }
}

// ----------------------------------------------------------------------------
// ERC20 Token Standard Interface
// ----------------------------------------------------------------------------
interface ERC20Interface {
    function totalSupply() external returns(uint);

    function balanceOf(address tokenOwner) external returns(uint balance);

    function allowance(address tokenOwner, address spender) external returns(uint remaining);

    function transfer(address to, uint tokens) external returns(bool success);

    function approve(address spender, uint tokens) external returns(bool success);

    function transferFrom(address from, address to, uint tokens) external returns(bool success);

    event Transfer(address indexed from, address indexed to, uint tokens);
    event Approval(address indexed tokenOwner, address indexed spender, uint tokens);
}

// ----------------------------------------------------------------------------
// DateTime API Interface
// ----------------------------------------------------------------------------
interface DateTimeAPI {

    function toTimestamp(uint16 year, uint8 month, uint8 day, uint8 hour, uint8 minute) external returns(uint timestamp);

}

// ----------------------------------------------------------------------------
// Main Contract definition
// ----------------------------------------------------------------------------
contract ICO {

// ----------------------------------------------------------------------------
// DateTime API Contract Addresses for each network
// ----------------------------------------------------------------------------
    DateTimeAPI dateTimeContract = DateTimeAPI(0x1a6184CD4C5Bea62B0116de7962EE7315B7bcBce);//Main
    //DateTimeAPI dateTimeContract = DateTimeAPI(0x71b6e049E78c75fC61480357CD5aA1B81E1b16E0);//Kovan
    //DateTimeAPI dateTimeContract = DateTimeAPI(0x670b2B167e13b131C491D87bA745dA41f07ecbc3);//Rinkeby
    //DateTimeAPI dateTimeContract = DateTimeAPI(0x1F0a2ba4B115bd3e4007533C52BBd30C17E8B222); //Ropsten

    using SafeMath
    for uint256;

    enum State {
        //This ico have these states
        preSale,
        ICO,
        finishing,
        extended,
        successful
    }

    //public variables

    //state related
    State public state = State.preSale; //Set initial stage
    

    //time related
    uint256 public startTime = dateTimeContract.toTimestamp(2019, 3, 20, 0, 0);
    uint256 public ICOdeadline = dateTimeContract.toTimestamp(2019, 6, 5, 23, 59);
    uint256 public completedAt;

    //token related
    ERC20Interface public tokenReward;
    uint256 public presaleLimit = 200000000 * 10 ** 18; //200.000.000 Tokens
    uint256 public ICOLimit = 360000000 * 10 ** 18; //360.000.000 Tokens

    //funding related
    uint256 public totalRaised; //eth in wei
    uint256 public totalDistributed; //tokens distributed
    uint256 public totalReferral; //total tokens for referrals
    mapping(address => uint256) public referralBalance; //referral ledger
    uint256[7] public rates = [1000, 800, 750, 700, 650, 600, 500];
    
    //info
    address public creator;
    address payable public beneficiary;
    string public version = '0.3';

    //events for log
    event LogFundingReceived(address _addr, uint _amount, uint _currentTotal);
    event LogBeneficiaryPaid(address _beneficiaryAddress);
    event LogFundingSuccessful(uint _totalRaised);
    event LogFunderInitialized(
        address _creator,
        uint256 _ICOdeadline);
    event LogContributorsPayout(address _addr, uint _amount);
    event LogStateCheck(State current);

    modifier notFinished() {
        require(state != State.successful);
        _;
    }

    /**
     * @notice ICO constructor
     * @param _addressOfTokenUsedAsReward is the token to distributed
     * @param _beneficiary is the address that will receive funds collected
     */
    constructor(ERC20Interface _addressOfTokenUsedAsReward, address payable _beneficiary) public {

        creator = msg.sender;
        tokenReward = _addressOfTokenUsedAsReward;
        beneficiary = _beneficiary;

        emit LogFunderInitialized(
            creator,
            ICOdeadline);

    }

    /**
     * @notice contribution handler
     * @param referralAddress is the address of the referral for this purchase
     */
    function contribute(address referralAddress) public notFinished payable {

        //require that the ico start date is reached
        require(now >= startTime,"Too early for the sale begin");

        uint256 tokenBought = 0;

        totalRaised = totalRaised.add(msg.value); //Track funds received

        //Rate of exchange depends on stage
        if (state == State.preSale) {

            if (now <= dateTimeContract.toTimestamp(2019, 3, 22, 23, 59)) { //>start date <22/3/2019 23:59 GMT

                tokenBought = msg.value.mul(rates[0]);

            } else if (now <= dateTimeContract.toTimestamp(2019, 3, 28, 23, 59)) { //>22/3/2019 23:59 GMT <28/3/2019 23:59 GMT

                tokenBought = msg.value.mul(rates[1]);

            } else { //>28/3/2019 23:59 GMT <11/4/2019 23:59 GMT

                tokenBought = msg.value.mul(rates[2]);

            }

        } else if (state == State.ICO) {

            //require ico stage has begin
            require(now > dateTimeContract.toTimestamp(2019, 4, 20, 0, 0),"Too early for the ICO begin"); 

            if (now <= dateTimeContract.toTimestamp(2019, 4, 22, 23, 59)) { //>20/4/2019 00:00 GMT <22/4/2019 23:59 GMT

                tokenBought = msg.value.mul(rates[3]);

            } else if (now <= dateTimeContract.toTimestamp(2019, 4, 28, 23, 59)) { //>22/4/2019 23:59 GMT <28/4/2019 23:59 GMT

                tokenBought = msg.value.mul(rates[4]);

            } else if (now <= dateTimeContract.toTimestamp(2019, 5, 4, 23, 59)) { //>28/4/2019 23:59 GMT <5/5/2019 23:59 GMT

                tokenBought = msg.value.mul(rates[5]);

            } else { //>5/5/2019 23:59 GMT <5/6/2019 23:59 GMT

                tokenBought = msg.value.mul(rates[6]);

            }

        } else if (state == State.finishing) { //Poll being made

            revert("Purchases disabled while extension Poll");

        } else { //extension approved, 30 more days from approval

            tokenBought = msg.value.mul(rates[6]);

        }

        //+10% Bonus for high contributor
        if (msg.value >= 100 ether) {
            tokenBought = tokenBought.mul(11);
            tokenBought = tokenBought.div(10);
        }

        //3% for referral
        //Can be claimed at the end of ICO
        if (referralAddress != address(0) && referralAddress != msg.sender) {
            uint256 bounty = tokenBought.mul(3);
            bounty = bounty.div(100);
            totalReferral = totalReferral.add(bounty);
            referralBalance[referralAddress] = referralBalance[referralAddress].add(bounty);
        }

        if (state == State.preSale) {

            require(totalDistributed.add(tokenBought.add(totalReferral)) <= presaleLimit, "Presale Limit exceded");

        } else {

            require(totalDistributed.add(tokenBought.add(totalReferral)) <= ICOLimit, "ICO Limit exceded");

        }

        //Automatic retrieve only after a trust threshold
        if (totalRaised >= 4000 ether) {

            beneficiary.transfer(address(this).balance);

            emit LogBeneficiaryPaid(beneficiary);
        }

        totalDistributed = totalDistributed.add(tokenBought); //update total token distribution

        require(tokenReward.transfer(msg.sender, tokenBought), "Transfer could not be made");

        emit LogFundingReceived(msg.sender, msg.value, totalRaised);
        emit LogContributorsPayout(msg.sender, tokenBought);

        checkIfFundingCompleteOrExpired();
    }

    /**
     * @notice check status
     */
    function checkIfFundingCompleteOrExpired() public {

        //If we reach presale time limit 11/4/2019 23:59 GMT
        if (state == State.preSale && now > dateTimeContract.toTimestamp(2019, 4, 11, 23, 59)) {

            //change state to ICO
            state = State.ICO;

        } else if (state == State.ICO && now > ICOdeadline) { //If we reach the ICO deadline

            //change state to finishing for extension poll
            state = State.finishing;

        } else if (state == State.extended && now > ICOdeadline) { //If it was extended, check until extension expires

            state = State.successful; //ico becomes Successful
            completedAt = now; //ICO is complete

            emit LogFundingSuccessful(totalRaised); //we log the finish
            finished(); //and execute closure

        }

        emit LogStateCheck(state);

    }

    /**
     * @notice closure handler
     */
    function finished() public { //When finished, eth are transfered to beneficiary

        //Only on sucess
        require(state == State.successful, "Wrong Stage");

        beneficiary.transfer(address(this).balance);

        emit LogBeneficiaryPaid(beneficiary);

    }

    /**
     * @notice referral bounty claim
     */
    function claimReferral() public {

        //Only on sucess
        require(state == State.successful, "Wrong Stage");

        uint256 bounty = referralBalance[msg.sender]; //check, get balance
        referralBalance[msg.sender] = 0; //effect, clear balance

        //interact
        require(tokenReward.transfer(msg.sender, bounty), "Transfer could not be made");

        //log
        emit LogContributorsPayout(msg.sender, bounty);
    }

    /**
     * @notice remaining tokens retrieve
     */
    function retrieveTokens() public {

        //Only creator
        require(msg.sender == creator,"You are not the creator");
        //Only on success
        require(state == State.successful, "Wrong Stage");
        //Only after 30 days claim period for referrals
        require(now >= completedAt.add(30 days), "Too early to retrieve");

        uint256 remanent = tokenReward.balanceOf(address(this));

        require(tokenReward.transfer(beneficiary, remanent), "Transfer could not be made");
    }

    /**
     * @notice extension poll result handler
     * @param pollResult a boolean value of approved(true) or denied(false)
     */
    function extension(bool pollResult) public {

        //Only creator
        require(msg.sender == creator,"You are not the creator");
        //Only on poll stage
        require(state == State.finishing, "Wrong Stage");

        //poll results
        if (pollResult == true) { //Approved
            //extended stage
            state = State.extended;
            //extension is 30Days
            ICOdeadline = now.add(30 days);
        } else { //Denied
            //ico becomes Successful
            state = State.successful;
            //ICO is complete, stamp it
            completedAt = now;

            emit LogFundingSuccessful(totalRaised); //we log the finish
            finished(); //and execute closure

        }
    }

    /*
     * @notice direct payments handler
     */
    function() external payable {

        contribute(address(0)); //no referral

    }
}