From 883558c051ec5fffde2268d44011f7da2f09f8cb Mon Sep 17 00:00:00 2001
From: sammyette <torchedsammy@gmail.com>
Date: Tue, 17 Jan 2023 22:22:54 -0400
Subject: [PATCH] feat: kill lua context of ctrl c during commander execution

---
 exec.go | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/exec.go b/exec.go
index 6e24719..1d4b07d 100644
--- a/exec.go
+++ b/exec.go
@@ -8,6 +8,7 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"os/signal"
 	"path/filepath"
 	"runtime"
 	"strings"
@@ -323,7 +324,29 @@ func execHandle(bg bool) interp.ExecHandlerFunc {
 		}
 
 		if commands[args[0]] != nil {
-			luaexitcode, err := rt.Call1(l.MainThread(), rt.FunctionValue(commands[args[0]]), rt.TableValue(luacmdArgs))
+			t := rt.NewThread(l)
+			sig := make(chan os.Signal)
+			exit := make(chan bool)
+
+			go func() {
+				defer func() {
+					if r := recover(); r != nil {
+						fmt.Println(r)
+					}
+				}()
+
+				signal.Notify(sig, os.Interrupt)
+				select {
+					case <-sig:
+						t.KillContext()
+						return
+					case <-exit:
+						return
+				}
+			}()
+
+			luaexitcode, err := rt.Call1(t, rt.FunctionValue(commands[args[0]]), rt.TableValue(luacmdArgs))
+			exit <- true
 			if err != nil {
 				fmt.Fprintln(os.Stderr, "Error in command:\n" + err.Error())
 				return interp.NewExitStatus(1)