InstaContract/contracts/bin/MoatAsset.sol

117 lines
3.5 KiB
Solidity

// withdraw the extra assets other than global balance (in case anyone donated for free) and then no need for seperate brokerage calculation
// IMPORTANT CHECK - decimals() - how the balance of tokens with less than 18 decimals are stored. Factor it.
// update the balance along with "transferAssets" functions and also check the for onlyAllowedResolver
// transfer assets to different address (create 2 different mappings) - 48 hour time to transfer all - send email for this
pragma solidity ^0.4.24;
import "openzeppelin-solidity/contracts/math/SafeMath.sol";
import "openzeppelin-solidity/contracts/token/ERC20/IERC20.sol";
interface AddressRegistry {
function getAddr(string name) external returns(address);
function isApprovedResolver(address user) external returns(bool);
}
contract Registry {
address public registryAddress;
AddressRegistry addrReg = AddressRegistry(registryAddress);
modifier onlyAllowedResolver(address user) {
require(
addrReg.isApprovedResolver(user),
"Permission Denied"
);
_;
}
function getAddress(string name) internal view returns(address addr) {
addr = addrReg.getAddr(name);
require(addr != address(0), "Invalid Address");
}
}
contract AssetDB is Registry {
using SafeMath for uint;
using SafeMath for uint256;
mapping(address => mapping(address => uint)) balances;
address eth = 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE;
function getBalance(
address assetHolder,
address tokenAddr
) public view returns (uint256 balance)
{
balance = balances[assetHolder][tokenAddr];
}
function deposit(address tknAddr, uint amount) public payable {
if (msg.value > 0) {
balances[msg.sender][eth] = balances[msg.sender][eth].add(msg.value);
} else {
IERC20 tokenFunctions = IERC20(tknAddr);
tokenFunctions.transferFrom(msg.sender, address(this), amount);
balances[msg.sender][tknAddr] = balances[msg.sender][tknAddr].add(amount);
}
}
function withdraw(address tknAddr, uint amount) public {
require(balances[msg.sender][tknAddr] >= amount, "Insufficient Balance");
balances[msg.sender][tknAddr] = balances[msg.sender][tknAddr].sub(amount);
if (tknAddr == eth) {
msg.sender.transfer(amount);
} else {
IERC20 tokenFunctions = IERC20(tknAddr);
tokenFunctions.transfer(msg.sender, amount);
}
}
function updateBalance(
address tokenAddr,
uint amount,
bool credit,
address user
) public onlyAllowedResolver(user)
{
if (credit) {
balances[user][tokenAddr] = balances[user][tokenAddr].add(amount);
} else {
balances[user][tokenAddr] = balances[user][tokenAddr].sub(amount);
}
}
function transferAssets(
address tokenAddress,
uint amount,
address sendTo,
address user
) public onlyAllowedResolver(user)
{
if (tokenAddress == eth) {
sendTo.transfer(amount);
} else {
IERC20 tokenFunctions = IERC20(tokenAddress);
tokenFunctions.transfer(sendTo, amount);
}
balances[user][tokenAddress] = balances[user][tokenAddress].sub(amount);
}
}
contract MoatAsset is AssetDB {
constructor(address rAddr) public {
registryAddress = rAddr;
}
function () public payable {}
}