[proxy] pass all available flow information to relevant methods of Policy type

1 files changed, 14 insertions, 4 deletions

diff --git a/src/hydrilla/proxy/policies/base.py b/src/hydrilla/proxy/policies/base.py
index b3d3172..fcdbf9d 100644
--- a/src/hydrilla/proxy/policies/base.py
+++ b/src/hydrilla/proxy/policies/base.py
@@ -61,10 +61,17 @@ class Policy(ABC):
 
     priority: t.ClassVar[PolicyPriority]
 
-    def should_process_request(self, parsed_url: ParsedUrl) -> bool:
+    def should_process_request(
+            self,
+            request_info: http_messages.BodylessRequestInfo
+    ) -> bool:
         return self._process_request
 
-    def should_process_response(self, parsed_url: ParsedUrl) -> bool:
+    def should_process_response(
+            self,
+            request_info:  http_messages.RequestInfo,
+            response_info: http_messages.BodylessResponseInfo
+    ) -> bool:
         return self._process_response
 
     def consume_request(self, request_info: http_messages.RequestInfo) \
@@ -73,8 +80,11 @@ class Policy(ABC):
             'This kind of policy does not consume requests.'
         )
 
-    def consume_response(self, response_info: http_messages.ResponseInfo) \
-        -> t.Optional[http_messages.ProducedResponse]:
+    def consume_response(
+            self,
+            request_info:  http_messages.RequestInfo,
+            response_info: http_messages.ResponseInfo
+    ) -> t.Optional[http_messages.ProducedResponse]:
         raise NotImplementedError(
             'This kind of policy does not consume responses.'
         )