From ef5afce2e2681431c0695ebdae61cd490bac8f45 Mon Sep 17 00:00:00 2001
From: 2mal3 <56305732+2mal3@users.noreply.github.com>
Date: Thu, 18 Jul 2024 11:33:25 +0200
Subject: [PATCH] chore: support longer shutdown times

---
 src/somnus/logic/stop.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/somnus/logic/stop.py b/src/somnus/logic/stop.py
index 1b16f21..f54fca1 100644
--- a/src/somnus/logic/stop.py
+++ b/src/somnus/logic/stop.py
@@ -41,6 +41,10 @@ async def _stop_mc_server(ssh: pxssh.pxssh, config: Config):
 
     log.debug("Sending stop command ...")
     ssh.sendline("stop")
+    messages = ["overworld", "the_end", "nether"]
+    for message in messages:
+        ssh.expect(message)
+        yield
     ssh.prompt()
     yield