import os
from functools import wraps
from uuid import uuid4

import boto3

from moto import mock_aws


def dynamodb_aws_verified(create_table: bool = True, add_range: bool = False):
    """
    Function that is verified to work against AWS.
    Can be run against AWS at any time by setting:
      MOTO_TEST_ALLOW_AWS_REQUEST=true

    If this environment variable is not set, the function runs in a `mock_aws` context.

    This decorator will:
      - Create a table
      - Run the test and pass the table_name as an argument
      - Delete the table
    """

    def inner(func):
        @wraps(func)
        def pagination_wrapper():
            table_name = "t" + str(uuid4())[0:6]

            allow_aws_request = (
                os.environ.get("MOTO_TEST_ALLOW_AWS_REQUEST", "false").lower() == "true"
            )

            if allow_aws_request:
                if create_table:
                    print(f"Test {func} will create DDB Table {table_name}")  # noqa
                    return create_table_and_test(table_name)
                else:
                    return func()
            else:
                with mock_aws():
                    if create_table:
                        return create_table_and_test(table_name)
                    else:
                        return func()

        def create_table_and_test(table_name):
            client = boto3.client("dynamodb", region_name="us-east-1")

            schema = [{"AttributeName": "pk", "KeyType": "HASH"}]
            defs = [{"AttributeName": "pk", "AttributeType": "S"}]
            if add_range:
                schema.append({"AttributeName": "sk", "KeyType": "RANGE"})
                defs.append({"AttributeName": "sk", "AttributeType": "S"})
            client.create_table(
                TableName=table_name,
                KeySchema=schema,
                AttributeDefinitions=defs,
                ProvisionedThroughput={"ReadCapacityUnits": 1, "WriteCapacityUnits": 5},
                Tags=[{"Key": "environment", "Value": "moto_tests"}],
            )
            waiter = client.get_waiter("table_exists")
            waiter.wait(TableName=table_name)
            try:
                resp = func(table_name)
            finally:
                ### CLEANUP ###
                client.delete_table(TableName=table_name)

            return resp

        return pagination_wrapper

    return inner