addons: an addonmanager.Loader is now passed to the load() event

1 files changed, 28 insertions, 1 deletions

diff --git a/mitmproxy/addonmanager.py b/mitmproxy/addonmanager.py
index b3a71f1b..1fc8e4dc 100644
--- a/mitmproxy/addonmanager.py
+++ b/mitmproxy/addonmanager.py
@@ -1,3 +1,5 @@
+import typing
+
 from mitmproxy import exceptions
 from mitmproxy import eventsequence
 from mitmproxy import controller
@@ -9,6 +11,30 @@ def _get_name(itm):
     return getattr(itm, "name", itm.__class__.__name__.lower())
 
 
+class Loader:
+    """
+        A loader object is passed to the load() event when addons start up.
+    """
+    def __init__(self, master):
+        self.master = master
+
+    def add_option(
+        self,
+        name: str,
+        typespec: type,
+        default: typing.Any,
+        help: str,
+        choices: typing.Optional[typing.Sequence[str]] = None
+    ) -> None:
+        self.master.options.add_option(
+            name,
+            typespec,
+            default,
+            help,
+            choices
+        )
+
+
 class AddonManager:
     def __init__(self, master):
         self.chain = []
@@ -41,8 +67,9 @@ class AddonManager:
         """
         self.chain.extend(addons)
         with self.master.handlecontext():
+            l = Loader(self.master)
             for i in addons:
-                self.invoke_addon(i, "load", self.master.options)
+                self.invoke_addon(i, "load", l)
 
     def remove(self, addon):
         """