import argparse
import os
import sys
import logging
import yaml
import subprocess

def path_to_aax(aax: str):
    if not os.path.isfile(aax):
        raise argparse.ArgumentTypeError(f"Pfad {aax} ist keine gueltige Datei.")

    if not aax.endswith('.aax'):
        raise argparse.ArgumentTypeError(f"Pfad {aax} ist keine gueltige aax-Datei.")

    return aax

def decrypt_aax_file(aax_path: str, profile:str):
    logging.info("Ich decryyypte...")

    try:
        #Probe auf die aax Datei fuer Checksum
        cmd_probe = f"ffprobe {aax_path} 2>&1 | grep checksum"
        result_probe = subprocess.run(cmd_probe, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
        tmp_checksum = result_probe.stdout
        checksum = tmp_checksum.split('== ')[1]
        checksum = checksum.split('\n')[0] # Ein Line-Break wird am Ende immer noch angehängt -> Muss weg sonst gibts beim nächsten CMD Probleme
        logging.debug(f"Checksum: {checksum}")

        #Decrypt der Checksum via Rainbowtables für UserId
        cmd_crack = f"./rcrack . -h {checksum} | grep hex"
        cwd_tables = os.path.join(os.getcwd(),'tables')
        result_crack = subprocess.run(cmd_crack, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, cwd=cwd_tables)
        tmp_profile_id = result_crack.stdout
        profile_id = tmp_profile_id.split('hex:')[1]
        profile_id = profile_id.split('\n')[0] # Ein Line-Break wird am Ende immer noch angehängt -> Muss weg
        logging.info(f"Profile-Id extrahiert: {profile_id}")

        #Erstellen von Profil
        with open('profiles.yml', 'w') as profile_yaml:
            yaml.dump({profile : profile_id}, profile_yaml, default_flow_style=False)

    except subprocess.CalledProcessError as err:
        logging.info(f"Fehler bei Prozess ausfuehrung. {err}")
    #HowTo-WiemanshinkriegtAbsurdewirklichwissenschaftlicheEmpfehlungenfralleLebenslagen_ep7.aax 2>&1 | grep checksum

def convert_to_m4b(aax_path:str, profile:str):
    logging.info("Ich converteeee...")


if __name__ == "__main__":
    argparser = argparse.ArgumentParser(description="Dieses Script konvertiert verschluesselte Audible Buecher zu unverschluesselte m4b-Audiobooks.")
    cmd_parser = argparser.add_subparsers(dest='cmd', title='Commands', description="Erlaubte Commands", required=True)

    decrypt_parser = cmd_parser.add_parser('decrypt', help="Extrahiert die User-spezifische ID zur Profil-Bildung von einer bestehenden aax-Datei.")
    decrypt_parser.add_argument('aax', type=path_to_aax, help='Path zu einem zu entschluesselnden aax file des neuen Profils')
    decrypt_parser.add_argument('profile', type=str, help='Profil-Name, zu welchem die extrahierte ID gespeichert wird. (ID ist notwendig fuer Entschluesselung der eigenen Hoerbuecher)')

    convert_parser = cmd_parser.add_parser('convert', help="Konvertiert ein aax-Audiobook in ein m4b-Audiobook.")
    convert_parser.add_argument('aax', type=path_to_aax, help='Path zu einem zu entschluesselnden aax file')
    convert_parser.add_argument('profile', type=str, help="Zugehoeriges Profile fuer entschluesselung")

    args: argparse.Namespace = argparser.parse_args(args=None if sys.argv[1:] else ['--help'])

    logging.basicConfig(level=logging.INFO, format="%(asctime)s | %(levelname)s | %(message)s", datefmt="%d-%m-%Y %H:%M:%S")

    profiles = None
    try:
        with open('profiles.yml', 'r') as file:
            profiles = yaml.safe_load(file)
    except IOError:
        with open('profiles.yml', 'w') as file:
            profiles = yaml.safe_load('')

    if args.cmd == "decrypt":
        logging.info(f"Decrypt wird gestartet...")
        aax_file = args.aax
        profile = args.profile
        if profiles is not None and profiles.get(args.profile) is not None:
            logging.info(f"Profil {args.profile} existiert bereits. Konvertierung kann starten.")
            sys.exit()

        decrypt_aax_file(aax_file, profile)

    elif args.cmd == "convert":
        if profiles is None or profiles.get(args.profiles) is None or args.profiles not in profiles:
            logging.info(f"Profil {args.profile} existiert nicht. Bitte ueber decrypt anlegen lassen.")
            sys.exit()

        logging.info(f"Convert wird gestartet...")
        aax_file = args.aax
        profile = args.profile

        convert_to_m4b(aax_file, profile)