mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2025-02-02 00:05:27 +00:00
111 lines
3.4 KiB
Python
111 lines
3.4 KiB
Python
import urwid
|
|
|
|
from mitmproxy.tools.console import common
|
|
from mitmproxy.tools.console import layoutwidget
|
|
import mitmproxy.tools.console.master # noqa
|
|
|
|
|
|
class FlowItem(urwid.WidgetWrap):
|
|
|
|
def __init__(self, master, flow):
|
|
self.master, self.flow = master, flow
|
|
w = self.get_text()
|
|
urwid.WidgetWrap.__init__(self, w)
|
|
|
|
def get_text(self):
|
|
cols, _ = self.master.ui.get_cols_rows()
|
|
layout = self.master.options.console_flowlist_layout
|
|
if layout == "list" or (layout == 'default' and cols < 100):
|
|
render_mode = common.RenderMode.LIST
|
|
else:
|
|
render_mode = common.RenderMode.TABLE
|
|
|
|
return common.format_flow(
|
|
self.flow,
|
|
render_mode=render_mode,
|
|
focused=self.flow is self.master.view.focus.flow,
|
|
hostheader=self.master.options.showhost,
|
|
)
|
|
|
|
def selectable(self):
|
|
return True
|
|
|
|
def mouse_event(self, size, event, button, col, row, focus):
|
|
if event == "mouse press" and button == 1:
|
|
self.master.commands.execute("console.view.flow @focus")
|
|
return True
|
|
|
|
def keypress(self, size, key):
|
|
return key
|
|
|
|
|
|
class FlowListWalker(urwid.ListWalker):
|
|
|
|
def __init__(self, master):
|
|
self.master = master
|
|
|
|
def positions(self, reverse=False):
|
|
# The stub implementation of positions can go once this issue is resolved:
|
|
# https://github.com/urwid/urwid/issues/294
|
|
ret = range(self.master.commands.execute("view.properties.length"))
|
|
if reverse:
|
|
return reversed(ret)
|
|
return ret
|
|
|
|
def view_changed(self):
|
|
self._modified()
|
|
|
|
def get_focus(self):
|
|
if not self.master.view.focus.flow:
|
|
return None, 0
|
|
f = FlowItem(self.master, self.master.view.focus.flow)
|
|
return f, self.master.view.focus.index
|
|
|
|
def set_focus(self, index):
|
|
if self.master.commands.execute("view.properties.inbounds %d" % index):
|
|
self.master.view.focus.index = index
|
|
|
|
def get_next(self, pos):
|
|
pos = pos + 1
|
|
if not self.master.commands.execute("view.properties.inbounds %d" % pos):
|
|
return None, None
|
|
f = FlowItem(self.master, self.master.view[pos])
|
|
return f, pos
|
|
|
|
def get_prev(self, pos):
|
|
pos = pos - 1
|
|
if not self.master.commands.execute("view.properties.inbounds %d" % pos):
|
|
return None, None
|
|
f = FlowItem(self.master, self.master.view[pos])
|
|
return f, pos
|
|
|
|
|
|
class FlowListBox(urwid.ListBox, layoutwidget.LayoutWidget):
|
|
title = "Flows"
|
|
keyctx = "flowlist"
|
|
|
|
def __init__(
|
|
self, master: "mitmproxy.tools.console.master.ConsoleMaster"
|
|
) -> None:
|
|
self.master: "mitmproxy.tools.console.master.ConsoleMaster" = master
|
|
super().__init__(FlowListWalker(master))
|
|
self.master.options.subscribe(
|
|
self.set_flowlist_layout,
|
|
["console_flowlist_layout"]
|
|
)
|
|
|
|
def keypress(self, size, key):
|
|
if key == "m_start":
|
|
self.master.commands.execute("view.focus.go 0")
|
|
elif key == "m_end":
|
|
self.master.commands.execute("view.focus.go -1")
|
|
elif key == "m_select":
|
|
self.master.commands.execute("console.view.flow @focus")
|
|
return urwid.ListBox.keypress(self, size, key)
|
|
|
|
def view_changed(self):
|
|
self.body.view_changed()
|
|
|
|
def set_flowlist_layout(self, opts, updated):
|
|
self.master.ui.clear()
|