"""

# Run awsume
awsume permissions-platform-qa

# Set up a python environment
rm -rf ./env/
python -mvenv env
./env/bin/pip3 install "secrets-manager>=1.4.6" -i https://pypi.theorchard.io/
./env/bin/pip3 install pytest
./env/bin/pip3 install "jwtauth[testing]==0.6.0" -i https://pypi.theorchard.io/

# Run the script
./env/bin/python3 PP-1127__login_from_secrets_manager__mfa_demo.py
"""

from jwtauth.testing import (
    SecretLookupInfo,
    login_from_secrets_manager,
    JwtAuthSecretsManager,
)

def pdp_test_bearer_token() -> None:
    """Example generating a token with `pdptest@theorchard.com`."""
    print("=================")
    print("Generating a token for pdptest@theorchard.com")
    secrets_manager = JwtAuthSecretsManager()

    token = login_from_secrets_manager(
        get_user_creds_args=SecretLookupInfo(
            environment="qa",
            service_name="pdp-integration-test",
            secret_name="PDP_TEST_USER_CREDENTIALS",
        ),
        get_auth0_creds_args=SecretLookupInfo(
            environment="qa",
            service_name="pdp-integration-test",
            secret_name="PDP_TEST_APP_AUTH0_CREDENTIALS",
        ),
        secrets_manager=secrets_manager,
    )

    print(f"Token (partial): {token[:10]}...")


def pdp_mfa_test_bearer_token() -> None:
    """Example to generate a token for an MFA enabled user:
    `pdptest+mfa@theorchard.com`."""
    print("=================")
    print("Generating a token for pdptest+mfa@theorchard.com")

    secrets_manager = JwtAuthSecretsManager()

    token = login_from_secrets_manager(
        get_user_creds_args=SecretLookupInfo(
            environment="qa",
            service_name="pdp-integration-test",
            secret_name="PDP_TEST_MFA_USER_CREDENTIALS",
        ),
        get_auth0_creds_args=SecretLookupInfo(
            environment="qa",
            service_name="pdp-integration-test",
            secret_name="PDP_TEST_APP_AUTH0_CREDENTIALS",
        ),
        secrets_manager=secrets_manager,
    )

    print("Token (complete)")
    print(token)


if __name__ == '__main__':
    pdp_test_bearer_token()
    pdp_mfa_test_bearer_token()