#!/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 simpleArbiter, z drugim pa na Internet. Na NATServer z ukazom ifconfig in z datoteko /etc/network/interfaces skonfiguriraj omrežne vmesnike tako da bo en vmesnik povezan v WAN (Internet) in en vmesnik na LAN (simpleArbiter). SimpleArbiter skonfiguriraj tako, da 
bo za privzeti prehod uporabljal NATServer. Nastavi DNS (etc/resolv.conf). če še ni, omogoči posredovanje IP naslovov. Nastavi NAT 
z uporabo paketa iptables. Na simpleArbiter z vlc preberi naslov, na katerem si lahko ogledaš kratek filmček.
""",
	'en':u"""
Set up two virtual machines - SimpleArbiter (using the disc simpleArbiter) and NATServer. NATServer should have two network adapters. The first is to connect to SimpleArbiter and the second is for connecting to the internet.
On NATServer use the instruction ifconfig and the file /etc/network/interfaces to configure the network adapters - the first one should be connected to WAN(Internet) and the second one to LAN(SimpleArbiter).
Configure SimpleArbiter to use NATServer as the default gateway. Setup DNS: /etc/resolv.conf, enable IP forwarding, using the iptables command. On SimpleArbiter, use vlc (network stream) to watch the video.
"""
}

computers = {
    'maliNetworkManager': {
        'disks': [
            {   'name': 'maliNetworkManager',
            },
            #{   'name': 'CDROM',
            #    'options':{'readonly': True},
            #    'parts': [],# no parts, no mounting.
            #}
        ],
        'network_interfaces': [{'network': 'net1'}],
        'flavor': 'm1.tiny',
        'config_drive': False

    },
    'maliBrezNetworkManager': {
        'disks': [
            {   'name': 'maliBrezNetworkManager',
            },
            #{   'name': 'CDROM',
            #    'options':{'readonly': True},
            #    'parts': [],# no parts, no mounting.
            #}
        ],
        'network_interfaces': [{'network': 'net1'}],
        'flavor': 'm1.tiny',
        'config_drive': False

    },
    'SimpleArbiter': {
        'disks': [
            {   'name': 'simpleArbiterDhcp',
                # 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
    }
}

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 NAT virtualke'}, 'w': False, 'public' : True, 'type' : 'username', 'generated' : True},
   'IP_NAT_passwd': {'descriptions': {'si': 'Password NAT virtualke'}, '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):
    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
    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

def prepare_disks(templates, params):
#    d = templates['simpleArbiterDhcp']
    pass