73 lines
2.2 KiB
Go
73 lines
2.2 KiB
Go
package ffmpeg
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strconv"
|
|
|
|
"git.vakhrushev.me/av/transcriber/internal/contract"
|
|
)
|
|
|
|
const ffprobeExecutable = "ffprobe"
|
|
|
|
type FfmpegMetaViewer struct {
|
|
}
|
|
|
|
// ffprobeOutput представляет структуру JSON-ответа от ffprobe
|
|
type ffprobeOutput struct {
|
|
Format struct {
|
|
Duration string `json:"duration"`
|
|
} `json:"format"`
|
|
}
|
|
|
|
func NewFfmpegMetaViewer() *FfmpegMetaViewer {
|
|
return &FfmpegMetaViewer{}
|
|
}
|
|
|
|
func (m *FfmpegMetaViewer) GetInfo(src string) (*contract.AudioInfo, error) {
|
|
// Проверяем существование исходного файла
|
|
if _, err := os.Stat(src); os.IsNotExist(err) {
|
|
return nil, fmt.Errorf("input file does not exist: %s", src)
|
|
}
|
|
|
|
// Проверяем, что ffprobe доступен в системе
|
|
if _, err := exec.LookPath(ffprobeExecutable); err != nil {
|
|
return nil, fmt.Errorf("ffprobe not found in PATH: %w", err)
|
|
}
|
|
|
|
// Создаем команду ffprobe для получения метаданных
|
|
cmd := exec.Command(ffprobeExecutable,
|
|
"-v", "quiet", // тихий режим (без лишнего вывода)
|
|
"-print_format", "json", // вывод в формате JSON
|
|
"-show_format", // показать информацию о формате
|
|
src, // входной файл
|
|
)
|
|
|
|
// Выполняем команду и получаем вывод
|
|
output, err := cmd.Output()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("ffprobe execution failed: %w", err)
|
|
}
|
|
|
|
// Парсим JSON-ответ
|
|
var probeResult ffprobeOutput
|
|
if err := json.Unmarshal(output, &probeResult); err != nil {
|
|
return nil, fmt.Errorf("failed to parse ffprobe output: %w", err)
|
|
}
|
|
|
|
// Конвертируем длительность из строки в секунды
|
|
durationFloat, err := strconv.ParseFloat(probeResult.Format.Duration, 64)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to parse duration: %w", err)
|
|
}
|
|
|
|
// Округляем до целых секунд
|
|
durationSeconds := int(durationFloat + 0.5) // +0.5 для правильного округления
|
|
|
|
return &contract.AudioInfo{
|
|
Seconds: durationSeconds,
|
|
}, nil
|
|
}
|