49 lines
1.0 KiB
Go
49 lines
1.0 KiB
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)
|
|
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 *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)
|
|
}
|
|
}
|