# -*- python -*-
from plugins.ipaddress import IPAddress
from plugins.clock     import Clock

from common import *

# TODO add IPv6 address
class MyIPAddress(Entry):
    def __init__(self, *args, ignore = None, public = True, both = False, **kwargs):
        self.ignore = ['lo'] if ignore is None else ignore
        self.text = None
        self.show_both = both
        self.show_public = public
        self.show_nic = None
        Entry.__init__(self, *args, **kwargs)
        def init():
            self.refresh()
            xasync(lambda : Clock(sync_to = 10 * Clock.MINUTES).continuous_sync(t(self.refresh)), name = 'ipaddress')
        xasync(init, name = 'ipaddress')
    
    def action(self, col, button, x, y):
        if button == LEFT_BUTTON:
            self.show_public = not self.show_public
            self.invalidate()
        elif button == MIDDLE_BUTTON:
            self.show_both = not self.show_both
            self.invalidate()
        elif button == RIGHT_BUTTON:
            xasync(self.refresh, name = 'ipaddress')
        elif button in (SCROLL_UP, SCROLL_DOWN):
            nic = self.show_nic
            if nic is None:
                return
            nics = list(sorted(self.text[0].keys()))
            try:
                nic = nics.index(nic)
            except:
                if len(nics) == 0:
                    self.show_nic = None
                    self.invalidate()
                else:
                    self.show_nic = nics[0]
                    self.invalidate()
            nic += +1 if button == SCROLL_UP else -1
            if 0 <= nic < len(nics):
                self.show_nic = nics[nic]
                self.invalidate()
    
    def refresh(self):
        text_private, text_public, text_both = {}, {}, {}
        ipa = IPAddress(*(self.ignore))
        for nic in ipa.nics.keys():
            (state, a4, a6, a) = ipa.nics[nic]
            if state == IPAddress.DOWN:
                label = '31'
            elif state == IPAddress.UP:
                label = '32'
            elif state == IPAddress.UNKNOWN:
                label = '33'
            else:
                label = '39'
            label = '\033[%sm%s\033[0m' % (label, nic)
            prv = pub = ''
            if a4 is not None:
                prv += ' ' + a4
            if a6 is not None:
                prv += ' ' + a6
            if a is not None:
                pub += ' ' + a
            finalise = lambda addrs : '%s:%s' % (label, (addrs if addrs != '' else ' no address'))
            text_private[nic] = finalise(prv)
            text_public[nic]  = finalise(pub)
            text_both[nic]    = finalise(pub + prv)
        self.text = (text_private, text_public, text_both)
        self.invalidate()
    
    def function(self):
        text = self.text
        if text is None:
            return '...'
        text = text[2 if self.show_both else (1 if self.show_public else 0)]
        nic = self.show_nic
        if nic is None or nic not in text:
            if len(text.keys()) == 0:
                self.show_nic = None
                return 'No NIC available'
            nic = sorted(text.keys())[0]
            self.show_nic = nic
        return text[nic]