#! /usr/bin/python
# Copyright 2019 Nokia
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License 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 cmframework.apis import cmuserconfig
from cmframework.apis import cmerror
from cmdatahandlers.api import configerror
from cmdatahandlers.network_profiles.config import config as net_prof_config

"""
This plugin is used to handle REC specific network profiles configs. Currently
its sole purpuse is to set default type for any provider networks.
"""


class recnetworkprofileshandler(cmuserconfig.CMUserConfigPlugin):
    defaul_sriov_net_type = 'caas'
    defaul_provider_interface_type = 'caas'

    def __init__(self):
        super(recnetworkprofileshandler, self).__init__()

    def handle(self, confman):
        try:
            self._set_default_type_for_provider_networks(confman)
        except configerror.ConfigError as exp:
            raise cmerror.CMError(str(exp))

    def _set_default_type_for_provider_networks(self, confman):
        netprofconf = confman.get_network_profiles_config_handler()
        network_profiles = netprofconf.get_network_profiles()
        for profile in network_profiles:
            try:
                self._set_default_provider_network_type(netprofconf, profile)
                self._set_default_sriov_provider_network_type(netprofconf, profile)
            except (net_prof_config.MissingProviderNetworkInterfaces, net_prof_config.MissingSriovProviderNetworks):
                pass

    def _set_default_provider_network_type(self, netprofconf, profile):
        for interface in netprofconf.get_profile_provider_network_interfaces(profile):
            try:
                netprofconf.get_profile_provider_network_interface_type(
                    profile, interface)
            except net_prof_config.MissingProviderNetworkInterfaceType:
                netprofconf.set_profile_provider_network_interface_type(
                        profile, interface, self.defaul_provider_interface_type)

    def _set_default_sriov_provider_network_type(self, netprofconf, profile):
        for sriov_net in netprofconf.get_profile_sriov_provider_networks(profile):
            if not netprofconf.get_profile_sriov_provider_network_type(
                    profile, sriov_net):
                netprofconf.set_profile_sriov_provider_network_type(
                        profile, sriov_net, self.defaul_sriov_net_type)