Add hooks functionality

This commit is contained in:
Ben Sarmiento
2023-10-28 13:19:32 +02:00
parent a9d017f481
commit 1cb7873a3b
5 changed files with 79 additions and 28 deletions

44
internal/torrent/hooks.go Normal file
View File

@@ -0,0 +1,44 @@
package torrent
import (
"bytes"
"fmt"
"log"
"os/exec"
"github.com/debridmediamanager.com/zurg/internal/config"
)
type ScriptExecutor struct {
Script 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(config config.ConfigInterface) {
executor := &ScriptExecutor{
Script: config.GetOnLibraryUpdate(),
}
output, err := executor.Execute()
if err != nil {
log.Printf("Failed to execute hook on_library_update:\n%v\n", err)
return
}
if output != "" {
log.Printf("Output of hook on_library_update:\n%s\n", output)
}
}

View File

@@ -202,6 +202,7 @@ func (t *TorrentManager) startRefreshJob() {
go t.repairAll(&wg)
}
go t.mapToDirectories()
go OnLibraryUpdateHook(t.config)
}
}