commander: add completion for flowspecs

1 files changed, 26 insertions, 0 deletions

diff --git a/mitmproxy/command.py b/mitmproxy/command.py
index 2d51317c..22c2376b 100644
--- a/mitmproxy/command.py
+++ b/mitmproxy/command.py
@@ -24,6 +24,32 @@ def lexer(s):
     return lex
 
 
+# This is an awkward location for these values, but it's better than having
+# the console core import and depend on an addon. FIXME: Add a way for
+# addons to add custom types and manage their completion and validation.
+valid_flow_prefixes = [
+    "@all",
+    "@focus",
+    "@shown",
+    "@hidden",
+    "@marked",
+    "@unmarked",
+    "~q",
+    "~s",
+    "~a",
+    "~hq",
+    "~hs",
+    "~b",
+    "~bq",
+    "~bs",
+    "~t",
+    "~d",
+    "~m",
+    "~u",
+    "~c",
+]
+
+
 Cuts = typing.Sequence[
     typing.Sequence[typing.Union[str, bytes]]
 ]