| ##===-- 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() |