#
#
#

from gi.repository import (Gtk, Gdk, GObject)

import ui
from logitech.unifying_receiver import status as _status


_RECEIVER_ICON_SIZE = Gtk.IconSize.BUTTON
_DEVICE_ICON_SIZE = Gtk.IconSize.DIALOG
_STATUS_ICON_SIZE = Gtk.IconSize.LARGE_TOOLBAR
_PLACEHOLDER = '~'

#
#
#

def _make_receiver_box(name):
	frame = Gtk.Frame()
	frame._device = None
	frame.set_name(name)

	icon_name = ui.get_icon(name, 'preferences-desktop-peripherals')
	icon = Gtk.Image.new_from_icon_name(icon_name, _RECEIVER_ICON_SIZE)
	icon.set_padding(2, 2)

	label = Gtk.Label('Scanning...')
	label.set_name('label')
	label.set_alignment(0, 0.5)

	pairing_icon = Gtk.Image.new_from_icon_name('network-wireless', Gtk.IconSize.MENU)
	pairing_icon.set_name('pairing-icon')
	pairing_icon.set_tooltip_text('The pairing lock is open.')

	toolbar = Gtk.Toolbar()
	toolbar.set_name('toolbar')
	toolbar.set_style(Gtk.ToolbarStyle.ICONS)
	toolbar.set_icon_size(Gtk.IconSize.MENU)
	toolbar.set_show_arrow(False)

	hbox = Gtk.HBox(homogeneous=False, spacing=8)
	hbox.pack_start(icon, False, False, 0)
	hbox.pack_start(label, True, True, 0)
	hbox.pack_start(pairing_icon, False, False, 0)
	hbox.pack_start(toolbar, False, False, 0)

	info_label = Gtk.Label('Querying ...')
	info_label.set_name('info-label')
	info_label.set_alignment(0, 0.5)
	info_label.set_padding(8, 2)
	info_label.set_selectable(True)

	info_box = Gtk.Frame()
	info_box.add(info_label)
	info_box.set_shadow_type(Gtk.ShadowType.ETCHED_IN)

	toggle_info_action = ui.action._toggle_action('info', 'Receiver info', _toggle_info_box, info_label, info_box, frame, _update_receiver_info_label)
	toolbar.insert(toggle_info_action.create_tool_item(), 0)
	toolbar.insert(ui.action.pair(frame).create_tool_item(), -1)
	# toolbar.insert(ui.action.about.create_tool_item(), -1)

	vbox = Gtk.VBox(homogeneous=False, spacing=4)
	vbox.set_border_width(4)
	vbox.pack_start(hbox, True, True, 0)
	vbox.pack_start(info_box, True, True, 0)

	frame.add(vbox)
	frame.show_all()

	info_box.set_visible(False)
	pairing_icon.set_visible(False)
	return frame


