2021-10-09 21:04:11 +02:00
|
|
|
# standard imports
|
|
|
|
import logging
|
2021-10-10 13:48:32 +02:00
|
|
|
import copy
|
2021-10-09 21:04:11 +02:00
|
|
|
|
2021-10-09 21:45:07 +02:00
|
|
|
# external imports
|
2021-10-10 12:50:18 +02:00
|
|
|
from chainlib.chain import ChainSpec
|
2021-10-10 15:37:26 +02:00
|
|
|
from chainlib.eth.tx import (
|
|
|
|
TxFormat,
|
|
|
|
TxFactory,
|
|
|
|
)
|
2021-10-09 21:45:07 +02:00
|
|
|
from chainlib.eth.connection import RPCConnection
|
2021-10-10 19:18:37 +02:00
|
|
|
from chainlib.eth.contract import (
|
|
|
|
ABIContractEncoder,
|
|
|
|
ABIContractType
|
|
|
|
)
|
|
|
|
from chainlib.eth.address import is_address
|
2021-10-10 12:50:18 +02:00
|
|
|
from eth_token_index import TokenUniqueSymbolIndex
|
2021-10-10 14:06:40 +02:00
|
|
|
from eth_address_declarator import Declarator
|
2021-10-09 21:45:07 +02:00
|
|
|
|
2021-10-09 21:04:11 +02:00
|
|
|
logg = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
|
|
|
class CICEth:
|
|
|
|
|
2021-10-10 12:50:18 +02:00
|
|
|
def __init__(self, chain_spec, resources, proofs, signer=None, rpc=None, nonce_oracle=None, fee_oracle=None):
|
2021-10-10 13:48:32 +02:00
|
|
|
"""resources will be modified
|
|
|
|
"""
|
2021-10-10 11:33:30 +02:00
|
|
|
self.resources = resources
|
2021-10-09 21:04:11 +02:00
|
|
|
self.proofs = proofs
|
2021-10-10 12:50:18 +02:00
|
|
|
self.chain_spec = chain_spec
|
|
|
|
self.signer = signer
|
|
|
|
self.rpc = rpc
|
|
|
|
self.nonce_oracle = nonce_oracle
|
|
|
|
self.fee_oracle = fee_oracle
|
2021-10-10 15:37:26 +02:00
|
|
|
self.token_details = None
|
2021-10-10 12:50:18 +02:00
|
|
|
self.token_address = None
|
2021-10-10 15:37:26 +02:00
|
|
|
self.token_code = None
|
2021-10-10 14:00:47 +02:00
|
|
|
self.outputs = []
|
2021-10-10 13:48:32 +02:00
|
|
|
self.tx_format = TxFormat.RAW_ARGS
|
|
|
|
if self.rpc != None:
|
|
|
|
self.tx_format = TxFormat.JSONRPC
|
|
|
|
elif self.signer != None:
|
|
|
|
self.tx_format = TxFormat.RLP_SIGNED
|
2021-10-09 21:04:11 +02:00
|
|
|
|
2021-10-10 12:50:18 +02:00
|
|
|
|
2021-10-10 19:18:37 +02:00
|
|
|
def prepare_token(self, name, symbol, precision, code, extra=[], extra_types=[], positions=None):
|
2021-10-10 15:37:26 +02:00
|
|
|
self.token_details = {
|
|
|
|
'name': name,
|
|
|
|
'symbol': symbol,
|
|
|
|
'precision': precision,
|
|
|
|
'code': code,
|
|
|
|
'extra': extra,
|
2021-10-10 19:18:37 +02:00
|
|
|
'extra_types': extra_types,
|
2021-10-10 15:37:26 +02:00
|
|
|
'positions': positions,
|
|
|
|
}
|
|
|
|
|
2021-10-10 19:18:37 +02:00
|
|
|
|
|
|
|
def __detect_arg_type(self, v):
|
|
|
|
typ = None
|
|
|
|
try:
|
|
|
|
int(v, 10)
|
|
|
|
typ = ABIContractType.UINT256
|
|
|
|
except TypeError:
|
|
|
|
pass
|
|
|
|
|
|
|
|
if typ == None:
|
|
|
|
try:
|
|
|
|
vv = strip_0x(v)
|
|
|
|
if is_address(vv):
|
|
|
|
typ = ABIContractType.ADDRESS
|
|
|
|
else:
|
|
|
|
typ = ABIContractType.BYTES32
|
|
|
|
except ValueError:
|
|
|
|
pass
|
|
|
|
|
|
|
|
if typ == None:
|
|
|
|
try:
|
|
|
|
v.encode('utf-8')
|
|
|
|
typ = ABIContractType.STRING
|
|
|
|
except ValueError:
|
|
|
|
pass
|
|
|
|
|
|
|
|
if typ == None:
|
|
|
|
raise ValueError('cannot automatically determine type for value {}'.format(v))
|
|
|
|
|
|
|
|
logg.info('argument {} parsed as abi contract type {}'.format(typ.value))
|
|
|
|
|
|
|
|
return typ
|
|
|
|
|
|
|
|
|
|
|
|
def __order_args(self):
|
|
|
|
args = [
|
|
|
|
self.token_details['name'],
|
|
|
|
self.token_details['symbol'],
|
|
|
|
self.token_details['precision'],
|
|
|
|
]
|
|
|
|
args_types = [
|
|
|
|
ABIContractType.STRING.value,
|
|
|
|
ABIContractType.STRING.value,
|
|
|
|
ABIContractType.UINT256.value,
|
|
|
|
]
|
|
|
|
|
|
|
|
for i, x in enumerate(self.token_details['extra']):
|
|
|
|
args.append(x)
|
|
|
|
typ = None
|
|
|
|
if self.token_details['extra_types'] != None:
|
|
|
|
typ = self.token_details['extra_types'][i]
|
|
|
|
else:
|
|
|
|
typ = self.__detect_arg_type(x)
|
2021-10-10 19:59:52 +02:00
|
|
|
args_types.append(typ)
|
2021-10-10 19:18:37 +02:00
|
|
|
|
|
|
|
positions = self.token_details['positions']
|
|
|
|
if positions == None:
|
|
|
|
positions = list(range(len(args)))
|
|
|
|
|
|
|
|
return (args, args_types, positions)
|
|
|
|
|
2021-10-10 15:37:26 +02:00
|
|
|
|
2021-10-10 12:50:18 +02:00
|
|
|
def process_token(self):
|
2021-10-10 19:18:37 +02:00
|
|
|
(args, args_types, positions) = self.__order_args()
|
|
|
|
|
2021-10-10 15:37:26 +02:00
|
|
|
enc = ABIContractEncoder()
|
|
|
|
|
2021-10-10 19:18:37 +02:00
|
|
|
for i in positions:
|
|
|
|
getattr(enc, args_types[i])(args[i])
|
|
|
|
|
|
|
|
code = self.token_details['code'] + enc.get()
|
2021-10-10 15:37:26 +02:00
|
|
|
|
|
|
|
signer_address = self.resources['token']['key_address']
|
|
|
|
c = TxFactory(self.chain_spec, signer=self.signer, nonce_oracle=self.nonce_oracle, gas_oracle=self.fee_oracle)
|
|
|
|
tx = c.template(signer_address, None, use_nonce=True)
|
|
|
|
tx = c.set_code(tx, code)
|
|
|
|
o = c.finalize(tx, self.tx_format)
|
|
|
|
|
|
|
|
r = None
|
|
|
|
if self.rpc != None:
|
|
|
|
r = self.rpc.do(o[1])
|
|
|
|
elif self.signer != None:
|
|
|
|
r = o[1]
|
2021-10-10 12:50:18 +02:00
|
|
|
|
2021-10-10 15:37:26 +02:00
|
|
|
if r == None:
|
|
|
|
r = code
|
|
|
|
|
|
|
|
self.outputs.append(r)
|
|
|
|
return r
|
2021-10-10 12:50:18 +02:00
|
|
|
|
2021-10-10 19:18:37 +02:00
|
|
|
|
2021-10-10 12:50:18 +02:00
|
|
|
def process_token_index(self):
|
2021-10-10 14:06:40 +02:00
|
|
|
c = TokenUniqueSymbolIndex(self.chain_spec, signer=self.signer)
|
2021-10-10 12:50:18 +02:00
|
|
|
|
|
|
|
contract_address = self.resources['token_index']['reference']
|
|
|
|
signer_address = self.resources['token_index']['key_address']
|
2021-10-10 13:20:00 +02:00
|
|
|
o = c.register(contract_address, signer_address, self.token_address, tx_format=self.tx_format)
|
2021-10-10 14:00:47 +02:00
|
|
|
r = None
|
2021-10-10 13:20:00 +02:00
|
|
|
if self.rpc != None:
|
2021-10-10 13:48:32 +02:00
|
|
|
r = self.rpc.do(o[1])
|
2021-10-10 13:20:00 +02:00
|
|
|
elif self.signer != None:
|
2021-10-10 14:00:47 +02:00
|
|
|
r = o[1]
|
2021-10-10 13:20:00 +02:00
|
|
|
else:
|
2021-10-10 14:00:47 +02:00
|
|
|
r = o
|
2021-10-10 13:48:32 +02:00
|
|
|
|
2021-10-10 14:00:47 +02:00
|
|
|
self.outputs.append(r)
|
|
|
|
return r
|
2021-10-10 13:48:32 +02:00
|
|
|
|
2021-10-10 12:50:18 +02:00
|
|
|
|
2021-10-10 14:06:40 +02:00
|
|
|
def process_address_declarator(self):
|
|
|
|
c = Declarator(self.chain_spec, signer=self.signer)
|
|
|
|
|
|
|
|
contract_address = self.resources['address_declarator']['reference']
|
|
|
|
signer_address = self.resources['address_declarator']['key_address']
|
|
|
|
|
|
|
|
r = []
|
|
|
|
for proof in self.proofs:
|
|
|
|
o = c.add_declaration(contract_address, signer_address, self.token_address, proof, tx_format=self.tx_format)
|
|
|
|
if self.rpc != None:
|
|
|
|
r.append(self.rpc.do(o[1]))
|
|
|
|
elif self.signer != None:
|
|
|
|
r.append(o[1])
|
|
|
|
else:
|
|
|
|
r.append(o)
|
|
|
|
|
|
|
|
self.outputs += r
|
|
|
|
return r
|
|
|
|
|
|
|
|
|
2021-10-10 12:50:18 +02:00
|
|
|
def process(self):
|
|
|
|
self.token_address = self.resources['token']['reference']
|
|
|
|
if self.token_address == None:
|
|
|
|
tasks.append('token')
|
2021-10-10 15:37:26 +02:00
|
|
|
|
|
|
|
for k in self.resources.keys():
|
|
|
|
if k == 'token':
|
|
|
|
continue
|
|
|
|
if self.resources[k]['reference'] != None:
|
|
|
|
tasks.append(k)
|
|
|
|
|
2021-10-10 12:50:18 +02:00
|
|
|
for task in tasks:
|
|
|
|
getattr(self, 'process_̈́ ' + task)()
|
2021-10-10 11:33:30 +02:00
|
|
|
|
|
|
|
|
|
|
|
def new(resources, proofs, signer_hint=None):
|
2021-10-10 12:50:18 +02:00
|
|
|
return CICEth(resources, proofs, signer=None)
|