1 files changed, 10 insertions, 0 deletions

diff --git a/WaitForSignal.hs b/WaitForSignal.hs
new file mode 100644
index 00000000..6252eaa6
--- /dev/null
+++ b/WaitForSignal.hs
@@ -0,0 +1,10 @@
+module WaitForSignal where
+
+import Control.Concurrent.STM
+import System.Posix.Signals
+
+waitForTermSignal = do
+    quitVar <- newEmptyTMVarIO
+    installHandler sigTERM (CatchOnce (atomically $ putTMVar quitVar True)) Nothing
+    installHandler sigINT (CatchOnce (atomically $ putTMVar quitVar True)) Nothing
+    atomically $ takeTMVar quitVar