def _make_device_box(index):
	frame = Gtk.Frame()
	frame._device = None
	frame.set_name(_PLACEHOLDER)

	icon_name = 'preferences-desktop-peripherals'
	icon = Gtk.Image.new_from_icon_name(icon_name, _DEVICE_ICON_SIZE)
	icon.set_name('icon')
	icon.set_alignment(0.5, 0)

	label = Gtk.Label('Initializing...')
	label.set_name('label')
	label.set_alignment(0, 0.5)
	label.set_padding(4, 4)

	battery_icon = Gtk.Image.new_from_icon_name('battery_unknown', _STATUS_ICON_SIZE)

	battery_label = Gtk.Label()
	battery_label.set_width_chars(6)
	battery_label.set_alignment(0, 0.5)

	light_icon = Gtk.Image.new_from_icon_name('light_unknown', _STATUS_ICON_SIZE)

	light_label = Gtk.Label()
	light_label.set_alignment(0, 0.5)
	light_label.set_width_chars(8)

	not_encrypted_icon = Gtk.Image.new_from_icon_name('security-low', _STATUS_ICON_SIZE)
	not_encrypted_icon.set_name('not-encrypted')
	not_encrypted_icon.set_tooltip_text('The link is not encrypted!')

	toolbar = Gtk.Toolbar()
	toolbar.set_name('toolbar')
	toolbar.set_style(Gtk.ToolbarStyle.ICONS)
	toolbar.set_icon_size(Gtk.IconSize.MENU)
	toolbar.set_show_arrow(False)
	toolbar.set_border_width(0)

	status_box = Gtk.HBox(homogeneous=False, spacing=0)
	status_box.set_name('status')
	status_box.pack_start(battery_icon, False, True, 0)
	status_box.pack_start(battery_label, False, True, 0)
	status_box.pack_start(light_icon, False, True, 0)
	status_box.pack_start(light_label, False, True, 0)
	status_box.pack_end(toolbar, False, False, 0)
	status_box.pack_end(not_encrypted_icon, False, False, 0)

	info_label = Gtk.Label('Querying ...')
	info_label.set_name('info-label')
	info_label.set_alignment(0, 0.5)
	info_label.set_padding(8, 2)
	info_label.set_selectable(True)
	info_label._fields = {}

	info_box = Gtk.Frame()
	info_box.add(info_label)

	toggle_info_action = ui.action._toggle_action('info', 'Device info', _toggle_info_box, info_label, info_box, frame, _update_device_info_label)
	toolbar.insert(toggle_info_action.create_tool_item(), 0)
	toolbar.insert(ui.action.unpair(frame).create_tool_item(), -1)

	vbox = Gtk.VBox(homogeneous=False, spacing=4)
	vbox.pack_start(label, True, True, 0)
	vbox.pack_start(status_box, True, True, 0)
	vbox.pack_start(info_box, True, True, 0)

	box = Gtk.HBox(homogeneous=False, spacing=4)
	box.set_border_width(4)
	box.pack_start(icon, False, False, 0)
	box.pack_start(vbox, True, True, 0)
	box.show_all()

	frame.add(box)
	info_box.set_visible(False)
	return frame


def toggle(window, trigger):
	if window.get_visible():
		position = window.get_position()
		window.hide()
		window.move(*position)
	else:
		if trigger and type(trigger) == Gtk.StatusIcon:
			x, y = window.get_position()
			if x == 0 and y == 0:
				x, y, _ = Gtk.StatusIcon.position_menu(Gtk.Menu(), trigger)
				window.move(x, y)
		window.present()
	return True

def _popup(window, trigger):
	if not window.get_visible():
		toggle(window, trigger)

def create(title, name, max_devices, systray=False):
	window = Gtk.Window()
	window.set_title(title)
	window.set_icon_name(ui.appicon(0))
	window.set_role('status-window')

	vbox = Gtk.VBox(homogeneous=False, spacing=4)
	vbox.set_border_width(4)

	rbox = _make_receiver_box(name)
	vbox.add(rbox)
	for i in range(1, 1 + max_devices):
		dbox = _make_device_box(i)
		vbox.add(dbox)
	vbox.set_visible(True)

	window.add(vbox)

	geometry = Gdk.Geometry()
	geometry.min_width = 320
	geometry.min_height = 32
	window.set_geometry_hints(vbox, geometry, Gdk.WindowHints.MIN_SIZE)
	window.set_resizable(False)

	window.toggle_visible = lambda i: toggle(window, i)
	window.popup = lambda i: _popup(window, i)

	if systray:
		window.set_keep_above(True)
		window.connect('delete-event', toggle)
	else:
		window.connect('delete-event', Gtk.main_quit)

	return window

#
#
#

def _update_device_info_label(label, dev):
	need_update = False

	if 'wpid' in label._fields:
		wpid = label._fields['wpid']
	else:
		wpid = label._fields['wpid'] = dev.wpid
		need_update = True

	if 'serial' in label._fields:
		serial = label._fields['serial']
	else:
		serial = label._fields['serial'] = dev.serial
		need_update = True

	if 'firmware' in label._fields:
		firmware = label._fields['firmware']
	else:
		if dev.status:
			firmware = label._fields['firmware'] = dev.firmware
			need_update = True
		else:
			firmware = None

	if 'hid' in label._fields:
		hid = label._fields['hid']
	else:
		if dev.status:
			hid = label._fields['hid'] = 'HID++ %1.1f' % dev.protocol
			need_update = True
		else:
			hid = None

	if need_update:
		items = [('Wireless PID', wpid), ('Serial', serial)]
		if hid:
			items += [('Protocol', hid)]
		if firmware:
			items += [(f.kind, f.name + ' ' + f.version) for f in firmware]

		label.set_markup('<small><tt>%s</tt></small>' % '\n'.join('%-12s: %s' % (item[0], str(item[1])) for item in items))


