squirrel.sol

pragma solidity ^0.5.1;

/* 
@title squirrel
@author Muhammed Nagy <me@muhnagy.com>
*/

contract squirrel {
    struct goal {uint256 target ; uint256 balance;}
    
    mapping (address => goal)  goals;
    uint256 count = 0;
    function createGoal(uint256 target) payable public returns (bool) {
        if (goals[msg.sender].target == 0) {
            count += 1;
        }
        goals[msg.sender].target = target;
        goals[msg.sender].balance += msg.value;
        return true;
    }
    
    function makePayment() payable public  returns (uint256) {
        goals[msg.sender].balance += msg.value;
        if (goals[msg.sender].balance >= goals[msg.sender].target) {
                msg.sender.transfer(goals[msg.sender].balance);
                goals[msg.sender].balance = 0;
                goals[msg.sender].target = 0;
                count -= 1;
        }
        return goals[msg.sender].balance;
    } 
    
    function getGoalsCount() public view returns (uint256) {
        return count;
    }
    function getInfo() public view returns (uint256, uint256){
        return (goals[msg.sender].target, goals[msg.sender].balance); 
    }
    
}