Add ffmpeg converter to ogg format

This commit is contained in:
2025-08-11 10:11:13 +03:00
parent 87d8b05efb
commit c0d55c2088
7 changed files with 100 additions and 2 deletions

View File

@@ -10,6 +10,7 @@ import (
"git.vakhrushev.me/av/transcriber/internal/entity"
"git.vakhrushev.me/av/transcriber/internal/repo"
"git.vakhrushev.me/av/transcriber/internal/repo/ffmpeg"
"github.com/gin-gonic/gin"
"github.com/google/uuid"
)
@@ -75,6 +76,7 @@ func (h *TranscribeHandler) CreateTranscribeJob(c *gin.Context) {
fileRecord := &entity.File{
Id: fileId,
Storage: entity.StorageLocal,
FileName: fileName,
Size: size,
CreatedAt: time.Now(),
}
@@ -136,8 +138,50 @@ func (h *TranscribeHandler) RunConversionJob(c *gin.Context) {
return
}
srcFile, err := h.fileRepo.GetByID(*job.FileID)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
srcFilePath := filepath.Join("data", "files", srcFile.FileName)
destFileId := uuid.New().String()
destFileName := fmt.Sprintf("%s%s", destFileId, ".ogg")
destFilePath := filepath.Join("data", "files", destFileName)
conv := ffmpeg.NewFileConverter()
err = conv.Convert(srcFilePath, destFilePath)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
stat, err := os.Stat(destFilePath)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
// Создаем запись в таблице files
destFileRecord := &entity.File{
Id: destFileId,
Storage: entity.StorageLocal,
FileName: destFileName,
Size: stat.Size(),
CreatedAt: time.Now(),
}
job.FileID = &destFileId
job.MoveToState(entity.StateConverted)
err = h.fileRepo.Create(destFileRecord)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
err = h.jobRepo.Save(job)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})

View File

@@ -12,6 +12,7 @@ const (
type File struct {
Id string
Storage string
FileName string
Size int64
CreatedAt time.Time
}

View File

@@ -20,3 +20,7 @@ type TranscriptJobRepository interface {
type ObjectStorage interface {
}
type FileConverter interface {
Convert(src, dest string) error
}

View File

@@ -0,0 +1,47 @@
package ffmpeg
import (
"fmt"
"os"
"os/exec"
)
type FileConverter struct {
}
func NewFileConverter() *FileConverter {
return &FileConverter{}
}
func (c *FileConverter) Convert(src, dest string) error {
// Проверяем существование исходного файла
if _, err := os.Stat(src); os.IsNotExist(err) {
return fmt.Errorf("input file does not exist: %s", src)
}
// Проверяем, что ffmpeg доступен в системе
if _, err := exec.LookPath("ffmpeg"); err != nil {
return fmt.Errorf("ffmpeg not found in PATH: %w", err)
}
// Создаем команду ffmpeg для конвертации в OGG
cmd := exec.Command("ffmpeg",
"-i", src, // входной файл
"-c:a", "libvorbis", // кодек Vorbis для OGG
"-q:a", "4", // качество аудио (0-10, где 4 - хорошее качество)
"-y", // перезаписать выходной файл если существует
dest, // выходной файл
)
// Выполняем команду
if err := cmd.Run(); err != nil {
return fmt.Errorf("ffmpeg conversion failed: %w", err)
}
// Проверяем, что выходной файл был создан
if _, err := os.Stat(dest); os.IsNotExist(err) {
return fmt.Errorf("output file was not created: %s", dest)
}
return nil
}

View File

@@ -21,6 +21,7 @@ func (repo *FileRepository) Create(file *entity.File) error {
record := goqu.Record{
"id": file.Id,
"storage": file.Storage,
"file_name": file.FileName,
"size": file.Size,
"created_at": file.CreatedAt,
}
@@ -39,14 +40,14 @@ func (repo *FileRepository) Create(file *entity.File) error {
}
func (repo *FileRepository) GetByID(id string) (*entity.File, error) {
query := repo.gq.From("files").Select("id", "storage", "size", "created_at").Where(goqu.C("id").Eq(id))
query := repo.gq.From("files").Select("id", "storage", "file_name", "size", "created_at").Where(goqu.C("id").Eq(id))
sql, args, err := query.ToSQL()
if err != nil {
return nil, fmt.Errorf("failed to build query: %w", err)
}
var file entity.File
err = repo.db.QueryRow(sql, args...).Scan(&file.Id, &file.Storage, &file.Size, &file.CreatedAt)
err = repo.db.QueryRow(sql, args...).Scan(&file.Id, &file.Storage, &file.FileName, &file.Size, &file.CreatedAt)
if err != nil {
return nil, fmt.Errorf("failed to get file: %w", err)
}

View File

@@ -2,6 +2,7 @@
CREATE TABLE files (
id TEXT PRIMARY KEY,
storage TEXT NOT NULL,
file_name TEXT NOT NULL,
size INTEGER NOT NULL,
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
);