from core.topo import Topo

class MpTwoInterfaceCongestionTopo(Topo):
    def __init__(self, topoBuilder, parameterFile):
        super().__init__(topoBuilder, parameterFile)

        print("Hello from topo two ifs cong")
        print("Expected topo:")
        print("c1----link0--------------|")
        print("|-------------r1--link1--r2-----s1")
        print("              |          |------s2")
        print("c2----link2----")

        self.client = self.addHost(Topo.clientName)
        self.clientCong = self.addHost(Topo.clientName + "Cong")
        self.server = self.addHost(Topo.serverName)
        self.serverCong = self.addHost(Topo.serverName + "Cong")
        self.router = self.addHost(Topo.routerName)
        self.routerCong = self.addHost(Topo.routerName + "Cong")
        self.switch = []

        # Link between c1 and r2
        self.switch.append(self.addOneSwitchPerLink(self.topoParam.linkCharacteristics[0]))
        self.addLink(self.client, self.switch[-1])
        self.addLink(self.switch[-1], self.router, **self.topoParam.linkCharacteristics[0].asDict())

        # Link between c1 and r1
        self.addLink(self.client, self.routerCong)

        # Link between c2 and r1
        self.switch.append(self.addOneSwitchPerLink(self.topoParam.linkCharacteristics[2]))
        self.addLink(self.clientCong, self.switch[-1])
        self.addLink(self.switch[-1], self.routerCong, **self.topoParam.linkCharacteristics[2].asDict())

        # Link between r1 and r2
        self.switch.append(self.addOneSwitchPerLink(self.topoParam.linkCharacteristics[1]))
        self.addLink(self.routerCong, self.switch[-1])
        self.addLink(self.switch[-1], self.router, **self.topoParam.linkCharacteristics[1].asDict())

        # Link between r2 and s1
        self.addLink(self.router, self.server)

        # Link between r2 and s2
        self.addLink(self.router, self.serverCong)

    def __str__(self):
        s = "Hello from topo two ifs cong \n"
        s = s + "c1----link0--------------| \n"
        s = s + "|-------------r1--link1--r2-----s1 \n"
        s = s + "              |          |------s2 \n"
        s = s + "c2----link2---- \n"
        return s

    def addOneSwitchPerLink(self, link):
        return self.addSwitch(Topo.switchNamePrefix + str(link.id))