# -*- python -*-
'''
xpybar – xmobar replacement written in python
Copyright © 2014, 2015, 2016, 2017, 2018, 2019  Mattias Andrée (m@maandreese)

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.mem

import time
import threading

from plugins.mem import Memory
from plugins.clock import Clock


OUTPUT, HEIGHT, YPOS, TOP = 0, 12, 24, True


clock = Clock(sync_to = Clock.SECONDS)

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


def redraw():
    def c(used, total):
        value = used * 100 / total
        colour = '39'
        if value >= 20:  colour = '32'
        if value >= 50:  colour = '33'
        if value >= 90:  colour = '31'
        return '\033[%sm%.0f\033[0m%%' % (colour, value)
    
    def u(b):
        unit = 0
        units = ['KB', 'MB', 'GB', 'TB', 'PB', 'EB']
        while (unit + 1 < len(units)) and (b >= 1024):
            b /= 1024
            unit += 1
        return '%.0f%s' % (b, units[unit])
    
    mem_ = Memory()
    text = 'Mem: %s (%s cached, %s buffers, %s free) │ Swp: %s (%s cached, %s free)'
    text %= (c(mem_.mem_used, mem_.mem_total), u(mem_.cached), u(mem_.buffers), u(mem_.mem_free),
             c(mem_.swap_used, mem_.swap_total), u(mem_.swap_cached), u(mem_.swap_free))
    if mem_.shmem is not None:
        text += ' │ Shm: %s' % c(mem_.shmem, mem_.mem_total)
    if mem_.slab is not None:
        text += ' │ Slab: %s' % c(mem_.slab, mem_.mem_total)
    if mem_.hardware_corrupted is not None:
        text += ' │ Crpt: %s' % c(mem_.hardware_corrupted, mem_.mem_total)
    
    bar.clear()
    bar.draw_coloured_text(0, 10, 0, 2, text)