marcobaobao
4c35b0b41f
introduced pipelines and abstracted download process.go in Downloader interface
67 lines
1.1 KiB
Go
67 lines
1.1 KiB
Go
package pipes
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"io"
|
|
"log/slog"
|
|
"os/exec"
|
|
"strings"
|
|
)
|
|
|
|
type Transcoder struct {
|
|
Args []string
|
|
}
|
|
|
|
func (t *Transcoder) Name() string { return "ffmpeg-transcoder" }
|
|
|
|
func (t *Transcoder) Connect(r io.Reader) (io.Reader, error) {
|
|
cmd := exec.Command("ffmpeg",
|
|
append([]string{"-i", "pipe:0"}, append(t.Args, "-f", "webm", "pipe:1")...)...,
|
|
)
|
|
|
|
stdin, err := cmd.StdinPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
stdout, err := cmd.StdoutPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
stderr, err := cmd.StderrPipe()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
go func() {
|
|
reader := bufio.NewReader(stderr)
|
|
var line string
|
|
|
|
for {
|
|
part, err := reader.ReadString('\r')
|
|
line += part
|
|
if err != nil {
|
|
break
|
|
}
|
|
|
|
line = strings.TrimRight(line, "\r\n")
|
|
slog.Info("ffmpeg transcoder", slog.String("log", line))
|
|
line = ""
|
|
}
|
|
}()
|
|
|
|
go func() {
|
|
defer stdin.Close()
|
|
_, err := io.Copy(stdin, r)
|
|
if err != nil && !errors.Is(err, io.EOF) {
|
|
slog.Error("transcoder stdin error", slog.Any("err", err))
|
|
}
|
|
}()
|
|
|
|
if err := cmd.Start(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return stdout, nil
|
|
}
|