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

# kpov_random_helpers should be imported by add_assignment.py
instructions = {
    'si':u"""
Ustvari tri navidezne racunalnike. Za prvega
uporabi sliko diska simpleArbiterDhcpGW, za drugega sliko diska
maliNetworkManager, za tretjega sliko diska maliBrezNetworkManager.
Racunalnike imenujmo enako kot slike diska.
Na maliBrezNetworkManager izklopi networkmanager in uporabi ifup / ifdown.
Na maliNetworkManager nastavi IP naslov {IP_NM} ter DNS {DNS_NM}.
Na maliBrezNetworkManager nastavi IP naslov {IP_static} ter DNS {DNS_static}.
DNS strežnikov. Naslove dobiš na simpleArbiterDhcpGW.
""",
    'en':u"""
"""
}

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': 'simpleArbiterDhcpGW',
                # 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} }

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_static': {'descriptions': {'si': 'Naslov maliBrezNetworkManager'}, 'w': False, 'public': True, 'type': 'IP', 'generated': True},
    'DNS_static': {'descriptions': {'si': 'DNS za maliBrezNetworkManager'}, 'w': False, 'public': True, 'type': 'IP', 'generated': True},
}

def task(IP_NM, DNS_NM, IP_static, DNS_static):
    from pexpect import pxssh
    import pexpect
    results = dict()
    peer_user = 'student'
    peer_passwd = 'vaje'
    # sA - network manager
    sA = pxssh.pxssh()
    sA.login(IP_NM, peer_user, peer_passwd)
    sA.sendline('nmcli d')
    sA.prompt()
    results['NM_nmcli'] = sA.before
    sA.sendline('nslookup www.arnes.si')
    sA.prompt()
    results['NM_nslookup'] = sA.before
    sA.logout()
    # sB - brez network manager
    sB = pxssh.pxssh()
    sB.login(IP_static, peer_user, peer_passwd)
    sB.sendline('nmcli d')
    sB.prompt()
    results['static_nmcli'] = sB.before
    sB.sendline('nslookup www.arnes.si')
    sB.prompt()
    results['static_nslookup'] = sB.before
    sB.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_subnet_gen(r, '172.23.128.0/18', 24)
    params['DNS_NM'] = r.choice(dns_servers)
    params['IP_NM'], params['IP_static'] = kpov_random_helpers.IPv4_addr_gen(r, net, 2)
    params['DNS_static'] = r.choice(dns_servers)
    return params

def task_check(results, params):
    import re
    score = 0
    hints = []
    #print results, params
    if results['NM_nslookup'].find(params['DNS_NM']) > -1:
    #    print "NM_nslookup OK"
        score += 2
    if results['static_nslookup'].find(params['DNS_static']) > -1:
    #    print "static_nslookup OK"
        score += 2
    if re.search(r'eth0 +802-.*connected', results['NM_nmcli']):
    #    print "NM_nmcli OK"
        score += 3
    if re.search(r'eth0 +802-.*unmanaged', results['static_nmcli']):
    #    print "static_nmcli OK"
        score += 3
    return score, hints

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