#!/usr/bin/env python3

import sys, re, os.path, json, ast
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 Id:

    def __init__(self, dunikey, node, pubkey='', username=''):
        self.dunikey = dunikey
        self.pubkey = pubkey if pubkey else get_privkey(dunikey, "pubsec").pubkey
        self.pubkey = pubkey
        self.username = username
        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, getBalance=False):
        # Build balance generation document
        if (getBalance):
            queryBuild = gql(
                """
                query ($pubkey: String!){
                    idty (pubkey: $pubkey) {
                        isMember
                        username
                    }
                    balance(script: $pubkey) {
                    amount
                }
                }
            """
            )
        else:
            queryBuild = gql(
                """
                query ($pubkey: String!){
                    idty (pubkey: $pubkey) {
                        isMember
                        username
                    }
                }
            """
            )
        
        paramsBuild = {
            "pubkey": self.pubkey
        }

        # Send balance document
        try:
            IDResult = 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 du solde:\n" + message + "\n")
            sys.exit(1)

        jsonBrut = IDResult['idty']
        if (getBalance):
            jsonBrut['balance'] = IDResult['balance']['amount']/100
        username = IDResult['idty']['username']
        isMember = IDResult['idty']['isMember']

        return json.dumps(jsonBrut, indent=2)