1 files changed, 90 insertions, 0 deletions
diff --git a/mitmproxy/libmproxy/console/window.py b/mitmproxy/libmproxy/console/window.py
new file mode 100644
index 00000000..47c284e4
--- /dev/null
+++ b/mitmproxy/libmproxy/console/window.py
@@ -0,0 +1,90 @@
+import urwid
+from . import signals
+
+
+class Window(urwid.Frame):
+
+    def __init__(self, master, body, header, footer, helpctx):
+        urwid.Frame.__init__(
+            self,
+            urwid.AttrWrap(body, "background"),
+            header = urwid.AttrWrap(header, "background") if header else None,
+            footer = urwid.AttrWrap(footer, "background") if footer else None
+        )
+        self.master = master
+        self.helpctx = helpctx
+        signals.focus.connect(self.sig_focus)
+
+    def sig_focus(self, sender, section):
+        self.focus_position = section
+
+    def mouse_event(self, *args, **kwargs):
+        # args: (size, event, button, col, row)
+        k = super(self.__class__, self).mouse_event(*args, **kwargs)
+        if not k:
+            if args[1] == "mouse drag":
+                signals.status_message.send(
+                    message = "Hold down shift, alt or ctrl to select text.",
+                    expire = 1
+                )
+            elif args[1] == "mouse press" and args[2] == 4:
+                self.keypress(args[0], "up")
+            elif args[1] == "mouse press" and args[2] == 5:
+                self.keypress(args[0], "down")
+            else:
+                return False
+            return True
+
+    def keypress(self, size, k):
+        k = super(self.__class__, self).keypress(size, k)
+        if k == "?":
+            self.master.view_help(self.helpctx)
+        elif k == "c":
+            if not self.master.client_playback:
+                signals.status_prompt_path.send(
+                    self,
+                    prompt = "Client replay",
+                    callback = self.master.client_playback_path
+                )
+            else:
+                signals.status_prompt_onekey.send(
+                    self,
+                    prompt = "Stop current client replay?",
+                    keys = (
+                        ("yes", "y"),
+                        ("no", "n"),
+                    ),
+                    callback = self.master.stop_client_playback_prompt,
+                )
+        elif k == "i":
+            signals.status_prompt.send(
+                self,
+                prompt = "Intercept filter",
+                text = self.master.state.intercept_txt,
+                callback = self.master.set_intercept
+            )
+        elif k == "o":
+            self.master.view_options()
+        elif k == "Q":
+            raise urwid.ExitMainLoop
+        elif k == "q":
+            signals.pop_view_state.send(self)
+        elif k == "S":
+            if not self.master.server_playback:
+                signals.status_prompt_path.send(
+                    self,
+                    prompt = "Server replay path",
+                    callback = self.master.server_playback_path
+                )
+            else:
+                signals.status_prompt_onekey.send(
+                    self,
+                    prompt = "Stop current server replay?",
+                    keys = (
+                        ("yes", "y"),
+                        ("no", "n"),
+                    ),
+                    callback = self.master.stop_server_playback_prompt,
+                )
+        else:
+            return k