Created
June 16, 2021 06:35
-
-
Save korrio/3ce6b6f0e253a4a4a85c83a11bce6f4c to your computer and use it in GitHub Desktop.
SimpleBank.sol
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
// SPDX-License-Identifier: MIT | |
pragma solidity ^0.6.0; | |
/** | |
* @dev Wrappers over Solidity's arithmetic operations with added overflow | |
* checks. | |
* | |
* Arithmetic operations in Solidity wrap on overflow. This can easily result | |
* in bugs, because programmers usually assume that an overflow raises an | |
* error, which is the standard behavior in high level programming languages. | |
* `SafeMath` restores this intuition by reverting the transaction when an | |
* operation overflows. | |
* | |
* Using this library instead of the unchecked operations eliminates an entire | |
* class of bugs, so it's recommended to use it always. | |
*/ | |
library SafeMath { | |
/** | |
* @dev Returns the addition of two unsigned integers, reverting on | |
* overflow. | |
* | |
* Counterpart to Solidity's `+` operator. | |
* | |
* Requirements: | |
* - Addition cannot overflow. | |
*/ | |
function add(uint256 a, uint256 b) internal pure returns (uint256) { | |
uint256 c = a + b; | |
require(c >= a, "SafeMath: addition overflow"); | |
return c; | |
} | |
/** | |
* @dev Returns the subtraction of two unsigned integers, reverting on | |
* overflow (when the result is negative). | |
* | |
* Counterpart to Solidity's `-` operator. | |
* | |
* Requirements: | |
* - Subtraction cannot overflow. | |
*/ | |
function sub(uint256 a, uint256 b) internal pure returns (uint256) { | |
require(b <= a, "SafeMath: subtraction overflow"); | |
uint256 c = a - b; | |
return c; | |
} | |
/** | |
* @dev Returns the multiplication of two unsigned integers, reverting on | |
* overflow. | |
* | |
* Counterpart to Solidity's `*` operator. | |
* | |
* Requirements: | |
* - Multiplication cannot overflow. | |
*/ | |
function mul(uint256 a, uint256 b) internal pure returns (uint256) { | |
// Gas optimization: this is cheaper than requiring 'a' not being zero, but the | |
// benefit is lost if 'b' is also tested. | |
// See: https://github.com/OpenZeppelin/openzeppelin-solidity/pull/522 | |
if (a == 0) { | |
return 0; | |
} | |
uint256 c = a * b; | |
require(c / a == b, "SafeMath: multiplication overflow"); | |
return c; | |
} | |
/** | |
* @dev Returns the integer division of two unsigned integers. Reverts on | |
* division by zero. The result is rounded towards zero. | |
* | |
* Counterpart to Solidity's `/` operator. Note: this function uses a | |
* `revert` opcode (which leaves remaining gas untouched) while Solidity | |
* uses an invalid opcode to revert (consuming all remaining gas). | |
* | |
* Requirements: | |
* - The divisor cannot be zero. | |
*/ | |
function div(uint256 a, uint256 b) internal pure returns (uint256) { | |
// Solidity only automatically asserts when dividing by 0 | |
require(b > 0, "SafeMath: division by zero"); | |
uint256 c = a / b; | |
// assert(a == b * c + a % b); // There is no case in which this doesn't hold | |
return c; | |
} | |
/** | |
* @dev Returns the remainder of dividing two unsigned integers. (unsigned integer modulo), | |
* Reverts when dividing by zero. | |
* | |
* Counterpart to Solidity's `%` operator. This function uses a `revert` | |
* opcode (which leaves remaining gas untouched) while Solidity uses an | |
* invalid opcode to revert (consuming all remaining gas). | |
* | |
* Requirements: | |
* - The divisor cannot be zero. | |
*/ | |
function mod(uint256 a, uint256 b) internal pure returns (uint256) { | |
require(b != 0, "SafeMath: modulo by zero"); | |
return a % b; | |
} | |
} | |
/// @title SimpleBank | |
/// @author nemild, kor, tot | |
/* 'contract' has similarities to 'class' in other languages (class variables, | |
inheritance, etc.) */ | |
contract SimpleBank { // CamelCase | |
using SafeMath for uint256; | |
// Declare state variables outside function, persist through life of contract | |
// dictionary that maps addresses to balances | |
mapping (address => uint256) private balances; | |
// Users in system | |
address[] accounts; | |
// Interest rate | |
uint256 rate = 3; | |
// "private" means that other contracts can't directly query balances | |
// but data is still viewable to other parties on blockchain | |
address public owner; | |
// 'public' makes externally readable (not writeable) by users or contracts | |
// Events - publicize actions to external listeners | |
event DepositMade(address accountAddress, uint amount); | |
// Constructor, can receive one or many variables here; only one allowed | |
constructor() public { | |
// msg provides details about the message that's sent to the contract | |
// msg.sender is contract caller (address of contract creator) | |
owner = msg.sender; | |
} | |
/// @notice Deposit ether into bank | |
/// @return The balance of the user after the deposit is made | |
function deposit() public payable returns (uint256) { | |
// Record account in array for looping | |
if (0 == balances[msg.sender]) { | |
accounts.push(msg.sender); | |
} | |
balances[msg.sender] = balances[msg.sender].add(msg.value); | |
// no "this." or "self." required with state variable | |
// all values set to data type's initial value by default | |
emit DepositMade(msg.sender, msg.value); // fire event | |
return balances[msg.sender]; | |
} | |
/// @notice Withdraw ether from bank | |
/// @dev This does not return any excess ether sent to it | |
/// @param withdrawAmount amount you want to withdraw | |
/// @return remainingBal The balance remaining for the user | |
function withdraw(uint withdrawAmount) public returns (uint256 remainingBal) { | |
require(balances[msg.sender] >= withdrawAmount); | |
balances[msg.sender] = balances[msg.sender].sub(withdrawAmount); | |
// Revert on failed | |
msg.sender.transfer(withdrawAmount); | |
return balances[msg.sender]; | |
} | |
/// @notice Get balance | |
/// @return The balance of the user | |
// 'constant' prevents function from editing state variables; | |
// allows function to run locally/off blockchain | |
function balance() public view returns (uint256) { | |
return balances[msg.sender]; | |
} | |
// Fallback function - Called if other functions don't match call or | |
// sent ether without data | |
// Typically, called when invalid data is sent | |
// Added so ether sent to this contract is reverted if the contract fails | |
// otherwise, the sender's money is transferred to contract | |
fallback () external { | |
revert(); // throw reverts state to before call | |
} | |
function calculateInterest(address user, uint256 _rate) private view returns(uint256) { | |
uint256 interest = balances[user].mul(_rate).div(100); | |
return interest; | |
} | |
function increaseYear() public { | |
for(uint256 i = 0; i < accounts.length; i++) { | |
address account = accounts[i]; | |
uint256 interest = calculateInterest(account, rate); | |
balances[account] = balances[account].add(interest); | |
} | |
} | |
function systemBalance() public view returns(uint256) { | |
return address(this).balance; | |
} | |
} | |
// ** END EXAMPLE ** |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment