# -*- python -*-
'''
xpybar – xmobar replacement written in python
Copyright © 2014, 2015, 2016, 2017, 2018  Mattias Andrée (maandree@kth.se)

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
'''

# A xpybar configuration example testing the features of plugins.df

import time
import threading

from plugins.df import Discs
from plugins.clock import Clock


df_ = Discs()


OUTPUT, HEIGHT, YPOS, TOP = 0, len(df_.filesystems.keys()) * 12, 24, True


clock = Clock(sync_to = 5 * Clock.SECONDS)

start_ = start
def start():
    start_()
    xasync(lambda : clock.continuous_sync(lambda : bar.invalidate()))


def redraw():
    texts = []
    
    fs_ = max(len(df_.mountpoints[mp].filesystem) for mp in df_.mountpoints.keys())
    mp_ = max(len(df_.mountpoints[mp].mountpoint) for mp in df_.mountpoints.keys())
    ft_ = max(len(df_.mountpoints[mp].fstype)     for mp in df_.mountpoints.keys())
    
    def usage(value):
        colour = '39'
        if value < 50:  colour = '32'
        if value > 75:  colour = '33'
        if value > 95:  colour = '31'
        if value > 99:  colour = '41;33'
        return '\033[%sm%6.2f\033[m%%' % (colour, value)
    
    def u(b):
        unit = 0
        units = ['B', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB']
        while (unit + 1 < len(units)) and (b >= 1024):
            b /= 1024
            unit += 1
        return '%7.2f%s' % (b, units[unit])
    
    for fs in sorted(df_.filesystems.keys()):
        d = df_.filesystems[fs]
        text = 'fs: %s%s │ mount: %s%s │ type: %s%s │ size: %9s │ used: %9s │ free: %9s │ usage: %s │ inodes: %10i │ iused: %10i │ ifree: %10i │ iusage: %s'
        text %= (d.filesystem, ' ' * (fs_ - len(d.filesystem)),
                 d.mountpoint, ' ' * (mp_ - len(d.mountpoint)),
                 d.fstype,     ' ' * (ft_ - len(d.fstype)),
                 u(d.blocks), u(d.used), u(d.available), usage(100 * d.used / d.blocks),
                 d.inodes, d.iused, d.ifree, usage(100 * d.iused / d.inodes))
        texts.append(text)
    
    text = '\n'.join(texts)
    bar.partial_clear(0, bar.width, 10, 0, 2, text)
    bar.draw_coloured_text(0, 10, 0, 2, text)
    # TODO this is very flickery