import string
from datetime import datetime, timedelta

import allure
import pytest
from frostfs_testlib import reporter
from frostfs_testlib.s3 import S3ClientWrapper, VersioningStatus
from frostfs_testlib.steps.s3 import s3_helper
from frostfs_testlib.storage.dataclasses.object_size import ObjectSize
from frostfs_testlib.utils import string_utils
from frostfs_testlib.utils.file_utils import generate_file

VALID_SYMBOLS_WITHOUT_DOT = string.ascii_lowercase + string.digits + "-"
VALID_AND_INVALID_SYMBOLS = string.ascii_letters + string.punctuation

# TODO: The dot symbol is temporarily not supported.
VALID_SYMBOLS_WITH_DOT = VALID_SYMBOLS_WITHOUT_DOT + "."


@pytest.mark.nightly
@pytest.mark.s3_gate
@pytest.mark.s3_gate_bucket
class TestS3GateBucket:
    @allure.title("Bucket API (s3_client={s3_client})")
    def test_s3_buckets(
        self,
        s3_client: S3ClientWrapper,
        simple_object_size: ObjectSize,
    ):
        """
        Test base S3 Bucket API (Create/List/Head/Delete).
        """

        file_path = generate_file(simple_object_size.value)
        file_name = s3_helper.object_key_from_file_path(file_path)

        with reporter.step("Create buckets"):
            bucket_1 = s3_client.create_bucket(object_lock_enabled_for_bucket=True)
            s3_helper.set_bucket_versioning(s3_client, bucket_1, VersioningStatus.ENABLED)
            bucket_2 = s3_client.create_bucket()

        with reporter.step("Check buckets are presented in the system"):
            buckets = s3_client.list_buckets()
            assert bucket_1 in buckets, f"Expected bucket {bucket_1} is in the list"
            assert bucket_2 in buckets, f"Expected bucket {bucket_2} is in the list"

        with reporter.step("Bucket must be empty"):
            for bucket in (bucket_1, bucket_2):
                with reporter.step("Verify default list command"):
                    objects_list = s3_client.list_objects(bucket)
                    assert not objects_list, f"Expected empty bucket, got {objects_list}"

                with reporter.step("Verify V2 list command"):
                    objects_list = s3_client.list_objects_v2(bucket)
                    assert not objects_list, f"Expected empty bucket, got {objects_list}"

        with reporter.step("Check buckets are visible with S3 head command"):
            s3_client.head_bucket(bucket_1)
            s3_client.head_bucket(bucket_2)

        with reporter.step("Check we can put/list object with S3 commands"):
            version_id = s3_client.put_object(bucket_1, file_path)
            s3_client.head_object(bucket_1, file_name)

            bucket_objects = s3_client.list_objects(bucket_1)
            assert file_name in bucket_objects, f"Expected file {file_name} in objects list {bucket_objects}"

        with reporter.step("Try to delete not empty bucket and get error"):
            with pytest.raises(Exception, match=r".*The bucket you tried to delete is not empty.*"):
                s3_client.delete_bucket(bucket_1)

            s3_client.head_bucket(bucket_1)

        with reporter.step("Delete empty bucket_2"):
            s3_client.delete_bucket(bucket_2)

        with reporter.step("Check bucket_2 is deleted"):
            with pytest.raises(Exception, match=r".*Not Found.*"):
                s3_client.head_bucket(bucket_2)

            buckets = s3_client.list_buckets()
            assert bucket_1 in buckets, f"Expected bucket {bucket_1} is in the list"
            assert bucket_2 not in buckets, f"Expected bucket {bucket_2} is not in the list"

        with reporter.step("Delete object from bucket_1"):
            s3_client.delete_object(bucket_1, file_name, version_id)
            s3_helper.check_objects_in_bucket(s3_client, bucket_1, expected_objects=[])

        with reporter.step("Delete bucket_1"):
            s3_client.delete_bucket(bucket_1)

        with reporter.step("Check bucket_1 deleted"):
            with pytest.raises(Exception, match=r".*Not Found.*"):
                s3_client.head_bucket(bucket_1)

    @allure.title("Create bucket with object lock (s3_client={s3_client})")
    def test_s3_bucket_object_lock(self, s3_client: S3ClientWrapper, simple_object_size: ObjectSize):
        file_path = generate_file(simple_object_size.value)
        file_name = s3_helper.object_key_from_file_path(file_path)

        with reporter.step("Create bucket with --no-object-lock-enabled-for-bucket"):
            bucket = s3_client.create_bucket(object_lock_enabled_for_bucket=False)
            date_obj = datetime.utcnow() + timedelta(days=1)
            with pytest.raises(Exception, match=r".*Object Lock configuration does not exist for this bucket.*"):
                # An error occurred (ObjectLockConfigurationNotFoundError) when calling the PutObject operation (reached max retries: 0):
                # Object Lock configuration does not exist for this bucket
                s3_client.put_object(
                    bucket,
                    file_path,
                    object_lock_mode="COMPLIANCE",
                    object_lock_retain_until_date=date_obj.strftime("%Y-%m-%dT%H:%M:%S"),
                )
        with reporter.step("Create bucket with --object-lock-enabled-for-bucket"):
            bucket_1 = s3_client.create_bucket(object_lock_enabled_for_bucket=True)
            date_obj_1 = datetime.utcnow() + timedelta(days=1)
            s3_client.put_object(
                bucket_1,
                file_path,
                object_lock_mode="COMPLIANCE",
                object_lock_retain_until_date=date_obj_1.strftime("%Y-%m-%dT%H:%M:%S"),
                object_lock_legal_hold_status="ON",
            )
            s3_helper.assert_object_lock_mode(s3_client, bucket_1, file_name, "COMPLIANCE", date_obj_1, "ON")

    @allure.title("Delete bucket (s3_client={s3_client})")
    def test_s3_delete_bucket(self, s3_client: S3ClientWrapper, simple_object_size: ObjectSize):
        file_path_1 = generate_file(simple_object_size.value)
        file_name_1 = s3_helper.object_key_from_file_path(file_path_1)
        file_path_2 = generate_file(simple_object_size.value)
        file_name_2 = s3_helper.object_key_from_file_path(file_path_2)
        bucket = s3_client.create_bucket()

        with reporter.step("Put two objects into bucket"):
            s3_client.put_object(bucket, file_path_1)
            s3_client.put_object(bucket, file_path_2)
            s3_helper.check_objects_in_bucket(s3_client, bucket, [file_name_1, file_name_2])

        with reporter.step("Try to delete not empty bucket and get error"):
            with pytest.raises(Exception, match=r".*The bucket you tried to delete is not empty.*"):
                s3_client.delete_bucket(bucket)

        with reporter.step("Delete object in bucket"):
            s3_client.delete_object(bucket, file_name_1)
            s3_client.delete_object(bucket, file_name_2)
            s3_helper.check_objects_in_bucket(s3_client, bucket, [])

        with reporter.step("Delete empty bucket"):
            s3_client.delete_bucket(bucket)
            with pytest.raises(Exception, match=r".*Not Found.*"):
                s3_client.head_bucket(bucket)

    @allure.title("Create bucket with valid name length (s3_client={s3_client}, length={length})")
    @pytest.mark.parametrize("length", [3, 4, 32, 62, 63])
    def test_s3_create_bucket_with_valid_length(self, s3_client: S3ClientWrapper, length: int):
        bucket_name = string_utils.random_string(length, VALID_SYMBOLS_WITHOUT_DOT)
        while not (bucket_name[0].isalnum() and bucket_name[-1].isalnum()):
            bucket_name = string_utils.random_string(length, VALID_SYMBOLS_WITHOUT_DOT)

        with reporter.step("Create bucket with valid name length"):
            s3_client.create_bucket(bucket_name)

        with reporter.step("Check bucket name in buckets"):
            assert bucket_name in s3_client.list_buckets()

    @allure.title("[NEGATIVE] Bucket with invalid name length should not be created (s3_client={s3_client}, length={length})")
    @pytest.mark.parametrize("length", [2, 64, 254, 255, 256])
    def test_s3_create_bucket_with_invalid_length(self, s3_client: S3ClientWrapper, length: int):
        bucket_name = string_utils.random_string(length, VALID_SYMBOLS_WITHOUT_DOT)
        while not (bucket_name[0].isalnum() and bucket_name[-1].isalnum()):
            bucket_name = string_utils.random_string(length, VALID_SYMBOLS_WITHOUT_DOT)

        with reporter.step("Create bucket with invalid name length and catch exception"):
            with pytest.raises(Exception, match=".*(?:InvalidBucketName|Invalid bucket name).*"):
                s3_client.create_bucket(bucket_name)

    @allure.title("[NEGATIVE] Bucket with invalid name should not be created (s3_client={s3_client}, bucket_name={bucket_name})")
    @pytest.mark.parametrize(
        "bucket_name",
        [
            "BUCKET-1",
            "buckeT-2",
            # The following case for AWS CLI is not handled correctly
            # "-bucket-3",
            "bucket-4-",
            ".bucket-5",
            "bucket-6.",
            "bucket..7",
            "bucket+8",
            "bucket_9",
            "bucket 10",
            "127.10.5.11",
            "xn--bucket-12",
            "bucket-13-s3alias",
            # The following names can be used in FrostFS but are prohibited by the AWS specification.
            # "sthree-bucket-14"
            # "sthree-configurator-bucket-15"
            # "amzn-s3-demo-bucket-16"
            # "sthree-bucket-17"
            # "bucket-18--ol-s3"
            # "bucket-19--x-s3"
            # "bucket-20.mrap"
        ],
    )
    def test_s3_create_bucket_with_invalid_name(self, s3_client: S3ClientWrapper, bucket_name: str):
        with reporter.step("Create bucket with invalid name and catch exception"):
            with pytest.raises(Exception, match=".*(?:InvalidBucketName|Invalid bucket name).*"):
                s3_client.create_bucket(bucket_name)

    @allure.title("[NEGATIVE] Delete non-empty bucket (s3_client={s3_client})")
    def test_s3_check_availability_non_empty_bucket_after_deleting(
        self,
        bucket: str,
        simple_object_size: ObjectSize,
        s3_client: S3ClientWrapper,
    ):
        object_path = generate_file(simple_object_size.value)
        object_name = s3_helper.object_key_from_file_path(object_path)

        with reporter.step("Put object into bucket"):
            s3_client.put_object(bucket, object_path)

        with reporter.step("Check that object appears in bucket"):
            objects = s3_client.list_objects(bucket)
            assert objects, f"Expected bucket with object, got empty {objects}"
            assert object_name in objects, f"Object {object_name} not found in bucket object list {objects}"

        with reporter.step("Try to delete not empty bucket and get error"):
            with pytest.raises(Exception, match=r".*The bucket you tried to delete is not empty.*"):
                s3_client.delete_bucket(bucket)

        with reporter.step("Check bucket availability"):
            objects = s3_client.list_objects(bucket)
            assert objects, f"Expected bucket with object, got empty {objects}"
            assert object_name in objects, f"Object {object_name} not found in bucket object list {objects}"