327 lines
12 KiB
Python
327 lines
12 KiB
Python
# vim: ft=python fileencoding=utf-8 sts=4 sw=4 et:
|
|
|
|
# Copyright 2014-2018 Florian Bruhin (The Compiler) <mail@qutebrowser.org>
|
|
#
|
|
# This file is part of qutebrowser.
|
|
#
|
|
# qutebrowser 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.
|
|
#
|
|
# qutebrowser 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 qutebrowser. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""KeyChainParser for "hint" and "normal" modes.
|
|
|
|
Module attributes:
|
|
STARTCHARS: Possible chars for starting a commandline input.
|
|
"""
|
|
|
|
import traceback
|
|
import enum
|
|
|
|
from PyQt5.QtCore import pyqtSlot, Qt
|
|
|
|
from qutebrowser.commands import cmdexc
|
|
from qutebrowser.config import config
|
|
from qutebrowser.keyinput import keyparser
|
|
from qutebrowser.utils import usertypes, log, message, objreg, utils
|
|
|
|
|
|
STARTCHARS = ":/?"
|
|
LastPress = enum.Enum('LastPress', ['none', 'filtertext', 'keystring'])
|
|
|
|
|
|
class NormalKeyParser(keyparser.CommandKeyParser):
|
|
|
|
"""KeyParser for normal mode with added STARTCHARS detection and more.
|
|
|
|
Attributes:
|
|
_partial_timer: Timer to clear partial keypresses.
|
|
"""
|
|
|
|
def __init__(self, win_id, parent=None):
|
|
super().__init__(win_id, parent, supports_count=True,
|
|
supports_chains=True)
|
|
self._read_config('normal')
|
|
self._partial_timer = usertypes.Timer(self, 'partial-match')
|
|
self._partial_timer.setSingleShot(True)
|
|
self._inhibited = False
|
|
self._inhibited_timer = usertypes.Timer(self, 'normal-inhibited')
|
|
self._inhibited_timer.setSingleShot(True)
|
|
|
|
def __repr__(self):
|
|
return utils.get_repr(self)
|
|
|
|
def _handle_single_key(self, e):
|
|
"""Override _handle_single_key to abort if the key is a startchar.
|
|
|
|
Args:
|
|
e: the KeyPressEvent from Qt.
|
|
|
|
Return:
|
|
A self.Match member.
|
|
"""
|
|
txt = e.text().strip()
|
|
if self._inhibited:
|
|
self._debug_log("Ignoring key '{}', because the normal mode is "
|
|
"currently inhibited.".format(txt))
|
|
return self.Match.none
|
|
match = super()._handle_single_key(e)
|
|
if match == self.Match.partial:
|
|
timeout = config.val.input.partial_timeout
|
|
if timeout != 0:
|
|
self._partial_timer.setInterval(timeout)
|
|
self._partial_timer.timeout.connect(self._clear_partial_match)
|
|
self._partial_timer.start()
|
|
return match
|
|
|
|
def set_inhibited_timeout(self, timeout):
|
|
"""Ignore keypresses for the given duration."""
|
|
if timeout != 0:
|
|
self._debug_log("Inhibiting the normal mode for {}ms.".format(
|
|
timeout))
|
|
self._inhibited = True
|
|
self._inhibited_timer.setInterval(timeout)
|
|
self._inhibited_timer.timeout.connect(self._clear_inhibited)
|
|
self._inhibited_timer.start()
|
|
|
|
@pyqtSlot()
|
|
def _clear_partial_match(self):
|
|
"""Clear a partial keystring after a timeout."""
|
|
self._debug_log("Clearing partial keystring {}".format(
|
|
self._keystring))
|
|
self._keystring = ''
|
|
self.keystring_updated.emit(self._keystring)
|
|
|
|
@pyqtSlot()
|
|
def _clear_inhibited(self):
|
|
"""Reset inhibition state after a timeout."""
|
|
self._debug_log("Releasing inhibition state of normal mode.")
|
|
self._inhibited = False
|
|
|
|
@pyqtSlot()
|
|
def _stop_timers(self):
|
|
super()._stop_timers()
|
|
self._partial_timer.stop()
|
|
try:
|
|
self._partial_timer.timeout.disconnect(self._clear_partial_match)
|
|
except TypeError:
|
|
# no connections
|
|
pass
|
|
self._inhibited_timer.stop()
|
|
try:
|
|
self._inhibited_timer.timeout.disconnect(self._clear_inhibited)
|
|
except TypeError:
|
|
# no connections
|
|
pass
|
|
|
|
|
|
class PromptKeyParser(keyparser.CommandKeyParser):
|
|
|
|
"""KeyParser for yes/no prompts."""
|
|
|
|
def __init__(self, win_id, parent=None):
|
|
super().__init__(win_id, parent, supports_count=False,
|
|
supports_chains=True)
|
|
# We don't want an extra section for this in the config, so we just
|
|
# abuse the prompt section.
|
|
self._read_config('prompt')
|
|
|
|
def __repr__(self):
|
|
return utils.get_repr(self)
|
|
|
|
|
|
class HintKeyParser(keyparser.CommandKeyParser):
|
|
|
|
"""KeyChainParser for hints.
|
|
|
|
Attributes:
|
|
_filtertext: The text to filter with.
|
|
_last_press: The nature of the last keypress, a LastPress member.
|
|
"""
|
|
|
|
def __init__(self, win_id, parent=None):
|
|
super().__init__(win_id, parent, supports_count=False,
|
|
supports_chains=True)
|
|
self._filtertext = ''
|
|
self._last_press = LastPress.none
|
|
self._read_config('hint')
|
|
self.keystring_updated.connect(self.on_keystring_updated)
|
|
|
|
def _handle_special_key(self, e):
|
|
"""Override _handle_special_key to handle string filtering.
|
|
|
|
Return True if the keypress has been handled, and False if not.
|
|
|
|
Args:
|
|
e: the KeyPressEvent from Qt.
|
|
|
|
Return:
|
|
True if event has been handled, False otherwise.
|
|
"""
|
|
log.keyboard.debug("Got special key 0x{:x} text {}".format(
|
|
e.key(), e.text()))
|
|
hintmanager = objreg.get('hintmanager', scope='tab',
|
|
window=self._win_id, tab='current')
|
|
if e.key() == Qt.Key_Backspace:
|
|
log.keyboard.debug("Got backspace, mode {}, filtertext '{}', "
|
|
"keystring '{}'".format(self._last_press,
|
|
self._filtertext,
|
|
self._keystring))
|
|
if self._last_press == LastPress.filtertext and self._filtertext:
|
|
self._filtertext = self._filtertext[:-1]
|
|
hintmanager.filter_hints(self._filtertext)
|
|
return True
|
|
elif self._last_press == LastPress.keystring and self._keystring:
|
|
self._keystring = self._keystring[:-1]
|
|
self.keystring_updated.emit(self._keystring)
|
|
if not self._keystring and self._filtertext:
|
|
# Switch back to hint filtering mode (this can happen only
|
|
# in numeric mode after the number has been deleted).
|
|
hintmanager.filter_hints(self._filtertext)
|
|
self._last_press = LastPress.filtertext
|
|
return True
|
|
else:
|
|
return super()._handle_special_key(e)
|
|
elif hintmanager.current_mode() != 'number':
|
|
return super()._handle_special_key(e)
|
|
elif not e.text():
|
|
return super()._handle_special_key(e)
|
|
else:
|
|
self._filtertext += e.text()
|
|
hintmanager.filter_hints(self._filtertext)
|
|
self._last_press = LastPress.filtertext
|
|
return True
|
|
|
|
def handle(self, e):
|
|
"""Handle a new keypress and call the respective handlers.
|
|
|
|
Args:
|
|
e: the KeyPressEvent from Qt
|
|
|
|
Returns:
|
|
True if the match has been handled, False otherwise.
|
|
"""
|
|
match = self._handle_single_key(e)
|
|
if match == self.Match.partial:
|
|
self.keystring_updated.emit(self._keystring)
|
|
self._last_press = LastPress.keystring
|
|
return True
|
|
elif match == self.Match.definitive:
|
|
self._last_press = LastPress.none
|
|
return True
|
|
elif match == self.Match.other:
|
|
return None
|
|
elif match == self.Match.none:
|
|
# We couldn't find a keychain so we check if it's a special key.
|
|
return self._handle_special_key(e)
|
|
else:
|
|
raise ValueError("Got invalid match type {}!".format(match))
|
|
|
|
def execute(self, cmdstr, keytype, count=None):
|
|
"""Handle a completed keychain."""
|
|
if not isinstance(keytype, self.Type):
|
|
raise TypeError("Type {} is no Type member!".format(keytype))
|
|
if keytype == self.Type.chain:
|
|
hintmanager = objreg.get('hintmanager', scope='tab',
|
|
window=self._win_id, tab='current')
|
|
hintmanager.handle_partial_key(cmdstr)
|
|
else:
|
|
# execute as command
|
|
super().execute(cmdstr, keytype, count)
|
|
|
|
def update_bindings(self, strings, preserve_filter=False):
|
|
"""Update bindings when the hint strings changed.
|
|
|
|
Args:
|
|
strings: A list of hint strings.
|
|
preserve_filter: Whether to keep the current value of
|
|
`self._filtertext`.
|
|
"""
|
|
self.bindings = {s: s for s in strings}
|
|
if not preserve_filter:
|
|
self._filtertext = ''
|
|
|
|
@pyqtSlot(str)
|
|
def on_keystring_updated(self, keystr):
|
|
"""Update hintmanager when the keystring was updated."""
|
|
hintmanager = objreg.get('hintmanager', scope='tab',
|
|
window=self._win_id, tab='current')
|
|
hintmanager.handle_partial_key(keystr)
|
|
|
|
|
|
class CaretKeyParser(keyparser.CommandKeyParser):
|
|
|
|
"""KeyParser for caret mode."""
|
|
|
|
passthrough = True
|
|
|
|
def __init__(self, win_id, parent=None):
|
|
super().__init__(win_id, parent, supports_count=True,
|
|
supports_chains=True)
|
|
self._read_config('caret')
|
|
|
|
|
|
class RegisterKeyParser(keyparser.CommandKeyParser):
|
|
|
|
"""KeyParser for modes that record a register key.
|
|
|
|
Attributes:
|
|
_mode: One of KeyMode.set_mark, KeyMode.jump_mark, KeyMode.record_macro
|
|
and KeyMode.run_macro.
|
|
"""
|
|
|
|
def __init__(self, win_id, mode, parent=None):
|
|
super().__init__(win_id, parent, supports_count=False,
|
|
supports_chains=False)
|
|
self._mode = mode
|
|
self._read_config('register')
|
|
|
|
def handle(self, e):
|
|
"""Override handle to always match the next key and use the register.
|
|
|
|
Args:
|
|
e: the KeyPressEvent from Qt.
|
|
|
|
Return:
|
|
True if event has been handled, False otherwise.
|
|
"""
|
|
if super().handle(e):
|
|
return True
|
|
|
|
key = e.text()
|
|
|
|
if key == '' or utils.keyevent_to_string(e) is None:
|
|
# this is not a proper register key, let it pass and keep going
|
|
return False
|
|
|
|
tabbed_browser = objreg.get('tabbed-browser', scope='window',
|
|
window=self._win_id)
|
|
macro_recorder = objreg.get('macro-recorder')
|
|
|
|
try:
|
|
if self._mode == usertypes.KeyMode.set_mark:
|
|
tabbed_browser.set_mark(key)
|
|
elif self._mode == usertypes.KeyMode.jump_mark:
|
|
tabbed_browser.jump_mark(key)
|
|
elif self._mode == usertypes.KeyMode.record_macro:
|
|
macro_recorder.record_macro(key)
|
|
elif self._mode == usertypes.KeyMode.run_macro:
|
|
macro_recorder.run_macro(self._win_id, key)
|
|
else:
|
|
raise ValueError(
|
|
"{} is not a valid register mode".format(self._mode))
|
|
except cmdexc.Error as err:
|
|
message.error(str(err), stack=traceback.format_exc())
|
|
|
|
self.request_leave.emit(self._mode, "valid register key", True)
|
|
|
|
return True
|