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

# kpov_random_helpers should be imported by add_assignment.py

instructions = {
    'si':u"""
Ustvari tri navidezne računalnike. Za enega (SimpleArbiter) uporabi sliko diska simpleArbiter. Na drugega (DHCP_server) postavi strežnika
DHCP in DNS. Poskrbi, da bo preostali računalnik (DHCP_client) dobil naslov, ki ga prebereš na SimpleArbiterju.
Poskrbi, da bo DNS strežnik za HOSTNAME_X vrnil naslov {HOSTNAME_X_IP}.
"""
}

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} }

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},
    'HOSTNAME_X_IP': {'descriptions': {'si': 'Naslov HOSTNAME_X'}, '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 = pxssh.pxssh()
    sB = pxssh.pxssh()
    sA.login(IP_NM, peer_user, peer_passwd)
    sB.login(IP_static, peer_user, peer_passwd)
    # sA
    # make sure NM is not handling eth0
	sA.sendline('nmcli -c no d')
    sA.prompt()
    results['NM_nmcli'] = sA.before
	
    results['NM_nslookup'] = sA.run('nslookup www.arnes.si')
    # sB
    # check whether NM is handling eth0
	sA.sendline('nmcli -c no d')
    sA.prompt()
    results['static_nmcli'] = sA.before
	
    results['static_nslookup'] = sB.run('nslookup www.arnes.si')
    sA.logout()
    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'], params['HOSTNAME_X_IP'] = kpov_random_helpers.IPv4_addr_gen(r, net, 3)
    params['DNS_static'] = r.choice(dns_servers)
    return params

def task_check(results, params):
    import re
    score = 0
    hints = []
    if results['NM_nslookup'].find('Server:\t\t{0}\r'.format(params['DNS_NM'])) > -1:
        score += 3
    else:
        hint += ["NM DNS wrong"]
    if results['static_nslookup'].find('Server:\t\t{0}\r'.format(params['DNS_static'])) > -1:
        score += 3
    else:
        hint += ["Static DNS wrong"]
	if re.search(r'e(th0|np0s3|ns3) +ethernet +connected', results['NM_nmcli']):
    #    print "NM_nmcli OK"
        score += 2
    else:
        hints += ['nmcli incorrect']
	if not re.search(r'e(th0|np0s3|ns3) +ethernet +connected', results['static_nmcli']):
    #    print "NM_nmcli OK"
        score += 2
    else:
        hints += ['static nmcli incorrect']
    return score, hints

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