#!/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}.
""",
    'en':u"""Create three virtual machines. Use simpleArbiterDhcpGW as the
disk image for the first, maliNetworkManager as the disk image for the second
and maliBrezNetworkManager for the third. The virtual machine names can match
the disks.

Turn off the networkmanager service on maliBrezNetworkManager. Use ifup/ifdown to
configure the network. The IP should be {IP_static} and the computer should use
the server {DNS_static} as it's DNS.

On maliNetworkManager, set the IP address to {IP_NM} and use {DNS_NM} as the DNS.
"""
}

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', 'en': 'IP address for maliNetworkManager'}, 'w': False, 'public':True, 'type': 'IP', 'generated': True},
    'DNS_NM': {'descriptions': {'si': 'DNS za maliNetworkManager', 'en': 'DNS for maliNetworkManager'}, 'w': False, 'public':True, 'type': 'IP', 'generated': True},
    'IP_static': {'descriptions': {'si': 'Naslov maliBrezNetworkManager', 'en': 'IP address for maliBrezNetworkManager'}, 'w': False, 'public': True, 'type': 'IP', 'generated': True},
    'DNS_static': {'descriptions': {'si': 'DNS za maliBrezNetworkManager', 'en':'DNS for 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'], global_params)