ytdlp package

ytdlp/init.go

@@ -0,0 +1,12 @@
+package ytdlp
+
+import "github.com/sirupsen/logrus"
+
+var log *logrus.Logger
+
+func Init(logger *logrus.Logger) error {
+	log = logger.WithFields(logrus.Fields{
+		"component": "ytdlp",
+	}).Logger
+	return nil
+}

ytdlp/ytdlp.go

@@ -0,0 +1,35 @@
+package ytdlp
+
+import (
+	"bytes"
+	"fmt"
+	"os/exec"
+	"strings"
+)
+
+// runs ffprobe with the provided args and returns (stdout, stderr, error)
+func Run(args ...string) ([]byte, []byte, error) {
+	ytdlp := "yt-dlp"
+	log.Infoln(ytdlp, strings.Join(args, " "))
+	cmd := exec.Command(ytdlp, args...)
+	var stdout bytes.Buffer
+	var stderr bytes.Buffer
+	cmd.Stdout = &stdout
+	cmd.Stderr = &stderr
+	err := cmd.Run()
+
+	if err != nil {
+		log.Errorf("yt-dlp error: %v", err)
+	}
+	log.Infoln("stdout:", stdout.String())
+	log.Infoln("stderr:", stderr.String())
+	return stdout.Bytes(), stderr.Bytes(), err
+}
+
+func Clip(src, dst string, from, to float64) error {
+	_, _, err := Run("-i", src,
+		"-ss", fmt.Sprintf("%f", from),
+		"-to", fmt.Sprintf("%f", to),
+		dst)
+	return err
+}