from core.api.grpc import client, wrappers
from core.api.grpc.wrappers import NodeType, Position

# interface helper
iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001::/64")

# create grpc client and connect
core = client.CoreGrpcClient()
core.connect()

# add session
session = core.create_session()

# create nodes
position = Position(x=200, y=200)
switch = session.add_node(1, _type=NodeType.SWITCH, position=position)
position = Position(x=100, y=100)
node1 = session.add_node(2, position=position)
position = Position(x=300, y=100)
node2 = session.add_node(3, position=position)

# create links
node1_iface1 = iface_helper.create_iface(node1.id, 0)
switch_iface1 = wrappers.Interface(0)
session.add_link(node1=node1, node2=switch, iface1=node1_iface1, iface2=switch_iface1)
node2_iface1 = iface_helper.create_iface(node2.id, 0)
switch_iface2 = wrappers.Interface(1)
session.add_link(node1=node2, node2=switch, iface1=node2_iface1, iface2=switch_iface2)

# start session
core.start_session(session)