From aeaaba50f5b4e483f5f06edfb3145ac6071646f3 Mon Sep 17 00:00:00 2001
From: Chord <chord@tuta.io>
Date: Tue, 26 Jul 2016 02:54:38 -0400
Subject: [PATCH] Add shutdown hook for cleanup

---
 pslogin/src/main/scala/PsLogin.scala | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pslogin/src/main/scala/PsLogin.scala b/pslogin/src/main/scala/PsLogin.scala
index 2780a4d2..a338300f 100644
--- a/pslogin/src/main/scala/PsLogin.scala
+++ b/pslogin/src/main/scala/PsLogin.scala
@@ -165,6 +165,12 @@ object PsLogin {
 
     logger.info(s"NOTE: Set client.ini to point to ${LoginConfig.serverIpAddress.getHostAddress}:$loginServerPort")
 
+    // Add our shutdown hook (this works for Control+C as well, but not in Cygwin)
+    sys addShutdownHook {
+      // TODO: clean up active sessions and close resources safely
+      logger.info("Login server now shutting down...")
+    }
+
     // Wait forever until the actor system shuts down
     Await.result(system.whenTerminated, Duration.Inf)
   }