291 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			Python
		
	
	
	
			
		
		
	
	
			291 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			Python
		
	
	
	
| # -*- python-mode -*-
 | |
| 
 | |
| ## Copyright (C) 2012-2013  Daniel Pavel
 | |
| ##
 | |
| ## 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 2 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, write to the Free Software Foundation, Inc.,
 | |
| ## 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 | |
| 
 | |
| from logging import DEBUG as _DEBUG
 | |
| from logging import getLogger
 | |
| 
 | |
| from gi.repository import GLib, Gtk
 | |
| from logitech_receiver import hidpp10 as _hidpp10
 | |
| from logitech_receiver.status import KEYS as _K
 | |
| from solaar.i18n import _, ngettext
 | |
| 
 | |
| from . import icons as _icons
 | |
| 
 | |
| _log = getLogger(__name__)
 | |
| del getLogger
 | |
| 
 | |
| #
 | |
| #
 | |
| #
 | |
| _PAIRING_TIMEOUT = 30  # seconds
 | |
| _STATUS_CHECK = 500  # milliseconds
 | |
| 
 | |
| address = kind = authentication = name = passcode = None
 | |
| 
 | |
| 
 | |
| def _create_page(assistant, kind, header=None, icon_name=None, text=None):
 | |
|     p = Gtk.VBox(False, 8)
 | |
|     assistant.append_page(p)
 | |
|     assistant.set_page_type(p, kind)
 | |
| 
 | |
|     if header:
 | |
|         item = Gtk.HBox(False, 16)
 | |
|         p.pack_start(item, False, True, 0)
 | |
| 
 | |
|         label = Gtk.Label(header)
 | |
|         label.set_alignment(0, 0)
 | |
|         label.set_line_wrap(True)
 | |
|         item.pack_start(label, True, True, 0)
 | |
| 
 | |
|         if icon_name:
 | |
|             icon = Gtk.Image.new_from_icon_name(icon_name, Gtk.IconSize.DIALOG)
 | |
|             icon.set_alignment(1, 0)
 | |
|             item.pack_start(icon, False, False, 0)
 | |
| 
 | |
|     if text:
 | |
|         label = Gtk.Label(text)
 | |
|         label.set_alignment(0, 0)
 | |
|         label.set_line_wrap(True)
 | |
|         p.pack_start(label, False, False, 0)
 | |
| 
 | |
|     p.show_all()
 | |
|     return p
 | |
| 
 | |
| 
 | |
| def _check_lock_state(assistant, receiver, count=2):
 | |
|     global address, kind, authentication, name, passcode
 | |
| 
 | |
|     if not assistant.is_drawable():
 | |
|         if _log.isEnabledFor(_DEBUG):
 | |
|             _log.debug('assistant %s destroyed, bailing out', assistant)
 | |
|         return False
 | |
| 
 | |
|     if receiver.status.get(_K.ERROR):
 | |
|         # receiver.status.new_device = _fake_device(receiver)
 | |
|         _pairing_failed(assistant, receiver, receiver.status.pop(_K.ERROR))
 | |
|         return False
 | |
| 
 | |
|     if receiver.status.new_device:
 | |
|         device, receiver.status.new_device = receiver.status.new_device, None
 | |
|         _pairing_succeeded(assistant, receiver, device)
 | |
|         return False
 | |
|     elif receiver.status.device_address and receiver.status.device_name and not address:
 | |
|         address = receiver.status.device_address
 | |
|         name = receiver.status.device_name
 | |
|         kind = receiver.status.device_kind
 | |
|         authentication = receiver.status.device_authentication
 | |
|         name = receiver.status.device_name
 | |
|         if receiver.pair_device(
 | |
|             address=address, authentication=authentication, entropy=20 if kind == _hidpp10.DEVICE_KIND.keyboard else 10
 | |
|         ):
 | |
|             return True
 | |
|         else:
 | |
|             _pairing_failed(assistant, receiver, 'failed to open pairing lock')
 | |
|             return False
 | |
|     elif address and receiver.status.device_passkey and not passcode:
 | |
|         passcode = receiver.status.device_passkey
 | |
|         _show_passcode(assistant, receiver, passcode)
 | |
|         return True
 | |
| 
 | |
|     if not receiver.status.lock_open and not receiver.status.discovering:
 | |
|         if count > 0:
 | |
|             # the actual device notification may arrive later so have a little patience
 | |
|             GLib.timeout_add(_STATUS_CHECK, _check_lock_state, assistant, receiver, count - 1)
 | |
|         else:
 | |
|             _pairing_failed(assistant, receiver, 'failed to open pairing lock')
 | |
|         return False
 | |
| 
 | |
|     return True
 | |
| 
 | |
| 
 | |
