#!/bin/bash /usr/scripts/python.sh
# -*- coding: utf-8 -*-

"""
Mise a jour periodique des listes d'ip et d'id disponibles.

Copyright (C) Alexandre Bos, largement pompe sur ldap_crans.py
Licence : GPLv2
"""

import sys
from config import NETs
from iptools import AddrInNet
try:
    from dialog import Dialog
except ImportError:
    Dialog = None

repertoire =  '/usr/scripts/var/numeros_disponibles/'

"""
 Un petit hack de rien du tout pour s'assurer qu'on n'attribue
 pas ces adresses. Certains services risquent de continuer
 d'essayer de se connecter a ces adresses
"""
ancien_vlan_adm = ['10.231.136.0/24']


def lister_ip_dispo(plage):
    f = open(repertoire + 'ip_' + plage)
    lignes = f.readlines()
    liste = []
    for l in lignes:
        liste.append(l.strip())
    return liste

def lister_ip_utilisees():
    liste = []
    from ldap_crans import crans_ldap
    db = crans_ldap()
    r = db.search('mid=*')
    machines = r['machineFixe'] + r['machineWifi'] + r['machineCrans'] + r['borneWifi']
    for m in machines:
        liste.append(m.ip())
    return liste

def update_ip(plage, occupees):
    net = NETs[plage]
    pool_ip = []                # Pool d'IP à tester
    for ne in net:
        ip = ne.split('/')[0]
        ip = ip.split('.')
        n = []
        for i in ip:
            n.append(int(i))
        while 1:
            if n[3] < 254:
                n[3] += 1
            else:
                n[2] += 1
                n[3] = 1
                if n[2] == 255: break
            ip = "%d.%d.%d.%d" % tuple(n)
            if not AddrInNet(ip, ne):
                # On est allé trop loin
                break
            pool_ip.append(ip)

    resultat = ''.join('%s\n' % ip for ip in pool_ip if ip not in occupees and not AddrInNet(ip, ancien_vlan_adm))

    f = open(repertoire + 'ip_' + plage,'w')
    f.write(resultat)
    f.close

def update_ip_fixe(occupees):
    update_ip('fil', 'ip_fil', occupees)

def update_ip_wifi_adh(occupees):
    update_ip('wifi-adh','ip_wifi-adh', occupees)

if __name__ == "__main__":
    if "--cron" in sys.argv:
        cron = True
    else:
        cron = False
    if not cron:
        dlg = Dialog()
        dlg.gauge_start(text="Recherche des machines...", backtitle="numeros_disponibles")
    ip_occupees = lister_ip_utilisees()
    done = 1
    for net in NETs.keys():
        if not cron:
            dlg.gauge_update(int(done*100/(len(NETs)+1)), text="IP libres dans %s" % net, update_text=True)
        update_ip(net, ip_occupees)
        done += 1
    if not cron:
        dlg.gauge_update(100, text="Fini !", update_text=True)
        dlg.gauge_stop()