From 41ff244abc26517a789b992a74d09f4a852e75ad Mon Sep 17 00:00:00 2001
From: Tom van der Lee <tom@vanderlee.io>
Date: Wed, 14 Dec 2022 12:02:16 +0100
Subject: Fixed issue with resending

---
 ttun/__main__.py |  2 +-
 ttun/client.py   | 11 ++++++++---
 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,
-- 
cgit v1.2.3