Init: generated with claude-sonnet-4

handlers/transcribe.go

@@ -0,0 +1,118 @@
+package handlers
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+	"os"
+	"path/filepath"
+	"time"
+
+	"git.vakhrushev.me/av/transcriber/database"
+	"git.vakhrushev.me/av/transcriber/models"
+	"github.com/gin-gonic/gin"
+	"github.com/google/uuid"
+)
+
+type TranscribeHandler struct {
+	db *database.DB
+}
+
+func NewTranscribeHandler(db *database.DB) *TranscribeHandler {
+	return &TranscribeHandler{db: db}
+}
+
+type TranscribeResponse struct {
+	JobID  string `json:"job_id"`
+	FileID string `json:"file_id"`
+	Status string `json:"status"`
+}
+
+func (h *TranscribeHandler) UploadAndTranscribe(c *gin.Context) {
+	// Получаем файл из формы
+	file, header, err := c.Request.FormFile("audio")
+	if err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "No audio file provided"})
+		return
+	}
+	defer file.Close()
+
+	// Генерируем UUID для файла
+	fileID := uuid.New().String()
+
+	// Определяем расширение файла
+	ext := filepath.Ext(header.Filename)
+	if ext == "" {
+		ext = ".audio" // fallback если расширение не определено
+	}
+
+	// Создаем путь для сохранения файла
+	fileName := fmt.Sprintf("%s%s", fileID, ext)
+	filePath := filepath.Join("data", "files", fileName)
+
+	// Создаем файл на диске
+	dst, err := os.Create(filePath)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create file"})
+		return
+	}
+	defer dst.Close()
+
+	// Копируем содержимое загруженного файла
+	size, err := io.Copy(dst, file)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to save file"})
+		return
+	}
+
+	// Создаем запись в таблице files
+	fileRecord := &models.File{
+		ID:        fileID,
+		Type:      header.Header.Get("Content-Type"),
+		Size:      size,
+		CreatedAt: time.Now(),
+	}
+
+	if err := h.db.CreateFile(fileRecord); err != nil {
+		// Удаляем файл если не удалось создать запись в БД
+		os.Remove(filePath)
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to save file record"})
+		return
+	}
+
+	// Создаем запись в таблице transcribe_jobs
+	jobID := uuid.New().String()
+	job := &models.TranscribeJob{
+		ID:        jobID,
+		Status:    models.StatusPending,
+		FileID:    fileID,
+		CreatedAt: time.Now(),
+		UpdatedAt: time.Now(),
+	}
+
+	if err := h.db.CreateTranscribeJob(job); err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": "Failed to create transcribe job"})
+		return
+	}
+
+	// Возвращаем успешный ответ
+	response := TranscribeResponse{
+		JobID:  jobID,
+		FileID: fileID,
+		Status: models.StatusPending,
+	}
+
+	c.JSON(http.StatusCreated, response)
+}
+
+func (h *TranscribeHandler) GetJobStatus(c *gin.Context) {
+	jobID := c.Param("id")
+
+	job, err := h.db.GetTranscribeJobByID(jobID)
+	if err != nil {
+		c.JSON(http.StatusNotFound, gin.H{"error": "Job not found"})
+		return
+	}
+
+	c.JSON(http.StatusOK, job)
+}