From d3777a7fcddd9d175c067756421f6dcf8a0e786c Mon Sep 17 00:00:00 2001 From: Ahren Stevens-Taylor Date: Thu, 6 May 2021 08:15:16 +0100 Subject: [PATCH] #42: Expose Work ID to script environment (#43) feat(): expose hook ID to script environment --- pkg/worker/work_runner.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pkg/worker/work_runner.go b/pkg/worker/work_runner.go index 6c163ba..ad4e8f2 100644 --- a/pkg/worker/work_runner.go +++ b/pkg/worker/work_runner.go @@ -2,6 +2,7 @@ package worker import ( "bufio" + "fmt" "io" "os" "os/exec" @@ -38,7 +39,8 @@ func Run(work *model.WorkRequest) error { // Exec script with args... cmd := exec.Command(binary, work.Payload) // with env variables... - cmd.Env = append(os.Environ(), work.Args...) + workEnv := append(os.Environ(), fmt.Sprintf("hook_id=%d", work.ID)) + cmd.Env = append(workEnv, work.Args...) // using a process group... cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}