#!/usr/bin/env python

import argparse
import base64
import cbor2
from cryptography import x509
from cryptography.hazmat.primitives.asymmetric import ec
from cryptography.hazmat.primitives.asymmetric.utils import \
        encode_dss_signature
import json
import os
import sys
from textwrap import wrap
import zlib

from . import base45


def read_qrcode(file) -> str:
    """
    Lit le contenu du fichier.
    Peut être combiné avec zbar pour directement lire le contenu du QRCode.
    """
    with file:
        content = file.read()
    content = content.replace('\n', '')
    return content


def analyse_qrcode(qrcode: str, additional_info: bool = False,
        check_signature: bool = True) -> None:
    """
    Analyse les données du QR code pour extraire les données
    et vérifier la signature.

    Si `additional_info` est vrai, les informations de
    vaccination/test sont affichées.
    """

    if not qrcode.startswith('HC1:'):
        raise ValueError("QR code invalide.")

    plain_input = qrcode.replace('HC1:', '')

    # QR Code en base 45
    compressed_cose = base45.b45decode(plain_input)

    # Décompression ZIP si nécessaire
    cose = compressed_cose
    if compressed_cose[0] == 0x78:
        fb = compressed_cose[1]
        if fb == 0x01 or fb == 0x5E or fb == 0x9C or fb == 0xDA:
            cose = zlib.decompress(compressed_cose)

    # Chargement de l'objet CBOR
    obj = cbor2.loads(cose)
    header = cbor2.loads(obj.value[0])
    payload = cbor2.loads(obj.value[2])
    signature = obj.value[3]

    if check_signature:
        # Récupération du certificat utilisé
        kid = base64.b64encode(header[4]).decode()
        cert_name = kid.replace('/', '_')

        base_dir = os.path.dirname(__file__)
        cert_path = os.path.join(base_dir, 'certs', f"{cert_name}.pem")
        if not os.path.isfile(cert_path):
            print(f"Le certificat {kid} n'a pas été trouvé.")
            print("Utilisez l'option --dontcheck pour sauter "
                  "la vérification.")
            print("Si vous disposez du certificat, installez-le dans",
                  cert_path)
            exit(1)
        with open(os.path.join(base_dir, 'certs', f"{cert_name}.pem")) as f:
            cert_asc = f.read()

        cert = x509.load_pem_x509_certificate(cert_asc.encode())

        public_key = cert.public_key()

        # Calcul de la bonne signature et des données signées
        data = cbor2.dumps(
                ["Signature1", cbor2.dumps(header),
                 bytes(), cbor2.dumps(payload)]
        )

        l = len(signature)
        r = int.from_bytes(signature[:l // 2], 'big')
        s = int.from_bytes(signature[l // 2:], 'big')
        signature = encode_dss_signature(r, s)
        try:
            # Vérification de la signature
            public_key.verify(signature, data,
                              ec.ECDSA(cert.signature_hash_algorithm))
            valid = True
        except:
            valid = False
    else:
        valid = True
        print("Attention : la signature du QR code n'a pas été vérifiée.")

    # TODO: Vérifier les données d'un test

    if valid:
        print("Pass sanitaire valide")

        # Information utile
        p = payload[-260][1]

        print("Nom :", p['nam']['fn'])
        print("Prénom :", p['nam']['gn'])
        print("Date de naissance :", p['dob'])
        if additional_info:
            if 'v' in p:
                # Vaccination
                # TODO Meilleur affichage
                print("Informations de vaccination :", p['v'])
            else:
                # TODO Gérer les tests positifs / négatifs
                print("Informations de test :", p)
    else:
        print("Pass sanitaire invalide")


def main():
    parser = argparse.ArgumentParser()

    parser.add_argument('file', nargs='?', type=argparse.FileType('r'),
                        default=sys.stdin,
                        help="QR Code à lire, en format texte.")
    parser.add_argument('--full', '-f', action='store_true',
                        help="Affiche toutes les informations.")
    parser.add_argument('--dontcheck', action='store_true',
                        help="Ne pas vérifier la signature.")

    args = parser.parse_args()

    qrcode = read_qrcode(args.file)
    analyse_qrcode(qrcode, args.full, not args.dontcheck)