#!/usr/bin/env python3

import sys, re, os.path, json, ast, time
from datetime import datetime
from termcolor import colored
from lib.natools import fmt, sign, get_privkey
from gql import gql, Client
from gql.transport.aiohttp import AIOHTTPTransport

PUBKEY_REGEX = "(?![OIl])[1-9A-Za-z]{42,45}"

class History:

    def __init__(self, dunikey, node, pubkey, useMempool=False):
        self.dunikey = dunikey
        self.pubkey = pubkey if pubkey else get_privkey(dunikey, "pubsec").pubkey
        self.useMempool = useMempool
        self.node = node
        if not re.match(PUBKEY_REGEX, self.pubkey) or len(self.pubkey) > 45:
            sys.stderr.write("La clé publique n'est pas au bon format.\n")
            sys.exit(1)

        # Define Duniter GVA node
        transport = AIOHTTPTransport(url=node)
        self.client = Client(transport=transport, fetch_schema_from_transport=True)

    def sendDoc(self):
        # Build TX generation document
        queryBuild = gql(
            """
            query ($pubkey: String!){
                transactionsHistory(pubkey: $pubkey) {
                    received {
                        writtenTime
                        issuers
                        outputs
                        comment
                    }
                    sent {
                        writtenTime
                        issuers
                        outputs
                        comment
                    }
                   receiving {
                        issuers
                        outputs
                        comment
                    }
                    sending {
                        issuers
                        outputs
                        comment
                    }
                }
                balance(script: $pubkey) {
                    amount
                    base
                }
            }
        """
        )
        paramsBuild = {
            "pubkey": self.pubkey
        }

        # Send TX document
        try:
            self.historyDoc = self.client.execute(queryBuild, variable_values=paramsBuild)
        except Exception as e:
            message = ast.literal_eval(str(e))["message"]
            sys.stderr.write("Echec de récupération de l'historique:\n" + message + "\n")
            sys.exit(1)


    def parseHistory(self):
        trans = []
        i = 0
 
        for sens in 'received','sent','receiving','sending':
            res = self.historyDoc['transactionsHistory'][sens]
            for bloc in res:
                output = bloc['outputs'][0]
                trans.append(i)
                trans[i] = []
                trans[i].append(sens)
                if sens in ('receiving','sending'):
                    trans[i].append(int(time.time()))
                else:
                    trans[i].append(bloc['writtenTime'])
                if sens in ('sent','sending'):
                    trans[i].append(output.split("SIG(")[1].replace(')',''))
                else:
                    trans[i].append(bloc['issuers'][0])
                amount = int(output.split(':')[0])
                base = int(output.split(':')[1])
                trans[i].append(amount*pow(10,base)/100)
                trans[i].append(bloc['comment'])
                i += 1

        trans.sort(key=lambda x: x[1])

        # Get balance
        balance = self.historyDoc['balance']['amount']
        baseBalance = self.historyDoc['balance']['base']
        balance = balance*pow(10,baseBalance)/100

        # Get terminal size
        rows = int(os.popen('stty size', 'r').read().split()[1])

        print('-'.center(rows, '-'))
        print("\033[1m{: <20} | {: <45} | {: <7} | {: <30}\033[0m".format("        Date","            De / À (clé publique)","Montant (Ḡ1)","Commentaire"))
        for t in trans:
            if t[0] == "received":
                color = "green"
            elif t[0] == "receiving":
                color = "yellow"
            elif t[0] == "sending":
                color = "red"
            else: color = "blue"

            date = datetime.fromtimestamp(t[1]).strftime("%d/%m/%Y à %H:%M")
            print('-'.center(rows, '-'))
            print(colored("| {: <18} | {: <45} | {: <12} | {: <30}".format(date, *t[2:]), color))
        print('-'.center(rows, '-'))
        print('\033[1mSolde du compte: {0} Ḡ1\033[0m'.format(balance).center(rows, ' '))
        print('-'.center(rows, '-'))
        print(colored('Reçus', 'green'), '-', colored('En cours de réception', 'yellow'), '-', colored('Envoyé', 'blue'), '-', colored("En cours d'envoi", 'red'))

    def history(self):
        self.sendDoc()
        self.parseHistory()