merging contract interaction script

DistributedCollective · Apr 15, 2021 · 947954f · 947954f
2 parents cf1cd68 + 8786149
commit 947954f
Show file tree

Hide file tree

Showing 7 changed files with 379 additions and 6 deletions.
diff --git a/scripts/contractInteraction/RBTCWrapperProxy.json b/scripts/contractInteraction/RBTCWrapperProxy.json
@@ -10,6 +10,11 @@
         "internalType": "address",
         "name": "_sovrynSwapNetworkAddress",
         "type": "address"
+      },
+      {
+        "internalType": "contract IContractRegistry",
+        "name": "_registry",
+        "type": "address"
       }
     ],
     "payable": false,
@@ -122,6 +127,25 @@
     "name": "LiquidityRemovedFromV1",
     "type": "event"
   },
+  {
+    "anonymous": false,
+    "inputs": [
+      {
+        "indexed": true,
+        "internalType": "address",
+        "name": "_prevOwner",
+        "type": "address"
+      },
+      {
+        "indexed": true,
+        "internalType": "address",
+        "name": "_newOwner",
+        "type": "address"
+      }
+    ],
+    "name": "OwnerUpdate",
+    "type": "event"
+  },
   {
     "anonymous": false,
     "inputs": [
@@ -158,6 +182,114 @@
     "stateMutability": "payable",
     "type": "fallback"
   },
+  {
+    "constant": false,
+    "inputs": [],
+    "name": "acceptOwnership",
+    "outputs": [],
+    "payable": false,
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "constant": true,
+    "inputs": [],
+    "name": "newOwner",
+    "outputs": [
+      {
+        "internalType": "address",
+        "name": "",
+        "type": "address"
+      }
+    ],
+    "payable": false,
+    "stateMutability": "view",
+    "type": "function"
+  },
+  {
+    "constant": true,
+    "inputs": [],
+    "name": "onlyOwnerCanUpdateRegistry",
+    "outputs": [
+      {
+        "internalType": "bool",
+        "name": "",
+        "type": "bool"
+      }
+    ],
+    "payable": false,
+    "stateMutability": "view",
+    "type": "function"
+  },
+  {
+    "constant": true,
+    "inputs": [],
+    "name": "owner",
+    "outputs": [
+      {
+        "internalType": "address",
+        "name": "",
+        "type": "address"
+      }
+    ],
+    "payable": false,
+    "stateMutability": "view",
+    "type": "function"
+  },
+  {
+    "constant": true,
+    "inputs": [],
+    "name": "prevRegistry",
+    "outputs": [
+      {
+        "internalType": "contract IContractRegistry",
+        "name": "",
+        "type": "address"
+      }
+    ],
+    "payable": false,
+    "stateMutability": "view",
+    "type": "function"
+  },
+  {
+    "constant": true,
+    "inputs": [],
+    "name": "registry",
+    "outputs": [
+      {
+        "internalType": "contract IContractRegistry",
+        "name": "",
+        "type": "address"
+      }
+    ],
+    "payable": false,
+    "stateMutability": "view",
+    "type": "function"
+  },
+  {
+    "constant": false,
+    "inputs": [],
+    "name": "restoreRegistry",
+    "outputs": [],
+    "payable": false,
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "constant": false,
+    "inputs": [
+      {
+        "internalType": "bool",
+        "name": "_onlyOwnerCanUpdateRegistry",
+        "type": "bool"
+      }
+    ],
+    "name": "restrictRegistryUpdate",
+    "outputs": [],
+    "payable": false,
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
   {
     "constant": true,
     "inputs": [],
@@ -173,6 +305,30 @@
     "stateMutability": "view",
     "type": "function"
   },
+  {
+    "constant": false,
+    "inputs": [
+      {
+        "internalType": "address",
+        "name": "_newOwner",
+        "type": "address"
+      }
+    ],
+    "name": "transferOwnership",
+    "outputs": [],
+    "payable": false,
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "constant": false,
+    "inputs": [],
+    "name": "updateRegistry",
+    "outputs": [],
+    "payable": false,
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
   {
     "constant": true,
     "inputs": [],

diff --git a/scripts/contractInteraction/contract_interaction.py b/scripts/contractInteraction/contract_interaction.py
@@ -73,14 +73,24 @@ def main():
 
     # createProposalSIP0015()
 
+    # transferSOVtoTokenSender()
+
+    # transferSOVtoScriptAccount()
     #transferSOVtoTokenSender()
     #readBalanceFromAMM()
     #checkRates()
 
-    #testV1Converter(contracts["ConverterSOV"], contracts["WRBTC"], contracts["SOV"])
+    # testV1Converter(contracts["ConverterSOV"], contracts["WRBTC"], contracts["SOV"])
     # transferSOVtoTokenSender()
     # addLiquidityV1(contracts["WRBTCtoSOVConverter"], [contracts['WRBTC'], contracts['SOV']], [1 * 10**16, 67 * 10**18])
-    #addLiquidityV1UsingWrapper(contracts["WRBTCtoSOVConverter"], [contracts['WRBTC'], contracts['SOV']], [1 * 10**16, 67 * 10**18])
+    # addLiquidityV1UsingWrapper(contracts["WRBTCtoSOVConverter"], [contracts['WRBTC'], contracts['SOV']], [1 * 10**16, 67 * 10**18])
+
+    # addLiquidityV1FromMultisigUsingWrapper(contracts["WRBTCtoSOVConverter"], [contracts['WRBTC'], contracts['SOV']], [1 * 10**15, 67 * 10**17])
+    # addLiquidityV1FromMultisigUsingWrapper(contracts["WRBTCtoSOVConverter"], [contracts['WRBTC'], contracts['SOV']], [30 * 10**18, 200000 * 10**18])
+
+    removeLiquidityV1toMultisigUsingWrapper(contracts["WRBTCtoSOVConverter"], 5 * 10**17, [contracts['WRBTC'], contracts['SOV']])
+    # 2986.175 × 99% ~ 2957
+    # removeLiquidityV1toMultisigUsingWrapper(contracts["WRBTCtoSOVConverter"], 2957 * 10**18, [contracts['WRBTC'], contracts['SOV']])
 
     '''''
     startRate = 1e8/15000 *1e10
@@ -1107,11 +1117,28 @@ def transferSOVtoTokenSender():
     txId = tx.events["Submission"]["transactionId"]
     print(txId)
 
+def transferSOVtoScriptAccount():
+    # 5825.7 SOV
+    amount = 58257 * 10**17
+
+    # TODO set receiver address
+    receiver = "0x27D55f5668eF4438635bdCE0aDCA083507E77752"
+    if (receiver == ""):
+        raise Exception("Invalid address")
+    SOVtoken = Contract.from_abi("SOV", address=contracts['SOV'], abi=SOV.abi, owner=acct)
+    data = SOVtoken.transfer.encode_input(receiver, amount)
+    print(data)
+
+    multisig = Contract.from_abi("MultiSig", address=contracts['multisig'], abi=MultiSigWallet.abi, owner=acct)
+    tx = multisig.submitTransaction(SOVtoken.address,0,data)
+    txId = tx.events["Submission"]["transactionId"]
+    print(txId)
+
 def setSupportedToken(tokenAddress):
     sovryn = Contract.from_abi("sovryn", address=contracts['sovrynProtocol'], abi=interface.ISovrynBrownie.abi, owner=acct)
     multisig = Contract.from_abi("MultiSig", address=contracts['multisig'], abi=MultiSigWallet.abi, owner=acct)
 
-    data = sovryn.setSupportedTokens.encode_input([tokenAddress],[True]) 
+    data = sovryn.setSupportedTokens.encode_input([tokenAddress],[True])
     tx = multisig.submitTransaction(sovryn.address,0,data)
     txId = tx.events["Submission"]["transactionId"]
     print(txId)
@@ -1147,7 +1174,7 @@ def testV1Converter(converterAddress, reserve1, reserve2):
 
     tokenContract2 = Contract.from_abi("Token", address=reserve2, abi=TestToken.abi, owner=acct)
     tokenContract2.approve(converter.address, bal2/50)
-    accountBalance = tokenContract2.balanceOf(acct)   
+    accountBalance = tokenContract2.balanceOf(acct)
 
     converter.addLiquidity([reserve1, reserve2],[bal1/100, bal2/50],1)
 
@@ -1202,4 +1229,54 @@ def getTargetAmountFromAMM(_sourceReserveBalance, _sourceReserveWeight, _targetR
 
     targetAmount = sovrynSwapFormula.crossReserveTargetAmount(_sourceReserveBalance, _sourceReserveWeight, _targetReserveBalance, _targetReserveWeight, _amount)
 
-    print(targetAmount)
+    print(targetAmount)
+def addLiquidityV1FromMultisigUsingWrapper(converter, tokens, amounts):
+    abiFile =  open('./scripts/contractInteraction/RBTCWrapperProxy.json')
+    abi = json.load(abiFile)
+    wrapperProxy = Contract.from_abi("RBTCWrapperProxy", address=contracts['RBTCWrapperProxy'], abi=abi, owner=acct)
+    multisig = Contract.from_abi("MultiSig", address=contracts['multisig'], abi=MultiSigWallet.abi, owner=acct)
+
+    # approve
+    token = Contract.from_abi("ERC20", address=tokens[1], abi=ERC20.abi, owner=acct)
+    data = token.approve.encode_input(wrapperProxy.address, amounts[1])
+    print(data)
+
+    tx = multisig.submitTransaction(token.address,0,data)
+    txId = tx.events["Submission"]["transactionId"]
+    print(txId)
+
+    # addLiquidityToV1
+    data = wrapperProxy.addLiquidityToV1.encode_input(converter, tokens, amounts, 1)
+    print(data)
+
+    tx = multisig.submitTransaction(wrapperProxy.address,amounts[0],data)
+    txId = tx.events["Submission"]["transactionId"]
+    print(txId)
+
+def removeLiquidityV1toMultisigUsingWrapper(converter, amount, tokens):
+    abiFile =  open('./scripts/contractInteraction/RBTCWrapperProxy.json')
+    abi = json.load(abiFile)
+    wrapperProxy = Contract.from_abi("RBTCWrapperProxy", address=contracts['RBTCWrapperProxy'], abi=abi, owner=acct)
+    multisig = Contract.from_abi("MultiSig", address=contracts['multisig'], abi=MultiSigWallet.abi, owner=acct)
+
+    converterAbiFile =  open('./scripts/contractInteraction/LiquidityPoolV1Converter.json')
+    converterAbi = json.load(converterAbiFile)
+    converterContract = Contract.from_abi("LiquidityPoolV1Converter", address=converter, abi=converterAbi, owner=acct)
+    poolToken = converterContract.anchor()
+
+    # approve
+    token = Contract.from_abi("ERC20", address=poolToken, abi=ERC20.abi, owner=acct)
+    data = token.approve.encode_input(wrapperProxy.address, amount)
+    print(data)
+
+    tx = multisig.submitTransaction(token.address,0,data)
+    txId = tx.events["Submission"]["transactionId"]
+    print(txId)
+
+    # removeLiquidityFromV1
+    data = wrapperProxy.removeLiquidityFromV1.encode_input(converter, amount, tokens, [1,1])
+    print(data)
+
+    tx = multisig.submitTransaction(wrapperProxy.address,0,data)
+    txId = tx.events["Submission"]["transactionId"]
+    print(txId)
diff --git a/scripts/contractInteraction/mainnet_contracts.json b/scripts/contractInteraction/mainnet_contracts.json
@@ -53,5 +53,5 @@
     "TokenSender": "0x31Ec06eB090Af6EEbfF5cF52084e9BCc3D7d9248",
 
     "WRBTCtoSOVConverter": "0x1D2c04967e8B17168310fd7035cC219dE477BE82",
-    "RBTCWrapperProxy":  "0xe9D09d83De186d0915342de98C34500602105aCc"
+    "RBTCWrapperProxy":  "0xA3B6E18B9A4ECAE44C7355458Ae7Db8874018C22"
 }