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
|
# Copyright 2000-2008 Michael Hudson-Doyle <micahel@gmail.com>
# Armin Rigo
#
# All Rights Reserved
#
#
# Permission to use, copy, modify, and distribute this software and
# its documentation for any purpose is hereby granted without fee,
# provided that the above copyright notice appear in all copies and
# that both that copyright notice and this permission notice appear in
# supporting documentation.
#
# THE AUTHOR MICHAEL HUDSON DISCLAIMS ALL WARRANTIES WITH REGARD TO
# THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
# AND FITNESS, IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL,
# INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
# RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF
# CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
# CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
# Bah, this would be easier to test if curses/terminfo didn't have so
# much non-introspectable global state.
from pyrepl import keymap
from pyrepl.console import Event
from pyrepl import curses
from .trace import trace
from termios import tcgetattr, VERASE
import os
try:
unicode
except NameError:
unicode = str
_keynames = {
"delete" : "kdch1",
"down" : "kcud1",
"end" : "kend",
"enter" : "kent",
"f1" : "kf1", "f2" : "kf2", "f3" : "kf3", "f4" : "kf4",
"f5" : "kf5", "f6" : "kf6", "f7" : "kf7", "f8" : "kf8",
"f9" : "kf9", "f10" : "kf10", "f11" : "kf11", "f12" : "kf12",
"f13" : "kf13", "f14" : "kf14", "f15" : "kf15", "f16" : "kf16",
"f17" : "kf17", "f18" : "kf18", "f19" : "kf19", "f20" : "kf20",
"home" : "khome",
"insert" : "kich1",
"left" : "kcub1",
"page down" : "knp",
"page up" : "kpp",
"right" : "kcuf1",
"up" : "kcuu1",
}
def general_keycodes():
keycodes = {}
for key, tiname in _keynames.items():
keycode = curses.tigetstr(tiname)
trace('key {key} tiname {tiname} keycode {keycode!r}', **locals())
if keycode:
keycodes[keycode] = key
return keycodes
def EventQueue(fd, encoding):
keycodes = general_keycodes()
if os.isatty(fd):
backspace = tcgetattr(fd)[6][VERASE]
keycodes[backspace] = unicode('backspace')
k = keymap.compile_keymap(keycodes)
trace('keymap {k!r}', k=k)
return EncodedQueue(k, encoding)
class EncodedQueue(object):
def __init__(self, keymap, encoding):
self.k = self.ck = keymap
self.events = []
self.buf = bytearray()
self.encoding=encoding
def get(self):
if self.events:
return self.events.pop(0)
else:
return None
def empty(self):
return not self.events
def flush_buf(self):
old = self.buf
self.buf = bytearray()
return bytes(old)
def insert(self, event):
trace('added event {event}', event=event)
self.events.append(event)
def push(self, char):
self.buf.append(ord(char))
if char in self.k:
if self.k is self.ck:
#sanity check, buffer is empty when a special key comes
assert len(self.buf) == 1
k = self.k[char]
trace('found map {k!r}', k=k)
if isinstance(k, dict):
self.k = k
else:
self.insert(Event('key', k, self.flush_buf()))
self.k = self.ck
else:
try:
decoded = bytes(self.buf).decode(self.encoding)
except:
return
self.insert(Event('key', decoded, self.flush_buf()))
self.k = self.ck
|
