#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
#  dda_dict.py

from VLAN import VLAN
from glob import glob
from os import path

verzei = glob("vlans/*")



#
# Eingabe
#
print ("""\n\n Welche(r) Rechner/Drucker soll(en) modifiziert werden.
Mehrere Rechner mit Leerzeichen trennen. \n\n """)

devices = "e4001 e4002 e4003"
#macadr = "11:11:11 22:22:22 33:33:33"
macadr = ""
#devices = input("Eingabe der Geräte Namen: ").lower()
#macadr = input("Optional Eingabe der MAC Adresse in der dazugehörigen Reihenfolge: ").upper()

l_devices = devices.split()
l_macadr = macadr.split()
if len(l_macadr) > 0 and len(l_macadr) != len(l_devices):
    raise OSError("Die Eingabelisten sind ungleich lang!")


while True:
    vlanid = input("Eingabe der VLAN Nr.: ")
    try:
        i_vlanid = int(vlanid)
    except ValueError:
        print("Keine Nummer! Bitte nur Zahlen eingeben ...")
    # Alles in Ordnung:
    else:
        vlanid = "{:04d}".format(i_vlanid)    # Eingabe: 200 -> 0200
        break


# Gleich testen, ob Datei existiert:
vlanname = "vlans/vlan{}.in".format(vlanid)
if not path.exists(vlanname):
    raise FileNotFoundError("VLAN-Datei '{}' wurde nicht gefunden!".format(vlanname))



#
# Fkt. IP generieren
#
def ipgen(vlanid):
    """ IPs generieren für die jeweilige übergebene VLAN ID """
    l_ipgen = []
    
    # "0200" -> 200
    s_vlanid = str(int(vlanid))
    
    if len(s_vlanid) > 3:    # wenn 1400 -> 14
        s_vlanid = s_vlanid[:2]
    else:
        s_vlanid = s_vlanid[0]
        
    for zaehler in range(1, 254):
        ip = "172.16.{}.{}".format(s_vlanid, zaehler)
        l_ipgen.append(ip)
        
    return (l_ipgen)



#
# MAC-Adr finden
#
if len(l_macadr) == 0:
    for vlandat in verzei:
        vlan = VLAN(vlandat, l_devices)
        # Lese die MAC Adresse aus der Datei falls diese die Devices aus der Eingabe enthält
        l_macadr = vlan.read_vlan_mac(l_macadr)

if not l_macadr:
    raise OSError("Zu diesen E-Nummer(n) wurde keine MAC-Adr. gefunden")

print(l_macadr)



#
# vlan Dateien einlesen
#
for vlandat in verzei:
    vlan = VLAN(vlandat, l_devices)
    d_vlan = vlan.read_vlan_make_dict()
    print (vlandat)

    if vlanname == vlandat:
        #
        # Das VLAN dem die Eingabe hinzugefügt wird
        #
        print ("Die Eingabe wird dem Ziel-Vlan hinzugefügt")

        # Im VLAN belegte IPs auflisten
        l_ip_vlan = []
        macip_d_vlan = list(d_vlan.values())
        for ipx in macip_d_vlan:
            l_ip_vlan.append(ipx[1])

        l_ipgen = ipgen(vlanid)
        
        # IPs filtern, von den generierte IPs 1-253 die IPs aus VLAN-Datei abziehen = noch freie IPs
        m_ipgen = set(l_ipgen)
        m_ipvlan = set(l_ip_vlan)
        m_freeips = m_ipgen.difference(m_ipvlan)
        l_freeips = list(m_freeips)
        
        #
        # Daten zusammenfügen und ins Dict des gesuchten VLANs eintragen
        #
        d_vlan_fkt = d_vlan.copy()
        print ("Zaehler für while-Schleife erstellen")
        durchlaeufe = len(l_devices)
        zaehler = 0
        while zaehler < durchlaeufe:
            dev = l_devices[zaehler]
            mac = l_macadr[zaehler]
            freeip = l_freeips[zaehler]
            d_vlan_fkt[dev] = [mac, freeip]
            zaehler += 1
   
        #
        # Schreibe Ziel VLAN-Datei
        #
        vlan.write_vlandat(d_vlan_fkt)
        
    else:
        #
        # Die restlichen VLANs aus denen die Eingabe abgezogen wird
        #
        d_vlan_fkt = d_vlan.copy()
        durchlaeufe = len(l_devices)
        print ("Anzahl Schleifen durchläufe", (durchlaeufe))
        zaehler = 0
        while zaehler < durchlaeufe:
                try:
                    # Lese die MAC Adresse aus der Datei falls diese die Devices aus der Eingabe enthält
                    print (d_vlan_fkt.pop(l_devices[zaehler]))
                    zaehler += 1
                except KeyError:
                    print("Das Device {} wurde nicht im {} gefunden ...".format(l_devices[zaehler], vlandat))
                    zaehler += 1

        if d_vlan_fkt != d_vlan:
            #
            # Schreibe VLAN Datei für die restlichen VLANs
            #
            vlan.write_vlandat(d_vlan_fkt)
            
            
"""
    if len(l_macadr) == 0:
        print ("Zaehler für while-Schleife erstellen")
        durchlaeufe = len(l_devices)
        zaehler = 0
        d_vlan_fkt = d_vlan.copy()
        while zaehler < durchlaeufe:
            try:
                # Lese die MAC Adresse aus der Datei falls diese die Devices aus der Eingabe enthält
                print (d_vlan_fkt.pop(l_devices[zaehler]))
                macadr = d_vlan_fkt.pop(l_devices[zaehler])
                print(macadr)
                zaehler += 1
            except KeyError:
                print("Das Device {} wurde nicht im {} gefunden ...".format(l_devices[zaehler], vlandat))
                zaehler += 1
            # Lese die MAC Adresse aus der Datei falls diese die Devices aus der Eingabe enthält
            #print (d_vlan_fkt.pop(l_devices[zaehler]))
            #macadr = d_vlan_fkt.pop(l_devices[zaehler])
            #d_vlan_fkt.pop(l_devices[zaehler])
            #print(macadr)
            #print(l_macadr)
"""