feat: added hooks to the plugin manager for http pre/post header requests.

Signed-off-by: Teryl Taylor <[email protected]>

Author: Teryl Taylor

mcpgateway/plugins/framework/manager.py

@@ -42,6 +42,8 @@
     Config,
     GlobalContext,
     HookType,
+    HttpHeaderPayload,
+    HttpHeaderPayloadResult,
     PluginCondition,
     PluginContext,
     PluginContextTable,
@@ -431,6 +433,54 @@ async def post_resource_fetch(plugin: PluginRef, payload: ResourcePostFetchPaylo
     return await plugin.plugin.resource_post_fetch(payload, context)
 
 
+async def pre_http_forwarding_call(plugin: PluginRef, payload: HttpHeaderPayload, context: PluginContext) -> HttpHeaderPayloadResult:
+    """Call plugin's HTTP pre-forwarding call hook.
+
+    Args:
+        plugin: The plugin to execute.
+        payload: The set of HTTP headers to be analyzed.
+        context: The plugin context.
+
+    Returns:
+        Modified HTTP headers with processing status.
+
+    Examples:
+        >>> from mcpgateway.plugins.framework.base import PluginRef
+        >>> from mcpgateway.plugins.framework import Plugin, HttpHeaderPayload, PluginContext, GlobalContext
+        >>> # Assuming you have a plugin instance:
+        >>> # plugin_ref = PluginRef(my_plugin)
+        >>> payload = HttpHeaderPayload({"Authorization": "Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ=="})
+        >>> context = PluginContext(request_id="123")
+        >>> # In async context:
+        >>> # result = await pre_http_forwarding_call(plugin_ref, payload, context)
+    """
+    return await plugin.plugin.http_pre_forwarding_call(payload, context)
+
+
+async def post_http_forwarding_call(plugin: PluginRef, payload: HttpHeaderPayload, context: PluginContext) -> HttpHeaderPayloadResult:
+    """Call plugin's HTTP post-forwarding call hook.
+
+    Args:
+        plugin: The plugin to execute.
+        payload: The set of HTTP headers to be analyzed.
+        context: The plugin context.
+
+    Returns:
+        Modified HTTP headers with processing status.
+
+    Examples:
+        >>> from mcpgateway.plugins.framework.base import PluginRef
+        >>> from mcpgateway.plugins.framework import Plugin, HttpHeaderPayload, PluginContext, GlobalContext
+        >>> # Assuming you have a plugin instance:
+        >>> # plugin_ref = PluginRef(my_plugin)
+        >>> payload = HttpHeaderPayload({"Authorization": "Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ=="})
+        >>> context = PluginContext(request_id="123")
+        >>> # In async context:
+        >>> # result = await post_http_forwarding_call(plugin_ref, payload, context)
+    """
+    return await plugin.plugin.http_post_forwarding_call(payload, context)
+
+
 class PluginManager:
     """Plugin manager for managing the plugin lifecycle.
 
@@ -475,6 +525,7 @@ class PluginManager:
     _post_tool_executor: PluginExecutor[ToolPostInvokePayload] = PluginExecutor[ToolPostInvokePayload]()
     _resource_pre_executor: PluginExecutor[ResourcePreFetchPayload] = PluginExecutor[ResourcePreFetchPayload]()
     _resource_post_executor: PluginExecutor[ResourcePostFetchPayload] = PluginExecutor[ResourcePostFetchPayload]()
+    _http_executor: PluginExecutor[HttpHeaderPayload] = PluginExecutor[HttpHeaderPayload]()
 
     # Context cleanup tracking
     _context_store: Dict[str, Tuple[PluginContextTable, float]] = {}
@@ -962,3 +1013,69 @@ async def resource_post_fetch(
             del self._context_store[global_context.request_id]
 
         return result
+
+    async def http_pre_forwarding_call(
+        self, payload: HttpHeaderPayload, global_context: GlobalContext, local_contexts: Optional[PluginContextTable] = None
+    ) -> tuple[HttpHeaderPayloadResult, PluginContextTable | None]:
+        """Execute pre-fetch hooks before an operation is called.
+
+        Args:
+            payload: The http payload containing http headers passed from requests to responses.
+            global_context: Shared context for all plugins with request metadata.
+            local_contexts: Optional existing contexts from previous hook executions.
+
+        Returns:
+            A tuple containing:
+            - HttpHeaderPayloadResult with processing status and modified headers
+            - PluginContextTable with plugin contexts for state management
+        """
+        # Get plugins configured for this hook
+        plugins = self._registry.get_plugins_for_hook(HookType.HTTP_PRE_FORWARDING_CALL)
+
+        def compare(payload: HttpHeaderPayload, conditions: list[PluginCondition], context: GlobalContext):
+            return True
+
+        # Execute plugins
+        result = await self._http_executor.execute(plugins, payload, global_context, pre_http_forwarding_call, compare, local_contexts)
+
+        # Store context for potential post-fetch
+        if result[1]:
+            self._context_store[global_context.request_id] = (result[1], time.time())
+
+        # Periodic cleanup
+        await self._cleanup_old_contexts()
+
+        return result
+
+    async def http_post_forwarding_call(
+        self, payload: HttpHeaderPayload, global_context: GlobalContext, local_contexts: Optional[PluginContextTable] = None
+    ) -> tuple[HttpHeaderPayloadResult, PluginContextTable | None]:
+        """Execute post-fetch hooks after an operation is complete.
+
+        Args:
+            payload: The http payload containing http headers passed from requests to responses.
+            global_context: Shared context for all plugins with request metadata.
+            local_contexts: Optional existing contexts from previous hook executions.
+
+        Returns:
+            A tuple containing:
+            - HttpHeaderPayloadResult with processing status and modified headers
+            - PluginContextTable with plugin contexts for state management
+        """
+        # Get plugins configured for this hook
+        plugins = self._registry.get_plugins_for_hook(HookType.HTTP_POST_FORWARDING_CALL)
+
+        def compare(payload: HttpHeaderPayload, conditions: list[PluginCondition], context: GlobalContext):
+            return True
+
+        # Execute plugins
+        result = await self._http_executor.execute(plugins, payload, global_context, post_http_forwarding_call, compare, local_contexts)
+
+        # Store context for potential post-fetch
+        if result[1]:
+            self._context_store[global_context.request_id] = (result[1], time.time())
+
+        # Periodic cleanup
+        await self._cleanup_old_contexts()
+
+        return result

mcpgateway/plugins/framework/models.py

@@ -30,6 +30,8 @@ class HookType(str, Enum):
     """MCP Forge Gateway hook points.
 
     Attributes:
+        http_pre_forwarding_call: The pre hook before an http forwarding request.
+        http_post_forwarding_call: The post hook after an http forwarding request.
         prompt_pre_fetch: The prompt pre hook.
         prompt_post_fetch: The prompt post hook.
         tool_pre_invoke: The tool pre invoke hook.
@@ -48,6 +50,8 @@ class HookType(str, Enum):
         [<HookType.PROMPT_PRE_FETCH: 'prompt_pre_fetch'>, <HookType.PROMPT_POST_FETCH: 'prompt_post_fetch'>, <HookType.TOOL_PRE_INVOKE: 'tool_pre_invoke'>, <HookType.TOOL_POST_INVOKE: 'tool_post_invoke'>, ...]
     """
 
+    HTTP_PRE_FORWARDING_CALL = "http_pre_forwarding_call"
+    HTTP_POST_FORWARDING_CALL = "http_post_forwarding_call"
     PROMPT_PRE_FETCH = "prompt_pre_fetch"
     PROMPT_POST_FETCH = "prompt_post_fetch"
     TOOL_PRE_INVOKE = "tool_pre_invoke"