From 461055a7e996f91c114061dcf3608b7af8155e6f Mon Sep 17 00:00:00 2001
From: Niklas Elsbrock <mail@nelsbrock.de>
Date: Sun, 17 Jul 2022 21:10:49 +0200
Subject: [PATCH] add graceful shutdown on `SIGTERM` for *nix

---
 src/bin/spamcontest.rs | 35 +++++++++++++++++++++++++++++++++--
 1 file changed, 33 insertions(+), 2 deletions(-)

diff --git a/src/bin/spamcontest.rs b/src/bin/spamcontest.rs
index 519a95c..25a28e6 100644
--- a/src/bin/spamcontest.rs
+++ b/src/bin/spamcontest.rs
@@ -1,7 +1,10 @@
 use log::{debug, error, LevelFilter};
 use serenity::prelude::*;
 use spamcontest::Handler;
-use std::{env, process};
+use std::{env, io, process};
+
+#[cfg(unix)]
+use tokio::pin;
 
 const TOKEN_VAR_KEY: &str = "DISCORD_TOKEN";
 
@@ -34,7 +37,10 @@ async fn main() {
 
     let shard_manager = client.shard_manager.clone();
     tokio::spawn(async move {
-        tokio::signal::ctrl_c().await.expect("ctrl+c handler error");
+        wait_for_shutdown_signal()
+            .await
+            .expect("error on waiting for shutdown signal");
+
         debug!("Shutting down...");
         shard_manager.lock().await.shutdown_all().await;
     });
@@ -44,3 +50,28 @@ async fn main() {
         process::exit(2);
     }
 }
+
+async fn wait_for_shutdown_signal() -> io::Result<()> {
+    let ctrl_c = tokio::signal::ctrl_c();
+
+    #[cfg(unix)]
+    {
+        use tokio::signal::unix;
+
+        let sigterm = async {
+            unix::signal(unix::SignalKind::terminate())?.recv().await;
+            Ok(())
+        };
+
+        pin!(ctrl_c);
+        pin!(sigterm);
+
+        tokio::select! {
+            result = &mut ctrl_c => { result }
+            result = &mut sigterm => { result }
+        }
+    }
+
+    #[cfg(not(unix))]
+    ctrl_c.await
+}