# Copyright 2017 VMware, Inc.
# 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. 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 oslo_config import cfg

from vmware_nsx._i18n import _
from vmware_nsx.common import availability_zones as common_az
from vmware_nsx.common import config
from vmware_nsx.common import exceptions as nsx_exc


DEFAULT_NAME = common_az.DEFAULT_NAME


class NsxV3AvailabilityZone(common_az.ConfiguredAvailabilityZone):

    def init_from_config_line(self, config_line):
        # Not supported for nsx_v3 (old configuration)
        raise nsx_exc.NsxInvalidConfiguration(
            opt_name="availability_zones",
            opt_value=config_line,
            reason=_("Expected a list of names"))

    def init_from_config_section(self, az_name):
        az_info = config.get_nsxv3_az_opts(self.name)

        # The optional parameters will get the global values if not
        # defined for this AZ
        self.metadata_proxy = az_info.get('metadata_proxy')
        if not self.metadata_proxy:
            raise nsx_exc.NsxInvalidConfiguration(
                opt_name="metadata_proxy",
                opt_value='None',
                reason=(_("metadata_proxy for availability zone %s "
                          "must be defined") % az_name))

        self.dhcp_profile = az_info.get('dhcp_profile')
        if not self.dhcp_profile:
            raise nsx_exc.NsxInvalidConfiguration(
                opt_name="dhcp_profile",
                opt_value='None',
                reason=(_("dhcp_profile for availability zone %s "
                          "must be defined") % az_name))

        self.native_metadata_route = az_info.get('native_metadata_route')
        if self.native_metadata_route is None:
            self.native_metadata_route = cfg.CONF.nsx_v3.native_metadata_route

        self.dns_domain = az_info.get('dns_domain')
        if self.dns_domain is None:
            self.dns_domain = cfg.CONF.nsx_v3.dns_domain

        self.nameservers = az_info.get('nameservers')
        if self.nameservers is None:
            self.nameservers = cfg.CONF.nsx_v3.nameservers

        self.default_overlay_tz = az_info.get('default_overlay_tz')
        if self.default_overlay_tz is None:
            self.default_overlay_tz = cfg.CONF.nsx_v3.default_overlay_tz

        self.default_vlan_tz = az_info.get('default_vlan_tz')
        if self.default_vlan_tz is None:
            self.default_vlan_tz = cfg.CONF.nsx_v3.default_vlan_tz

    def init_default_az(self):
        # use the default configuration
        self.metadata_proxy = cfg.CONF.nsx_v3.metadata_proxy
        self.dhcp_profile = cfg.CONF.nsx_v3.dhcp_profile
        self.native_metadata_route = cfg.CONF.nsx_v3.native_metadata_route
        self.dns_domain = cfg.CONF.nsx_v3.dns_domain
        self.nameservers = cfg.CONF.nsx_v3.nameservers
        self.default_overlay_tz = cfg.CONF.nsx_v3.default_overlay_tz
        self.default_vlan_tz = cfg.CONF.nsx_v3.default_vlan_tz

    def translate_configured_names_to_uuids(self, nsxlib):
        # Mandatory configurations (in AZ or inherited from global values)
        # Unless this is the default AZ, and metadata is disabled.
        if self.dhcp_profile:
            dhcp_id = nsxlib.native_dhcp_profile.get_id_by_name_or_id(
                self.dhcp_profile)
            self._native_dhcp_profile_uuid = dhcp_id
        else:
            self._native_dhcp_profile_uuid = None

        if self.metadata_proxy:
            proxy_id = nsxlib.native_md_proxy.get_id_by_name_or_id(
                self.metadata_proxy)
            self._native_md_proxy_uuid = proxy_id
        else:
            self._native_md_proxy_uuid = None

        if self.default_overlay_tz:
            tz_id = nsxlib.transport_zone.get_id_by_name_or_id(
                self.default_overlay_tz)
            self._default_overlay_tz_uuid = tz_id
        else:
            self._default_overlay_tz_uuid = None

        # Optional configurations (may be None)
        if self.default_vlan_tz:
            tz_id = nsxlib.transport_zone.get_id_by_name_or_id(
                self.default_vlan_tz)
            self._default_vlan_tz_uuid = tz_id
        else:
            self._default_vlan_tz_uuid = None


class NsxV3AvailabilityZones(common_az.ConfiguredAvailabilityZones):

    def __init__(self):
        super(NsxV3AvailabilityZones, self).__init__(
            cfg.CONF.nsx_v3.availability_zones,
            NsxV3AvailabilityZone)