feat: kill lua context of ctrl c during commander execution

pull/313/head
sammyette 2023-01-17 22:22:54 -04:00
parent 959030f70d
commit 883558c051
Signed by: sammyette
GPG Key ID: 904FC49417B44DCD
1 changed files with 24 additions and 1 deletions

25
exec.go
View File

@ -8,6 +8,7 @@ import (
"fmt" "fmt"
"io" "io"
"os" "os"
"os/signal"
"path/filepath" "path/filepath"
"runtime" "runtime"
"strings" "strings"
@ -323,7 +324,29 @@ func execHandle(bg bool) interp.ExecHandlerFunc {
} }
if commands[args[0]] != nil { 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 { if err != nil {
fmt.Fprintln(os.Stderr, "Error in command:\n" + err.Error()) fmt.Fprintln(os.Stderr, "Error in command:\n" + err.Error())
return interp.NewExitStatus(1) return interp.NewExitStatus(1)