Solaar/lib/solaar/ui/status_icon.py

#
#
#

from __future__ import absolute_import, division, print_function, unicode_literals

from gi.repository import Gtk, GdkPixbuf

from solaar import NAME
from . import action as _action, icons as _icons
from logitech.unifying_receiver import status as _status

#
#
#

def create(activate_callback):
	assert activate_callback

	icon = Gtk.StatusIcon()
	icon.set_title(NAME)
	icon.set_name(NAME)
	icon.set_from_icon_name(_icons.APP_ICON[0])
	icon._devices_info = []
	icon._receivers = set()

	icon.set_tooltip_text(NAME)
	icon.connect('activate', activate_callback)

	menu = Gtk.Menu()

	# per-device menu entries will be generated as-needed

	menu.append(Gtk.SeparatorMenuItem.new())
	menu.append(_action.about.create_menu_item())
	menu.append(_action.make('application-exit', 'Quit', Gtk.main_quit).create_menu_item())
	menu.show_all()

	icon._menu = menu

	icon.connect('popup_menu',
					lambda icon, button, time, menu:
						menu.popup(None, None, icon.position_menu, icon, button, time),
					menu)
	return icon

#
#
#

def _generate_tooltip_lines(icon):
	yield '<b>%s</b>' % NAME
	yield ''

	for _, serial, name, status in icon._devices_info:
		yield '<b>%s</b>' % name

		p = str(status)
		if p:  # does it have any properties to print?
			if status:
				yield '\t%s' % p
			else:
				yield '\t%s <small>(inactive)</small>' % p
		else:
			if status:
				yield '\t<small>no status</small>'
			else:
				yield '\t<small>(inactive)</small>'
		yield ''


_PIXMAPS = {}
def _icon_with_battery(level, active):
	battery_icon = _icons.battery(level)
	name = '%s-%s' % (battery_icon, active)
	if name not in _PIXMAPS:
		mask = _icons.icon_file(_icons.APP_ICON[2], 128)
		assert mask
		mask = GdkPixbuf.Pixbuf.new_from_file(mask)
		assert mask.get_width() == 128 and mask.get_height() == 128
		mask.saturate_and_pixelate(mask, 0.7, False)

		battery = _icons.icon_file(battery_icon, 128)
		assert battery
		battery = GdkPixbuf.Pixbuf.new_from_file(battery)
		assert battery.get_width() == 128 and battery.get_height() == 128
		if not active:
			battery.saturate_and_pixelate(battery, 0, True)

		# TODO can the masking be done at runtime?
		battery.composite(mask, 0, 7, 80, 121, -32, 7, 1, 1, GdkPixbuf.InterpType.NEAREST, 255)
		_PIXMAPS[name] = mask

	return _PIXMAPS[name]

def _update_image(icon):
	if not icon._receivers:
		icon.set_from_icon_name(_icons.APP_ICON[-1])
		return

	battery_status = None
	battery_level = 1000

	for _, serial, name, status in icon._devices_info:
		level = status.get(_status.BATTERY_LEVEL)
		if level is not None and level < battery_level:
			battery_status = status
			battery_level = level

	if battery_status is None:
		icon.set_from_icon_name(_icons.APP_ICON[1])
	else:
		icon.set_from_pixbuf(_icon_with_battery(battery_level, bool(battery_status)))

#
#
#

def _device_index(icon, device):
	if device.receiver.serial in icon._receivers:
		for index, (rserial, serial, name, _) in enumerate(icon._devices_info):
			if rserial == device.receiver.serial and serial == device.serial:
				return index


def _add_device(icon, device):
	index = len(icon._devices_info)
	device_info = (device.receiver.serial, device.serial, device.name, device.status)
	icon._devices_info.append(device_info)

	menu_item = Gtk.ImageMenuItem.new_with_label(device.name)
	icon._menu.insert(menu_item, index)
	menu_item.set_image(Gtk.Image())
	menu_item.show_all()

	return index


def _remove_device(icon, index):
	# print ("remove device", index)
	assert index is not None
	del icon._devices_info[index]
	menu_items = icon._menu.get_children()
	icon._menu.remove(menu_items[index])


def _remove_receiver(icon, receiver):
	icon._receivers.remove(receiver.serial)
	index = 0
	while index < len(icon._devices_info):
		rserial, _, _, _ = icon._devices_info[index]
		# print ("remove receiver", index, rserial)
		if rserial == receiver.serial:
			_remove_device(icon, index)
		else:
			index += 1


def _update_menu_item(icon, index, device_status):
	menu_items = icon._menu.get_children()
	menu_item = menu_items[index]

	image = menu_item.get_image()
	battery_level = device_status.get(_status.BATTERY_LEVEL) if device_status else None
	image.set_from_icon_name(_icons.battery(battery_level), Gtk.IconSize.MENU)
	image.set_sensitive(battery_level is not None)

#
#
#

def update(icon, device=None):
	# print ("icon update", device)

	if device is not None:
		if device.kind is None:
			# receiver
			receiver = device
			if receiver:
				icon._receivers.add(receiver.serial)
			else:
				_remove_receiver(icon, receiver)
		else:
			# peripheral
			index = _device_index(icon, device)
			if device.status is None:
				# was just unpaired
				assert index is not None
				_remove_device(icon, index)
			else:
				if index is None:
					index = _add_device(icon, device)
				_update_menu_item(icon, index, device.status)

	tooltip_lines = _generate_tooltip_lines(icon)
	icon.set_tooltip_markup('\n'.join(tooltip_lines).rstrip('\n'))
	_update_image(icon)