From 0b8e607882932a10118e99fb5109fb192664b6a4 Mon Sep 17 00:00:00 2001
From: Alistair Young <avatar@arkane-systems.net>
Date: Tue, 22 Feb 2022 23:07:24 -0600
Subject: [PATCH] Shutdown routine

---
 binsrc/genie/genie | 132 ++++++++++++++++-----------------------------
 1 file changed, 47 insertions(+), 85 deletions(-)

diff --git a/binsrc/genie/genie b/binsrc/genie/genie
index 7e7413f..575ce15 100755
--- a/binsrc/genie/genie
+++ b/binsrc/genie/genie
@@ -630,91 +630,53 @@ def do_command(commandline):
         return sp
 
 def do_shutdown():
-    print ('todo: shutdown command')
-
-#         // Shut down the bottle and clean up.
-#         public static int ShutdownHandler (bool verbose)
-#         {
-#             // Get the bottle state.
-#             var state = new BottleStatus (verbose);
-
-#             if (state.Status == Status.NoBottlePresent)
-#             {
-#                 Console.WriteLine ("genie: no bottle exists.");
-#                 return EINVAL;
-#             }
-
-#             if (state.StartedWithinBottle)
-#             {
-#                 Console.WriteLine ("genie: cannot shut down bottle from inside bottle; exiting.");
-#                 return EINVAL;
-#             }
-
-#             if (state.BottleWillExist || state.BottleClosingDown)
-#             {
-#                 Console.WriteLine ("genie: bottle in transitional state; please wait until this is complete.");
-#             }
-
-#             using (var r = new RootPrivilege())
-#             {
-#                 StopBottle (verbose);
-#             }
-
-#             return 0;
-#         }
-
-#         // Do the work of shutting down the bottle.
-#         public static void StopBottle (bool verbose)
-#         {
-#             FlagFiles.ShutdownFile = true;
-#             FlagFiles.RunFile = false;
-
-#             if (verbose)
-#                 Console.WriteLine ("genie: running systemctl poweroff within bottle");
-
-#             var systemdPid = Helpers.GetSystemdPid();
-#             var sd = Process.GetProcessById (systemdPid);
-
-#             Helpers.ChainInside (systemdPid,
-#                 new string[] {"systemctl", "poweroff"},
-#                 "running command failed; nsenter systemctl poweroff");
-
-#             // Wait for systemd to exit.
-#             Console.Write ("Waiting for systemd exit...");
-
-#             int timeout = Config.SystemdStartupTimeout;
-
-#             while (!sd.WaitForExit(1000))
-#             {
-#                 Console.Write (".");
-#                 timeout--;
-
-#                 if (timeout < 0)
-#                 {
-#                     Console.WriteLine("\n\nTimed out waiting for systemd to exit.\nThis may indicate a systemd configuration error.\nAttempting to continue.");
-#                     break;
-#                 }
-#             }
-
-#             Console.WriteLine();
-
-#             // We reverse the processes we performed to pre-startup the bottle as the post-shutdown, in reverse order.
-#             if (Config.AppArmorNamespace)
-#                 UnconfigureAppArmorNamespace (verbose);
-
-#             RemountBinfmts (verbose);
-
-#             if (Config.ResolvedStub)
-#                 Helpers.RemoveResolvSymlink (verbose);
-
-#             if (Config.UpdateHostname)
-#             {
-#                 Thread.Sleep (500);
-#                 Helpers.DropHostname (verbose);
-#             }
-
-#             FlagFiles.ShutdownFile = false;
-#         }
+    """Shutdown the genie bottle and clean up."""
+    sdp = find_systemd()
+
+    if sdp == 0:
+        sys.exit ("genie: no bottle exists")
+
+    if sdp == 1:
+        sys.exit ("genie: cannot shut down bottle from inside bottle")
+
+    state = get_systemd_state (sdp)
+
+    if 'starting' in state or 'stopping' in state:
+        sys.exit (f"genie: bottle is currently {state}; please wait until it is in a stable state")
+
+    if verbose:
+        print ("genie: running systemctl poweroff within bottle")
+
+    with nsenter.Namespace (sdp, 'pid'):
+        subprocess.run (["systemctl", "poweroff"])
+
+    # Wait for systemd to exit.
+    print ("Waiting for systemd to exit...", end="", flush=True)
+
+    timeout = config_system_timeout()
+
+    while find_systemd() != 0 and timeout > 0:
+        time.sleep (1)
+        print (".", end="", flush=True)
+
+        timeout -= 1
+
+    print ("")
+
+    if (timeout <= 0):
+        print ("genie: systemd did not exit after {config_system_timeout()} seconds")
+        print ("genie: this may be due to a problem with your systemd configuration")
+        print ("genie: attempting to continue")
+
+    # Reverse the processes we performed to prepare the bottle as the post-shutdown
+    # cleanup, only in reverse.
+    if os.path.exists('/sys/module/apparmor'):
+        apparmor_unconfigure()
+
+    binfmts_mount()
+
+    if config_update_hostname():
+        hostname_restore()    
 
 def do_is_running():
     """Display whether the bottle is running or not."""