import json
import boto3
from datetime import datetime, timedelta
import os
from os import getenv

KINESIS_STREAM_NAME = os.environ["KINESIS_STREAM_NAME"]
TRANSCRIBER_CALL_EVENT_TABLE_NAME = os.environ["TRANSCRIBER_CALL_EVENT_TABLE_NAME"]
DYNAMODB_EXPIRATION_IN_DAYS = getenv("DYNAMODB_EXPIRATION_IN_DAYS", "90")

kdsClient = boto3.client('kinesis')
dynamoClient = boto3.resource('dynamodb')
dynamoTable = dynamoClient.Table(TRANSCRIBER_CALL_EVENT_TABLE_NAME)

voiceTaskCache = {}
callDetailCache = {}

def get_ttl():
    return int((datetime.utcnow() + timedelta(days=int(DYNAMODB_EXPIRATION_IN_DAYS))).timestamp())

def get_call_record(callId):
    if callId in callDetailCache:
        return callDetailCache[callId]
    
    pk = "cd#" + callId
    sk = "BOTH"
    response = dynamoTable.get_item(
        Key = {
            'PK': pk,
            'SK': sk
        }
    )
    callDetailCache[callId] = response['Item']
    return response['Item']

def get_callId_for_voiceTask(voiceToneAnalysisTaskId):
    if voiceToneAnalysisTaskId in voiceTaskCache:
        return voiceTaskCache[voiceToneAnalysisTaskId]
    
    pk = "vta#" + voiceToneAnalysisTaskId
    response = dynamoTable.get_item(
        Key = {
            'PK': pk,
            'SK': 'VTA'
        }
    )
    
    callId = response['Item']['CallId']
    voiceTaskCache[voiceToneAnalysisTaskId] = callId
    
    return callId

def lambda_handler(event, context):
    print("Received call analytics event")
    print(json.dumps(event))
    detail = event['detail']

    if detail['detailStatus'] == 'VoiceToneAnalysisSuccessful':
        print("Received VoiceToneAnalysisSuccessful event")
        callId = get_callId_for_voiceTask(detail['taskId'])
        callRecord = get_call_record(get_callId_for_voiceTask(detail['taskId']))
        callData = json.loads(callRecord['CallData'])
        
        callStartTimeStr = callData['callStreamingStartTime']
        callStartTime = datetime.strptime(callStartTimeStr,'%Y-%m-%dT%H:%M:%S.%fZ')
        
        timestampStr = datetime.now().strftime('%Y-%m-%dT%H:%M:%SZ')
        
        participant = 'CALLER_VOICETONE' if detail['isCaller'] != True else 'AGENT_VOICETONE'
        sentiment = detail['voiceToneAnalysisDetails']['currentAverageVoiceTone']['voiceToneLabel'].upper()
        sentimentWeighted = detail['voiceToneAnalysisDetails']['currentAverageVoiceTone']['voiceToneScore']['positive'] * 1 + detail['voiceToneAnalysisDetails']['currentAverageVoiceTone']['voiceToneScore']['negative'] * -1

        sentimentScore = {
            'Positive': detail['voiceToneAnalysisDetails']['currentAverageVoiceTone']['voiceToneScore']['positive'],
            'Negative': detail['voiceToneAnalysisDetails']['currentAverageVoiceTone']['voiceToneScore']['negative'],
            'Neutral': detail['voiceToneAnalysisDetails']['currentAverageVoiceTone']['voiceToneScore']['neutral'],
            'Mixed': 0
        }

        segmentStartTimeStr = detail['voiceToneAnalysisDetails']['currentAverageVoiceTone']['startTime']
        segmentStartTime = datetime.strptime(segmentStartTimeStr,'%Y-%m-%dT%H:%M:%S.%fZ')
        
        segmentEndTimeStr = detail['voiceToneAnalysisDetails']['currentAverageVoiceTone']['endTime']
        segmentEndTime = datetime.strptime(segmentEndTimeStr,'%Y-%m-%dT%H:%M:%S.%fZ')

        endMillis = (segmentEndTime - callStartTime).total_seconds() * 1000
        #startMillis = (segmentStartTime - callStartTime).total_seconds() * 1000
        startMillis = endMillis - 5000
        
        putObj = {
            'EventType': 'ADD_TRANSCRIPT_SEGMENT',
            'CallId': callId,
            'UtteranceEvent': {
                'UtteranceId': event['id'][3:],
                'ParticipantRole': participant,
                'isPartial': False,
                'Transcript':'voice tone',
                'Sentiment': sentiment,
                'SentimentWeighted': sentimentWeighted,
                'SentimentScore': sentimentScore,
                'BeginOffsetMillis': startMillis,
                'EndOffsetMillis': endMillis,
            },
            'CreatedAt': timestampStr,
            'UpdatedAt': timestampStr
        }
        
        print(json.dumps(putObj))
        
        response = kdsClient.put_record(
            StreamName=KINESIS_STREAM_NAME,
            Data=json.dumps(putObj),
            PartitionKey=callId,
        )
        
    return