import boto3
from src.system_var.constants import (AWS_RESOURCE_S3, S3_CHAR_CODE,
                                      S3_RESPONSE_BODY)
from src.system_var.environments import (CRM_BACKUP_BUCKET, CRM_CONFIG_BUCKET,
                                         CRM_IMPORT_DATA_BACKUP_FOLDER,
                                         CRM_IMPORT_DATA_FOLDER,
                                         IMPORT_DATA_BUCKET,
                                         LAST_FETCH_DATE_FOLDER,
                                         OBJECT_INFO_FILENAME,
                                         OBJECT_INFO_FOLDER,
                                         PROCESS_RESULT_FOLDER,
                                         RESPONSE_JSON_BACKUP_FOLDER)


class S3Client:
    def __init__(self, bucket_name: str) -> None:
        self.__s3_client = boto3.client(AWS_RESOURCE_S3)
        self.__s3_bucket = bucket_name

    def get_object(self, object_key: str) -> str:
        response = self.__s3_client.get_object(Bucket=self.__s3_bucket, Key=object_key)
        body = response[S3_RESPONSE_BODY].read()
        return body.decode(S3_CHAR_CODE)

    def put_object(self, object_key: str, local_file_path: str) -> None:
        self.__s3_client.upload_file(Filename=local_file_path, Bucket=self.__s3_bucket, Key=object_key)
        return

    def copy(self, src_bucket: str, src_key: str, dest_bucket: str, dest_key: str) -> None:
        copy_source = {'Bucket': src_bucket, 'Key': src_key}
        self.__s3_client.copy_object(CopySource=copy_source, Bucket=dest_bucket, Key=dest_key)
        return


class ConfigBucket:
    __s3_client: S3Client = None

    def __init__(self) -> None:
        self.__s3_client = S3Client(CRM_CONFIG_BUCKET)

    def __str__(self) -> str:
        return CRM_CONFIG_BUCKET

    def get_object_info_file(self) -> str:
        return self.__s3_client.get_object(f'{OBJECT_INFO_FOLDER}/{OBJECT_INFO_FILENAME}')

    def get_last_fetch_datetime_file(self, file_key: str) -> str:
        return self.__s3_client.get_object(f'{LAST_FETCH_DATE_FOLDER}/{file_key}')

    def put_last_fetch_datetime_file(self, file_key: str, local_file_path: str) -> None:
        self.__s3_client.put_object(
            f'{LAST_FETCH_DATE_FOLDER}/{file_key}', local_file_path)
        return


class DataBucket:
    __s3_client: S3Client = None

    def __init__(self) -> None:
        self.__s3_client = S3Client(IMPORT_DATA_BUCKET)

    def __str__(self) -> str:
        return IMPORT_DATA_BUCKET

    def put_csv(self, file_key: str, local_file_path: str) -> None:
        object_key = f'{CRM_IMPORT_DATA_FOLDER}/{file_key}'
        self.__s3_client.put_object(object_key, local_file_path)
        return

    def put_csv_from(self, src_bucket: str, src_key: str):
        dest_filename = src_key.split('/')[-1]
        self.__s3_client.copy(src_bucket, src_key, str(self), f'{CRM_IMPORT_DATA_FOLDER}/{dest_filename}')
        return


class BackupBucket:
    __s3_client: S3Client = None

    def __init__(self) -> None:
        self.__s3_client = S3Client(CRM_BACKUP_BUCKET)

    def __str__(self) -> str:
        return CRM_BACKUP_BUCKET

    def put_response_json(self, file_key: str, local_file_path: str) -> None:
        object_key = f'{RESPONSE_JSON_BACKUP_FOLDER}/{file_key}'
        self.__s3_client.put_object(object_key, local_file_path)
        return

    def put_csv(self, file_key: str, local_file_path: str) -> None:
        object_key = f'{CRM_IMPORT_DATA_BACKUP_FOLDER}/{file_key}'
        self.__s3_client.put_object(object_key, local_file_path)
        return

    def put_result_json(self, file_key: str, local_file_path: str) -> None:
        object_key = f'{PROCESS_RESULT_FOLDER}/{file_key}'
        self.__s3_client.put_object(object_key, local_file_path)
        return