import click

from atlas_um.pgdb import DNAAccount
from atlas_um.tokens.bearer_tokens import DNABearerToken
from atlas_um.tokens.services import (
    CreateBearerTokenService,
    CreateRefreshTokenService,
)


@click.option("--ttl", default=DNABearerToken.DEFAULT_TIME_TO_LIVE)
@click.argument("name_or_email")
def generate_token_for_dna_account(name_or_email, ttl):
    """
    Utility command to generate DNA bearer tokens
    for local development needs.
    """

    dna_account = DNAAccount.query.search(name_or_email).first()
    if not dna_account:
        click.echo("There is no account with for provided search criteria")
        return
    dna_bearer_token = CreateBearerTokenService.execute(
        dna_account=dna_account,
        ttl=ttl,
    ).value
    dna_refresh_token = CreateRefreshTokenService.execute(
        dna_account=dna_account,
    ).value
    click.echo("DNA bearer token:")
    click.echo(dna_bearer_token)
    click.echo("DNA refresh token:")
    click.echo(dna_refresh_token)