cic-monitor/ge_render/__init__.py

# standard imports
import logging
import os

# external imports
from eth_accounts_index import AccountsIndex
from erc20_faucet import Faucet
from cic_contracts.erc20 import ERC20
from chainlib.eth.error import RequestMismatchException
from chainlib.eth.address import AddressChecksum
from chainlib.eth.constant import ZERO_ADDRESS
from hexathon import uniform

logg = logging.getLogger(__name__)

faucet_token_cache = {}
token_cache = {}


def get_method(chain_str, tx):
    data = tx.payload
    s = None
    if len(data) < 8:
        raise ValueError('not minimum signature length')

    data = uniform(data[:8])

    chainmap = addrmap.get(chain_str)
    if chainmap != None:
        contractmap = chainmap.get(uniform(tx.inputs[0]))
        if contractmap != None:
            m = contractmap.get(data)
            if m != None:
                return m

    chainmap = sigmap.get(chain_str)
    if chainmap != None:
        return chainmap.get(data)
        

def apply(c, result, chain_spec, conn, block, tx, db_session=None):
    s = result.get()
    try:
        m = get_method(str(chain_spec), tx)
    except ValueError as e:
        return True

    if m == None: 
        return True

    r = m(c, result, chain_spec, conn, block, tx)
    return r


def account_registry_add(c, result, chain_spec, conn, block, tx):
    try:
        o = AccountsIndex.parse_add_request(tx.payload)
        s = '{} block {} tx {} account registration for {}'.format(c, block.number, tx.index, o[0])
        result.set(s)
    except RequestMismatchException as e:
        return True

    return False


def ge_faucet_gift(i, result, chain_spec, conn, block, tx):
    
    c = Faucet(chain_spec)
    o = c.token_amount(tx.inputs[0], sender_address=ZERO_ADDRESS, height=block.number)
    r = conn.do(o)
    v = c.parse_token_amount(r)
    logg.info('retrieved token amount {} at block height {} for faucet {}'.format(v, block.number, tx.inputs[0]))

    token = faucet_token_cache.get(tx.inputs[0])
    if token == None:
        o = c.token(tx.inputs[0], sender_address=ZERO_ADDRESS)
        r = conn.do(o)
        token = c.parse_token(r)
        token = uniform(token)
        faucet_token_cache[tx.inputs[0]] = token
        logg.info('found token {} for faucet {}'.format(v, token, tx.inputs[0]))

    token_symbol = token_cache.get(token)
    if token_symbol == None:
        c = ERC20(chain_spec)
        o = c.symbol(token, sender_address=ZERO_ADDRESS)
        r = conn.do(o)
        token_symbol = c.parse_symbol(r)
        token_cache[token] = token_symbol
        logg.info('resolved token {} to symbol'.format(v, block.number, tx.inputs[0]))

    try:
        o = Faucet.parse_give_to_request(tx.payload)
        s = '{} GE faucet was triggered for {} {} by {} on block {} tx {}'.format(i, v, token_symbol, o[0], block.number, tx.index)
        result.set(s)
    except RequestMismatchException as e:
        return True

    return False



sigmap = {
        'evm:kitabu:5050:sarafu': {
            '0a3b0a4f': account_registry_add,
            },
        }

addrmap = {
        'evm:kitabu:5050:sarafu': {
            'd462aadb7251d418e6a9e4f205928f678e1c6b3b': {
                '63e4bff4': ge_faucet_gift,
                    },
                }
        }