#!/usr/bin/env python3
'''
xpybar – xmobar replacement written in python
Copyright © 2014  Mattias Andrée (maandree@member.fsf.org)

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/>.
'''

import subprocess
import Xlib.display, Xlib.Xatom, Xlib.ext.randr, Xlib.X


display = None
screen = None
screen_i = None


def open_x(screen_no = None):
    global display, screen, screen_i
    display = Xlib.display.Display()
    screen_i = screen_no if screen_no is not None else display.get_default_screen()
    screen = display.screen(screen_i)


def get_monitors():
    global screen_i
    p = subprocess.Popen(['xrandr'], stdout = subprocess.PIPE)
    p = p.communicate()[0].decode('utf-8', 'replace')
    s = -1
    rc = []
    prim = None
    for line in p.split('\n'):
        if line.startswith('Screen '):
            s = int(line[len('Screen '):].split(':')[0])
        elif s == screen_i:
            if ' connected ' in line:
                m = line.replace('-', '+-').replace('++', '+')
                p = ' primary ' in m
                m = m.replace(' primary ', ' ')
                m = m.split(' ')[2].replace('+', 'x').split('x')
                m = [int(x) for x in m]
                if p and (prim is None):
                    prim = m
                else:
                    rc.append(m)
    if prim is not None:
        rc = [prim] + rc
    return rc


def get_display():
    global display
    return display


def get_screen():
    global screen
    return screen


def create_panel(width, height, left, top, panel_height, at_top):
    global display, screen
    ypos = top if at_top else (height - top - panel_height)
    window = screen.root.create_window(left, ypos, width, panel_height, 0, screen.root_depth,
                                       Xlib.X.InputOutput, Xlib.X.CopyFromParent,
                                       event_mask = (
                                           Xlib.X.StructureNotifyMask |
                                           Xlib.X.ButtonReleaseMask
                                       ),
                                       colormap = Xlib.X.CopyFromParent)
    
    top_    = lambda x, y, w, h : [0, 0, y + h, 0, 0, 0, 0, 0, x, x + w, 0, 0]
    bottom_ = lambda x, y, w, h : [0, 0, 0, y + h, 0, 0, 0, 0, 0, 0, x, x + w]
    
    window.set_wm_name('xpybar')
    window.set_wm_icon_name('xpybar')
    window.set_wm_class('bar', 'xpybar')
    
    _CARD = display.intern_atom("CARDINAL")
    _PSTRUT = display.intern_atom("_NET_WM_STRUT_PARTIAL")
    window.change_property(_PSTRUT, _CARD, 32, (top_ if at_top else bottom_)(left, top, width, panel_height))
    
    _ATOM = display.intern_atom("ATOM")
    _TYPE = display.intern_atom("_NET_WM_WINDOW_TYPE")
    _DOCK = display.intern_atom("_NET_WM_WINDOW_TYPE_DOCK")
    window.change_property(_TYPE, _ATOM, 32, [_DOCK])
    
    return window


def close_x():
    global display
    display.flush()
    display.close()