console: add a -l flag to set the limit on startup

libmproxy/cmdline.py

@@ -599,6 +599,11 @@ def mitmproxy():
         type=str, dest="intercept", default=None,
         help="Intercept filter expression."
     )
+    group.add_argument(
+        "-l", "--limit", action="store",
+        type=str, dest="limit", default=None,
+        help="Limit filter expression."
+    )
     return parser
 
 

libmproxy/console/__init__.py

@@ -116,6 +116,7 @@ class Options(object):
         "keepserving",
         "kill",
         "intercept",
+        "limit",
         "no_server",
         "refresh_server_playback",
         "rfile",
@@ -162,6 +163,9 @@ class ConsoleMaster(flow.FlowMaster):
             print >> sys.stderr, "Intercept error:", r
             sys.exit(1)
 
+        if options.limit:
+            self.set_limit(options.limit)
+
         r = self.set_stickycookie(options.stickycookie)
         if r:
             print >> sys.stderr, "Sticky cookies error:", r

libmproxy/console/grideditor.py

@@ -549,3 +549,9 @@ class CookieEditor(GridEditor):
     title = "Editing request Cookie header"
     columns = 2
     headings = ("Name", "Value")
+
+
+class SetCookieEditor(GridEditor):
+    title = "Editing request SetCookie header"
+    columns = 2
+    headings = ("Name", "Value")

libmproxy/main.py

@@ -92,6 +92,7 @@ def mitmproxy(args=None):  # pragma: nocover
     console_options.palette_transparent = options.palette_transparent
     console_options.eventlog = options.eventlog
     console_options.intercept = options.intercept
+    console_options.limit = options.limit
 
     server = get_server(console_options.no_server, proxy_config)