#
# All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
# its licensors.
#
# For complete copyright and license terms please see the LICENSE at the root of this
# distribution (the "License"). All use of this software is governed by the License,
# or, if provided, by the license below or the license accompanying this file. Do not
# remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#

import retry
import boto3_util
import metric_constant as c
import math
import uuid
import sys
import time
import os
import math
import util
import json
from botocore.exceptions import ClientError

class S3(object):

    def __init__(self, context={}, bucket=""):        
        self.__context = context        
        self.__resource = boto3_util.resource('s3', api_version='2006-03-01')
        self.__client = boto3_util.client('s3', api_version='2006-03-01')
        self.__bucket = bucket
  
    @property
    def client(self):
        return self.__client

    @property
    def resource(self):
        return self.__resource

    def list(self, prefix = None):
        paginator = self.__client.get_paginator( "list_objects_v2" )
        params = dict({})
        params['Bucket'] = self.__bucket        
        params['PaginationConfig'] = {            
            'PageSize': 500
        }
        if prefix is not None:
            params['Prefix'] = prefix
        page_iterator = paginator.paginate(**params)
        return page_iterator

    def delete(self, keys_to_delete):        
        split_size = 1000              
        keys_to_delete = util.split(keys_to_delete, split_size, self.__delete_wrapper)
        for set in keys_to_delete:                                                          
            params = dict({})
            params['Bucket'] = self.__bucket
            params['Delete'] = { 
                    "Objects": set                    
            }
            util.debug_print("Deleting {}".format(params))
            response = self.__client.delete_objects(**params)  
            util.debug_print("Delete response {}".format(response))            

    def read(self, key):        
        response = self.__client.get_object(
            Bucket=self.__bucket,
            Key=util.get_path_without_leading_seperator(key)
        )
        return response["Body"].read()
        
    def size(self, key):    
        try:
            response = self.__client.head_object(Bucket=self.__bucket, Key=util.get_path_without_leading_seperator(key))
        except ClientError as e:             
            raise e
                
        return response['ContentLength']

    def put_object(self, key, body, expiration=None):
        params = dict({})
        params['Key'] = util.get_path_without_leading_seperator(key)
        params['Body'] = body
        params['Bucket'] = self.__bucket
        if expiration:
            params['Expires'] = expiration
        return self.client.put_object(**params)

    def size_in_megabytes(self, key):
        return self.size(key)/ 1048576.0

    def get_object(self, key):
        params = dict({})
        params['Key'] = util.get_path_without_leading_seperator(key)
        params['Bucket'] = self.__bucket        
        return self.client.get_object(**params)

    def __delete_wrapper(self, key):
        return { "Key": util.get_path_without_leading_seperator(key) }