# Copyright (c) 2024 Cisco and/or its affiliates.
# 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.

"""IP Topology Library."""

from robot.libraries.BuiltIn import BuiltIn

from resources.libraries.python.IPUtil import IPUtil


class IPTopology:
    """IP Topology Library."""

    @staticmethod
    def initialize_ipv4_forwarding(count=1, pfs=2, route_prefix=32):
        """
        Custom setup of IPv4 forwarding with scalability of IP routes on all
        DUT nodes in 2-node / 3-node circular topology.

        :param count: Number of routes to configure.
        :param pfs: Number of physical interfaces to configure.
        :param route_prefix: Route prefix to configure.
        :type count: int
        :type pfs: int
        :type route_prefix: int
        """
        topology = BuiltIn().get_variable_value("&{topology_info}")
        dut = topology["duts"][-1]
        ifl = BuiltIn().get_variable_value("${int}")

        for l, i in zip(range(pfs // 2), range(1, pfs, 2)):
            dut1_int1 = BuiltIn().get_variable_value(f"${{DUT1_{ifl}{i}}}[0]")
            dut1_int2 = BuiltIn().get_variable_value(f"${{DUT1_{ifl}{i+1}}}[0]")
            dut_int1 = BuiltIn().get_variable_value(f"${{{dut}_{ifl}{i}}}[0]")
            dut_int2 = BuiltIn().get_variable_value(f"${{{dut}_{ifl}{i+1}}}[0]")

            IPUtil.vpp_add_ip_neighbor(
                topology["DUT1"], dut1_int1, f"1.{l}.1.1",
                topology[f"TG_pf{i}_mac"][0]
            )
            if dut == "DUT2":
                dut_mac1 = BuiltIn().get_variable_value(
                    f"${{{dut}_{ifl}{i}_mac}}[0]"
                )
                IPUtil.vpp_add_ip_neighbor(
                    topology["DUT1"], dut1_int2, f"3.{l}.3.2", dut_mac1
                )
                dut_mac2 = BuiltIn().get_variable_value(
                    f"${{DUT1_{ifl}{i+1}_mac}}[0]"
                )
                IPUtil.vpp_add_ip_neighbor(
                    topology["DUT2"], dut_int1, f"3.{l}.3.1", dut_mac2
                )
            IPUtil.vpp_add_ip_neighbor(
                topology[dut], dut_int2, f"2.{l}.2.1",
                topology[f"TG_pf{i+1}_mac"][0]
            )

            IPUtil.vpp_interface_set_ip_address(
                topology["DUT1"], dut1_int1, f"1.{l}.1.2", 30
            )
            if dut == "DUT2":
                IPUtil.vpp_interface_set_ip_address(
                    topology["DUT1"], dut1_int2, f"3.{l}.3.1", 30
                )
                IPUtil.vpp_interface_set_ip_address(
                    topology["DUT2"], dut_int1, f"3.{l}.3.2", 30
                )
            IPUtil.vpp_interface_set_ip_address(
                topology[dut], dut_int2, f"2.{l}.2.2", 30
            )

            IPUtil.vpp_route_add(
                topology["DUT1"], f"{i}0.0.0.0", route_prefix,
                gateway=f"1.{l}.1.1", interface=dut1_int1, count=count
            )
            if dut == "DUT2":
                IPUtil.vpp_route_add(
                    topology["DUT1"], f"{i+1}0.0.0.0", route_prefix,
                    gateway=f"3.{l}.3.2", interface=dut1_int2, count=count
                )
                IPUtil.vpp_route_add(
                    topology["DUT2"], f"{i}0.0.0.0", route_prefix,
                    gateway=f"3.{l}.3.1", interface=dut_int1, count=count
                )
            IPUtil.vpp_route_add(
                topology[dut], f"{i+1}0.0.0.0", route_prefix,
                gateway=f"2.{l}.2.1", interface=dut_int2, count=count
            )


    @staticmethod
    def initialize_ipv6_forwarding(count=1, pfs=2, route_prefix=128):
        """
        Custom setup of IPv6 forwarding with scalability of IP routes on all
        DUT nodes in 2-node / 3-node circular topology.

        :param count: Number of routes to configure.
        :param pfs: Number of physical interfaces to configure.
        :param route_prefix: Route prefix to configure.
        :type count: int
        :type pfs: int
        :type route_prefix: int
        """
        topology = BuiltIn().get_variable_value("&{topology_info}")
        dut = topology["duts"][-1]
        ifl = BuiltIn().get_variable_value("${int}")

        for l, i in zip(range(pfs // 2), range(1, pfs, 2)):
            dut1_int1 = BuiltIn().get_variable_value(f"${{DUT1_{ifl}{i}}}[0]")
            dut1_int2 = BuiltIn().get_variable_value(f"${{DUT1_{ifl}{i+1}}}[0]")
            dut_int1 = BuiltIn().get_variable_value(f"${{{dut}_{ifl}{i}}}[0]")
            dut_int2 = BuiltIn().get_variable_value(f"${{{dut}_{ifl}{i+1}}}[0]")

            IPUtil.vpp_add_ip_neighbor(
                topology["DUT1"], dut1_int1, f"2001:{l}::1",
                topology[f"TG_pf{i}_mac"][0]
            )
            if dut == "DUT2":
                dut_mac1 = BuiltIn().get_variable_value(
                    f"${{{dut}_{ifl}{i}_mac}}[0]"
                )
                IPUtil.vpp_add_ip_neighbor(
                    topology["DUT1"], dut1_int2, f"2003:{l}::2", dut_mac1
                )
                dut_mac2 = BuiltIn().get_variable_value(
                    f"${{DUT1_{ifl}{i+1}_mac}}[0]"
                )
                IPUtil.vpp_add_ip_neighbor(
                    topology["DUT2"], dut_int1, f"2003:{l}::1", dut_mac2
                )
            IPUtil.vpp_add_ip_neighbor(
                topology[dut], dut_int2, f"2002:{l}::1",
                topology[f"TG_pf{i+1}_mac"][0]
            )

            IPUtil.vpp_interface_set_ip_address(
                topology["DUT1"], dut1_int1, f"2001:{l}::2", 64
            )
            if dut == "DUT2":
                IPUtil.vpp_interface_set_ip_address(
                    topology["DUT1"], dut1_int2, f"2003:{l}::1", 64
                )
                IPUtil.vpp_interface_set_ip_address(
                    topology["DUT2"], dut_int1, f"2003:{l}::2", 64
                )
            IPUtil.vpp_interface_set_ip_address(
                topology[dut], dut_int2, f"2002:{l}::2", 64
            )

            IPUtil.vpp_route_add(
                topology["DUT1"], f"2{i}00::0", route_prefix,
                gateway=f"2001:{l}::1", interface=dut1_int1, count=count
            )
            if dut == "DUT2":
                IPUtil.vpp_route_add(
                    topology["DUT1"], f"2{i+1}00::0", route_prefix,
                    gateway=f"2003:{l}::2", interface=dut1_int2, count=count
                )
                IPUtil.vpp_route_add(
                    topology["DUT2"], f"2{i}00::0", route_prefix,
                    gateway=f"2003:{l}::1", interface=dut_int1, count=count
                )
            IPUtil.vpp_route_add(
                topology[dut], f"2{i+1}00::0", route_prefix,
                gateway=f"2002:{l}::1", interface=dut_int2, count=count
            )