###############################################################################
# Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy of this
# software and associated documentation files (the "Software"), to deal in the Software
# without restriction, including without limitation the rights to use, copy, modify,
# merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
# permit persons to whom the Software is furnished to do so.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
# INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
# PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
# SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
#
#
# For more information, see:
# https://docs.aws.amazon.com/AmazonRDS/latest/AuroraUserGuide/data-api.html
#
# Prerequisites:
#  - Required permissions to Secrets Manager and RDS Data API
#  - Latest boto3 SDK in local package. Run the following in Cloud9 shell:
#     - `pip install boto3 -t boto3`
#
###############################################################################

import os
import os.path
import sys

# Use latest boto3 from local environment
envLambdaTaskRoot = os.environ["LAMBDA_TASK_ROOT"]
sys.path.insert(0, envLambdaTaskRoot+"/boto3")

# Required imports
import botocore
import boto3

# Imports for your app
from datetime import datetime

# Update your cluster and secret ARNs
cluster_arn = 'arn:aws:rds:us-east-1:0000000000:cluster:my-cluster' 
secret_arn = 'arn:aws:secretsmanager:us-east-1:0000000000:secret:my-secret'

def lambda_handler(event, context):
    # Handle event here and call RDS Data API
    callRdsDataApi(datetime.now().strftime("%Y-%m-%dT%T.%fZ"), 'some string from event?')


def callRdsDataApi(timestamp, message):
    rdsData = boto3.client('rds-data')

    sqlStatement = """
                    INSERT INTO sampleTable(received_at, message)
                    VALUES(STR_TO_DATE(:time, "%Y-%m-%dT%T.%fZ"), :message)
                   """

    param1 = {'name':'time', 'value':{'stringValue': timestamp}}
    param2 = {'name':'message', 'value':{'stringValue': message}}
    paramSet = [param1, param2]
 
    response = rdsData.execute_statement(
            resourceArn = cluster_arn, 
            secretArn = secret_arn, 
            database = 'sampleDb', 
            sql = sqlStatement,
            parameters = paramSet)
    
    print (str(response));


def sampleSetup():
    rdsData = boto3.client('rds-data')
    
    # Create a database
    rdsData.execute_statement(
        resourceArn = cluster_arn,
        secretArn = secret_arn,
        sql = "CREATE DATABASE IF NOT EXISTS sampleDb")
        
    # Create a table
    rdsData.execute_statement(
        resourceArn = cluster_arn,
        secretArn = secret_arn,
        database = "sampleDb",
        sql = "CREATE TABLE IF NOT EXISTS sampleTable(received_at datetime, message varchar(255))")

# Create a sample database and table
sampleSetup()