// Allow ERC20 deposits
// withdraw the extra assets other than global balance (in case anyone donated for free) and then no need for seperate brokerage calculation
// IMPORTANT CHECK - 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

pragma solidity ^0.4.24;

interface AddressRegistry {
    function getAddr(string name) external returns(address);
    function isApprovedResolver(address user) external returns(bool);
}

interface token {
    function approve(address spender, uint256 value) external returns (bool);
    function transfer(address receiver, uint amount) external returns (bool);
    function transferFrom(address from, address to, uint amount) external returns (bool);
}


contract Registry {

    address public registryAddress;
    AddressRegistry aRegistry = AddressRegistry(registryAddress);

    modifier onlyAdmin() {
        require(
            msg.sender == getAddress("admin"),
            "Permission Denied"
        );
        _;
    }

    modifier onlyAllowedResolver(address user) {
        require(
            aRegistry.isApprovedResolver(user),
            "Permission Denied"
        );
        _;
    }

    function getAddress(string name) internal view returns(address addr) {
        addr = aRegistry.getAddr(name);
        require(addr != address(0), "Invalid Address");
    }
 
}


contract AssetDB is Registry {

    // AssetOwner >> TokenAddress >> Balance (as per respective decimals)
    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] += msg.value;
        } else {
            token tokenFunctions = token(tknAddr);
            tokenFunctions.transferFrom(msg.sender, address(this), amount);
            balances[msg.sender][eth] += msg.value;
        }
    }

    function withdraw(address addr, uint amt) public {
        require(balances[msg.sender][addr] >= amt, "Insufficient Balance");
        balances[msg.sender][addr] -= amt;
        if (addr == eth) {
            msg.sender.transfer(amt);
        } else {
            token tokenFunctions = token(addr);
            tokenFunctions.transfer(msg.sender, amt);
        }
    }

    function updateBalance(
        address tokenAddr,
        uint amt,
        bool add,
        address user
    ) public onlyAllowedResolver(user)
    {
        if (add) {
            balances[user][tokenAddr] += amt;
        } else {
            balances[user][tokenAddr] -= amt;
        }
    }

    function transferAssets(
        address tokenAddress,
        uint amount,
        address sendTo
    ) public onlyAllowedResolver 
    {
        if (tokenAddress == eth) {
            sendTo.transfer(amount);
        } else {
            token tokenFunctions = token(tokenAddress);
            tokenFunctions.transfer(sendTo, amount);
        }
    }

}


contract MoatAsset is AssetDB {

    constructor(address rAddr) public {
        registryAddress = rAddr;
    }

    // emit an event atleast
    function () public payable {
        deposit(eth);
    }

}