dsa-connectors/contracts/avalanche/mapping/benqi.sol

pragma solidity ^0.7.0;
pragma experimental ABIEncoderV2;

interface IndexInterface {
    function master() external view returns (address);
}

interface ConnectorsInterface {
    function chief(address) external view returns (bool);
}

interface QiTokenInterface {
    function isQiToken() external view returns (bool);
    function underlying() external view returns (address);
}

interface MappingControllerInterface {
    function hasRole(address,address) external view returns (bool);
}

abstract contract Helpers {

    struct TokenMap {
        address qitoken;
        address token;
    }

    event LogQiTokenAdded(string indexed name, address indexed token, address indexed qitoken);
    event LogQiTokenUpdated(string indexed name, address indexed token, address indexed qitoken);

    // InstaConnectorsV2
    ConnectorsInterface public constant connectors = ConnectorsInterface(0x127d8cD0E2b2E0366D522DeA53A787bfE9002C14);
    // InstaIndex Address.
    IndexInterface public constant instaIndex = IndexInterface(0x6CE3e607C808b4f4C26B7F6aDAeB619e49CAbb25);

    // InstaMappingController Address.
    MappingControllerInterface public constant mappingController = MappingControllerInterface(0xF2113d0c99f36D7D6F6c6FAf05E0863892255999);

    address public constant avaxAddr = 0xEeeeeEeeeEeEeeEeEeEeeEEEeeeeEeeeeeeeEEeE;

    mapping (string => TokenMap) public qiTokenMapping;

    modifier isChief {
        require(msg.sender == instaIndex.master() || connectors.chief(msg.sender), "not-an-chief");
        _;
    }

    modifier hasRoleOrIsChief {
        require(
            msg.sender == instaIndex.master() ||
                connectors.chief(msg.sender) ||
                mappingController.hasRole(address(this), msg.sender),
            "not-an-chief/controller"
        );
        _;
    }

    function _addQitokenMapping(
        string[] memory _names,
        address[] memory _tokens,
        address[] memory _qitokens
    ) internal {
        require(_names.length == _tokens.length, "addQitokenMapping: not same length");
        require(_names.length == _qitokens.length, "addQitokenMapping: not same length");

        for (uint i = 0; i < _qitokens.length; i++) {
            TokenMap memory _data = qiTokenMapping[_names[i]];

            require(_data.qitoken == address(0), "addQitokenMapping: mapping added already");
            require(_data.token == address(0), "addQitokenMapping: mapping added already");

            require(_tokens[i] != address(0), "addQitokenMapping: _tokens address not vaild");
            require(_qitokens[i] != address(0), "addQitokenMapping: _qitokens address not vaild");

            QiTokenInterface _qitokenContract = QiTokenInterface(_qitokens[i]);

            require(_qitokenContract.isQiToken(), "addQitokenMapping: not a qiToken");
            if (_tokens[i] != avaxAddr) {
                require(_qitokenContract.underlying() == _tokens[i], "addQitokenMapping: mapping mismatch");
            }

            qiTokenMapping[_names[i]] = TokenMap(
                _qitokens[i],
                _tokens[i]
            );
            emit LogQiTokenAdded(_names[i], _tokens[i], _qitokens[i]);
        }
    }

    function updateQitokenMapping(
        string[] calldata _names,
        address[] memory _tokens,
        address[] calldata _qitokens
    ) external isChief {

        require(_names.length == _tokens.length, "updateQitokenMapping: not same length");
        require(_names.length == _qitokens.length, "updateQitokenMapping: not same length");

        for (uint i = 0; i < _qitokens.length; i++) {
            TokenMap memory _data = qiTokenMapping[_names[i]];

            require(_data.qitoken != address(0), "updateQitokenMapping: mapping does not exist");
            require(_data.token != address(0), "updateQitokenMapping: mapping does not exist");

            require(_tokens[i] != address(0), "updateQitokenMapping: _tokens address not vaild");
            require(_qitokens[i] != address(0), "updateQitokenMapping: _qitokens address not vaild");

            QiTokenInterface _qitokenContract = QiTokenInterface(_qitokens[i]);

            require(_qitokenContract.isQiToken(), "updateQitokenMapping: not a qiToken");
            if (_tokens[i] != avaxAddr) {
                require(_qitokenContract.underlying() == _tokens[i], "addQitokenMapping: mapping mismatch");
            }

            qiTokenMapping[_names[i]] = TokenMap(
                _qitokens[i],
                _tokens[i]
            );
            emit LogQiTokenUpdated(_names[i], _tokens[i], _qitokens[i]);
        }
    }

    function addQitokenMapping(
        string[] memory _names,
        address[] memory _tokens,
        address[] memory _qitokens
    ) external hasRoleOrIsChief {
        _addQitokenMapping(_names, _tokens, _qitokens);
    }

    function getMapping(string memory _tokenId) external view returns (address, address) {
        TokenMap memory _data = qiTokenMapping[_tokenId];
        return (_data.token, _data.qitoken);
    }

}

contract InstaBenqiMappingAvalanche is Helpers {
    string constant public name = "Benqi-Mapping-v1.0";

    constructor(
        string[] memory _qitokenNames,
        address[] memory _tokens,
        address[] memory _qitokens
    ) {
        _addQitokenMapping(_qitokenNames, _tokens, _qitokens);
    }
}