# Copyright 2022 highstreet technologies GmbH
#
# 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.

#!/usr/bin/python
"""
Module containing a class representing a SMO as TAPI Node.
"""
from typing import List
from lxml import etree
from model.python.tapi_node import TapiNode
from model.python.tapi_node_edge_point import TapiNodeEdgePoint


class TapiNodeSmo(TapiNode):
    """
    Class representing a SMO as TAPI Node
    """
    __width: 0

    # constructor
    def __init__(self, parent: TapiNode, config):
        super().__init__(parent, config)

        # add O2 consumer interface
        nep_configuration = {
            "parent": self.identifier(),
            "nodeEdgePoint": {
                "interface": "o2", "cep": [{"protocol": "REST", "role": "consumer"}]
            }
        }
        self.add(TapiNodeEdgePoint(nep_configuration))

        # add A1 consumer interface
        nep_configuration = {
            "parent": self.identifier(),
            "nodeEdgePoint": {
                "interface": "a1", "cep": [{"protocol": "REST", "role": "consumer"}]
            }
        }
        self.add(TapiNodeEdgePoint(nep_configuration))

        # add O1/OAM NetConf Consumer interface
        nep_configuration = {
            "parent": self.identifier(),
            "nodeEdgePoint": {
                "interface": "oam", "cep": [{"protocol": "NETCONF", "role": "consumer"}]
            }
        }
        self.add(TapiNodeEdgePoint(nep_configuration))

        # add O1 VES Provider interface
        nep_configuration = {
            "parent": self.identifier(),
            "nodeEdgePoint": {
                "interface": "o1", "cep": [{"protocol": "VES", "role": "provider"}]
            }
        }
        self.add(TapiNodeEdgePoint(nep_configuration))

        # add O1 File Transfer Consumer interface
        nep_configuration = {
            "parent": self.identifier(),
            "nodeEdgePoint": {
                "interface": "o1", "cep": [{"protocol": "FILE", "role": "consumer"}]
            }
        }
        self.add(TapiNodeEdgePoint(nep_configuration))

    def __smo_component(self, x: int, y: int, label: str) -> etree.Element:
        group = etree.Element("g")
        group.attrib["class"] = " ".join(["node", label])

        width = (2 + 2) * (2.2*self.FONTSIZE)
        height = 2 * (2*self.FONTSIZE)

        rect = etree.Element("rect")
        rect.attrib["x"] = str(int(x - width/2))
        rect.attrib["y"] = str(int(y - height/2))
        rect.attrib["width"] = str(int(width))
        rect.attrib["height"] = str(int(height))
        rect.attrib["rx"] = str(self.FONTSIZE)
        rect.attrib["class"] = " ".join(["node", label])
        group.append(rect)

        labelElement = etree.Element('text')
        labelElement.attrib['x'] = str(x)
        # +4px for font-size 14px (think of chars like 'gjy')
        labelElement.attrib['y'] = str(y + 4)
        labelElement.attrib['class'] = " ".join(["node", label])
        labelElement.text = label.upper()
        group.append(labelElement)
        return group

    def svg(self, x: int, y: int) -> etree.Element:
        """
        Getter for a xml Element object representing the TAPI Node.
        :return TAPI Node as svg object.
        """
        super().svg(x, y)

        components = ["o2-controller", "non-rt-ric", "oam-controller",
                      "ves-collector", "file-server"]

        group = etree.Element("g")
        group.attrib["class"] = "node"
        title = etree.Element("title")
        title.text = "\n TAPI Node\n id: " + \
            self.identifier() + "\n name: " + self.name()
        group.append(title)

        width = (len(components)*5 +1) * (2.2*self.FONTSIZE)
        height = 2 * (2.2*self.FONTSIZE)

        rect = etree.Element("rect")
        rect.attrib["x"] = str(int(x - width/2))
        rect.attrib["y"] = str(int(y - height/2))
        rect.attrib["width"] = str(int(width))
        rect.attrib["height"] = str(int(height))
        rect.attrib["rx"] = str(self.FONTSIZE)
        rect.attrib["class"] = " ".join(
            ["node", self.function_label().lower()])
        group.append(rect)

        label = etree.Element('text')
        label.attrib['x'] = str(x)
        # +4px for font-size 14px (think of chars like 'gjy')
        label.attrib['y'] = str(y + 4)
        label.attrib['class'] = " ".join(
            ["node", self.function_label().lower()])
        label.text = self.function_label()
        group.append(label)

        for component in components:
            x_mapping = {
                "o2-controller": -4*6*self.FONTSIZE,
                "non-rt-ric": -2*6*self.FONTSIZE,
                "oam-controller": -0*6*self.FONTSIZE,
                "ves-collector": +2*6*self.FONTSIZE,
                "file-server": +4*6*self.FONTSIZE
            }
            comp_x = x + x_mapping[component]
            comp_y = y-0*self.FONTSIZE
            group.append(self.__smo_component(comp_x, comp_y, component))

        for nep in self.data()['owned-node-edge-point']:
            nep_x = x + \
                super().x_offset_by_cep_name(
                    nep.connection_edge_points()[0].name(), 0)
            nep_y = y + \
                super().y_offset_by_cep_name(
                    nep.connection_edge_points()[0].name())
            group.append(nep.svg(nep_x, nep_y))

        return group