| def _show_passcode(assistant, receiver, passkey):
 | |
|     if _log.isEnabledFor(_DEBUG):
 | |
|         _log.debug('%s show passkey: %s', receiver, passkey)
 | |
|     name = receiver.status.device_name
 | |
|     authentication = receiver.status.device_authentication
 | |
|     intro_text = _('%(receiver_name)s: pair new device') % {'receiver_name': receiver.name}
 | |
|     page_text = _('Enter passcode on %(name)s.') % {'name': name}
 | |
|     page_text += '\n'
 | |
|     if authentication & 0x01:
 | |
|         page_text += _('Type %(passcode)s and then press the enter key.') % {'passcode': receiver.status.device_passkey}
 | |
|     else:
 | |
|         passcode = ', '.join([
 | |
|             _('right') if bit == '1' else _('left') for bit in f'{int(receiver.status.device_passkey):010b}'
 | |
|         ])
 | |
|         page_text += _('Press %(code)s\nand then press left and right buttons simultaneously.') % {'code': passcode}
 | |
|     page = _create_page(assistant, Gtk.AssistantPageType.PROGRESS, intro_text, 'preferences-desktop-peripherals', page_text)
 | |
|     assistant.set_page_complete(page, True)
 | |
|     assistant.next_page()
 | |
| 
 | |
| 
 | |
| def _prepare(assistant, page, receiver):
 | |
|     index = assistant.get_current_page()
 | |
|     if _log.isEnabledFor(_DEBUG):
 | |
|         _log.debug('prepare %s %d %s', assistant, index, page)
 | |
| 
 | |
|     if index == 0:
 | |
|         if receiver.receiver_kind == 'bolt':
 | |
|             if receiver.discover(timeout=_PAIRING_TIMEOUT):
 | |
|                 assert receiver.status.new_device is None
 | |
|                 assert receiver.status.get(_K.ERROR) is None
 | |
|                 spinner = page.get_children()[-1]
 | |
|                 spinner.start()
 | |
|                 GLib.timeout_add(_STATUS_CHECK, _check_lock_state, assistant, receiver)
 | |
|                 assistant.set_page_complete(page, True)
 | |
|             else:
 | |
|                 GLib.idle_add(_pairing_failed, assistant, receiver, 'discovery did not start')
 | |
|         elif receiver.set_lock(False, timeout=_PAIRING_TIMEOUT):
 | |
|             assert receiver.status.new_device is None
 | |
|             assert receiver.status.get(_K.ERROR) is None
 | |
|             spinner = page.get_children()[-1]
 | |
|             spinner.start()
 | |
|             GLib.timeout_add(_STATUS_CHECK, _check_lock_state, assistant, receiver)
 | |
|             assistant.set_page_complete(page, True)
 | |
|         else:
 | |
|             GLib.idle_add(_pairing_failed, assistant, receiver, 'the pairing lock did not open')
 | |
|     else:
 | |
|         assistant.remove_page(0)
 | |
| 
 | |
| 
 | |
| def _finish(assistant, receiver):
 | |
|     if _log.isEnabledFor(_DEBUG):
 | |
|         _log.debug('finish %s', assistant)
 | |
|     assistant.destroy()
 | |
|     receiver.status.new_device = None
 | |
|     if receiver.status.lock_open:
 | |
|         if receiver.receiver_kind == 'bolt':
 | |
|             receiver.pair_device('cancel')
 | |
|         else:
 | |
|             receiver.set_lock()
 | |
|     if receiver.status.discovering:
 | |
|         receiver.discover(True)
 | |
|     if not receiver.status.lock_open and not receiver.status.discovering:
 | |
|         receiver.status[_K.ERROR] = None
 | |
| 
 | |
| 
 | |
| def _pairing_failed(assistant, receiver, error):
 | |
|     if _log.isEnabledFor(_DEBUG):
 | |
|         _log.debug('%s fail: %s', receiver, error)
 | |
| 
 | |
|     assistant.commit()
 | |
| 
 | |
|     header = _('Pairing failed') + ': ' + _(str(error)) + '.'
 | |
|     if 'timeout' in str(error):
 | |
|         text = _('Make sure your device is within range, and has a decent battery charge.')
 | |
|     elif str(error) == 'device not supported':
 | |
|         text = _('A new device was detected, but it is not compatible with this receiver.')
 | |
|     elif 'many' in str(error):
 | |
|         text = _('More paired devices than receiver can support.')
 | |
|     else:
 | |
|         text = _('No further details are available about the error.')
 | |
|     _create_page(assistant, Gtk.AssistantPageType.SUMMARY, header, 'dialog-error', text)
 | |
| 
 | |
|     assistant.next_page()
 | |
|     assistant.commit()
 | |
| 
 | |
| 
 | |
