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

# kpov_random_helpers should be imported by add_assignment.py
# TODO: finish this!

instructions = {
    'si':u"""
Postavi dva navidezna racunalnika: 
- SimpleArbiterDhcpRdate s slike diska simpleArbiterRdate ter 
-rdateClient.

Na rdateClient ustvarite uporabnika test z geslom test. Nastavite cas na 
rdateClient tako, da kot rdate strežnik uporabite SimpleArbiterDhcpRdate. 
Nato v domacem imeniku uporabnika test spravite program z imenom guliver, 
ki prebere 4 byte podatkov s standardnega vhoda in jih pretvori v 
predznaceno celo število, pri cemer naj uporablja zapis z debelim koncem 
(big endian).
Število naj program izpiše na standardni izhod.

V param_meta dodajte parameter, katerega vrednost bo nakljucen string 
sestavljen iz crk in stevilk.Tega parametra student ne bo mogel 
nastavljati sam. S tem parametrom se bo preverjalo ali je prišlo do
prepisovanja.

"""
}

computers = {
    'rdateClient': {
        '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': 'simpleArbiterDhcpRdate',
                # 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} }

# TODO: (polz) pick the parameters you need! ASAP!
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):
    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
    results['NM_nmcli'] = sA.run('nmcli d')
    results['NM_nslookup'] = sA.run('nslookup www.arnes.si')
    # sB
    # check whether NM is handling eth0
    results['static_nmcli'] = sB.run('nmcli d')
    results['static_nslookup'] = sB.run('nslookup www.arnes.si')
    sA.logout()
    sB.logout()
    return results
    
import datetime
import random
def gen_params(user_id, params_meta):
    # TODO: (polz) tole nima smisla. Za začetek datetime in random
    # ne bosta importana v test_task.py, kjer se bo tale koda izvajala.
    # Prosim, s kolegi se pogovorite in si izmislite, katere parametre bi
    # lahko uporabili.
    crke = "ABCDEFGHIJKLMNOPRSTUVZXYQW"
    time = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
    sifra = ""
    for x in time:
      for i in x:
         sifra+=random.choice(crke) + i + random.choice(crke) 
    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_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)
    params[user_id] = sifra
    return params

def task_check(results, params):
    import re
    score = 0
    if results['NM_nslookup'].find('Server:\t\t{0}\r'.format(params['DNS_NM'])) > -1:
        score += 3
    if results['static_nslookup'].find('Server:\t\t{0}\r'.format(params['DNS_static'])) > -1:
        score += 3
    if re.search(r'eth0 +802-.*connected', results['NM_nmcli']):
        score += 2
    if not re.search(r'eth0 +802-.*connected', results['static_nmcli']):
        score += 2
    #score = 0
    return score

def prepare_disks(templates, params):
#    d = templates['simpleArbiterDhcp']
    pass