108 lines
3.7 KiB
Python
108 lines
3.7 KiB
Python
# standard imports
|
|
import logging
|
|
|
|
# third-party imports
|
|
import web3
|
|
import celery
|
|
from cic_registry.error import UnknownContractError
|
|
|
|
# local imports
|
|
from .base import SyncFilter
|
|
from cic_eth.eth.token import unpack_transfer
|
|
from cic_eth.eth.token import unpack_transferfrom
|
|
from cic_eth.eth.token import ExtendedTx
|
|
from .base import SyncFilter
|
|
|
|
logg = logging.getLogger()
|
|
|
|
transfer_method_signature = '0xa9059cbb' # keccak256(transfer(address,uint256))
|
|
transferfrom_method_signature = '0x23b872dd' # keccak256(transferFrom(address,address,uint256))
|
|
giveto_method_signature = '0x63e4bff4' # keccak256(giveTo(address))
|
|
|
|
|
|
class CallbackFilter(SyncFilter):
|
|
|
|
trusted_addresses = []
|
|
|
|
def __init__(self, method, queue):
|
|
self.queue = queue
|
|
self.method = method
|
|
|
|
|
|
def call_back(self, transfer_type, result):
|
|
s = celery.signature(
|
|
self.method,
|
|
[
|
|
result,
|
|
transfer_type,
|
|
int(rcpt.status == 0),
|
|
],
|
|
queue=self.queue,
|
|
)
|
|
# s_translate = celery.signature(
|
|
# 'cic_eth.ext.address.translate',
|
|
# [
|
|
# result,
|
|
# self.trusted_addresses,
|
|
# chain_str,
|
|
# ],
|
|
# queue=self.queue,
|
|
# )
|
|
# s_translate.link(s)
|
|
# s_translate.apply_async()
|
|
s.apply_async()
|
|
|
|
|
|
def parse_data(self, tx, rcpt):
|
|
transfer_type = 'transfer'
|
|
transfer_data = None
|
|
method_signature = tx.input[:10]
|
|
|
|
if method_signature == transfer_method_signature:
|
|
transfer_data = unpack_transfer(tx.input)
|
|
transfer_data['from'] = tx['from']
|
|
transfer_data['token_address'] = tx['to']
|
|
|
|
elif method_signature == transferfrom_method_signature:
|
|
transfer_type = 'transferfrom'
|
|
transfer_data = unpack_transferfrom(tx.input)
|
|
transfer_data['token_address'] = tx['to']
|
|
|
|
# TODO: do not rely on logs here
|
|
elif method_signature == giveto_method_signature:
|
|
transfer_type = 'tokengift'
|
|
transfer_data = unpack_gift(tx.input)
|
|
for l in rcpt.logs:
|
|
if l.topics[0].hex() == '0x45c201a59ac545000ead84f30b2db67da23353aa1d58ac522c48505412143ffa':
|
|
transfer_data['value'] = web3.Web3.toInt(hexstr=l.data)
|
|
token_address_bytes = l.topics[2][32-20:]
|
|
transfer_data['token_address'] = web3.Web3.toChecksumAddress(token_address_bytes.hex())
|
|
transfer_data['from'] = rcpt.to
|
|
|
|
return (transfer_type, transfer_data)
|
|
|
|
|
|
def filter(self, w3, tx, rcpt, chain_spec, session=None):
|
|
logg.debug('applying callback filter "{}:{}"'.format(self.queue, self.method))
|
|
chain_str = str(chain_spec)
|
|
|
|
transfer_data = self.parse_data(tx, rcpt)
|
|
|
|
transfer_data = None
|
|
if len(tx.input) < 10:
|
|
logg.debug('callbacks filter data length not sufficient for method signature in tx {}, skipping'.format(tx['hash']))
|
|
return
|
|
|
|
logg.debug('checking callbacks filter input {}'.format(tx.input[:10]))
|
|
|
|
if transfer_data != None:
|
|
token_symbol = None
|
|
result = None
|
|
try:
|
|
tokentx = ExtendedTx(self.chain_spec)
|
|
tokentx.set_actors(transfer_data['from'], transfer_data['to'], self.trusted_addresses)
|
|
tokentx.set_tokens(transfer_data['token_address'], transfer_data['value'])
|
|
self.call_back(tokentx.to_dict())
|
|
except UnknownContractError:
|
|
logg.debug('callback filter {}:{} skipping "transfer" method on unknown contract {} tx {}'.format(tc.queue, tc.method, transfer_data['to'], tx.hash.hex()))
|