Instadapp/dsa-connectors
1
0
Fork 0
mirror of https://github.com/Instadapp/dsa-connectors.git synced 2024-07-29 22:37:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactoring + fixes

Browse Source
This commit is contained in:
pradyuman-verma 2022-06-25 19:12:04 +05:30
parent d280a09b64
commit eeaae50596
No known key found for this signature in database
GPG Key ID: E36FD6BC8923221F
4 changed files with 64 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
contracts/mainnet/connectors/socket/events.sol
View File

@ -3,11 +3,11 @@ pragma solidity ^0.7.0;
pragma experimental ABIEncoderV2; pragma experimental ABIEncoderV2;
contract Events { contract Events {
event LogSocketBridge ( event LogSocketBridge(
address token, address token,
uint256 amount, uint256 amount,
uint256 sourceChain, uint256 sourceChain,
uint256 targetChain, uint256 targetChain,
address recipient address recipient
); );
} }

33
contracts/mainnet/connectors/socket/helpers.sol Normal file
View File

@ -0,0 +1,33 @@
//SPDX-License-Identifier: MIT
pragma solidity ^0.7.6;
pragma experimental ABIEncoderV2;
import "./interface.sol";
import { Basic } from "../../common/basic.sol";
import { TokenInterface } from "../../common/interfaces.sol";
contract Helpers is Basic {
address internal immutable registry =
0xc30141B657f4216252dc59Af2e7CdB9D8792e1B0;
function _socketBridge(bytes memory _txData, uint256 _nativeTokenAmt)
internal
returns (bool _success)
{
(_success, ) = registry.call{ value: _nativeTokenAmt }(_txData);
}
/**
* @dev Gets Allowance target from registry.
* @param _route route number
*/
function getAllowanceTarget(uint256 _route)
internal
view
returns (address _allowanceTarget)
{
RouteData memory data = ISocketRegistry(registry).routes(_route);
_allowanceTarget = data.route;
require(_allowanceTarget != address(0), "allowanceTarget-not-valid");
}
}

17
contracts/mainnet/connectors/socket/interface.sol
View File

@ -2,13 +2,12 @@
pragma solidity ^0.7.6; pragma solidity ^0.7.6;
pragma experimental ABIEncoderV2; pragma experimental ABIEncoderV2;
interface ISocketRegistry { struct RouteData {
address route;
struct RouteData { bool isEnabled;
address route; bool isMiddleware;
bool isEnabled; }
bool isMiddleware;
} interface ISocketRegistry {
function routes(uint256) external view returns (RouteData memory);
function routes(uint256) external view returns(RouteData memory);
} }

60
contracts/mainnet/connectors/socket/main.sol
View File

@ -7,46 +7,10 @@ pragma experimental ABIEncoderV2;
* @dev Multi-chain Bridge Aggregator. * @dev Multi-chain Bridge Aggregator.
*/ */
import { Basic } from "../../common/basic.sol";
import { TokenInterface } from "../../common/interfaces.sol";
import "./events.sol"; import "./events.sol";
import "./interface.sol"; import "./helpers.sol";
abstract contract SocketConnectorBridge is Basic { abstract contract SocketResolver is Helpers {
address constant registry = 0xc30141B657f4216252dc59Af2e7CdB9D8792e1B0;
/**
* @dev socket API bridge handler
* @param _txData - contains data returned from socket build-tx API. Struct defined in interfaces.sol
* @param _ethAmt - Eth to bridge for .value()
*/
function socketBridge(bytes memory _txData, uint256 _ethAmt)
internal
returns (bool _success)
{
(_success, ) = registry.call{ value: _ethAmt }(_txData);
require(_success, "Socket-swap-failed");
}
}
abstract contract SocketConnectorResolver is SocketConnectorBridge {
/**
* @dev Gets Allowance target from registry.
* @param _route route number
*/
function getAllowanceTarget(uint256 _route)
internal
view
returns (address _allowanceTarget)
{
ISocketRegistry.RouteData memory data = ISocketRegistry(registry)
.routes(_route);
require(data.route != address(0), "allowanceTarget-not-valid");
return data.route;
}
}
abstract contract SocketConnector is SocketConnectorResolver {
/** /**
* @dev Bridge Token. * @dev Bridge Token.
* @notice Bridge Token on Socket. * @notice Bridge Token on Socket.
@ -69,16 +33,22 @@ abstract contract SocketConnector is SocketConnectorResolver {
payable payable
returns (string memory _eventName, bytes memory _eventParam) returns (string memory _eventName, bytes memory _eventParam)
{ {
uint256 _ethAmt; bool isNative = _token == ethAddr;
uint256 nativeTokenAmt;
if (_token == ethAddr) { if (isNative) {
_ethAmt = _amount; _amount = _amount == uint256(-1) ? address(this).balance : _amount;
nativeTokenAmt = _amount;
} else { } else {
TokenInterface _tokenContract = TokenInterface(_token); TokenInterface tokenContract = TokenInterface(_token);
_tokenContract.approve(getAllowanceTarget(_route), _amount);
_amount = _amount == uint256(-1)
? tokenContract.balanceOf(address(this))
: _amount;
tokenContract.approve(getAllowanceTarget(_route), _amount);
} }
socketBridge(_txData, _ethAmt); require(_socketBridge(_txData, nativeTokenAmt), "Socket-swap-failed");
uint256 _sourceChain; uint256 _sourceChain;
assembly { assembly {
@ -96,6 +66,6 @@ abstract contract SocketConnector is SocketConnectorResolver {
} }
} }
contract ConnectV2Socket is SocketConnector { contract ConnectV2Socket is SocketResolver {
string public constant name = "Socket-v1.0"; string public constant name = "Socket-v1.0";
} }