def _update_receiver_info_label(label, dev):
	if label.get_visible() and '\n' not in label.get_text():
		items = [('Serial', dev.serial)] + \
				[(f.kind, f.version) for f in dev.firmware]
		label.set_markup('<small><tt>%s</tt></small>' % '\n'.join('%-10s: %s' % (item[0], str(item[1])) for item in items))


def _toggle_info_box(action, label_widget, box_widget, frame, update_function):
	if action.get_active():
		box_widget.set_visible(True)
		GObject.timeout_add(50, update_function, label_widget, frame._device)
	else:
		box_widget.set_visible(False)


def _update_receiver_box(frame, receiver):
	label, pairing_icon, toolbar, info_label = ui.find_children(frame, 'label', 'pairing-icon', 'toolbar', 'info-label')

	if receiver.status is None:
		frame._device = None
		label.set_text('No receiver found.')
		pairing_icon.set_visible(False)
		toolbar.set_visible(False)
		toolbar.get_children()[0].set_active(False)
		info_label.set_text('')
	else:
		frame._device = receiver
		label.set_text(str(receiver.status))
		pairing_icon.set_visible(receiver.status.lock_open)
		toolbar.set_visible(True)


def _update_device_box(frame, dev):
	# print (dev.name, dev.kind)

	icon, label, info_label = ui.find_children(frame, 'icon', 'label', 'info-label')

	first_run = frame.get_name() != dev.name
	if first_run:
		frame._device = dev
		frame.set_name(dev.name)
		icon_name = ui.get_icon(dev.name, dev.kind)
		icon.set_from_icon_name(icon_name, _DEVICE_ICON_SIZE)
		label.set_markup('<b>' + dev.name + '</b>')

	status_icons = ui.find_children(frame, 'status').get_children()
	battery_icon, battery_label, light_icon, light_label, not_encrypted_icon = status_icons[0:5]

	battery_level = dev.status.get(_status.BATTERY_LEVEL)

	if not dev.status:
		label.set_sensitive(False)

		battery_icon.set_sensitive(False)
		battery_label.set_sensitive(False)
		if battery_level is None:
			battery_label.set_markup('<small>inactive</small>')
		else:
			battery_label.set_markup('%d%%' % battery_level)

		for c in status_icons[2:-1]:
			c.set_visible(False)

	else:
		label.set_sensitive(True)

		if battery_level is None:
			battery_icon.set_sensitive(False)
			battery_icon.set_from_icon_name('battery_unknown', _STATUS_ICON_SIZE)
			text = 'no status' if dev.protocol < 2.0 else 'waiting for status...'
			battery_label.set_markup('<small>%s</small>' % text)
			battery_label.set_sensitive(True)
		else:
			battery_icon.set_from_icon_name(ui.get_battery_icon(battery_level), _STATUS_ICON_SIZE)
			battery_icon.set_sensitive(True)
			battery_label.set_text('%d%%' % battery_level)
			battery_label.set_sensitive(True)

		battery_status = dev.status.get(_status.BATTERY_STATUS)
		battery_icon.set_tooltip_text(battery_status or '')

		light_level = dev.status.get(_status.LIGHT_LEVEL)
		if light_level is None:
			light_icon.set_visible(False)
			light_label.set_visible(False)
		else:
			icon_name = 'light_%03d' % (20 * ((light_level + 50) // 100))
			light_icon.set_from_icon_name(icon_name, _STATUS_ICON_SIZE)
			light_icon.set_visible(True)
			light_label.set_text('%d lux' % light_level)
			light_label.set_visible(True)

	not_encrypted_icon.set_visible(dev.status.get(_status.ENCRYPTED) == False)

	if first_run:
		frame.set_visible(True)
		GObject.timeout_add(5000, _update_device_info_label, info_label, dev)


def update(window, receiver, device=None):
	# print ("update", receiver, receiver.status, device)
	assert receiver is not None
	window.set_icon_name(ui.appicon(receiver.status))

	vbox = window.get_child()
	frames = list(vbox.get_children())

	if device is None:
		_update_receiver_box(frames[0], receiver)
		if not receiver.status:
			for frame in frames[1:]:
				frame.set_visible(False)
				frame.set_name(_PLACEHOLDER)
				frame._device = None
	else:
		frame = frames[device.number]
		if device.status is None:
			frame.set_visible(False)
			frame.set_name(_PLACEHOLDER)
			frame._device = None
		else:
			_update_device_box(frame, device)