From 41229be20da92dda7d3cd98a8f70b1465a2c98ab Mon Sep 17 00:00:00 2001
From: Fuwn <fuwnzy@gmail.com>
Date: Mon, 5 Jul 2021 18:30:03 -0700
Subject: feat(munch): setup signal handler

This commit adds a signal handler which detects Ctrl+Cs and then
properly shuts down all of Munch. Before, only one of Munch's Goroutines
would be shut down to the full leaving a dangling Goroutine, in this
case; the Worlds bot.
---
 munch.go                |  2 ++
 pkg/utilities/signal.go | 25 +++++++++++++++++++++++++
 2 files changed, 27 insertions(+)
 create mode 100644 pkg/utilities/signal.go

diff --git a/munch.go b/munch.go
index da1a28e..4624538 100644
--- a/munch.go
+++ b/munch.go
@@ -7,12 +7,14 @@ import (
 	"github.com/Whirlsplash/munch/pkg/config"
 	"github.com/Whirlsplash/munch/pkg/discord"
 	"github.com/Whirlsplash/munch/pkg/server"
+	"github.com/Whirlsplash/munch/pkg/utilities"
 	"github.com/spf13/viper"
 	"sync"
 )
 
 func main() {
 	config.Setup()
+	utilities.SetupSignalHandler()
 
 	var wg sync.WaitGroup
 
diff --git a/pkg/utilities/signal.go b/pkg/utilities/signal.go
new file mode 100644
index 0000000..00fbf1e
--- /dev/null
+++ b/pkg/utilities/signal.go
@@ -0,0 +1,25 @@
+// Copyright (C) 2021-2021 The Whirlsplash Collective
+// SPDX-License-Identifier: GPL-3.0-only
+
+package utilities
+
+import (
+	"log"
+	"os"
+	"os/signal"
+	"syscall"
+)
+
+func SetupSignalHandler() {
+	c := make(chan os.Signal)
+
+	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
+
+	go func() {
+		<-c
+
+		log.Println("Killing Munch with SignalHandler")
+
+		os.Exit(1)
+	}()
+}
-- 
cgit v1.2.3