#!/usr/bin/env python
#
#    NEPI, a framework to manage network experiments
#    Copyright (C) 2013 INRIA
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License version 2 as
#    published by the Free Software Foundation;
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# Author: Alina Quereilhac <alina.quereilhac@inria.fr>

# This very simple experiment will ssh in a linux box, and from there
# issue a ping command towards a landmark (default faraday.inria.fr)
#
# $ ping.py -u root mybox.domain.com
# 
# would do the equivalent of
# ssh root@mybox.domain.com ping -c3 faraday.inria.fr
#
# If nepi is not installed in your system, and you only have a git clone
# you might wish to do instead
#  
# $ cd <path-to-nepi>/src
# $ python ../examples/linux/ping.py -u root mybox.domain.com

# let's be ready for python3
from __future__ import print_function

import os
from argparse import ArgumentParser

from nepi.execution.ec import ExperimentController 

# 
default_landmark = "faraday.inria.fr"

parser = ArgumentParser()
parser.add_argument("-u", "--username", dest="username", 
                    help="Username to SSH to remote host")
parser.add_argument("-i", "--ssh-key", dest="ssh_key", 
                    help="Path to private SSH key to be used for connection")
parser.add_argument("-l", "--landmark", dest='landmark', default=default_landmark,
                    help="Set pings destination, default={}".format(default_landmark))
# this is required
parser.add_argument("hostname", type=str)

args = parser.parse_args()

hostname = args.hostname
username = args.username
ssh_key = args.ssh_key
landmark = args.landmark

ec = ExperimentController(exp_id = "ping-exp")
        
node = ec.register_resource("linux::Node")
ec.set(node, "hostname", hostname)
ec.set(node, "username", username)
ec.set(node, "identity", ssh_key)
ec.set(node, "cleanExperiment", True)
ec.set(node, "cleanProcesses", True)

app = ec.register_resource("linux::Application")
ec.set(app, "command", "ping -c3 {}".format(landmark))
ec.register_connection(app, node)

ec.deploy()

ec.wait_finished(app)

print(ec.trace(app, "stdout"))

ec.shutdown()