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

# kpov_random_helpers should be imported by add_assignment.py
# Poglej nek film na nekem določenem URL.
# (?md5 vsota filma?)

# Nastavi nek računalnik tako, da bo izvajal NAT.
#TODO: finish this

instructions = {
    'si':u"""
Postavi dva navidezna računalnika - SimpleArbiter z diskom simpleArbiter
ter NATServer. NATServer naj ima dva omrežna vmesnika - z enim naj bo povezan
na lokalno omrežje, na katerem naj bo tudi simpleArbiter, z drugim pa na 
Internet. Na NATServer skonfiguriraj omrežne vmesnike tako, da bo imel dostop
do Interneta in da bo imel na lokalnem omrežju {IP_NAT}. Na NATServer
ustvarite še uporabnika {IP_NAT_user}.

Poskrbi, da bo SimpleArbiter prek DHCP dobil naslov {IP_simple}.
Poskrbi, da bo NATServer deloval kot prehod za SimpleArbiter in izvajal NAT.
""",
	'en':u"""
Set up two virtual machines - SimpleArbiter (using the disc simpleArbiter) and
NATServer. NATServer should have two network adapters. Connect the first
adapter to SimpleArbiter and the second adapter to the Internet. Configure the
network in NATServer so that one interface is connected to the Internet while
the other is connected to SimpleArbiter and has the address {IP_NAT}. Create
a user called {IP_NAT_user} on NATServer.

Configure a DHCP server on NATServer so that SimpleArbiter gets the IP
{IP_simple}. Also, set up NAT on NATServer and set it as the gateway for 
SimpleArbiter.
"""
}

computers = {
    'NATServer': {
        'disks': [
            {   'name': 'student-NATServer',
                # attempt automount
            },
            #{   'name': 'CDROM',
            #    'options': {'readonly': True},
            #    'parts': [{'dev': 'b1', 'path': '/cdrom'}],
            #},
        ],
        'network_interfaces': [{'network': 'net1'}, {'network': 'test-net'}],
        'flavor': 'm1.tiny',
        'config_drive': False
    },
    'SimpleArbiter': {
        'disks': [
            {   'name': 'simpleArbiter',
            },
            #{   'name': 'CDROM',
            #    'options':{'readonly': True},
            #    'parts': [],# no parts, no mounting.
            #}
        ],
        'network_interfaces': [{'network': 'net1'}],
        'flavor': 'm1.tiny',
        'config_drive': False

    },

}

networks = { 'net1': {'public': False}, 'test-net': {'public': True} }

#ne potrebujemo dnsjev in ip za malibreznewtork manager?
params_meta = {
#    'IP_NM': {'descriptions': {'si': 'Naslov maliNetworkManager'}, 'w': False, 'public':True, 'type': 'IP', 'generated': True},
#   'DNS_NM': {'descriptions': {'si': 'DNS za maliNetworkManager'}, 'w': False, 'public':True, 'type': 'IP', 'generated': True},
   'IP_simple': {'descriptions': {'si': 'Naslov SimpleArbiter'}, 'w': False, 'public': True, 'type': 'IP', 'generated': True},
   'IP_NAT': {'descriptions': {'si': 'Naslov NAT'}, 'w': False, 'public': True,  'type': 'IP', 'generated': True },    
   'IP_NAT_user': {'descriptions': {'si': 'Username na NATServer'}, 'w': False, 'public' : True, 'type' : 'username', 'generated' : True},
   'IP_NAT_passwd': {'descriptions': {'si': 'Password na NATServer'}, 'w': True,'public' : True, 'type' : 'passwd', 'generated' : False},
#    'DNS_static': {'descriptions': {'si': 'DNS za maliBrezNetworkManager'}, 'w': False, 'public': True, 'type': 'IP', 'generated': True},
}

def task(IP_simple, IP_NAT, IP_NAT_user, IP_NAT_passwd):
    from pexpect import pxssh
    import pexpect
    results = dict()
    # Connect to NAT
    povezava = pxssh.pxssh()
    povezava.login(IP_NAT,IP_NAT_user,IP_NAT_passwd)  

    # Check if If IP_simple is connected to NAT
    results['IP_simple ping to NAT'] = run('ping -c 5 IP_NAT')
    # Check routing table on IP_simple
    results['IP_simple routing table'] = run('route -n')
    # Tracert Check if IP_simple is connected to internet
    results['IP_simple to internet'] = run('traceroute 8.8.8.8')
    # Check DNS
    results['IP_simple dns'] = run('nslookup www.google.com')
    # Check if IP_NAT ip forward is not 0
    povezava.prompt()
    povezava.sendline('cat /proc/sys/net/ipv4/ip_forward')
    povezava.prompt()
    results['IP_NAT ip forward'] = povezava.before
    povezava.logout();
    return results
    
def gen_params(user_id, params_meta):
    params = dict()
    r = random.Random(user_id)
    # IP_NM, DNS_NM, IP_static, DNS_static)
    # dns_servers = ['193.2.1.66', '193.2.1.72', '8.8.8.8', '8.8.4.4', '208.67.222.222', '208.67.220.220']
    net = kpov_random_helpers.IPv4_net_gen(r, 253, True, False)
    # params['DNS_NM'] = r.choice(dns_servers)
    params['IP_NAT'], params['IP_simple'] = kpov_random_helpers.IPv4_addr_gen(r, net, 2)
    # params['IP_NM'] 
    # params['IP_simple'] = kpov_random_helpers.IPv4_addr_gen(r, net, 1)
    # params['DNS_static'] = r.choice(dns_servers)
    return params

def task_check(results, params):
    import re
    score = 0
    hints = []
    if re.search(r'0% packet loss',results['IP_simple ping to NAT']):
	score += 2
    if results['IP_NAT ip forward']=='1':
	score += 2
    if not re.search(r'Network is unreachable',results['IP_simple to internet']):
	score += 2
    if re.search(r'94.140.66.250',results['IP_simple dns']):
	score += 2
    gateway='0.0.0.0\t'+params['IP_NAT']
    if re.search(gateway,results['IP_simple routing table']):
	score += 2		
    return score, hints

def prepare_disks(templates, task_params, global_params):
    write_default_config(templates['simpleArbiter'], global_params)