From 502de803b806d6259aff75d95915851616fbde01 Mon Sep 17 00:00:00 2001
From: 0xMRTT <0xMRTT@proton.me>
Date: Sat, 10 Jun 2023 15:05:33 +0200
Subject: [PATCH] feat: add systemd restart

---
 matrixai/__init__.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/matrixai/__init__.py b/matrixai/__init__.py
index a058291..ee9e660 100644
--- a/matrixai/__init__.py
+++ b/matrixai/__init__.py
@@ -20,14 +20,18 @@ PING_URL = os.environ.get(
     "https://healthchecks.projectsegfau.lt/ping/2f7f2cd3-2a8d-4fff-b5ca-f4d17d47f75b",
 )
 
+SYSTEMD_SERVICE_NAME = os.environ.get("SYSTEMD_SERVICE_NAME", "matrixai")
+
 import io
 
 import aiohttp
+import asyncio
+import os
 from langdetect import detect
 
 from enum import Enum
 import uuid
-
+import subprocess
 
 class Style(Enum):
     """
@@ -856,8 +860,12 @@ Help Message:
         ):
             await bot.api.send_text_message(room.room_id, bot_help_message)
 
-    bot.run()
-
+    try:
+        bot.run()
+    except Exception as e:
+        subprocess.run(["systemctl", "restart", "--user", SYSTEMD_SERVICE_NAME]) 
+        print("Restarting bot due to {e}")
+        return 1
 
 if __name__ == "__main__":
     sys.exit(run())