mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 16:17:49 +00:00
126 lines
4.4 KiB
Python
126 lines
4.4 KiB
Python
import urwid
|
|
from . import grideditor, signals, contentview
|
|
|
|
|
|
class Window(urwid.Frame):
|
|
def __init__(self, master, body, header, footer, helpctx):
|
|
urwid.Frame.__init__(self, body, header=header, footer=footer)
|
|
self.master = master
|
|
self.helpctx = helpctx
|
|
signals.focus.connect(self.sig_focus)
|
|
|
|
def sig_focus(self, sender, section):
|
|
self.focus_position = section
|
|
|
|
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 == "H":
|
|
self.master.view_grideditor(
|
|
grideditor.SetHeadersEditor(
|
|
self.master,
|
|
self.master.setheaders.get_specs(),
|
|
self.master.setheaders.set
|
|
)
|
|
)
|
|
elif k == "I":
|
|
self.master.view_grideditor(
|
|
grideditor.HostPatternEditor(
|
|
self.master,
|
|
[[x] for x in self.master.get_ignore_filter()],
|
|
self.master.edit_ignore_filter
|
|
)
|
|
)
|
|
elif k == "T":
|
|
self.master.view_grideditor(
|
|
grideditor.HostPatternEditor(
|
|
self.master,
|
|
[[x] for x in self.master.get_tcp_filter()],
|
|
self.master.edit_tcp_filter
|
|
)
|
|
)
|
|
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 == "M":
|
|
signals.status_prompt_onekey.send(
|
|
prompt = "Global default display mode",
|
|
keys = contentview.view_prompts,
|
|
callback = self.master.change_default_display_mode
|
|
)
|
|
elif k == "R":
|
|
self.master.view_grideditor(
|
|
grideditor.ReplaceEditor(
|
|
self.master,
|
|
self.master.replacehooks.get_specs(),
|
|
self.master.replacehooks.set
|
|
)
|
|
)
|
|
elif k == "s":
|
|
self.master.view_grideditor(
|
|
grideditor.ScriptEditor(
|
|
self.master,
|
|
[[i.command] for i in self.master.scripts],
|
|
self.master.edit_scripts
|
|
)
|
|
)
|
|
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,
|
|
)
|
|
elif k == "t":
|
|
signals.status_prompt.send(
|
|
prompt = "Sticky cookie filter",
|
|
text = self.master.stickycookie_txt,
|
|
callback = self.master.set_stickycookie
|
|
)
|
|
elif k == "u":
|
|
signals.status_prompt.send(
|
|
prompt = "Sticky auth filter",
|
|
text = self.master.stickyauth_txt,
|
|
callback = self.master.set_stickyauth
|
|
)
|
|
else:
|
|
return k
|