#!/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 poskrbi, da networkmanager ne bo več skrbel
za omrežni vmesnik, temveč bosta delovala ukaza ifup in 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.

Make sure that on maliBrezNetworkManager ifup/ifdown can be used to configure
the network and that NetworkManager, while still installed, will not manage 
the network interface. 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 -c no 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 -c no 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, '10.94.96.0/19', 25)
    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
    else:
        hints += ['NM nslookup incorrect']
    if results['static_nslookup'].find(params['DNS_static']) > -1:
    #    print "static_nslookup OK"
        score += 2
    else:
        hints += ['static nslookup incorrect']
    if re.search(r'e(th0|np0s3|ns3) +ethernet +connected', results['NM_nmcli']):
    #    print "NM_nmcli OK"
        score += 3
    else:
        hints += ['nmcli incorrect']
    if re.search(r'e(th0|np0s3|ns3) +ethernet +unmanaged', results['static_nmcli']):
    #    print "static_nmcli OK"
        score += 3
    else:
        hints += ['nmcli on malibrez incorrect']
    return score, hints

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