Fixed issue with resendingv1.5.2

2 files changed, 9 insertions, 4 deletions

diff --git a/ttun/__main__.py b/ttun/__main__.py
index cbf8e16..62664a3 100644
--- a/ttun/__main__.py
+++ b/ttun/__main__.py
@@ -66,7 +66,7 @@ def main():
 
     server = Server(
         config=client.config,
-        on_resend=client.proxy_request,
+        on_resend=client.resend,
         on_started=print_info,
     )
 
diff --git a/ttun/client.py b/ttun/client.py
index 9e597ab..5434f0e 100644
--- a/ttun/client.py
+++ b/ttun/client.py
@@ -72,10 +72,11 @@ class Client:
         if self.connection.open:
             return self.connection
 
+    def session(self):
+        return ClientSession(base_url=self.proxy_origin, cookie_jar=DummyCookieJar())
+
     async def handle_messages(self):
-        async with ClientSession(
-            base_url=self.proxy_origin, cookie_jar=DummyCookieJar()
-        ) as session:
+        async with self.session() as session:
             while True:
                 try:
                     request: RequestData = await self.receive()
@@ -87,6 +88,10 @@ class Client:
                 except ConnectionClosed:
                     break
 
+    async def resend(self, data: RequestData):
+        async with self.session() as session:
+            await self.proxy_request(session, data)
+
     async def proxy_request(
         self,
         session: ClientSession,