import asyncio
import json
import re
import warnings
from datetime import datetime, timezone
from json import JSONDecodeError

from web3 import Web3
from web3_multi_provider import FallbackProvider

from common import *
from logs import *

warnings.filterwarnings("ignore")
web3 = Web3(FallbackProvider(json.loads(os.getenv('RPC_SERVER'))))
log_contract = web3.eth.contract(
    os.getenv('LOG_CONTRACT_ADDRESS'),
    abi=json.load(open("./data/abi/{}".format(os.getenv('LOG_CONTRACT_ABI_FILE'))))
)
block_cache = {}


async def keep_chat_updated(event_filter, poll_interval):
    while True:
        for event in event_filter.get_new_entries():
            output_line(event)
        await asyncio.sleep(poll_interval)


def preload_chat(earliest_block, latest_block):
    event_logs = web3.eth.get_logs({
        "fromBlock": int(earliest_block),
        "toBlock": int(latest_block),
        "topics": [os.getenv('LOG_TOPIC')],
        "address": os.getenv('LOG_CONTRACT_ADDRESS')
    })
    for log in event_logs:
        output_line(log)


def load_chat(latest_block):
    event_filter = web3.eth.filter({
        "fromBlock": int(latest_block),
        "topics": [os.getenv('LOG_TOPIC')],
        "address": os.getenv('LOG_CONTRACT_ADDRESS')
    })
    loop = asyncio.new_event_loop()
    asyncio.set_event_loop(loop)
    while True:
        try:
            asyncio.run(keep_chat_updated(event_filter, 2))
        except KeyboardInterrupt:
            break


def output_line(log):
    try:
        tx_receipt = web3.eth.get_transaction_receipt(log.transactionHash)
        decoded_data = log_contract.events.LogEvent().process_receipt(tx_receipt)
    except Exception as e:
        return
    if decoded_data[0]['args']['Aura'] and decoded_data[0]['args']['Soul']:
        # get the block and save it to cache
        block_number = tx_receipt['blockNumber']
        if block_number not in block_cache:
            block_cache[block_number] = web3.eth.get_block(block_number)

        # log the message and output to console
        message = "{}:{} {}".format(decoded_data[0]['args']['Soul'], decoded_data[0]['args']['Aura'], decoded_data[0]['args']['LogLine'])
        log_message(os.getenv('CHANNEL_NAME'), message)
        timestamp = datetime.fromtimestamp(block_cache[block_number]['timestamp'], tz=timezone.utc)
        print("[{}] {}".format(timestamp.strftime("%Y-%m-%d %H:%M:%S"), message if os.getenv('SHOW_AURA_AND_SOUL') != '0' else decoded_data[0]['args']['LogLine']))

        # save latest block number
        room = json.load(open(room_file := "{}/rooms/{}.json".format(os.getenv('DATA_FOLDER'), os.getenv('CHANNEL_NAME')), 'r'))
        room['last_block'] = tx_receipt['blockNumber']
        open(room_file, 'w').write(json.dumps(room))


if __name__ == '__main__':
    # create the log and rooms folder if it doesn't exist
    os.makedirs("{}/logs/".format(os.getenv('DATA_FOLDER')), exist_ok=True)
    os.makedirs("{}/rooms/".format(os.getenv('DATA_FOLDER')), exist_ok=True)

    # create the rooms file if it doesn't exist
    try:
        room = json.load(open(room_file := "{}/rooms/{}.json".format(os.getenv('DATA_FOLDER'), os.getenv('CHANNEL_NAME')), 'r'))  # TODO: change this to target an address
    except (JSONDecodeError, FileNotFoundError):
        room = {
            "label": os.getenv('CHANNEL_NAME'),
            "last_block": 0,
            "preloaded": False,
        }
        open(room_file, 'w').write(json.dumps(room))
    log_file = "{}/logs/{}.log".format(os.getenv('DATA_FOLDER'), "VOID")
    # grab the entire log if the log file doesn't exist
    if not room['preloaded']:
        if os.path.exists(log_file):
            os.remove(log_file)
        latest_block = web3.eth.get_block('latest')
        block_cache[latest_block.number] = latest_block
        start_block_number = os.getenv('START_BLOCK')
        preload_chat(start_block_number, latest_block.number)
        room['preloaded'] = True
        room['last_block'] = latest_block.number
    else:
        # print past log to console
        logs = read_file_to_list(log_file)
        for i, log in enumerate(logs):
            if os.getenv('SHOW_AURA_AND_SOUL') != '0':
                print(log)
            else:
                match = re.match(r'^.*( [\d]+:[\d]+) .*$', log)
                if match:
                    print(log.replace(match.group(1), ''))
    open(room_file, 'w').write(json.dumps(room))
    # check for new messages
    load_chat(room['last_block'])