From 55b95d70b24cfa1b4703ae442a3c6d1781cc95aa Mon Sep 17 00:00:00 2001
From: Wojtek Kosior <koszko@koszko.org>
Date: Thu, 20 Oct 2022 21:57:47 +0200
Subject: [proxy] rework internal HTTP headers representation

---
 src/hydrilla/proxy/policies/base.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

(limited to 'src/hydrilla/proxy/policies/base.py')

diff --git a/src/hydrilla/proxy/policies/base.py b/src/hydrilla/proxy/policies/base.py
index fcdbf9d..8ea792f 100644
--- a/src/hydrilla/proxy/policies/base.py
+++ b/src/hydrilla/proxy/policies/base.py
@@ -48,9 +48,9 @@ class PolicyPriority(int, enum.Enum):
     _TWO   = 2
     _THREE = 3
 
-ProducedMessage = t.Union[
-    http_messages.ProducedRequest,
-    http_messages.ProducedResponse
+MessageInfo = t.Union[
+    http_messages.RequestInfo,
+    http_messages.ResponseInfo
 ]
 
 class Policy(ABC):
@@ -75,7 +75,7 @@ class Policy(ABC):
         return self._process_response
 
     def consume_request(self, request_info: http_messages.RequestInfo) \
-        -> t.Optional[ProducedMessage]:
+        -> t.Optional[MessageInfo]:
         raise NotImplementedError(
             'This kind of policy does not consume requests.'
         )
@@ -84,7 +84,7 @@ class Policy(ABC):
             self,
             request_info:  http_messages.RequestInfo,
             response_info: http_messages.ResponseInfo
-    ) -> t.Optional[http_messages.ProducedResponse]:
+    ) -> t.Optional[http_messages.ResponseInfo]:
         raise NotImplementedError(
             'This kind of policy does not consume responses.'
         )
@@ -109,6 +109,11 @@ class PolicyFactory(ABC):
             sorting_keys.get(other.__class__.__name__, 999)
 
 sorting_order = (
+    'WebUIMainPolicyFactory',
+    'WebUILandingPolicyFactory',
+
+    'MitmItPagePolicyFactory',
+
     'PayloadResourcePolicyFactory',
 
     'PayloadPolicyFactory',
-- 
cgit v1.2.3