check for system signals

main.go

@@ -6,14 +6,21 @@ import (
 	"io/ioutil"
 	"os"
 	"os/exec"
+	"os/signal"
 	"path/filepath"
 	"strings"
+	"syscall"
 )
 
 var (
 	envFile = flag.String("env", "$HOME/.config/i3-wrapper/environment", "environment file with variable definitions")
 )
 
+var (
+	i3      *os.Process
+	running bool
+)
+
 func main() {
 	logFile, err := os.Create(filepath.Join(os.Getenv("HOME"), ".i3.log"))
 	if err != nil {
@@ -24,7 +31,18 @@ func main() {
 	os.Stdout = logFile
 	os.Stderr = logFile
 
-	for {
+	fmt.Println("environment file:", os.ExpandEnv(*envFile))
+
+	sigs := make(chan os.Signal, 1)
+	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
+	go func() {
+		sig := <-sigs
+		running = false
+		i3.Signal(sig)
+	}()
+
+	running = true
+	for running {
 		cmd := exec.Command("i3")
 		cmd.Stdin = os.Stdin
 		cmd.Stdout = os.Stdout
@@ -41,6 +59,7 @@ func main() {
 		if err != nil {
 			fmt.Fprintln(os.Stderr, err)
 		}
+		i3 = cmd.Process
 
 		err = cmd.Wait()
 		if err != nil {