dsa-connectors/contracts/polygon/connectors/basic/main.sol

//SPDX-License-Identifier: MIT
pragma solidity ^0.7.0;

/**
 * @title Basic.
 * @dev Deposit & Withdraw from DSA.
 */

import { SafeERC20 } from "@openzeppelin/contracts/token/ERC20/SafeERC20.sol";
import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";

import { DSMath } from "../../common/math.sol";
import { Basic } from "../../common/basic.sol";
import { Events } from "./events.sol";

abstract contract BasicResolver is Events, DSMath, Basic {
	using SafeERC20 for IERC20;

	/**
	 * @dev Deposit Assets To Smart Account.
	 * @notice Deposit a token to DSA
	 * @param token The address of the token to deposit. (For MATIC: 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE)
	 * @param amt The amount of tokens to deposit. (For max: `uint256(-1)` (Not valid for MATIC))
	 * @param getId ID to retrieve amt.
	 * @param setId ID stores the amount of tokens deposited.
	 */
	function deposit(
		address token,
		uint256 amt,
		uint256 getId,
		uint256 setId
	)
		public
		payable
		returns (string memory _eventName, bytes memory _eventParam)
	{
		uint256 _amt = getUint(getId, amt);
		if (token != maticAddr) {
			IERC20 tokenContract = IERC20(token);
			_amt = _amt == uint256(-1)
				? tokenContract.balanceOf(msg.sender)
				: _amt;
			tokenContract.safeTransferFrom(msg.sender, address(this), _amt);
		} else {
			require(
				msg.value == _amt || _amt == uint256(-1),
				"invalid-ether-amount"
			);
			_amt = msg.value;
		}
		setUint(setId, _amt);

		_eventName = "LogDeposit(address,uint256,uint256,uint256)";
		_eventParam = abi.encode(token, _amt, getId, setId);
	}

	/**
	 * @dev Deposit Assets To Smart Account From any user.
	 * @notice Deposit a token to DSA from any user.
	 * @param token The address of the token to deposit. (Note: MATIC is not supported. Use `deposit()`)
	 * @param amt The amount of tokens to deposit. (For max: `uint256(-1)`)
	 * @param from The address depositing the token.
	 * @param getId ID to retrieve amt.
	 * @param setId ID stores the amount of tokens deposited.
	 */
	function depositFrom(
		address token,
		uint256 amt,
		address from,
		uint256 getId,
		uint256 setId
	)
		public
		payable
		returns (string memory _eventName, bytes memory _eventParam)
	{
		uint256 _amt = getUint(getId, amt);
		require(token != maticAddr, "eth-not-supported");
		IERC20 tokenContract = IERC20(token);
		_amt = _amt == uint256(-1) ? tokenContract.balanceOf(from) : _amt;
		tokenContract.safeTransferFrom(from, address(this), _amt);

		setUint(setId, _amt);

		_eventName = "LogDepositFrom(address,uint256,address,uint256,uint256)";
		_eventParam = abi.encode(token, _amt, from, getId, setId);
	}

	/**
	 * @dev Withdraw Assets from Smart  Account
	 * @notice Withdraw a token from DSA
	 * @param token The address of the token to withdraw. (For MATIC: 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE)
	 * @param amt The amount of tokens to withdraw. (For max: `uint256(-1)`)
	 * @param to The address to receive the token upon withdrawal
	 * @param getId ID to retrieve amt.
	 * @param setId ID stores the amount of tokens withdrawn.
	 */
	function withdraw(
		address token,
		uint256 amt,
		address payable to,
		uint256 getId,
		uint256 setId
	)
		public
		payable
		returns (string memory _eventName, bytes memory _eventParam)
	{
		uint256 _amt = getUint(getId, amt);
		if (token == maticAddr) {
			_amt = _amt == uint256(-1) ? address(this).balance : _amt;
			to.call{ value: _amt }("");
		} else {
			IERC20 tokenContract = IERC20(token);
			_amt = _amt == uint256(-1)
				? tokenContract.balanceOf(address(this))
				: _amt;
			tokenContract.safeTransfer(to, _amt);
		}
		setUint(setId, _amt);

		_eventName = "LogWithdraw(address,uint256,address,uint256,uint256)";
		_eventParam = abi.encode(token, _amt, to, getId, setId);
	}
}

contract ConnectV2BasicPolygon is BasicResolver {
	string public constant name = "Basic-v1.1";
}