input-remapper/inputremapper/gui/controller.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# input-remapper - GUI for device specific keyboard mappings
# Copyright (C) 2022 sezanzeb <proxima@sezanzeb.de>
#
# This file is part of input-remapper.
#
# input-remapper 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 3 of the License, or
# (at your option) any later version.
#
# input-remapper 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 input-remapper.  If not, see <https://www.gnu.org/licenses/>.
from __future__ import annotations  # needed for the TYPE_CHECKING import

import re
from functools import partial
from typing import (
    TYPE_CHECKING,
    Optional,
    Union,
    Literal,
    Sequence,
    Dict,
    Callable,
    List,
)

from evdev.ecodes import EV_KEY, EV_REL, EV_ABS
from gi.repository import Gtk

from inputremapper.configs.mapping import MappingData, UIMapping
from inputremapper.event_combination import EventCombination
from inputremapper.exceptions import DataManagementError
from inputremapper.gui.data_manager import DataManager, DEFAULT_PRESET_NAME
from inputremapper.gui.gettext import _
from inputremapper.gui.helper import is_helper_running
from inputremapper.gui.messages.message_broker import (
    MessageBroker,
    MessageType,
)
from inputremapper.gui.messages.message_data import (
    PresetData,
    StatusData,
    CombinationRecorded,
    UserConfirmRequest,
    DoStackSwitch,
)
from inputremapper.gui.utils import CTX_APPLY, CTX_ERROR, CTX_WARNING, CTX_MAPPING
from inputremapper.injection.injector import (
    RUNNING,
    FAILED,
    NO_GRAB,
    UPGRADE_EVDEV,
    STARTING,
    STOPPED,
    InjectorState,
)
from inputremapper.input_event import InputEvent
from inputremapper.logger import logger

if TYPE_CHECKING:
    # avoids gtk import error in tests
    from inputremapper.gui.user_interface import UserInterface


MAPPING_DEFAULTS = {"target_uinput": "keyboard"}


