# Copyright 2014 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"). You
# may not use this file except in compliance with the License. A copy of
# the License is located at
#
#     http://aws.amazon.com/apache2.0/
#
# or in the "license" file accompanying this file. This file is
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
# ANY KIND, either express or implied. See the License for the specific
# language governing permissions and limitations under the License.

from awscli.customizations.emr import constants, exceptions


def validate_and_build_instance_fleets(parsed_instance_fleets):
    """
    Helper method that converts --instance-fleets option value in
    create-cluster to Amazon Elastic MapReduce InstanceFleetConfig
    data type.
    """
    instance_fleets = []
    for instance_fleet in parsed_instance_fleets:
        instance_fleet_config = {}

        keys = instance_fleet.keys()

        if 'Name' in keys:
            instance_fleet_config['Name'] = instance_fleet['Name']
        else:
            instance_fleet_config['Name'] = instance_fleet['InstanceFleetType']
        instance_fleet_config['InstanceFleetType'] = instance_fleet[
            'InstanceFleetType'
        ]

        if 'TargetOnDemandCapacity' in keys:
            instance_fleet_config['TargetOnDemandCapacity'] = instance_fleet[
                'TargetOnDemandCapacity'
            ]

        if 'TargetSpotCapacity' in keys:
            instance_fleet_config['TargetSpotCapacity'] = instance_fleet[
                'TargetSpotCapacity'
            ]

        if 'InstanceTypeConfigs' in keys:
            instance_fleet_config['InstanceTypeConfigs'] = instance_fleet[
                'InstanceTypeConfigs'
            ]

        if 'LaunchSpecifications' in keys:
            instanceFleetProvisioningSpecifications = instance_fleet[
                'LaunchSpecifications'
            ]
            instance_fleet_config['LaunchSpecifications'] = {}

            if 'SpotSpecification' in instanceFleetProvisioningSpecifications:
                instance_fleet_config['LaunchSpecifications'][
                    'SpotSpecification'
                ] = instanceFleetProvisioningSpecifications[
                    'SpotSpecification'
                ]

            if (
                'OnDemandSpecification'
                in instanceFleetProvisioningSpecifications
            ):
                instance_fleet_config['LaunchSpecifications'][
                    'OnDemandSpecification'
                ] = instanceFleetProvisioningSpecifications[
                    'OnDemandSpecification'
                ]

        if 'ResizeSpecifications' in keys:
            instanceFleetResizeSpecifications = instance_fleet[
                'ResizeSpecifications'
            ]
            instance_fleet_config['ResizeSpecifications'] = {}

            if 'SpotResizeSpecification' in instanceFleetResizeSpecifications:
                instance_fleet_config['ResizeSpecifications'][
                    'SpotResizeSpecification'
                ] = instanceFleetResizeSpecifications[
                    'SpotResizeSpecification'
                ]

            if (
                'OnDemandResizeSpecification'
                in instanceFleetResizeSpecifications
            ):
                instance_fleet_config['ResizeSpecifications'][
                    'OnDemandResizeSpecification'
                ] = instanceFleetResizeSpecifications[
                    'OnDemandResizeSpecification'
                ]

        if 'Context' in keys:
            instance_fleet_config['Context'] = instance_fleet['Context']

        instance_fleets.append(instance_fleet_config)
    return instance_fleets
