#!/usr/bin/env python
# -*- coding: utf-8 -*-

#
# Experiment Topology:
#
#  n1 --- n2 --- n3
# 0.1  0.2  1.1  1.2
#    

from nepi.core.design import ExperimentDescription, FactoriesProvider
from nepi.core.execute import ExperimentController
import getpass
import tempfile
import time

user = getpass.getuser()
root_dir = tempfile.mkdtemp()

exp_desc = ExperimentDescription()

testbed_id = "netns"
netns_provider = FactoriesProvider(testbed_id)
netns_desc = exp_desc.add_testbed_description(netns_provider)
netns_desc.set_attribute_value("homeDirectory", root_dir)
#netns_desc.set_attribute_value("enableDebug", True)

node1 = netns_desc.create("Node")
node2 = netns_desc.create("Node")
node3 = netns_desc.create("Node")

iface12 = netns_desc.create("P2PNodeInterface")
iface12.set_attribute_value("up", True)
node1.connector("devs").connect(iface12.connector("node"))

iface21 = netns_desc.create("P2PNodeInterface")
iface21.set_attribute_value("up", True)
node2.connector("devs").connect(iface21.connector("node"))

iface23 = netns_desc.create("P2PNodeInterface")
iface23.set_attribute_value("up", True)
node2.connector("devs").connect(iface23.connector("node"))

iface32 = netns_desc.create("P2PNodeInterface")
iface32.set_attribute_value("up", True)
node3.connector("devs").connect(iface32.connector("node"))

iface12.connector("p2p").connect(iface21.connector("p2p"))
iface23.connector("p2p").connect(iface32.connector("p2p"))

ip12 = iface12.add_address()
ip12.set_attribute_value("Address", "192.168.0.1")
ip12.set_attribute_value("NetPrefix", 30)

ip21 = iface21.add_address()
ip21.set_attribute_value("Address", "192.168.0.2")
ip21.set_attribute_value("NetPrefix", 30)

ip23 = iface23.add_address()
ip23.set_attribute_value("Address", "192.168.1.1")
ip23.set_attribute_value("NetPrefix", 30)

ip32 = iface32.add_address()
ip32.set_attribute_value("Address", "192.168.1.2")
ip32.set_attribute_value("NetPrefix", 30)

route1 = node1.add_route()
route1.set_attribute_value("Destination", "192.168.1.0")
route1.set_attribute_value("NetPrefix", 30)
route1.set_attribute_value("NextHop", "192.168.0.2")

route2 = node3.add_route()
route2.set_attribute_value("Destination", "192.168.0.0")
route2.set_attribute_value("NetPrefix", 30)
route2.set_attribute_value("NextHop", "192.168.1.1")

app1 = netns_desc.create("Application")
app1.set_attribute_value("command", "ping -qc 3 192.168.1.2")
app1.set_attribute_value("user", user)
app1.connector("node").connect(node1.connector("apps"))

app1.enable_trace("stdout")

app2 = netns_desc.create("Application")
app2.set_attribute_value("command", "ping -qc 3 192.168.0.1")
app2.set_attribute_value("user", user)
app2.connector("node").connect(node3.connector("apps"))

app2.enable_trace("stdout")

xml = exp_desc.to_xml()

controller = ExperimentController(xml, root_dir)
controller.start()
while not (controller.is_finished(app1.guid) and \
        controller.is_finished(app2.guid)):
    time.sleep(0.5)

result1 = controller.trace(app1.guid, "stdout")
result2 = controller.trace(app2.guid, "stdout")

controller.stop()
controller.shutdown()

print result1
print result2