class Controller:
    """implements the behaviour of the gui"""

    def __init__(self, message_broker: MessageBroker, data_manager: DataManager):
        self.message_broker = message_broker
        self.data_manager = data_manager
        self.gui: Optional[UserInterface] = None

        self.button_left_warn = False
        self._attach_to_events()

    def set_gui(self, gui: UserInterface):
        self.gui = gui

    def _attach_to_events(self) -> None:
        self.message_broker.subscribe(MessageType.groups, self._on_groups_changed)
        self.message_broker.subscribe(MessageType.preset, self._on_preset_changed)
        self.message_broker.subscribe(MessageType.init, self._on_init)
        self.message_broker.subscribe(
            MessageType.preset, self._send_mapping_errors_as_status_msg
        )
        self.message_broker.subscribe(
            MessageType.mapping, self._send_mapping_errors_as_status_msg
        )

    def _on_init(self, __):
        """initialize the gui and the data_manager"""
        # make sure we get a groups_changed event when everything is ready
        # this might not be necessary if the helper takes longer to provide the
        # initial groups
        self.data_manager.send_groups()
        self.data_manager.send_uinputs()
        if not is_helper_running():
            self.show_status(CTX_ERROR, _("The helper did not start"))

    def _on_groups_changed(self, _):
        """load the newest group as soon as everyone got notified
        about the updated groups"""
        group_key = self.get_a_group()
        if group_key:
            self.load_group(self.get_a_group())

    def _on_preset_changed(self, data: PresetData):
        """load a mapping as soon as everyone got notified about the new preset"""
        if data.mappings:
            mappings = list(data.mappings)
            mappings.sort(
                key=lambda mapping: (
                    mapping.format_name() or mapping.event_combination.beautify()
                )
            )
            combination = mappings[0].event_combination
            self.load_mapping(combination)
            self.load_event(combination[0])
        else:
            # send an empty mapping to make sure the ui is reset to default values
            self.message_broker.send(MappingData(**MAPPING_DEFAULTS))

    def _on_combination_recorded(self, data: CombinationRecorded):
        self.update_combination(data.combination)

    def _send_mapping_errors_as_status_msg(self, *__):
        """send mapping ValidationErrors to the MessageBroker."""
        if not self.data_manager.active_preset:
            return
        if self.data_manager.active_preset.is_valid():
            self.message_broker.send(StatusData(CTX_MAPPING))
            return

        for mapping in self.data_manager.active_preset:
            if not mapping.get_error():
                continue

            position = mapping.format_name()
            msg = _("Mapping error at %s, hover for info") % position
            self.show_status(CTX_MAPPING, msg, self._get_ui_error_string(mapping))

    @staticmethod
    def _get_ui_error_string(mapping: UIMapping) -> str:
        """get a human readable error message from a mapping error"""
        error_string = str(mapping.get_error())

        # check all the different error messages which are not useful for the user
        if (
            "output_symbol is a macro:" in error_string
            or "output_symbol and output_code mismatch:" in error_string
        ) and mapping.event_combination.has_input_axis():
            return _(
                "Remove the macro or key from the macro input field "
                "when specifying an analog output"
            )
        elif (
            "output_symbol is a macro:" in error_string
            or "output_symbol and output_code mismatch:" in error_string
        ) and not mapping.event_combination.has_input_axis():
            return _(
                "Remove the Analog Output Axis "
                "when specifying an macro or key output"
            )
        if "missing output axis:" in error_string:
            message = _(
                "The input specifies an analog axis, but no output axis is selected."
            )
            if mapping.output_symbol is not None:
                event = [
                    event for event in mapping.event_combination if event.value == 0
                ][0]
                message += _(
                    "\nIf you mean to create a key or macro mapping "
                    "go to the advanced input configuration"
                    ' and set a "Trigger Threshold" for '
                    f'"{event.description()}"'
                )
            return message

        if "missing macro or key:" in error_string and mapping.output_symbol is None:
            message = _(
                "The input specifies a key or macro input, but no macro or key is "
                "programmed."
            )
            if mapping.output_type in (EV_ABS, EV_REL):
                message += _(
                    "\nIf you mean to create an analog axis mapping go to the "
                    'advanced input configuration and set a input to "Use as Analog".'
                )
            return message

        return error_string

    def get_a_preset(self) -> str:
        """attempts to get the newest preset in the current group
        creates a new preset if that fails"""
        try:
            return self.data_manager.get_newest_preset_name()
        except FileNotFoundError:
            pass
        self.data_manager.create_preset(self.data_manager.get_available_preset_name())
        return self.data_manager.get_newest_preset_name()

    def get_a_group(self) -> Optional[str]:
        """attempts to get the group with the newest preset
        returns any if that fails"""
        try:
            return self.data_manager.get_newest_group_key()
        except FileNotFoundError:
            pass

        keys = self.data_manager.get_group_keys()
        return keys[0] if keys else None

    def copy_preset(self):
        """create a copy of the active preset and name it `preset_name copy`"""
        name = self.data_manager.active_preset.name
        match = re.search(" copy *\d*$", name)
        if match:
            name = name[: match.start()]

        self.data_manager.copy_preset(
            self.data_manager.get_available_preset_name(f"{name} copy")
        )
        self.message_broker.send(DoStackSwitch(1))

    def update_combination(self, combination: EventCombination):
        """update the event_combination of the active mapping"""
        try:
            self.data_manager.update_mapping(event_combination=combination)
            self.save()
        except KeyError:
            self.show_status(
                CTX_MAPPING,
                f'"{combination.beautify()}" already mapped to something else',
            )
            return

        if combination.is_problematic():
            self.show_status(
                CTX_WARNING,
                _("ctrl, alt and shift may not combine properly"),
                _("Your system might reinterpret combinations ")
                + _("with those after they are injected, and by doing so ")
                + _("break them."),
            )

    def move_event_in_combination(
        self, event: InputEvent, direction: Union[Literal["up"], Literal["down"]]
    ):
        """move the active_event up or down in the event_combination of the
        active_mapping"""
        if (
            not self.data_manager.active_mapping
            or len(self.data_manager.active_mapping.event_combination) == 1
        ):
            return
        combination: Sequence[
            InputEvent
        ] = self.data_manager.active_mapping.event_combination

        i = combination.index(event)
        if (
            i + 1 == len(combination)
            and direction == "down"
            or i == 0
            and direction == "up"
        ):
            return

        if direction == "up":
            combination = (
                list(combination[: i - 1])
                + [event]
                + [combination[i - 1]]
                + list(combination[i + 1 :])
            )
        elif direction == "down":
            combination = (
                list(combination[:i])
                + [combination[i + 1]]
                + [event]
                + list(combination[i + 2 :])
            )
        else:
            raise ValueError(f"unknown direction: {direction}")
        self.update_combination(EventCombination(combination))
        self.load_event(event)

    def load_event(self, event: InputEvent):
        """load an InputEvent form the active mapping event combination"""
        self.data_manager.load_event(event)

    def update_event(self, new_event: InputEvent):
        """modify the active event"""
        try:
            self.data_manager.update_event(new_event)
        except KeyError:
            # we need to synchronize the gui
            self.data_manager.send_mapping()
            self.data_manager.send_event()

    def remove_event(self):
        """remove the active InputEvent from the active mapping event combination"""
        if not self.data_manager.active_mapping or not self.data_manager.active_event:
            return

        combination = list(self.data_manager.active_mapping.event_combination)
        combination.remove(self.data_manager.active_event)
        try:
            self.data_manager.update_mapping(
                event_combination=EventCombination(combination)
            )
            self.load_event(combination[0])
            self.save()
        except (KeyError, ValueError):
            # we need to synchronize the gui
            self.data_manager.send_mapping()
            self.data_manager.send_event()

    def set_event_as_analog(self, analog: bool):
        """use the active event as an analog input"""
        assert self.data_manager.active_event is not None
        event = self.data_manager.active_event
        if event.type == EV_KEY:
            pass

        elif analog:
            try:
                self.data_manager.update_event(event.modify(value=0))
                return
            except KeyError:
                pass
        else:
            try_values = {EV_REL: [1, -1], EV_ABS: [10, -10]}
            for value in try_values[event.type]:
                try:
                    self.data_manager.update_event(event.modify(value=value))
                    return
                except KeyError:
                    pass

        # didn't update successfully
        # we need to synchronize the gui
        self.data_manager.send_mapping()
        self.data_manager.send_event()

    def load_groups(self):
        """refresh the groups"""
        self.data_manager.refresh_groups()

    def load_group(self, group_key: str):
        """load the group and then a preset of that group"""
        self.data_manager.load_group(group_key)
        self.load_preset(self.get_a_preset())

    def load_preset(self, name: str):
        """load the preset"""
        self.data_manager.load_preset(name)
        # self.load_mapping(...) # not needed because we have on_preset_changed()

    def rename_preset(self, new_name: str):
        """rename the active_preset"""
        if (
            not self.data_manager.active_preset
            or not new_name
            or new_name == self.data_manager.active_preset.name
        ):
            return
        name = self.data_manager.get_available_preset_name(new_name)
        self.data_manager.rename_preset(name)

    def add_preset(self, name: str = DEFAULT_PRESET_NAME):
        """create a new preset, add it to the active_group and name it `new preset n`"""
        name = self.data_manager.get_available_preset_name(name)
        try:
            self.data_manager.create_preset(name)
            self.data_manager.load_preset(name)
        except PermissionError as e:
            self.show_status(CTX_ERROR, _("Permission denied!"), str(e))

    def delete_preset(self):
        """delete the active_preset from the disc"""

        def f(answer: bool):
            if answer:
                self.data_manager.delete_preset()
                self.data_manager.load_preset(self.get_a_preset())
                self.message_broker.send(DoStackSwitch(1))

        if not self.data_manager.active_preset:
            return
        msg = (
            _('Are you sure you want to delete the preset "%s"?')
            % self.data_manager.active_preset.name
        )
        self.message_broker.send(UserConfirmRequest(msg, f))

    def load_mapping(self, event_combination: EventCombination):
        """load the mapping with the given event_combination form the active_preset"""
        self.data_manager.load_mapping(event_combination)
        self.load_event(event_combination[0])

    def update_mapping(self, **kwargs):
        """update the active_mapping with the given keywords and values"""
        if "mapping_type" in kwargs.keys():
            if not (kwargs := self._change_mapping_type(kwargs)):
                # we need to synchronize the gui
                self.data_manager.send_mapping()
                self.data_manager.send_event()
                return

        self.data_manager.update_mapping(**kwargs)
        self.save()

    def create_mapping(self):
        """create a new empty mapping in the active_preset"""
        try:
            self.data_manager.create_mapping()
        except KeyError:
            # there is already an empty mapping
            return

        self.data_manager.load_mapping(combination=EventCombination.empty_combination())
        self.data_manager.update_mapping(**MAPPING_DEFAULTS)

    def delete_mapping(self):
        """remove the active_mapping form the active_preset"""

        def f(answer: bool):
            if answer:
                self.data_manager.delete_mapping()
                self.save()

        if not self.data_manager.active_mapping:
            return
        self.message_broker.send(
            UserConfirmRequest(_("Are you sure you want to delete this mapping?"), f)
        )

    def set_autoload(self, autoload: bool):
        """set the autoload state for the active_preset and active_group"""
        self.data_manager.set_autoload(autoload)
        self.data_manager.refresh_service_config_path()

    def save(self):
        """save all data to the disc"""
        try:
            self.data_manager.save()
        except PermissionError as e:
            self.show_status(CTX_ERROR, _("Permission denied!"), str(e))

    def start_key_recording(self):
        """Record the input of the active_group

        Updates the active_mapping.event_combination with the recorded events.
        """
        self.message_broker.signal(MessageType.recording_started)  # TODO test

        state = self.data_manager.get_state()
        if state == RUNNING or state == STARTING:
            self.message_broker.signal(MessageType.recording_finished)
            self.show_status(CTX_ERROR, _('Use "Stop" to stop before editing'))
            return

        logger.debug("Recording Keys")

        def f(_):
            self.message_broker.unsubscribe(f)
            self.message_broker.unsubscribe(self._on_combination_recorded)
            self.gui.connect_shortcuts()

        self.gui.disconnect_shortcuts()
        self.message_broker.subscribe(
            MessageType.combination_recorded, self._on_combination_recorded
        )
        self.message_broker.subscribe(MessageType.recording_finished, f)
        self.data_manager.start_combination_recording()

    def stop_key_recording(self):
        """stop recording the input"""
        logger.debug("Stopping Key recording")
        self.data_manager.stop_combination_recording()

    def start_injecting(self):
        """inject the active_preset for the active_group"""
        if len(self.data_manager.active_preset) == 0:
            logger.error(_("Cannot apply empty preset file"))
            # also helpful for first time use
            self.show_status(CTX_ERROR, _("You need to add keys and save first"))
            return

        if not self.button_left_warn:
            if self.data_manager.active_preset.dangerously_mapped_btn_left():
                self.show_status(
                    CTX_ERROR,
                    "This would disable your click button",
                    "Map a button to BTN_LEFT to avoid this.\n"
                    "To overwrite this warning, press apply again.",
                )
                self.button_left_warn = True
                return

        # todo: warn about unreleased keys
        self.button_left_warn = False
        self.message_broker.subscribe(
            MessageType.injector_state, self.show_injector_result
        )
        self.show_status(CTX_APPLY, _("Starting injection..."))
        if not self.data_manager.start_injecting():
            self.message_broker.unsubscribe(self.show_injector_result)
            self.show_status(
                CTX_APPLY,
                _("Failed to apply preset %s") % self.data_manager.active_preset.name,
            )

    def show_injector_result(self, msg: InjectorState):
        """Show if the injection was successfully started."""
        self.message_broker.unsubscribe(self.show_injector_result)
        state = msg.state

        def running():
            msg = _("Applied preset %s") % self.data_manager.active_preset.name
            if self.data_manager.active_preset.get_mapping(
                EventCombination(InputEvent.btn_left())
            ):
                msg += _(", CTRL + DEL to stop")
            self.show_status(CTX_APPLY, msg)
            logger.info(
                'Group "%s" is currently mapped', self.data_manager.active_group.key
            )

        assert self.data_manager.active_preset  # make mypy happy
        state_calls: Dict[int, Callable] = {
            RUNNING: running,
            FAILED: partial(
                self.show_status,
                CTX_ERROR,
                _("Failed to apply preset %s") % self.data_manager.active_preset.name,
            ),
            NO_GRAB: partial(
                self.show_status,
                CTX_ERROR,
                "The device was not grabbed",
                "Either another application is already grabbing it or "
                "your preset doesn't contain anything that is sent by the "
                "device.",
            ),
            UPGRADE_EVDEV: partial(
                self.show_status,
                CTX_ERROR,
                "Upgrade python-evdev",
                "Your python-evdev version is too old.",
            ),
        }
        state_calls[state]()

    def stop_injecting(self):
        """stop injecting any preset for the active_group"""

        def show_result(msg: InjectorState):
            self.message_broker.unsubscribe(show_result)
            assert msg.state == STOPPED
            self.show_status(CTX_APPLY, _("Applied the system default"))

        try:
            self.message_broker.subscribe(MessageType.injector_state, show_result)
            self.data_manager.stop_injecting()
        except DataManagementError:
            self.message_broker.unsubscribe(show_result)

    def show_status(
        self, ctx_id: int, msg: Optional[str] = None, tooltip: Optional[str] = None
    ):
        """send a status message to the ui to show it in the status-bar"""
        self.message_broker.send(StatusData(ctx_id, msg, tooltip))

    def is_empty_mapping(self) -> bool:
        """check if the active_mapping is empty"""
        return (
            self.data_manager.active_mapping == UIMapping(**MAPPING_DEFAULTS)
            or self.data_manager.active_mapping is None
        )

    def refresh_groups(self):
        """reload the connected devices and send them as a groups message

        runs asynchronously"""
        self.data_manager.refresh_groups()

    def close(self):
        """safely close the application"""
        logger.debug("Closing Application")
        self.save()
        self.message_broker.signal(MessageType.terminate)
        logger.debug("Quitting")
        Gtk.main_quit()

    def set_focus(self, component):
        """focus the given component"""
        self.gui.window.set_focus(component)

    def _change_mapping_type(self, kwargs):
        """query the user to update the mapping in order to change the mapping type"""
        mapping = self.data_manager.active_mapping

        if mapping is None:
            return kwargs

        if kwargs["mapping_type"] == mapping.mapping_type:
            return kwargs

        if kwargs["mapping_type"] == "analog":
            msg = f"You are about to change the mapping to analog."
            if mapping.output_symbol:
                msg += (
                    f'\nThis will remove "{mapping.output_symbol}" '
                    f"from the text input!"
                )

            if not [e for e in mapping.event_combination if e.value == 0]:
                # there is no analog input configured, let's try to autoconfigure it
                events: List[InputEvent] = list(mapping.event_combination)
                for i, e in enumerate(events):
                    if e.type in [EV_ABS, EV_REL]:
                        events[i] = e.modify(value=0)
                        kwargs["event_combination"] = EventCombination(events)
                        msg += (
                            f'\nThe input "{e.description()}" '
                            f"will be used as analog input."
                        )
                        break
                else:
                    # not possible to autoconfigure inform the user
                    msg += "\nYou need to record an analog input."

            elif not mapping.output_symbol:
                return kwargs

            answer = None

            def f(a: bool):
                nonlocal answer
                answer = a

            self.message_broker.send(UserConfirmRequest(msg, f))
            if answer:
                kwargs["output_symbol"] = None
                return kwargs
            else:
                return None

        if kwargs["mapping_type"] == "key_macro":
            try:
                analog_input = [e for e in mapping.event_combination if e.value == 0][0]
            except IndexError:
                kwargs["output_type"] = None
                kwargs["output_code"] = None
                return kwargs

            answer = None

            def f(a: bool):
                nonlocal answer
                answer = a

            self.message_broker.send(
                UserConfirmRequest(
                    f"You are about to change the mapping to a Key or Macro mapping!\n"
                    f"Go to the advanced input configuration and set a "
                    f'"Trigger Threshold" for "{analog_input.description()}".',
                    f,
                )
            )
            if answer:
                kwargs["output_type"] = None
                kwargs["output_code"] = None
                return kwargs
            else:
                return None

        return kwargs