from xdg import xdg_config_home
from utils import open_fake_terminal, close_fake_terminal, run
import audible, click, sys

def handle_aax(aax_path):
    authcode_path = xdg_config_home().joinpath('knock', 'aax', 'authcode')

    # make the config dir if it doesn't exist
    authcode_path.parent.mkdir(parents=True, exist_ok=True)

    m4b_path = aax_path.with_suffix('.m4b')

    if m4b_path.exists():
        click.echo(f"Error: {m4b_path} must be moved out of the way or deleted.", err=True)
        sys.exit(1)

    if not authcode_path.exists():
        click.echo('This device does not have an Audible decryption key.')

        email = click.prompt("Enter your Audible account's email address")
        password = click.prompt("Enter your Audible account's password", hide_input=True)
        locale = click.prompt("Enter your locale (e.g. 'us', 'ca', 'jp', etc)")

        open_fake_terminal(f'audible.auth.Authenticator.from_login("{email}", "{locale}").get_activation_bytes()')

        try:
            authcode = audible.auth.Authenticator.from_login(
                username=email,
                password=password,
                locale=locale
            ).get_activation_bytes()
            authcode_path.write_text(authcode)
        except Exception as error:
            click.echo(error, err=True)
            close_fake_terminal(1)

        close_fake_terminal(0)

    click.echo('Decrypting the file...')

    authcode = authcode_path.read_text()

    run([
        'ffmpeg',
        '-activation_bytes', authcode,
        '-i', str(aax_path),
        '-c', 'copy', str(m4b_path),
        '-loglevel', 'error'
    ])

    click.secho(f'DRM-free M4B file created:\n{aax_path.with_suffix(".m4b")}', fg='green')