## 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.

import logging

from gi.repository import GLib
from gi.repository import Gtk
from logitech_receiver import hidpp10_constants as _hidpp10_constants

from solaar.i18n import _
from solaar.i18n import ngettext

from . import icons as _icons

logger = logging.getLogger(__name__)

#
#
#
_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 logger.isEnabledFor(logging.DEBUG):
            logger.debug("assistant %s destroyed, bailing out", assistant)
        return False

    if receiver.pairing.error:
        # receiver.pairing.new_device = _fake_device(receiver)
        _pairing_failed(assistant, receiver, receiver.pairing.error)
        receiver.pairing.error = None
        return False

    if receiver.pairing.new_device:
        receiver.remaining_pairings(False)  # Update remaining pairings
        device, receiver.pairing.new_device = receiver.pairing.new_device, None
        _pairing_succeeded(assistant, receiver, device)
        return False
    elif receiver.pairing.device_address and receiver.pairing.device_name and not address:
        address = receiver.pairing.device_address
        name = receiver.pairing.device_name
        kind = receiver.pairing.device_kind
        authentication = receiver.pairing.device_authentication
        name = receiver.pairing.device_name
        entropy = 10
        if kind == _hidpp10_constants.DEVICE_KIND.keyboard:
            entropy = 20
        if receiver.pair_device(
            address=address,
            authentication=authentication,
            entropy=entropy,
        ):
            return True
        else:
            _pairing_failed(assistant, receiver, "failed to open pairing lock")
            return False
    elif address and receiver.pairing.device_passkey and not passcode:
        passcode = receiver.pairing.device_passkey
        _show_passcode(assistant, receiver, passcode)
        return True

    if not receiver.pairing.lock_open and not receiver.pairing.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 logger.isEnabledFor(logging.DEBUG):
        logger.debug("%s show passkey: %s", receiver, passkey)
    name = receiver.pairing.device_name
    authentication = receiver.pairing.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.pairing.device_passkey}
    else:
        passcode = ", ".join(
            [_("right") if bit == "1" else _("left") for bit in f"{int(receiver.pairing.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 logger.isEnabledFor(logging.DEBUG):
        logger.debug("prepare %s %d %s", assistant, index, page)

    if index == 0:
        if receiver.receiver_kind == "bolt":
            if receiver.discover(timeout=_PAIRING_TIMEOUT):
                assert receiver.pairing.new_device is None
                assert receiver.pairing.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.pairing.new_device is None
            assert receiver.pairing.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 logger.isEnabledFor(logging.DEBUG):
        logger.debug("finish %s", assistant)
    assistant.destroy()
    receiver.pairing.new_device = None
    if receiver.pairing.lock_open:
        if receiver.receiver_kind == "bolt":
            receiver.pair_device("cancel")
        else:
            receiver.set_lock()
    if receiver.pairing.discovering:
        receiver.discover(True)
    if not receiver.pairing.lock_open and not receiver.pairing.discovering:
        receiver.pairing.error = None


def _pairing_failed(assistant, receiver, error):
    if logger.isEnabledFor(logging.DEBUG):
        logger.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 logger.isEnabledFor(logging.DEBUG):
        logger.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(f"<b>{device.name}</b>")
    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.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 == "unifying":
        page_text = _("Unifying receivers are only compatible with Unifying devices.")
    elif receiver.receiver_kind == "bolt":
        page_text = _("Bolt receivers are only compatible with Bolt devices.")
    else:
        page_text = _("Other receivers are only compatible with a few devices.")
    page_text += "\n"
    page_text += _("The device must not be paired with a nearby powered-on receiver.")
    page_text += "\n\n"

    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