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
|
##===-- commandwin.py ----------------------------------------*- Python -*-===##
##
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
##
##===----------------------------------------------------------------------===##
import cui
import curses
import lldb
from itertools import islice
class History(object):
def __init__(self):
self.data = {}
self.pos = 0
self.tempEntry = ""
def previous(self, curr):
if self.pos == len(self.data):
self.tempEntry = curr
if self.pos < 0:
return ""
if self.pos == 0:
self.pos -= 1
return ""
if self.pos > 0:
self.pos -= 1
return self.data[self.pos]
def next(self):
if self.pos < len(self.data):
self.pos += 1
if self.pos < len(self.data):
return self.data[self.pos]
elif self.tempEntry != "":
return self.tempEntry
else:
return ""
def add(self, c):
self.tempEntry = ""
self.pos = len(self.data)
if self.pos == 0 or self.data[self.pos - 1] != c:
self.data[self.pos] = c
self.pos += 1
class CommandWin(cui.TitledWin):
def __init__(self, driver, x, y, w, h):
super(CommandWin, self).__init__(x, y, w, h, "Commands")
self.command = ""
self.data = ""
driver.setSize(w, h)
self.win.scrollok(1)
self.driver = driver
self.history = History()
def enterCallback(content):
self.handleCommand(content)
def tabCompleteCallback(content):
self.data = content
matches = lldb.SBStringList()
commandinterpreter = self.getCommandInterpreter()
commandinterpreter.HandleCompletion(
self.data, self.el.index, 0, -1, matches
)
if matches.GetSize() == 2:
self.el.content += matches.GetStringAtIndex(0)
self.el.index = len(self.el.content)
self.el.draw()
else:
self.win.move(self.el.starty, self.el.startx)
self.win.scroll(1)
self.win.addstr("Available Completions:")
self.win.scroll(1)
for m in islice(matches, 1, None):
self.win.addstr(self.win.getyx()[0], 0, m)
self.win.scroll(1)
self.el.draw()
self.startline = self.win.getmaxyx()[0] - 2
self.el = cui.CursesEditLine(
self.win, self.history, enterCallback, tabCompleteCallback
)
self.el.prompt = self.driver.getPrompt()
self.el.showPrompt(self.startline, 0)
def handleCommand(self, cmd):
# enter!
self.win.scroll(1) # TODO: scroll more for longer commands
if cmd == "":
cmd = self.history.previous("")
elif cmd in ("q", "quit"):
self.driver.terminate()
return
self.history.add(cmd)
ret = self.driver.handleCommand(cmd)
if ret.Succeeded():
out = ret.GetOutput()
attr = curses.A_NORMAL
else:
out = ret.GetError()
attr = curses.color_pair(3) # red on black
self.win.addstr(self.startline, 0, out + "\n", attr)
self.win.scroll(1)
self.el.showPrompt(self.startline, 0)
def handleEvent(self, event):
if isinstance(event, int):
if event == curses.ascii.EOT and self.el.content == "":
# When the command is empty, treat CTRL-D as EOF.
self.driver.terminate()
return
self.el.handleEvent(event)
def getCommandInterpreter(self):
return self.driver.getCommandInterpreter()
|
