enable live-reloading for contentviews (mitmproxy#4856)

Author: mhils

mitmproxy/contentviews/__init__.py

@@ -15,6 +15,8 @@
 from typing import List, Union
 from typing import Optional
 
+import blinker
+
 from mitmproxy import flow
 from mitmproxy import http
 from mitmproxy.utils import strutils
@@ -29,6 +31,9 @@
 
 views: List[View] = []
 
+on_add = blinker.Signal()
+"""A new contentview has been added."""
+
 
 def get(name: str) -> Optional[View]:
     for i in views:
@@ -44,6 +49,7 @@ def add(view: View) -> None:
             raise ValueError("Duplicate view: " + view.name)
 
     views.append(view)
+    on_add.send(view)
 
 
 def remove(view: View) -> None:

mitmproxy/tools/console/flowview.py

@@ -49,6 +49,7 @@ def __init__(self, master):
         super().__init__([])
         self.show()
         self.last_displayed_body = None
+        contentviews.on_add.connect(self.contentview_added)
 
     @property
     def view(self):
@@ -58,6 +59,12 @@ def view(self):
     def flow(self) -> mitmproxy.flow.Flow:
         return self.master.view.focus.flow
 
+    def contentview_added(self, view):
+        # this is called when a contentview addon is live-reloaded.
+        # we clear our cache and then rerender
+        self._get_content_view.cache_clear()
+        self.show()
+
     def focus_changed(self):
         f = self.flow
         if f: