Add ffmpeg converter to ogg format
This commit is contained in:
@@ -10,6 +10,7 @@ import (
|
|||||||
|
|
||||||
"git.vakhrushev.me/av/transcriber/internal/entity"
|
"git.vakhrushev.me/av/transcriber/internal/entity"
|
||||||
"git.vakhrushev.me/av/transcriber/internal/repo"
|
"git.vakhrushev.me/av/transcriber/internal/repo"
|
||||||
|
"git.vakhrushev.me/av/transcriber/internal/repo/ffmpeg"
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
)
|
)
|
||||||
@@ -75,6 +76,7 @@ func (h *TranscribeHandler) CreateTranscribeJob(c *gin.Context) {
|
|||||||
fileRecord := &entity.File{
|
fileRecord := &entity.File{
|
||||||
Id: fileId,
|
Id: fileId,
|
||||||
Storage: entity.StorageLocal,
|
Storage: entity.StorageLocal,
|
||||||
|
FileName: fileName,
|
||||||
Size: size,
|
Size: size,
|
||||||
CreatedAt: time.Now(),
|
CreatedAt: time.Now(),
|
||||||
}
|
}
|
||||||
@@ -136,8 +138,50 @@ func (h *TranscribeHandler) RunConversionJob(c *gin.Context) {
|
|||||||
return
|
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)
|
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)
|
err = h.jobRepo.Save(job)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||||
|
@@ -12,6 +12,7 @@ const (
|
|||||||
type File struct {
|
type File struct {
|
||||||
Id string
|
Id string
|
||||||
Storage string
|
Storage string
|
||||||
|
FileName string
|
||||||
Size int64
|
Size int64
|
||||||
CreatedAt time.Time
|
CreatedAt time.Time
|
||||||
}
|
}
|
||||||
|
@@ -20,3 +20,7 @@ type TranscriptJobRepository interface {
|
|||||||
|
|
||||||
type ObjectStorage interface {
|
type ObjectStorage interface {
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type FileConverter interface {
|
||||||
|
Convert(src, dest string) error
|
||||||
|
}
|
||||||
|
47
internal/repo/ffmpeg/conv.go
Normal file
47
internal/repo/ffmpeg/conv.go
Normal 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
|
||||||
|
}
|
@@ -21,6 +21,7 @@ func (repo *FileRepository) Create(file *entity.File) error {
|
|||||||
record := goqu.Record{
|
record := goqu.Record{
|
||||||
"id": file.Id,
|
"id": file.Id,
|
||||||
"storage": file.Storage,
|
"storage": file.Storage,
|
||||||
|
"file_name": file.FileName,
|
||||||
"size": file.Size,
|
"size": file.Size,
|
||||||
"created_at": file.CreatedAt,
|
"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) {
|
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()
|
sql, args, err := query.ToSQL()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to build query: %w", err)
|
return nil, fmt.Errorf("failed to build query: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
var file entity.File
|
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 {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("failed to get file: %w", err)
|
return nil, fmt.Errorf("failed to get file: %w", err)
|
||||||
}
|
}
|
@@ -2,6 +2,7 @@
|
|||||||
CREATE TABLE files (
|
CREATE TABLE files (
|
||||||
id TEXT PRIMARY KEY,
|
id TEXT PRIMARY KEY,
|
||||||
storage TEXT NOT NULL,
|
storage TEXT NOT NULL,
|
||||||
|
file_name TEXT NOT NULL,
|
||||||
size INTEGER NOT NULL,
|
size INTEGER NOT NULL,
|
||||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||||
);
|
);
|
||||||
|
Reference in New Issue
Block a user