#!/usr/bin/env python

# An example script using the rumba package

from rumba.model import *

# import testbed plugins
import rumba.testbeds.emulab as emulab
import rumba.testbeds.jfed as jfed
import rumba.testbeds.faketestbed as fake
import rumba.testbeds.qemu as qemu

# import prototype plugins
import rumba.prototypes.ouroboros as our
import rumba.prototypes.rlite as rl
import rumba.prototypes.irati as irati

import rumba.log as log


log.set_logging_level('DEBUG')

n1 = NormalDIF("n1", policies = {"rmt.pff": "lfa",
                                 "security-manager": "passwd"})

e1 = ShimEthDIF("e1")

a = Node("a",
         difs = [n1, e1],
         dif_registrations = {n1 : [e1]},
         registrations = {"a.crap" : [n1]},
         bindings = {"a.crap" : "/usr/bin/crap"})

b = Node("b",
         difs = [e1, n1],
         dif_registrations = {n1 : [e1]})

tb = qemu.Testbed(exp_name = "example1",
                  username = "root",
                  password = "root",
                  bzimage = '/home/vmaffione/git/rlite/demo/buildroot/bzImage',
                  initramfs = '/home/vmaffione/git/rlite/demo/buildroot/rootfs.cpio')

exp = rl.Experiment(tb, nodes = [a, b])

print(exp)

try:
    exp.swap_in()
    exp.bootstrap_prototype()
finally:
    exp.swap_out()