# -*- python -*-
from plugins.powersupply import PowerSupply

from common import *

class MyBattery(Entry):
    COMPACT  = 0
    NORMAL   = 1
    DETAILED = 2
    
    def __init__(self, *args, show_all = False, show_battery = 0, details = 1, batteries = None, **kwargs):
        self.show_all = show_all
        self.show_battery = show_battery
        self.supply_filter = batteries
        self.details = details
        self.supplies = None
        self.batteries = None
        self.reload()
        Entry.__init__(self, *args, **kwargs)
    
    def reload(self):
        name = None
        newsupplies = {}
        newbatteries = []
        if self.batteries is not None:
            if isinstance(self.batteries, int):
                name = self.batteries[self.show_battery]
            else:
                name = self.show_battery
        for supply in PowerSupply.supplies():
            if self.supply_filter is not None:
                if supply not in self.supply_filter:
                    continue
            supply = PowerSupply(supply)
            if supply.type != 'Battery':
                continue
            newsupplies[supply.name] = supply
            newbatteries.append(supply.name)
        if self.supply_filter is not None:
            newbatteries = [b for b in self.supply_filter if b in newsupplies]
        if name is None:
            index = 0
        elif isinstance(name, int):
            index = name % len(newbatteries) if len(newbatteries) > 0 else 0
        elif name in self.batteries:
            index = self.batteries.index(index)
        else:
            index = 0
        self.supplies, self.batteries, self.show_battery = newsupplies, newbatteries, index
    
    def action(self, col, button, x, y):
        if button == LEFT_BUTTON:
            self.show_all = not self.show_all
            self.invalidate()
        elif button == MIDDLE_BUTTON:
            self.details += 1
            self.details %= 3
            self.invalidate()
        elif button == RIGHT_BUTTON:
            self.reload()
            self.invalidate()
        elif button == SCROLL_UP:
            if len(self.batteries) > 0:
                self.show_battery += 1
                self.show_battery %= len(self.batteries)
            self.invalidate()
        elif button == SCROLL_DOWN:
            if len(self.batteries) > 0:
                self.show_battery -= 1
                if self.show_battery < 0:
                    self.show_battery += len(self.batteries)
            self.invalidate()
    
    def colourise(self, text):
        try:
            value = text.replace(',', '.')
            value = float(value) if '.' in value else int(value)
            colour = '39'
            if value < 80:  colour = '32'
            if value < 50:  colour = '33'
            if value < 15:  colour = '31'
            if value <  5:  colour = '7;31'
            return '\033[%sm%s\033[0m' % (colour, text)
        except:
            return text
    
    def battery(self, bat):
        bat = self.supplies[bat]
        name = bat.name[:1].upper() + bat.name[1:].lower()
        if bat.is_online():
            try:
                charge   = (bat.get_charge(),  bat.get_charge_full(), bat.charge_full_design)
                current  = bat.get_current()
                status   = bat.get_status().lower()
                left     = None
                if charge[0] == charge[1]:
                    capacity = '100'
                else:
                    capacity = '%.1f' % (charge[0] / charge[1] * 100)
                capacity = self.colourise(capacity)
                
                if current > 0:
                    if status == 'charging':
                        left = (charge[1] - charge[0]) / current
                    elif status == 'discharging':
                        left = charge[0] / current
                    if left is not None:
                        hours = int(left)
                        minutes = int((left - hours) * 60 + 0.5)
                        if minutes >= 60:
                            minutes -= 60
                            hours += 1
                        left = '%ih%02im until %s' % (hours, minutes, status.replace('ing', 'ed'))
                
                if self.details == MyBattery.COMPACT:
                    if status == 'full':
                        return 'full'
                    elif status == 'charging':
                        status = '+'
                    elif status == 'discharging':
                        status = '-'
                    else:
                        status = '?'
                    if left is None:
                        return '%s%%%s' % (capacity, status)
                    else:
                        return '%s%%%s %s' % (capacity, status, left)
                elif self.details == MyBattery.NORMAL:
                    if left is None:
                        return '%s: %s%% %s' % (name, capacity, status)
                    else:
                        return '%s: %s%% %s %s' % (name, capacity, status, left)
                else:
                    if charge[1] == charge[2]:
                        health = '100'
                    else:
                        health = '%.1f' % (charge[1] / charge[2] * 100)
                    health = self.colourise(health)
                    volt = bat.get_voltage() / bat.voltage_min_design
                    if volt < 1:
                        volt = '\033[31m%.1f\033[0m' % volt
                    elif volt < 1.05:
                        volt = '\033[33m%.1f\033[0m' % volt
                    else:
                        volt = '%.1f' % volt
                    if left is None:
                        return '%s: %s%% %s, %s%% healthy, %sx voltage' % (name, capacity, status, health, volt)
                    else:
                        return '%s: %s%% %s %s, %s%% healthy, %sx voltage' % (name, capacity, status, left, health, volt)
            except Exception as e:
                return (repr(e))
                return ('%s: ???' % name) if self.details > 0 else '???'
        else:
            return ('%s: offline' % name) if self.details > 0 else 'offline'
    
    def function(self):
        if len(self.batteries) == 0:
            return 'no batteries available'
        
        if self.show_all:
            return SEPARATOR.join(self.battery(bat) for bat in self.batteries)
        else:
            return self.battery(self.batteries[self.show_battery])