commit 7e931e6047ab7e22ba8e08de58f54e7a687bd28d
Author: Sambuddha Basu <sambuddhabasu1(a)gmail.com>
Date: Sun Jun 19 18:16:41 2016 -0700
Interpreter panel supports GETINFO requests
---
nyx/controller.py | 8 +++---
nyx/panel/interpreter.py | 72 ++++++++++++++++++++++++++++++++++++++++++++++++
nyx/panel/interpretor.py | 49 --------------------------------
nyx/tor_interpreter.py | 23 ++++++++++++++++
4 files changed, 99 insertions(+), 53 deletions(-)
diff --git a/nyx/controller.py b/nyx/controller.py
index c167c9a..54316a0 100644
--- a/nyx/controller.py
+++ b/nyx/controller.py
@@ -20,7 +20,7 @@ import nyx.panel.graph
import nyx.panel.header
import nyx.panel.log
import nyx.panel.torrc
-import nyx.panel.interpretor
+import nyx.panel.interpreter
import stem
@@ -47,7 +47,7 @@ CONFIG = conf.config_dict('nyx', {
'features.panels.show.connection': True,
'features.panels.show.config': True,
'features.panels.show.torrc': True,
- 'features.panels.show.interpretor': True,
+ 'features.panels.show.interpreter': True,
'features.redrawRate': 5,
'features.refreshRate': 5,
'features.confirmQuit': True,
@@ -120,8 +120,8 @@ class Controller(object):
if CONFIG['features.panels.show.torrc']:
self._page_panels.append([nyx.panel.torrc.TorrcPanel()])
- if CONFIG['features.panels.show.interpretor']:
- self._page_panels.append([nyx.panel.interpretor.InterpretorPanel()])
+ if CONFIG['features.panels.show.interpreter']:
+ self._page_panels.append([nyx.panel.interpreter.InterpreterPanel()])
self.quit_signal = False
self._page = 0
diff --git a/nyx/panel/interpreter.py b/nyx/panel/interpreter.py
new file mode 100644
index 0000000..2769529
--- /dev/null
+++ b/nyx/panel/interpreter.py
@@ -0,0 +1,72 @@
+"""
+Panel providing raw control port access with syntax hilighting, usage
+information, tab completion, and other usability features.
+"""
+
+import nyx.controller
+import nyx.curses
+
+from nyx.curses import GREEN, CYAN, BOLD, HIGHLIGHT
+from nyx.tor_interpreter import handle_query
+from nyx import panel
+
+
+USAGE_INFO = "to use this panel press enter"
+PROMPT = ">>> "
+PROMPT_LINE = [[(PROMPT, GREEN, BOLD), (USAGE_INFO, CYAN, BOLD)]]
+
+class InterpreterPanel(panel.Panel):
+ """
+ Renders the interpreter panel with a prompt providing raw control port
+ access.
+ """
+
+ def __init__(self):
+ panel.Panel.__init__(self, 'interpreter')
+
+ self._is_input_mode = False
+
+ def key_handlers(self):
+ def _execute_command():
+ self._is_input_mode = True
+
+ while self._is_input_mode:
+ self.redraw(True)
+ user_input = nyx.curses.str_input(len(PROMPT), self.top + len(PROMPT_LINE))
+ user_input, is_done = user_input.strip(), False
+
+ if not user_input:
+ is_done = True
+ else:
+ input_entry, output_entry = handle_query(user_input)
+ input_entry.insert(0, (PROMPT, GREEN, BOLD))
+ PROMPT_LINE.insert(len(PROMPT_LINE) - 1, input_entry)
+ for line in output_entry:
+ PROMPT_LINE.insert(len(PROMPT_LINE) - 1, [line])
+
+ if is_done:
+ self._is_input_mode = False
+ self.redraw(True)
+
+ return (
+ nyx.panel.KeyHandler('enter', 'execute a command', _execute_command, key_func = lambda key: key.is_selection()),
+ )
+
+ def draw(self, width, height):
+ usage_msg = " (enter \"/help\" for usage or a blank line to stop)" if self._is_input_mode else ""
+ self.addstr(0, 0, 'Control Interpreter%s:' % usage_msg, HIGHLIGHT)
+
+ x_offset = 0
+ draw_line = 1
+ for entry in PROMPT_LINE:
+ cursor = x_offset
+
+ msg, color, attr = None, None, None
+ for line in entry:
+ if len(line) == 2:
+ self.addstr(draw_line, cursor, line[0], line[1])
+ elif len(line) == 3:
+ self.addstr(draw_line, cursor, line[0], line[1], line[2])
+ cursor += len(line[0])
+
+ draw_line += 1
diff --git a/nyx/panel/interpretor.py b/nyx/panel/interpretor.py
deleted file mode 100644
index 995490c..0000000
--- a/nyx/panel/interpretor.py
+++ /dev/null
@@ -1,49 +0,0 @@
-"""
-Panel providing raw control port access with syntax hilighting, usage
-information, tab completion, and other usability features.
-"""
-
-import nyx.curses
-
-from nyx.curses import GREEN, CYAN, BOLD, HIGHLIGHT
-from nyx import panel
-
-
-USAGE_INFO = "to use this panel press enter"
-PROMPT = ">>> "
-PROMPT_LINE = [[(PROMPT, GREEN, BOLD), (USAGE_INFO, CYAN, BOLD)]]
-
-class InterpretorPanel(panel.Panel):
- """
- Renders the current torrc or nyxrc with syntax highlighting in a scrollable
- area.
- """
-
- def __init__(self):
- panel.Panel.__init__(self, 'interpretor')
-
- self._is_input_mode = False
-
- def key_handlers(self):
- def _execute_command():
- self._is_input_mode ^= True
- self.redraw(True)
-
- return (
- nyx.panel.KeyHandler('enter', 'execute a command', _execute_command, key_func = lambda key: key.is_selection()),
- )
-
- def draw(self, width, height):
- usage_msg = " (enter \"/help\" for usage or a blank line to stop)" if self._is_input_mode else ""
- self.addstr(0, 0, 'Control Interpretor%s:' % usage_msg, HIGHLIGHT)
-
- x_offset = 0
- draw_line = 1
- for entry in PROMPT_LINE:
- cursor = x_offset
-
- for msg, color, attr in entry:
- self.addstr(draw_line, cursor, msg, color, attr)
- cursor += len(msg)
-
- draw_line += 1
diff --git a/nyx/tor_interpreter.py b/nyx/tor_interpreter.py
new file mode 100644
index 0000000..64d6554
--- /dev/null
+++ b/nyx/tor_interpreter.py
@@ -0,0 +1,23 @@
+from nyx.curses import GREEN, CYAN, BOLD, HIGHLIGHT
+from nyx import tor_controller
+
+
+def handle_query(user_input):
+ user_input = user_input.strip()
+
+ input_entry, output_entry = [], []
+
+ if " " in user_input: cmd, arg = user_input.split(" ", 1)
+ else: cmd, arg = user_input, ""
+
+ cmd = cmd.upper()
+ input_entry.append((cmd + " ", GREEN, BOLD))
+ if arg:
+ input_entry.append((arg, CYAN, BOLD))
+
+ if cmd == "GETINFO":
+ resp = tor_controller().get_info(arg)
+ for line in resp.split('\n'):
+ output_entry.append((line, CYAN,))
+
+ return input_entry, output_entry