import argparse
import logging

import boto3
import botocore

logger = logging.getLogger(__name__)


def invoke_validate_email_domains_lambda(env: str, lambda_account_id: str) -> None:
    botocore_config = botocore.config.Config(
        retries={"max_attempts": 0},
        read_timeout=1000,
        connect_timeout=300,
        tcp_keepalive=True,
    )
    lambda_client = boto3.client("lambda", config=botocore_config)
    lambda_client.invoke(
        FunctionName=f"{lambda_account_id}:function:{env}-lambda-audience-validate-email-domains",
        InvocationType="Event",
        Payload=b"{}",
    )


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Invoke validate-email-domains lambda")
    parser.add_argument(
        "-e",
        "--environment",
        type=str,
        help="Environment (qa, prod)",
        default="qa",
        choices=["qa", "prod"],
    )
    parser.add_argument(
        "-lambda_account_id",
        type=str,
        help="Lambda account id",
    )
    args = parser.parse_args()

    invoke_validate_email_domains_lambda(
        env=args.environment,
        lambda_account_id=args.lambda_account_id,
    )