# ruff: noqa: T201 (allow print)
import json

import boto3

from m2mconfig.connectors.assume_role_secrets_manager import AssumeRoleSecretsManager

Environment = "qa"
QA_AWS_ACCOUNT_IDS = ["591204808501", "285943604611"]

SECRET_NAMES = [
    "qa/lambda-test-m2m-client/M2M_AUTH0_CLIENT_CREDENTIALS",
    "qa/this-asterisk/M2M_AUTH0_CLIENT_CREDENTIALS",
    "qa/lambda-test-m2m-client/SOME_OTHER_SECRET",
]
secret_value = {
    "grant_type": "client_credentials",
    "audience": "api_audience",
    "client_id": "some-auth0-client-id",
    "client_secret": "some-secret-string",
}

sts_client = boto3.client("sts")
managers = {
    account: AssumeRoleSecretsManager(
        Environment, sts_client, account, "assume-auth0-m2m-config"
    )
    for account in QA_AWS_ACCOUNT_IDS
}

for account, manager in managers.items():
    print("#####################################")
    print(f"Testing AWS Account Id: {account}")
    print("#####################################")
    for secret_name in SECRET_NAMES:
        print(f"~~{secret_name}~~")
        try:
            does_secret_exist = manager.does_secret_exist(secret_name)
            if does_secret_exist:
                print(">>>>IT EXISTS")
                print(f">>>>GET: {manager.get_secret_string(secret_name)}")
                print(
                    f">>>>SET: {manager.save_secret_string(secret_name, json.dumps(secret_value))}"
                )
                print(f">>>>GET: {manager.get_secret_string(secret_name)}")
        except Exception as e:
            print(f">>>>ERROR: {e}")
        print("~~~~~~~~~~~~~~~~~~~~~~~~")