# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0
import os
import json
import uuid
from urllib.request import Request, urlopen
from datetime import datetime
import boto3
from botocore.exceptions import ClientError
import awsapi_cached_client

class Metrics(object):

    event_type = ''
    send_metrics_option = 'No'
    solution_version = ''
    solution_uuid = None
    session = None
    region = None
    ssm_client = None
    metrics_parameter_name = '/Solutions/SO0111/anonymous_metrics_uuid'

    def __init__(self, event_type):
        self.session = boto3.session.Session()
        self.region = self.session.region_name

        self.ssm_client = self.connect_to_ssm()

        if not self.send_anonymous_metrics_enabled():
            return

        self.event_type = event_type

        self.__get_solution_uuid()

        try:
            solution_version_parm = '/Solutions/SO0111/version'
            solution_version_from_ssm = self.ssm_client.get_parameter(
                Name=solution_version_parm
            ).get('Parameter').get('Value')
        except ClientError as ex:
            exception_type = ex.response['Error']['Code']
            if exception_type == 'ParameterNotFound':
                solution_version_from_ssm = 'unknown'
            else:
                print(ex)
        except Exception as e:
            print(e)
            raise

        self.solution_version = solution_version_from_ssm

    def send_anonymous_metrics_enabled(self):
        is_enabled = False # default value
        try:
            ssm_parm = '/Solutions/SO0111/sendAnonymousMetrics'
            send_anonymous_metrics_from_ssm = self.ssm_client.get_parameter(
                Name=ssm_parm
            ).get('Parameter').get('Value').lower()

            if send_anonymous_metrics_from_ssm != 'yes' and send_anonymous_metrics_from_ssm != 'no':
                print(f'Unexpected value for {ssm_parm}: {send_anonymous_metrics_from_ssm}. Defaulting to "no"')
            elif send_anonymous_metrics_from_ssm == 'yes':
                is_enabled = True

        except Exception as e:
            print(e)

        return is_enabled

    def connect_to_ssm(self):
        try:
            if not self.ssm_client:
                new_ssm_client = awsapi_cached_client.AWSCachedClient(self.region).get_connection('ssm')
                return new_ssm_client
        except Exception as e:
            print(f'Could not connect to ssm: {str(e)}')

    def __update_solution_uuid(self, new_uuid):
        self.ssm_client.put_parameter(
            Name=self.metrics_parameter_name,
            Description='Unique Id for anonymous metrics collection',
            Value=new_uuid,
            Type='String'
        )

    def __get_solution_uuid(self):
        try:
            solution_uuid_from_ssm = self.ssm_client.get_parameter(
                Name=self.metrics_parameter_name
            ).get('Parameter').get('Value')
            self.solution_uuid = solution_uuid_from_ssm
        except ClientError as ex:
            exception_type = ex.response['Error']['Code']
            if exception_type == 'ParameterNotFound':
                self.solution_uuid = str(uuid.uuid4())
                self.__update_solution_uuid(self.solution_uuid)
            else:
                print(ex)
                raise
        except Exception as e:
            print(e)
            raise

    def get_metrics_from_finding(self, finding):

        try:
            if finding is not None:
                metrics_data = {
                    'generator_id': finding.get('GeneratorId'),
                    'type': finding.get('Title'),
                    'productArn': finding.get('ProductArn'),
                    'finding_triggered_by': self.event_type,
                    'region': self.region
                }
            else:
                metrics_data = {}
            return metrics_data
        except Exception as excep:
            print(excep)
            return {}

    def send_metrics(self, metrics_data):

        try:
            if metrics_data is not None and self.send_anonymous_metrics_enabled():
                usage_data = {
                    'Solution': 'SO0111',
                    'UUID': self.solution_uuid,
                    'TimeStamp': str(datetime.utcnow().isoformat()),
                    'Data': metrics_data,
                    'Version': self.solution_version
                }
                print(f'Sending metrics data {json.dumps(usage_data)}')
                self.post_metrics_to_api(usage_data)

            else:
                return
        except Exception as excep:
            print(excep)

    def post_metrics_to_api(self, request_data):
        url = 'https://metrics.awssolutionsbuilder.com/generic'
        req = Request(url, method='POST', data=bytes(json.dumps(
            request_data), encoding='utf8'), headers={'Content-Type': 'application/json'})
        urlopen(req) # nosec