84 lines
1.9 KiB
Go
84 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
"ytdlp-site/ffmpeg"
|
|
)
|
|
|
|
type FFProbeOutput struct {
|
|
Streams []struct {
|
|
CodecName string `json:"codec_name"`
|
|
} `json:"streams"`
|
|
}
|
|
|
|
func getAudioFormat(filename string) (string, error) {
|
|
output, _, err := ffmpeg.Ffprobe("-v", "quiet", "-print_format", "json", "-show_streams", filename)
|
|
if err != nil {
|
|
log.Errorln("ffprobe error:", err)
|
|
return "", err
|
|
}
|
|
|
|
var ffprobeOutput FFProbeOutput
|
|
err = json.Unmarshal(output, &ffprobeOutput)
|
|
if err != nil {
|
|
log.Errorln("failed to parse ffprobe output:", err)
|
|
return "", err
|
|
}
|
|
|
|
numStreams := len(ffprobeOutput.Streams)
|
|
if numStreams > 1 || numStreams <= 0 {
|
|
log.Error(numStreams, "streams in ffprobe output", numStreams)
|
|
return "", err
|
|
}
|
|
|
|
return ffprobeOutput.Streams[0].CodecName, nil
|
|
}
|
|
|
|
func getStreamBitrate(path string, stream int) (uint, error) {
|
|
ffprobeArgs := []string{
|
|
"-v", "quiet",
|
|
"-select_streams", fmt.Sprintf("a:%d", stream),
|
|
"-show_entries", "stream=bit_rate",
|
|
"-of", "default=noprint_wrappers=1:nokey=1",
|
|
path}
|
|
|
|
stdout, _, err := ffmpeg.Ffprobe(ffprobeArgs...)
|
|
if err != nil {
|
|
fmt.Println("ffprobe error:", err, string(stdout))
|
|
return 0, err
|
|
}
|
|
bitrateStr := strings.TrimSpace(string(stdout))
|
|
|
|
bitrate, err := strconv.ParseUint(bitrateStr, 10, 32)
|
|
if err != nil {
|
|
fmt.Println("parse bitrate error:", err)
|
|
return 0, err
|
|
}
|
|
return uint(bitrate), nil
|
|
}
|
|
|
|
func getFormatBitrate(path string) (uint, error) {
|
|
ffprobeArgs := []string{
|
|
"-v", "quiet",
|
|
"-show_entries", "format=bit_rate",
|
|
"-of", "default=noprint_wrappers=1:nokey=1",
|
|
path}
|
|
|
|
stdout, _, err := ffmpeg.Ffprobe(ffprobeArgs...)
|
|
if err != nil {
|
|
fmt.Println("ffprobe error:", err, string(stdout))
|
|
return 0, err
|
|
}
|
|
bitrateStr := strings.TrimSpace(string(stdout))
|
|
|
|
bitrate, err := strconv.ParseUint(bitrateStr, 10, 32)
|
|
if err != nil {
|
|
fmt.Println("parse bitrate error:", err)
|
|
return 0, err
|
|
}
|
|
return uint(bitrate), nil
|
|
}
|