From 82e50061b1f0e862d9156bbdacaab55257478cc7 Mon Sep 17 00:00:00 2001
From: nemunaire <nemunaire@nemunai.re>
Date: Tue, 12 May 2015 10:41:13 +0200
Subject: [PATCH] Catch SIGHUP: deep reload

---
 nemubot/__main__.py | 34 ++++++++++++++++++++++++++++++----
 1 file changed, 30 insertions(+), 4 deletions(-)

diff --git a/nemubot/__main__.py b/nemubot/__main__.py
index 4f739f8..757d3b0 100644
--- a/nemubot/__main__.py
+++ b/nemubot/__main__.py
@@ -155,14 +155,40 @@ def main():
             __import__(module)
 
     # Signals handling
-    def sighandler(signum, frame):
+    def sigtermhandler(signum, frame):
+        """On SIGTERM and SIGINT, quit nicely"""
         context.quit()
-    signal.signal(signal.SIGINT, sighandler)
-    signal.signal(signal.SIGTERM, sighandler)
+    signal.signal(signal.SIGINT, sigtermhandler)
+    signal.signal(signal.SIGTERM, sigtermhandler)
 
+    def sighuphandler(signum, frame):
+        """On SIGHUP, perform a deep reload"""
+        import imp
+
+        # Reload nemubot Python modules
+        imp.reload(nemubot)
+        nemubot.reload()
+
+        # Hotswap context
+        import nemubot.bot
+        context = nemubot.bot.hotswap(context)
+
+        # Reload configuration file
+        for path in args.files:
+            if os.path.isfile(path):
+                context.sync_queue.put_nowait(["loadconf", path])
+    signal.signal(signal.SIGHUP, sighuphandler)
+
+    # Here we go!
     context.start()
-    context.join()
 
+    # context can change when performing an hotswap, always join the latest context
+    oldcontext = None
+    while oldcontext != context:
+        oldcontext = context
+        context.join()
+
+    # Wait for consumers
     logger.info("Waiting for other threads shuts down...")
     sys.exit(0)