feat: use plugins as context manager

muffin/plugins.py

@@ -64,6 +64,14 @@ def __repr__(self) -> str:
         """Human readable representation."""
         return f"<muffin.Plugin: { self.name }>"
 
+    async def __aenter__(self):
+        if self.startup is not None:
+            await self.startup()
+
+    async def __aexit__(self, exc_type, exc, tb):
+        if self.shutdown is not None:
+            await self.shutdown()
+
     @property
     def app(self) -> Application:
         """Get the application."""

tests/common/test_plugin.py

@@ -115,3 +115,25 @@ class Plugin(BasePlugin):
         "plugin1": p1,
         "plugin2": p2,
     }
+
+
+async def test_plugin_as_context_manager():
+    from muffin.plugins import BasePlugin
+
+    events = []
+
+    class Plugin(BasePlugin):
+        name = "plugin"
+
+        async def startup(self):
+            events.append("start")
+
+        async def shutdown(self):
+            events.append("end")
+
+    plug = Plugin()
+    async with plug:
+        pass
+
+    assert "start" in events
+    assert "end" in events