59 lines
1.1 KiB
Go
59 lines
1.1 KiB
Go
//go:build !windows
|
|
// +build !windows
|
|
|
|
package utils
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"os/exec"
|
|
)
|
|
|
|
var (
|
|
ErrInvalidInput = errors.New("invalid ffmpeg input")
|
|
)
|
|
|
|
// ServeTranscodedStream passes an input file or io.Reader to ffmpeg and writes the output directly
|
|
// to our io.Writer.
|
|
func ServeTranscodedStream(w io.Writer, input interface{}, ff *exec.Cmd) error {
|
|
// Pipe streaming is not great as explained here
|
|
// https://video.stackexchange.com/questions/34087/ffmpeg-fails-on-pipe-to-pipe-video-decoding.
|
|
// That's why if we have the option to pass the file directly to ffmpeg, we should.
|
|
var in string
|
|
switch f := input.(type) {
|
|
case string:
|
|
in = f
|
|
case io.Reader:
|
|
in = "pipe:0"
|
|
default:
|
|
return ErrInvalidInput
|
|
}
|
|
|
|
if ff != nil && ff.Process != nil {
|
|
_ = ff.Process.Kill()
|
|
}
|
|
|
|
cmd := exec.Command(
|
|
"ffmpeg",
|
|
"-re",
|
|
"-i", in,
|
|
"-vcodec", "h264",
|
|
"-acodec", "aac",
|
|
"-ac", "2",
|
|
"-vf", "format=yuv420p",
|
|
"-movflags", "+faststart",
|
|
"-f", "flv",
|
|
"pipe:1",
|
|
)
|
|
|
|
*ff = *cmd
|
|
|
|
if in == "pipe:0" {
|
|
ff.Stdin = input.(io.Reader)
|
|
}
|
|
|
|
ff.Stdout = w
|
|
|
|
return ff.Run()
|
|
}
|