/
keybinding.lua
239 lines (201 loc) · 7.68 KB
/
keybinding.lua
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
--[[- A system for handling keyboard shortcuts. Given a table of keybindings and
their corresponding action, @{metis.input.keybinding} will process events and
invoke the appropriate action when a key is pressed.
## Keybinding syntax
Keybindings are written in an Emacs-esque notation, specifying the modifier keys
and then the actual key. For instance, <kbd>C-M-underscore</kbd> means the
"Control", "Meta" (or "Alt") and "underscore" keys must be pressed.
@usage
```lua
local kb = require "metis.input.keybinding".create {
["C-M-x"] = function() print("Pressed Ctrl, Alt and 'x'") end,
char = function(c) print("Typed character " .. c) end,
}
while true do kb:event(os.pullEvent()) end
```
]]
local exp = require "cc.expect"
local expect, field = exp.expect, exp.field
local modifiers = {
[keys.leftCtrl] = true, [keys.rightCtrl] = true,
[keys.leftAlt] = true, [keys.rightAlt] = true,
[keys.leftShift] = true, [keys.rightShift] = true,
}
local function update_modifier(self)
local down = self._down
local modifier = 0
if down[keys.leftCtrl] or down[keys.rightCtrl] then modifier = modifier + 1 end
if down[keys.leftAlt] or down[keys.rightAlt] then modifier = modifier + 2 end
if down[keys.leftShift] or down[keys.rightShift] then modifier = modifier + 4 end
self._modifier = modifier
end
local function void() end
local function parse_binding(str)
local pos, len = 1, #str
local modifier = 0
while pos <= len do
local mod = str:match("^([A-Z])%-", pos)
if not mod then break end
if mod == "C" then
modifier = bit32.bor(modifier, 1)
elseif mod == "M" then
modifier = bit32.bor(modifier, 2)
elseif mod == "S" then
modifier = bit32.bor(modifier, 4)
else
error(("Unknown modifier %q in binding %q"):format(modifier, str))
end
pos = pos + 2
end
if pos > len then error(("Malformed binding %q"):format(str)) end
local name = str:sub(pos, len)
local key = keys[name]
if type(key) ~= "number" then
error(("Unknown key %q in binding %q"):format(name, str))
end
return modifier, key
end
--- A mapping of various key combinations to functions.
--
-- This is used by a @{Keybindings} instance to dispatch
--
-- @type Keymap
-- @see create_keymap to create a new @{Keymap}.
local Keymap = {}
local keymap_mt = { __name = "Keymap", __index = Keymap }
--[[- Create a new keymap.
@tparam { [string] = function }|Keymap ... One or more tables of key bindings. These
can either be:
- An existing @{Keymap} instance or.
- A mapping of keybinding names (as described in the module description) to functions.
The special name "char" may be used when a character is typed.
Later keybindings override existing ones.
@tparam[2] Keymap ... An existing keymap which this one will extend.
@treturn Keymap The constructed keymap handler.
]]
local function create_keymap(...)
local bindings = { [0] = {}, {}, {}, {}, {}, {}, {}, {} }
local meta_chars = {}
local char = void
for i = 1, select('#', ...) do
local arg = select(i, ...)
expect(i, arg, "table")
if getmetatable(arg) == keymap_mt then
for mod, mod_bindings in pairs(arg._bindings) do
for key, fn in pairs(mod_bindings) do bindings[mod][key] = fn end
end
for k in pairs(arg._meta_chars) do meta_chars[k] = true end
if arg._on_char and arg._on_char ~= void then char = arg._on_char end
else
char = field(arg, "char", "function", "nil") or char
local local_bindings = { [0] = {}, {}, {}, {}, {}, {}, {}, {} }
for k in pairs(arg) do
if k ~= "char" then
if type(k) ~= "string" then error(("Bad key %s of type %s"):format(k, type(k)), 2) end
local action = field(arg, k, "function")
local mod, key = parse_binding(k)
if local_bindings[mod][key] then error(("Duplicate bindings for %s"):format(k), 2) end
local_bindings[mod][key] = true
bindings[mod][key] = action
if mod == 2 then
local name = keys.getName(key)
if name and #name == 1 then meta_chars[name] = true end
end
end
end
end
end
return setmetatable({ _bindings = bindings, _meta_chars = meta_chars, _on_char = char }, keymap_mt)
end
local empty_keymap = create_keymap()
--- The keybinding processor. This accepts events, determines what keys are
-- pressed, and dispatches the appropriate action.
--
-- @type Keybindings
local Keybindings = {}
local keybindings_mt = { __name = Keybindings, __index = Keybindings }
--- Process a `char` event.
--
-- @tparam string chr The character which has been typed.
-- @param ... Additional arguments which will be passed to the associated action.
-- @return The result of the associated keybinding's action, or @{nil}.
function Keybindings:char(chr, ...)
-- If we've got the alt key down, and we're not pressing a key which is used
-- in binding with alt, treat it as a normal char. This is a little ugly, as
-- it means that unhandled keys are "typed", but ensures that Alt Gr correctly
-- types characters.
local modifier = self._modifier
local keymap = self._keymap
if bit32.band(modifier, 2) == 0 or not keymap._meta_chars[chr] then
return keymap._on_char(chr, ...)
end
end
--- Process a `key` event.
--
-- @tparam number key The key which has been pressed.
-- @param ... Additional arguments which will be passed to the associated action.
-- @return The result of the associated keybinding's action, or @{nil}.
function Keybindings:key(key, ...)
if modifiers[key] then
self._down[key] = true
update_modifier(self)
end
local fn = self._keymap._bindings[self._modifier][key]
if fn then return fn(...) end
end
--- Process a `key_up` event.
--
-- @tparam number key The key which has been released.
function Keybindings:key_up(key)
if modifiers[key] then
self._down[key] = false
update_modifier(self)
end
end
--- Process an event. This dispatches to the @{Keybindings:key},
-- @{Keybindings:key_up} or @{Keybindings:char} as appropriate.
--
-- Repeat key events will be passed to @{Keybindings:key} by default. If this is
-- not desired, your keybindings should either check if this event is a repeat
-- (the first argument will be @{true}) or roll your own version of `event`.
--
-- @tparam string event The name of the event.
-- @param ... Additional event arguments.
-- @usage kb:event(os.pullEvent())
function Keybindings:event(event, ...)
if event == "key" then
return self:key(...)
elseif event == "key_up" then
return self:key_up(...)
elseif event == "char" then
return self:char(...)
end
end
--- Update the underlying @{Keymap} of this keybindings instance.
--
-- @tparam Keymap|{ [string] = function } keymap The keymap to use for this instance.
function Keybindings:set_keymap(keymap)
expect(1, keymap, "table")
if getmetatable(keymap) ~= keymap_mt then keymap = create_keymap(keymap) end
self._keymap = keymap
end
--- Reset the set of pressed keys within the keybinding manager. This may be
-- used if a window becomes unfocused, and so `key_up` events will no longer be
-- sent to it.
function Keybindings:reset()
for k in pairs(self._down) do self._down[k] = false end
update_modifier(self)
end
--- Create a new group of keybindings.
--
-- @tparam[opt] Keymap|{ [string] = function } keymap The keymap to use for this
-- instance. Can be changed later with @{Keybindings:set_keymap}.
-- @treturn Keybindings The constructed keybinding handler.
local function create(keymap)
expect(1, keymap, "table", "nil")
if keymap == nil then keymap = empty_keymap
elseif getmetatable(keymap) ~= keymap_mt then keymap = create_keymap(keymap)
end
return setmetatable({ _down = {}, _modifier = 0, _keymap = keymap }, keybindings_mt)
end
return { create_keymap = create_keymap, create = create }