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

# kpov_util should be imported by add_assignment.py
instructions = {
    'si':"""
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':"""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):
    import collections

    tests = (
        ('nmcli', 'nmcli -c no d'),
        ('nslookup', 'nslookup www.arnes.si'),
    )

    results = collections.defaultdict(str)
    for name, host in [('nm', IP_NM), ('static', IP_static)]:
        host_results = kpov_util.ssh_test(host, 'student', 'vaje', tests)
        for key, value in host_results.items():
            results[key+'-'+name] = value
    return results
    
def gen_params(user_id, params_meta):
    params = dict()
    r = random.Random(user_id)
    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_util.IPv4_subnet_gen(r, '10.94.96.0/19', 25)
    params['DNS_NM'] = r.choice(dns_servers)
    params['IP_NM'], params['IP_static'] = kpov_util.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 = []
    if results['ssh-nm'] is True:
        score += 1
    else:
        hints += ['mali ssh failed: ' + results['ssh-nm']]
    if results['ssh-static'] is True:
        score += 1
    else:
        hints += ['malibrez ssh failed: ' + results['ssh-static']]
    if params['DNS_NM'] in results['nslookup-nm']:
        score += 2
    else:
        hints += ['NM nslookup incorrect']
    if params['DNS_static'] in results['nslookup-static']:
        score += 2
    else:
        hints += ['static nslookup incorrect']
    if re.search(r'e(th0|np0s3|ns3) +ethernet +connected', results['nmcli-nm']):
        score += 2
    else:
        hints += ['nmcli incorrect']
    if re.search(r'e(th0|np0s3|ns3) +ethernet +unmanaged', results['nmcli-static']):
        score += 2
    else:
        hints += ['nmcli on malibrez incorrect']
    return score, hints

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