Source code for libqtile.widget.systray

# Copyright (c) 2010 Aldo Cortesi
# Copyright (c) 2010-2011 dequis
# Copyright (c) 2010, 2012 roger
# Copyright (c) 2011 Mounier Florian
# Copyright (c) 2011-2012, 2014 Tycho Andersen
# Copyright (c) 2012 dmpayton
# Copyright (c) 2012-2013 Craig Barnes
# Copyright (c) 2013 hbc
# Copyright (c) 2013 Tao Sauvage
# Copyright (c) 2014 Sean Vig
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
import xcffib
from xcffib.xproto import (
    ClientMessageData,
    ClientMessageEvent,
    EventMask,
    SetMode,
)

from libqtile import bar
from libqtile.backend.x11 import window
from libqtile.widget import base

XEMBED_PROTOCOL_VERSION = 0


class Icon(window._Window):
    _window_mask = EventMask.StructureNotify | \
        EventMask.PropertyChange | \
        EventMask.Exposure

    def __init__(self, win, qtile, systray):
        window._Window.__init__(self, win, qtile)
        self.systray = systray
        self.update_size()

    def update_size(self):
        icon_size = self.systray.icon_size
        self.update_hints()

        width = self.hints.get("min_width", icon_size)
        height = self.hints.get("min_height", icon_size)

        width = max(width, icon_size)
        height = max(height, icon_size)

        if height > icon_size:
            width = width * icon_size // height
            height = icon_size

        self.width = width
        self.height = height
        return False

    def handle_PropertyNotify(self, e):  # noqa: N802
        name = self.qtile.core.conn.atoms.get_name(e.atom)
        if name == "_XEMBED_INFO":
            info = self.window.get_property('_XEMBED_INFO', unpack=int)
            if info and info[1]:
                self.systray.bar.draw()

        return False

    def handle_DestroyNotify(self, event):  # noqa: N802
        wid = event.window
        del(self.qtile.windows_map[wid])
        del(self.systray.icons[wid])
        self.systray.bar.draw()
        return False

    handle_UnmapNotify = handle_DestroyNotify  # noqa: N815


[docs]class Systray(window._Window, base._Widget): """A widget that manages system tray""" _window_mask = EventMask.StructureNotify | \ EventMask.Exposure orientations = base.ORIENTATION_HORIZONTAL defaults = [ ('icon_size', 20, 'Icon width'), ('padding', 5, 'Padding between icons'), ] def __init__(self, **config): base._Widget.__init__(self, bar.CALCULATED, **config) self.add_defaults(Systray.defaults) self.icons = {} self.screen = 0 def calculate_length(self): width = sum(i.width for i in self.icons.values()) width += self.padding * len(self.icons) return width def _configure(self, qtile, bar): base._Widget._configure(self, qtile, bar) if self.configured: return self.conn = conn = qtile.core.conn win = conn.create_window(-1, -1, 1, 1) window._Window.__init__(self, window.XWindow(conn, win.wid), qtile) qtile.windows_map[win.wid] = self # Even when we have multiple "Screen"s, we are setting up as the system # tray on a particular X display, that is the screen we need to # reference in the atom if qtile.current_screen: self.screen = qtile.current_screen.index self.bar = bar atoms = conn.atoms conn.conn.core.SetSelectionOwner( win.wid, atoms['_NET_SYSTEM_TRAY_S{:d}'.format(self.screen)], xcffib.CurrentTime ) data = [ xcffib.CurrentTime, atoms['_NET_SYSTEM_TRAY_S{:d}'.format(self.screen)], win.wid, 0, 0 ] union = ClientMessageData.synthetic(data, "I" * 5) event = ClientMessageEvent.synthetic( format=32, window=qtile.core._root.wid, type=atoms['MANAGER'], data=union ) qtile.core._root.send_event(event, mask=EventMask.StructureNotify) def handle_ClientMessage(self, event): # noqa: N802 atoms = self.conn.atoms opcode = event.type data = event.data.data32 message = data[1] wid = data[2] parent = self.bar.window.window if opcode == atoms['_NET_SYSTEM_TRAY_OPCODE'] and message == 0: w = window.XWindow(self.conn, wid) icon = Icon(w, self.qtile, self) self.icons[wid] = icon self.qtile.windows_map[wid] = icon self.conn.conn.core.ChangeSaveSet(SetMode.Insert, wid) self.conn.conn.core.ReparentWindow(wid, parent.wid, 0, 0) self.conn.conn.flush() info = icon.window.get_property('_XEMBED_INFO', unpack=int) if not info: self.bar.draw() return False if info[1]: self.bar.draw() return False def draw(self): xoffset = self.padding self.drawer.clear(self.background or self.bar.background) self.drawer.draw(offsetx=self.offset, width=self.length) for pos, icon in enumerate(self.icons.values()): icon.window.set_attribute(backpixmap=self.drawer.pixmap) icon.place( self.offset + xoffset, self.bar.height // 2 - self.icon_size // 2, icon.width, self.icon_size, 0, None ) if icon.hidden: icon.unhide() data = [ self.conn.atoms["_XEMBED_EMBEDDED_NOTIFY"], xcffib.xproto.Time.CurrentTime, 0, self.bar.window.wid, XEMBED_PROTOCOL_VERSION ] u = xcffib.xproto.ClientMessageData.synthetic(data, "I" * 5) event = xcffib.xproto.ClientMessageEvent.synthetic( format=32, window=icon.wid, type=self.conn.atoms["_XEMBED"], data=u ) self.window.send_event(event) xoffset += icon.width + self.padding def finalize(self): base._Widget.finalize(self) atoms = self.conn.atoms self.conn.conn.core.SetSelectionOwner( 0, atoms['_NET_SYSTEM_TRAY_S{:d}'.format(self.screen)], xcffib.CurrentTime, ) self.hide()