#

# Copyright (c) 2020 Red Hat, Inc.

#

# This file is part of nmstate

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU Lesser General Public License as published by

# the Free Software Foundation, either version 2.1 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Lesser General Public License for more details.

#

# You should have received a copy of the GNU Lesser General Public License

# along with this program. If not, see <https://www.gnu.org/licenses/>.

#

import logging

from libnmstate.schema import InfiniBand

from libnmstate.schema import Interface

from libnmstate.schema import InterfaceType

from .common import NM

_NM_IB_MODE_DATAGRAM = "datagram"

_NM_IB_MODE_CONNECTED = "connected"

def get_infiniband_config(nm_profile):

    if nm_profile:

        ib_setting = nm_profile.get_setting_infiniband()

        if ib_setting:

            mode = _nm_ib_mode_to_nmstate(ib_setting.get_transport_mode())

            if not mode:

                logging.warning(

                    "Unknown InfiniBand transport mode "

                    f"{ib_setting.get_transport_mode()} for interface "

                    f"{nm_profile.get_interface_name()}"

                )

                return {}

            pkey = ib_setting.get_p_key()

            if pkey == -1:

                pkey = str(InfiniBand.DEFAULT_PKEY)

            else:

                pkey = hex(pkey)

            base_iface = ib_setting.get_parent()

            if base_iface is None:

                base_iface = ""

            return {

                Interface.TYPE: InterfaceType.INFINIBAND,

                InfiniBand.CONFIG_SUBTREE: {

                    InfiniBand.PKEY: pkey,

                    InfiniBand.MODE: mode,

                    InfiniBand.BASE_IFACE: base_iface,

                },

            }

    return {}

def _nm_ib_mode_to_nmstate(nm_ib_mode):

    if nm_ib_mode == _NM_IB_MODE_DATAGRAM:

        return InfiniBand.Mode.DATAGRAM

    elif nm_ib_mode == _NM_IB_MODE_CONNECTED:

        return InfiniBand.Mode.CONNECTED

    else:

        return None

def create_setting(iface_info, base_con_profile, original_iface_info):

    ib_config = iface_info.get(InfiniBand.CONFIG_SUBTREE)

    if not ib_config:

        return None

    ib_setting = None

    if base_con_profile:

        ib_setting = base_con_profile.get_setting_infiniband()

        if ib_setting:

            ib_setting = ib_setting.duplicate()

    if not ib_setting:

        ib_setting = NM.SettingInfiniband.new()

    if Interface.MTU in original_iface_info:

        ib_setting.props.mtu = original_iface_info[Interface.MTU]

    if Interface.MAC in original_iface_info:

        ib_setting.props.mac_address = original_iface_info[Interface.MAC]

    if ib_config[InfiniBand.PKEY] == InfiniBand.DEFAULT_PKEY:

        ib_setting.props.p_key = -1

    else:

        ib_setting.props.p_key = ib_config[InfiniBand.PKEY]

    ib_setting.props.transport_mode = ib_config[InfiniBand.MODE]

    if InfiniBand.BASE_IFACE in ib_config:

        if ib_config[InfiniBand.BASE_IFACE]:

            ib_setting.props.parent = ib_config[InfiniBand.BASE_IFACE]

        else:

            ib_setting.props.parent = None

    return ib_setting