#42: Expose Work ID to script environment (#43)

feat(): expose hook ID to script environment
ncarlier · May 6, 2021 · d3777a7 · d3777a7
1 parent f29a174
commit d3777a7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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}