#!/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.

"""

    'en':u"""
Setup two virtual machines: 
-SimpleArbiterDhcpRdate from disk image simpleArbiterRdate and 
-rdateClient.

On rdateClient create a user "test" with password "test". Set time on rdateClient 
so that it will use SimpleArbiterDhcpRdate as it's rdate server. After that create 
a program named guliver in the root directory of user "test". This program has to 
read 4bytes from standard input and convert them to signed number using big endian rule.
Write this number to standard output. 

Generate a random string from both letters and numbers and add it as a parameter 
to param_meta. This parameter will not be accessible to students and will be used 
to check if student copied his work from someone else.
"""

}

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