| def _pairing_succeeded(assistant, receiver, device):
 | |
|     assert device
 | |
|     if _log.isEnabledFor(_DEBUG):
 | |
|         _log.debug('%s success: %s', receiver, device)
 | |
| 
 | |
|     page = _create_page(assistant, Gtk.AssistantPageType.SUMMARY)
 | |
| 
 | |
|     header = Gtk.Label(_('Found a new device:'))
 | |
|     header.set_alignment(0.5, 0)
 | |
|     page.pack_start(header, False, False, 0)
 | |
| 
 | |
|     device_icon = Gtk.Image()
 | |
|     icon_set = _icons.device_icon_set(device.name, device.kind)
 | |
|     device_icon.set_from_icon_set(icon_set, Gtk.IconSize.LARGE)
 | |
|     device_icon.set_alignment(0.5, 1)
 | |
|     page.pack_start(device_icon, True, True, 0)
 | |
| 
 | |
|     device_label = Gtk.Label()
 | |
|     device_label.set_markup('<b>%s</b>' % device.name)
 | |
|     device_label.set_alignment(0.5, 0)
 | |
|     page.pack_start(device_label, True, True, 0)
 | |
| 
 | |
|     hbox = Gtk.HBox(False, 8)
 | |
|     hbox.pack_start(Gtk.Label(' '), False, False, 0)
 | |
|     hbox.set_property('expand', False)
 | |
|     hbox.set_property('halign', Gtk.Align.CENTER)
 | |
|     page.pack_start(hbox, False, False, 0)
 | |
| 
 | |
|     def _check_encrypted(dev):
 | |
|         if assistant.is_drawable():
 | |
|             if device.status.get(_K.LINK_ENCRYPTED) is False:
 | |
|                 hbox.pack_start(Gtk.Image.new_from_icon_name('security-low', Gtk.IconSize.MENU), False, False, 0)
 | |
|                 hbox.pack_start(Gtk.Label(_('The wireless link is not encrypted') + '!'), False, False, 0)
 | |
|                 hbox.show_all()
 | |
|             else:
 | |
|                 return True
 | |
| 
 | |
|     GLib.timeout_add(_STATUS_CHECK, _check_encrypted, device)
 | |
| 
 | |
|     page.show_all()
 | |
| 
 | |
|     assistant.next_page()
 | |
|     assistant.commit()
 | |
| 
 | |
| 
 | |
| def create(receiver):
 | |
|     assert receiver is not None
 | |
|     assert receiver.kind is None
 | |
| 
 | |
|     global address, kind, authentication, name, passcode
 | |
|     address = name = kind = authentication = passcode = None
 | |
| 
 | |
|     assistant = Gtk.Assistant()
 | |
|     assistant.set_title(_('%(receiver_name)s: pair new device') % {'receiver_name': receiver.name})
 | |
|     assistant.set_icon_name('list-add')
 | |
| 
 | |
|     assistant.set_size_request(400, 240)
 | |
|     assistant.set_resizable(False)
 | |
|     assistant.set_role('pair-device')
 | |
| 
 | |
|     if receiver.receiver_kind == 'bolt':
 | |
|         page_text = _('Press a pairing button or key until the pairing light flashes quickly.')
 | |
|         page_text += '\n'
 | |
|         page_text += _('You may have to first turn the device off and on again.')
 | |
|     else:
 | |
|         page_text = _('Turn on the device you want to pair.')
 | |
|         page_text += '\n'
 | |
|         page_text += _('If the device is already turned on, turn it off and on again.')
 | |
|     if receiver.remaining_pairings() and receiver.remaining_pairings() >= 0:
 | |
|         page_text += ngettext(
 | |
|             '\n\nThis receiver has %d pairing remaining.', '\n\nThis receiver has %d pairings remaining.',
 | |
|             receiver.remaining_pairings()
 | |
|         ) % receiver.remaining_pairings()
 | |
|         page_text += _('\nCancelling at this point will not use up a pairing.')
 | |
| 
 | |
|     intro_text = _('%(receiver_name)s: pair new device') % {'receiver_name': receiver.name}
 | |
| 
 | |
|     page_intro = _create_page(
 | |
|         assistant, Gtk.AssistantPageType.PROGRESS, intro_text, 'preferences-desktop-peripherals', page_text
 | |
|     )
 | |
|     spinner = Gtk.Spinner()
 | |
|     spinner.set_visible(True)
 | |
|     page_intro.pack_end(spinner, True, True, 24)
 | |
| 
 | |
|     assistant.connect('prepare', _prepare, receiver)
 | |
|     assistant.connect('cancel', _finish, receiver)
 | |
|     assistant.connect('close', _finish, receiver)
 | |
| 
 | |
|     return assistant
 |