#!/usr/bin/env pyston3

import argparse
import logging
import os
import os.path
import rx.operators as ops
import rx
import sys
import traceback
import typing

from solana.publickey import PublicKey

sys.path.insert(0, os.path.abspath(
    os.path.join(os.path.dirname(__file__), "..")))
import mango  # nopep8

# We explicitly want argument parsing to be outside the main try-except block because some arguments
# (like --help) will cause an exit, which our except: block traps.
parser = argparse.ArgumentParser(
    description="Run the Transaction Scout to display information about a specific transaction.")
mango.Context.add_command_line_parameters(parser)
parser.add_argument("--since-signature", type=str,
                    help="The signature of the transaction to look up")
parser.add_argument("--instruction-type", type=lambda ins: mango.InstructionType[ins],
                    choices=list(mango.InstructionType),
                    help="The signature of the transaction to look up")
parser.add_argument("--sender", type=PublicKey,
                    help="Only transactions sent by this PublicKey will be returned")
parser.add_argument("--notify-transactions", type=mango.parse_subscription_target, action="append", default=[],
                    help="The notification target for transaction information")
parser.add_argument("--notify-successful-transactions", type=mango.parse_subscription_target,
                    action="append", default=[], help="The notification target for successful transactions")
parser.add_argument("--notify-failed-transactions", type=mango.parse_subscription_target,
                    action="append", default=[], help="The notification target for failed transactions")
parser.add_argument("--notify-errors", type=mango.parse_subscription_target, action="append", default=[],
                    help="The notification target for errors")
parser.add_argument("--summarise", action="store_true", default=False,
                    help="create a short summary rather than the full TransactionScout details")
args = parser.parse_args()

logging.getLogger().setLevel(args.log_level)
for notify in args.notify_errors:
    handler = mango.NotificationHandler(notify)
    handler.setLevel(logging.ERROR)
    logging.getLogger().addHandler(handler)

logging.warning(mango.WARNING_DISCLAIMER_TEXT)


def summariser(context: mango.Context) -> typing.Callable[[mango.TransactionScout], str]:
    def summarise(transaction_scout: mango.TransactionScout) -> str:
        instruction_details: typing.List[str] = []
        for ins in transaction_scout.instructions:
            params = ins.describe_parameters()
            if params == "":
                instruction_details += [f"[{ins.instruction_type.name}]"]
            else:
                instruction_details += [f"[{ins.instruction_type.name}: {params}]"]

        instructions = ", ".join(instruction_details)
        changes = mango.OwnedTokenValue.changes(
            transaction_scout.pre_token_balances, transaction_scout.post_token_balances)

        in_tokens = []
        for ins in transaction_scout.instructions:
            if ins.token_in_account is not None:
                in_tokens += [mango.OwnedTokenValue.find_by_owner(changes, ins.token_in_account)]

        out_tokens = []
        for ins in transaction_scout.instructions:
            if ins.token_out_account is not None:
                out_tokens += [mango.OwnedTokenValue.find_by_owner(changes, ins.token_out_account)]

        changed_tokens = in_tokens + out_tokens
        changed_tokens_text = ", ".join(
            [f"{tok.token_value.value:,.8f} {tok.token_value.token.name}" for tok in changed_tokens]) or "None"

        return f"« 🥭 {transaction_scout.timestamp} {transaction_scout.group_name} {instructions}\n    From: {transaction_scout.sender}\n    Token Changes: {changed_tokens_text}\n    {transaction_scout.signatures} »"
    return summarise


try:
    since_signature = args.since_signature
    instruction_type = args.instruction_type
    sender = args.sender

    context = mango.Context.from_command_line_parameters(args)

    logging.info(f"Context: {context}")
    logging.info(f"Since signature: {since_signature}")
    logging.info(f"Filter to instruction type: {instruction_type}")

    first_item_capturer = mango.CaptureFirstItem()
    signatures = mango.fetch_all_recent_transaction_signatures(context)
    pipeline = rx.from_(signatures).pipe(
        ops.map(first_item_capturer.capture_if_first),
        #  ops.map(debug_print_item("Signature:")),
        ops.take_while(lambda sig: sig != since_signature),
        ops.map(lambda sig: mango.TransactionScout.load_if_available(context, sig)),
        ops.filter(lambda item: item is not None),
        #  ops.take(100),
    )

    if sender is not None:
        pipeline = pipeline.pipe(
            ops.filter(lambda item: item.sender == sender)
        )

    if instruction_type is not None:
        pipeline = pipeline.pipe(
            ops.filter(lambda item: item.has_any_instruction_of_type(
                instruction_type))
        )

    if args.summarise:
        pipeline = pipeline.pipe(
            ops.map(summariser(context))
        )

    fan_out = rx.subject.Subject()
    fan_out.subscribe(mango.PrintingObserverSubscriber(False))

    for notify in args.notify_transactions:
        fan_out.subscribe(on_next=notify.send)
    for notification_target in args.notify_successful_transactions:
        filtering = mango.FilteringNotificationTarget(
            notification_target, lambda item: isinstance(item, mango.TransactionScout) and item.succeeded)
        fan_out.subscribe(on_next=filtering.send)
    for notification_target in args.notify_failed_transactions:
        filtering = mango.FilteringNotificationTarget(notification_target, lambda item: isinstance(
            item, mango.TransactionScout) and not item.succeeded)
        fan_out.subscribe(on_next=filtering.send)

    pipeline.subscribe(fan_out)

    if first_item_capturer.captured is not None:
        with open("report.state", "w") as state_file:
            state_file.write(first_item_capturer.captured)
except Exception as exception:
    logging.critical(
        f"report-transactions stopped because of exception: {exception} - {traceback.format_exc()}")
except:
    logging.critical(
        f"report-transactions stopped because of uncatchable error: {traceback.format_exc()}")