mirror of
https://github.com/xfrr/goffmpeg.git
synced 2025-10-06 00:16:50 +08:00
Added skip audio and video, also ran a go fmt for accurate tab spacing
This commit is contained in:
Rohit Garg
@@ -1,247 +1,247 @@
|
||||
package transcoder
|
||||
|
||||
import (
|
||||
"github.com/xfrr/goffmpeg/models"
|
||||
"github.com/xfrr/goffmpeg/ffmpeg"
|
||||
"github.com/xfrr/goffmpeg/utils"
|
||||
"errors"
|
||||
"os"
|
||||
"os/exec"
|
||||
"fmt"
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"bufio"
|
||||
"strings"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"io"
|
||||
"bufio"
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"github.com/xfrr/goffmpeg/ffmpeg"
|
||||
"github.com/xfrr/goffmpeg/models"
|
||||
"github.com/xfrr/goffmpeg/utils"
|
||||
"io"
|
||||
"os"
|
||||
"os/exec"
|
||||
"regexp"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Transcoder struct {
|
||||
stdErrPipe io.ReadCloser
|
||||
process *exec.Cmd
|
||||
mediafile *models.Mediafile
|
||||
configuration ffmpeg.Configuration
|
||||
stdErrPipe io.ReadCloser
|
||||
process *exec.Cmd
|
||||
mediafile *models.Mediafile
|
||||
configuration ffmpeg.Configuration
|
||||
}
|
||||
|
||||
func (t *Transcoder) SetProcessStderrPipe(v io.ReadCloser) {
|
||||
t.stdErrPipe = v
|
||||
t.stdErrPipe = v
|
||||
}
|
||||
|
||||
func (t *Transcoder) SetProcess(cmd *exec.Cmd) {
|
||||
t.process = cmd
|
||||
t.process = cmd
|
||||
}
|
||||
|
||||
func (t *Transcoder) SetMediaFile(v *models.Mediafile) {
|
||||
t.mediafile = v
|
||||
t.mediafile = v
|
||||
}
|
||||
|
||||
func (t *Transcoder) SetConfiguration(v ffmpeg.Configuration) {
|
||||
t.configuration = v
|
||||
t.configuration = v
|
||||
}
|
||||
|
||||
/*** GETTERS ***/
|
||||
func (t Transcoder) Process() *exec.Cmd{
|
||||
return t.process
|
||||
func (t Transcoder) Process() *exec.Cmd {
|
||||
return t.process
|
||||
}
|
||||
|
||||
func (t Transcoder) MediaFile() *models.Mediafile {
|
||||
return t.mediafile
|
||||
return t.mediafile
|
||||
}
|
||||
|
||||
func (t Transcoder) FFmpegExec() string {
|
||||
return t.configuration.FfmpegBin
|
||||
return t.configuration.FfmpegBin
|
||||
}
|
||||
|
||||
func (t Transcoder) FFprobeExec() string {
|
||||
return t.configuration.FfprobeBin
|
||||
return t.configuration.FfprobeBin
|
||||
}
|
||||
|
||||
func (t Transcoder) GetCommand() []string {
|
||||
media := t.mediafile
|
||||
rcommand := append([]string{"-y"}, media.ToStrCommand()...)
|
||||
return rcommand
|
||||
media := t.mediafile
|
||||
rcommand := append([]string{"-y"}, media.ToStrCommand()...)
|
||||
return rcommand
|
||||
}
|
||||
|
||||
/*** FUNCTIONS ***/
|
||||
|
||||
func (t *Transcoder) Initialize(inputPath string, outputPath string) (error) {
|
||||
func (t *Transcoder) Initialize(inputPath string, outputPath string) error {
|
||||
|
||||
configuration, err := ffmpeg.Configure()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return err
|
||||
}
|
||||
configuration, err := ffmpeg.Configure()
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
return err
|
||||
}
|
||||
|
||||
if inputPath == "" {
|
||||
return errors.New("error: transcoder.Initialize -> inputPath missing")
|
||||
}
|
||||
if inputPath == "" {
|
||||
return errors.New("error: transcoder.Initialize -> inputPath missing")
|
||||
}
|
||||
|
||||
_, err = os.Stat(inputPath)
|
||||
if os.IsNotExist(err) {
|
||||
return errors.New("error: transcoder.Initialize -> input file not found")
|
||||
}
|
||||
_, err = os.Stat(inputPath)
|
||||
if os.IsNotExist(err) {
|
||||
return errors.New("error: transcoder.Initialize -> input file not found")
|
||||
}
|
||||
|
||||
command := []string{"-i", inputPath, "-print_format", "json", "-show_format", "-show_streams", "-show_error"}
|
||||
command := []string{"-i", inputPath, "-print_format", "json", "-show_format", "-show_streams", "-show_error"}
|
||||
|
||||
cmd := exec.Command(configuration.FfprobeBin, command...)
|
||||
cmd := exec.Command(configuration.FfprobeBin, command...)
|
||||
|
||||
var out bytes.Buffer
|
||||
cmd.Stdout = &out
|
||||
var out bytes.Buffer
|
||||
cmd.Stdout = &out
|
||||
|
||||
err = cmd.Run()
|
||||
if err != nil {
|
||||
return fmt.Errorf("Failed FFPROBE (%s) with %s, message %s", command, err, out.String())
|
||||
}
|
||||
err = cmd.Run()
|
||||
if err != nil {
|
||||
return fmt.Errorf("Failed FFPROBE (%s) with %s, message %s", command, err, out.String())
|
||||
}
|
||||
|
||||
var Metadata models.Metadata
|
||||
var Metadata models.Metadata
|
||||
|
||||
if err = json.Unmarshal([]byte(out.String()), &Metadata); err != nil {
|
||||
return err
|
||||
}
|
||||
if err = json.Unmarshal([]byte(out.String()), &Metadata); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// Set new Mediafile
|
||||
MediaFile := new(models.Mediafile)
|
||||
MediaFile.SetMetadata(Metadata)
|
||||
MediaFile.SetInputPath(inputPath)
|
||||
MediaFile.SetOutputPath(outputPath)
|
||||
// Set transcoder configuration
|
||||
// Set new Mediafile
|
||||
MediaFile := new(models.Mediafile)
|
||||
MediaFile.SetMetadata(Metadata)
|
||||
MediaFile.SetInputPath(inputPath)
|
||||
MediaFile.SetOutputPath(outputPath)
|
||||
// Set transcoder configuration
|
||||
|
||||
t.SetMediaFile(MediaFile)
|
||||
t.SetConfiguration(configuration)
|
||||
t.SetMediaFile(MediaFile)
|
||||
t.SetConfiguration(configuration)
|
||||
|
||||
return nil
|
||||
return nil
|
||||
|
||||
}
|
||||
|
||||
func (t *Transcoder) Run(progress bool) <-chan error {
|
||||
done := make(chan error)
|
||||
command := t.GetCommand()
|
||||
if !progress {
|
||||
command = append([]string{"-nostats", "-loglevel", "0"}, command...)
|
||||
}
|
||||
proc := exec.Command(t.configuration.FfmpegBin, command...)
|
||||
if progress {
|
||||
errStream, err := proc.StderrPipe()
|
||||
if err != nil {
|
||||
fmt.Println("Progress not available: "+ err.Error())
|
||||
} else {
|
||||
t.SetProcessStderrPipe(errStream)
|
||||
}
|
||||
}
|
||||
done := make(chan error)
|
||||
command := t.GetCommand()
|
||||
if !progress {
|
||||
command = append([]string{"-nostats", "-loglevel", "0"}, command...)
|
||||
}
|
||||
proc := exec.Command(t.configuration.FfmpegBin, command...)
|
||||
if progress {
|
||||
errStream, err := proc.StderrPipe()
|
||||
if err != nil {
|
||||
fmt.Println("Progress not available: " + err.Error())
|
||||
} else {
|
||||
t.SetProcessStderrPipe(errStream)
|
||||
}
|
||||
}
|
||||
|
||||
out := &bytes.Buffer{}
|
||||
proc.Stdout = out
|
||||
out := &bytes.Buffer{}
|
||||
proc.Stdout = out
|
||||
|
||||
err := proc.Start()
|
||||
t.SetProcess(proc)
|
||||
go func(err error, out *bytes.Buffer) {
|
||||
if err != nil {
|
||||
done <- fmt.Errorf("Failed Start FFMPEG (%s) with %s, message %s", command, err, out.String())
|
||||
close(done)
|
||||
return
|
||||
}
|
||||
err = proc.Wait()
|
||||
if err != nil {
|
||||
err = fmt.Errorf("Failed Finish FFMPEG (%s) with %s message %s", command, err, out.String())
|
||||
}
|
||||
done <- err
|
||||
close(done)
|
||||
}(err, out)
|
||||
err := proc.Start()
|
||||
t.SetProcess(proc)
|
||||
go func(err error, out *bytes.Buffer) {
|
||||
if err != nil {
|
||||
done <- fmt.Errorf("Failed Start FFMPEG (%s) with %s, message %s", command, err, out.String())
|
||||
close(done)
|
||||
return
|
||||
}
|
||||
err = proc.Wait()
|
||||
if err != nil {
|
||||
err = fmt.Errorf("Failed Finish FFMPEG (%s) with %s message %s", command, err, out.String())
|
||||
}
|
||||
done <- err
|
||||
close(done)
|
||||
}(err, out)
|
||||
|
||||
return done
|
||||
return done
|
||||
}
|
||||
|
||||
func (t Transcoder) Output() <-chan models.Progress {
|
||||
out := make(chan models.Progress)
|
||||
out := make(chan models.Progress)
|
||||
|
||||
go func() {
|
||||
defer close(out)
|
||||
if t.stdErrPipe == nil {
|
||||
out <- models.Progress{}
|
||||
return
|
||||
} else {
|
||||
defer t.stdErrPipe.Close()
|
||||
}
|
||||
scanner := bufio.NewScanner(t.stdErrPipe)
|
||||
go func() {
|
||||
defer close(out)
|
||||
if t.stdErrPipe == nil {
|
||||
out <- models.Progress{}
|
||||
return
|
||||
} else {
|
||||
defer t.stdErrPipe.Close()
|
||||
}
|
||||
scanner := bufio.NewScanner(t.stdErrPipe)
|
||||
|
||||
split := func(data []byte, atEOF bool) (advance int, token []byte, spliterror error) {
|
||||
if atEOF && len(data) == 0 {
|
||||
return 0, nil, nil
|
||||
}
|
||||
if i := bytes.IndexByte(data, '\n'); i >= 0 {
|
||||
// We have a full newline-terminated line.
|
||||
return i + 1, data[0:i], nil
|
||||
}
|
||||
if i := bytes.IndexByte(data, '\r'); i >= 0 {
|
||||
// We have a cr terminated line
|
||||
return i + 1, data[0:i], nil
|
||||
}
|
||||
if atEOF {
|
||||
return len(data), data, nil
|
||||
}
|
||||
split := func(data []byte, atEOF bool) (advance int, token []byte, spliterror error) {
|
||||
if atEOF && len(data) == 0 {
|
||||
return 0, nil, nil
|
||||
}
|
||||
if i := bytes.IndexByte(data, '\n'); i >= 0 {
|
||||
// We have a full newline-terminated line.
|
||||
return i + 1, data[0:i], nil
|
||||
}
|
||||
if i := bytes.IndexByte(data, '\r'); i >= 0 {
|
||||
// We have a cr terminated line
|
||||
return i + 1, data[0:i], nil
|
||||
}
|
||||
if atEOF {
|
||||
return len(data), data, nil
|
||||
}
|
||||
|
||||
return 0, nil, nil
|
||||
}
|
||||
return 0, nil, nil
|
||||
}
|
||||
|
||||
scanner.Split(split)
|
||||
buf := make([]byte, 2)
|
||||
scanner.Buffer(buf, bufio.MaxScanTokenSize)
|
||||
scanner.Split(split)
|
||||
buf := make([]byte, 2)
|
||||
scanner.Buffer(buf, bufio.MaxScanTokenSize)
|
||||
|
||||
for scanner.Scan() {
|
||||
Progress := new(models.Progress)
|
||||
line := scanner.Text()
|
||||
if strings.Contains(line, "frame=") && strings.Contains(line, "time=") && strings.Contains(line, "bitrate=") {
|
||||
var re= regexp.MustCompile(`=\s+`)
|
||||
st := re.ReplaceAllString(line, `=`)
|
||||
for scanner.Scan() {
|
||||
Progress := new(models.Progress)
|
||||
line := scanner.Text()
|
||||
if strings.Contains(line, "frame=") && strings.Contains(line, "time=") && strings.Contains(line, "bitrate=") {
|
||||
var re = regexp.MustCompile(`=\s+`)
|
||||
st := re.ReplaceAllString(line, `=`)
|
||||
|
||||
f := strings.Fields(st)
|
||||
var framesProcessed string
|
||||
var currentTime string
|
||||
var currentBitrate string
|
||||
var currentSpeed string
|
||||
f := strings.Fields(st)
|
||||
var framesProcessed string
|
||||
var currentTime string
|
||||
var currentBitrate string
|
||||
var currentSpeed string
|
||||
|
||||
for j := 0; j < len(f); j++ {
|
||||
field := f[j]
|
||||
fieldSplit := strings.Split(field, "=")
|
||||
for j := 0; j < len(f); j++ {
|
||||
field := f[j]
|
||||
fieldSplit := strings.Split(field, "=")
|
||||
|
||||
if len(fieldSplit) > 1 {
|
||||
fieldname := strings.Split(field, "=")[0]
|
||||
fieldvalue := strings.Split(field, "=")[1]
|
||||
if len(fieldSplit) > 1 {
|
||||
fieldname := strings.Split(field, "=")[0]
|
||||
fieldvalue := strings.Split(field, "=")[1]
|
||||
|
||||
if fieldname == "frame" {
|
||||
framesProcessed = fieldvalue
|
||||
}
|
||||
if fieldname == "frame" {
|
||||
framesProcessed = fieldvalue
|
||||
}
|
||||
|
||||
if fieldname == "time" {
|
||||
currentTime = fieldvalue
|
||||
}
|
||||
if fieldname == "time" {
|
||||
currentTime = fieldvalue
|
||||
}
|
||||
|
||||
if fieldname == "bitrate" {
|
||||
currentBitrate = fieldvalue
|
||||
}
|
||||
if fieldname == "speed" {
|
||||
currentSpeed = fieldvalue
|
||||
}
|
||||
}
|
||||
}
|
||||
if fieldname == "bitrate" {
|
||||
currentBitrate = fieldvalue
|
||||
}
|
||||
if fieldname == "speed" {
|
||||
currentSpeed = fieldvalue
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
timesec := utils.DurToSec(currentTime)
|
||||
dursec, _ := strconv.ParseFloat(t.MediaFile().Metadata().Format.Duration, 64)
|
||||
//live stream check
|
||||
if dursec != 0 {
|
||||
// Progress calculation
|
||||
progress := (timesec * 100) / dursec
|
||||
Progress.Progress = progress
|
||||
}
|
||||
Progress.CurrentBitrate = currentBitrate
|
||||
Progress.FramesProcessed = framesProcessed
|
||||
Progress.CurrentTime = currentTime
|
||||
Progress.Speed = currentSpeed
|
||||
out <- *Progress
|
||||
}
|
||||
}
|
||||
}()
|
||||
timesec := utils.DurToSec(currentTime)
|
||||
dursec, _ := strconv.ParseFloat(t.MediaFile().Metadata().Format.Duration, 64)
|
||||
//live stream check
|
||||
if dursec != 0 {
|
||||
// Progress calculation
|
||||
progress := (timesec * 100) / dursec
|
||||
Progress.Progress = progress
|
||||
}
|
||||
Progress.CurrentBitrate = currentBitrate
|
||||
Progress.FramesProcessed = framesProcessed
|
||||
Progress.CurrentTime = currentTime
|
||||
Progress.Speed = currentSpeed
|
||||
out <- *Progress
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
return out
|
||||
return out
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user