diff --git a/mautrix_telegram/abstract_user.py b/mautrix_telegram/abstract_user.py index 2eb96260..d3edf7ce 100644 --- a/mautrix_telegram/abstract_user.py +++ b/mautrix_telegram/abstract_user.py @@ -241,6 +241,24 @@ async def _init_client(self) -> None: use_ipv6=self.config["telegram.connection.use_ipv6"], ) self.client.add_event_handler(self._update_catch) + self._schedule_reconnect() + + def _schedule_reconnect(self) -> None: + reconnect_interval = self.config["telegram.force_refresh_interval_seconds"] + if not reconnect_interval or reconnect_interval == 0: + return + refresh_time = time.time() + reconnect_interval + self.log.info( + "Scheduling forced reconnect in %d seconds. Connection will be refreshed at %s", + reconnect_interval, + time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(refresh_time)), + ) + self.loop.call_later(reconnect_interval, lambda: background_task.create(self._reconnect())) + + async def _reconnect(self) -> None: + self.log.info("Reconnecting to Telegram...") + await self.stop() + await self.start() @abstractmethod async def on_signed_out(self, err: UnauthorizedError | AuthKeyError) -> None: diff --git a/mautrix_telegram/config.py b/mautrix_telegram/config.py index 14148881..84ed3d07 100644 --- a/mautrix_telegram/config.py +++ b/mautrix_telegram/config.py @@ -264,6 +264,7 @@ def do_update(self, helper: ConfigUpdateHelper) -> None: copy("telegram.catch_up") copy("telegram.sequential_updates") copy("telegram.exit_on_update_error") + copy("telegram.force_refresh_interval_seconds") copy("telegram.connection.timeout") copy("telegram.connection.retries") diff --git a/mautrix_telegram/example-config.yaml b/mautrix_telegram/example-config.yaml index b23083e4..06b0e946 100644 --- a/mautrix_telegram/example-config.yaml +++ b/mautrix_telegram/example-config.yaml @@ -581,6 +581,8 @@ telegram: # Should incoming updates be handled sequentially to make sure order is preserved on Matrix? sequential_updates: true exit_on_update_error: false + # Interval to force refresh the connection (full reconnect). 0 disables it. + force_refresh_interval_seconds: 0 # Telethon connection options. connection: