Files
zurg/internal/torrent/hooks.go
2023-12-06 19:18:04 +01:00

49 lines
1.1 KiB
Go

package torrent
import (
"bytes"
"fmt"
"os/exec"
"github.com/debridmediamanager/zurg/internal/config"
"github.com/debridmediamanager/zurg/pkg/logutil"
)
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)
cmd.Args = append(cmd.Args, "zurg")
cmd.Args = append(cmd.Args, se.Args...)
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 *logutil.Logger) {
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)
}
}