From 2f5694cffb871f0663e74b0b87793a73c87a379c Mon Sep 17 00:00:00 2001
From: Alex Kotov <kotovalexarian@gmail.com>
Date: Sun, 8 Jan 2023 23:28:13 +0400
Subject: [PATCH] Add an application class

---
 mirrortea/__main__.py | 37 +++++++++++++++++++++----------------
 1 file changed, 21 insertions(+), 16 deletions(-)

diff --git a/mirrortea/__main__.py b/mirrortea/__main__.py
index 2dcf01a..926643d 100644
--- a/mirrortea/__main__.py
+++ b/mirrortea/__main__.py
@@ -10,23 +10,28 @@ MATRIX_FULL_USER_ID   = os.environ['MATRIX_FULL_USER_ID']
 MATRIX_PASSWORD       = os.environ['MATRIX_PASSWORD']
 TELEGRAM_BOT_TOKEN    = os.environ['TELEGRAM_BOT_TOKEN']
 
-async def main():
-    try:
-        matrix_loop = None
+def main():
+    asyncio.run(Application().run())
 
-        matrix_loop = MatrixLoop(MATRIX_HOMESERVER_URL, MATRIX_FULL_USER_ID,
-                                 MATRIX_PASSWORD)
-        await matrix_loop.prepare()
-
-        telegram_loop = TelegramLoop(TELEGRAM_BOT_TOKEN)
-
-        await asyncio.gather(
-            matrix_loop.run(),
-            telegram_loop.run(),
+class Application:
+    def __init__(self):
+        self.matrix_loop = MatrixLoop(
+            MATRIX_HOMESERVER_URL,
+            MATRIX_FULL_USER_ID,
+            MATRIX_PASSWORD,
         )
-    finally:
-        if matrix_loop:
-            await matrix_loop.finish()
+        self.telegram_loop = TelegramLoop(TELEGRAM_BOT_TOKEN)
+
+    async def run(self):
+        try:
+            await self.matrix_loop.prepare()
+            await asyncio.gather(
+                self.matrix_loop.run(),
+                self.telegram_loop.run(),
+            )
+        finally:
+            if self.matrix_loop:
+                await self.matrix_loop.finish()
 
 class MatrixLoop:
     def __init__(self, homeserver_url, full_user_id, password):
@@ -59,4 +64,4 @@ class TelegramLoop:
         print(msg, file=sys.stderr)
 
 if __name__ == '__main__':
-    asyncio.run(main())
+    main()