Files
zurg/internal/torrent/hooks.go
2023-11-23 00:22:10 +01:00

47 lines
988 B
Go

package torrent
import (
"bytes"
"fmt"
"os/exec"
"github.com/debridmediamanager.com/zurg/internal/config"
"go.uber.org/zap"
)
type ScriptExecutor struct {
Script string
Args []string
}
func (se *ScriptExecutor) Execute() (string, error) {
if se.Script == "" {
return "", nil
}
cmd := exec.Command("/bin/sh", "-c", se.Script)
var out bytes.Buffer
var stderr bytes.Buffer
cmd.Stdout = &out
cmd.Stderr = &stderr
err := cmd.Run()
if err != nil {
return "", fmt.Errorf("error executing script: %v; stderr: %s", err, stderr.String())
}
return out.String(), nil
}
func OnLibraryUpdateHook(paths []string, config config.ConfigInterface, log *zap.SugaredLogger) {
executor := &ScriptExecutor{
Script: config.GetOnLibraryUpdate(),
Args: paths,
}
output, err := executor.Execute()
if err != nil {
log.Errorf("Failed to execute hook on_library_update: %v", err)
return
}
if output != "" {
log.Infof("Output of hook on_library_update:\n%s", output)
}
}