transcriber/internal/controller/http/transcribe.go

package http

import (
	"context"
	"log/slog"
	"net/http"
	"time"

	"github.com/pocketbase/pocketbase/apis"
	"github.com/pocketbase/pocketbase/core"
	"github.com/pocketbase/pocketbase/tools/router"

	"git.vakhrushev.me/av/transcriber/internal/contract"
	"git.vakhrushev.me/av/transcriber/internal/entity"
	"git.vakhrushev.me/av/transcriber/internal/service"
)

type TranscribeHandler struct {
	jobRepo    contract.TranscriptJobRepository
	trsService *service.TranscribeService
	logger     *slog.Logger
}

func NewTranscribeHandler(jobRepo contract.TranscriptJobRepository, trsService *service.TranscribeService, logger *slog.Logger) *TranscribeHandler {
	if logger == nil {
		logger = slog.Default()
	}
	return &TranscribeHandler{jobRepo: jobRepo, trsService: trsService, logger: logger}
}

type CreateTranscribeJobResponse struct {
	JobID string `json:"job_id"`
	State string `json:"status"`
}

type GetTranscribeJobResponse struct {
	JobID             string    `json:"job_id"`
	State             string    `json:"status"`
	CreatedAt         time.Time `json:"created_at"`
	TranscriptionText *string   `json:"transcription_text,omitempty"`
}

// Register вешает маршруты сервиса на роутер хранилища. Порт у сервиса и у
// панели один, поэтому и роутер один; имена полей ответа и коды при переезде
// сохранены — публичный контракт API объявлен необратимым.
func (h *TranscribeHandler) Register(r *router.Router[*core.RequestEvent]) {
	api := r.Group("/api")

	// Оба адреса уходят за аутентификацию. Слой предъявления стоит перед
	// проверкой и действует только здесь: собственная поверхность хранилища под
	// него не подпадает, часть её защищена ровно тем, что браузер заголовка сам
	// не шлёт.
	api.Bind(SessionFromCookie())
	api.Bind(apis.RequireAuth())

	// Умолчание роутера хранилища — 32 МиБ на тело, и оно отсекало бы запись
	// раньше обработчика, без строки в журнале приёма. Приём размеру не судья,
	// поэтому предел тела равен потолку самой записи.
	api.POST("/audio", h.CreateTranscribeJob).Bind(apis.BodyLimit(entity.MaxRecordSize))
	api.GET("/status/{id}", h.GetTranscribeJobStatus)
}

func (h *TranscribeHandler) CreateTranscribeJob(e *core.RequestEvent) error {
	// Получаем файл из формы
	file, header, err := e.Request.FormFile("audio")
	if err != nil {
		return e.JSON(http.StatusBadRequest, map[string]string{"error": "No audio file provided"})
	}
	defer func() {
		if err := file.Close(); err != nil {
			h.logger.Error("Failed to close uploaded file", "error", err)
		}
	}()

	// Запись доехала целиком, поэтому задача заводится независимо от того,
	// дождётся ли отправитель ответа: на контексте запроса приём терял бы
	// полностью загруженную запись от одного обрыва соединения, а забрать
	// результат он может и позже — по `GET /status/{id}`. Значения контекста
	// (журнал запроса, сессия) при этом сохраняются, теряется только отмена.
	ctx := context.WithoutCancel(e.Request.Context())

	job, err := h.trsService.CreateJobFromApi(ctx, file, header.Filename)
	if err != nil {
		// Второй раз отказ не логируем: приём назван конвенцией логирующей
		// границей и уже написал о нём. Транспорт переводит ошибку в ответ.
		return e.JSON(http.StatusInternalServerError, map[string]string{"error": "Failed to create transcibe job"})
	}

	// Возвращаем успешный ответ
	return e.JSON(http.StatusCreated, CreateTranscribeJobResponse{
		JobID: job.Id,
		State: job.State,
	})
}

func (h *TranscribeHandler) GetTranscribeJobStatus(e *core.RequestEvent) error {
	jobID := e.Request.PathValue("id")

	job, err := h.jobRepo.GetByID(jobID)
	if err != nil {
		return e.JSON(http.StatusNotFound, map[string]string{"error": "Job not found"})
	}

	return e.JSON(http.StatusOK, GetTranscribeJobResponse{
		JobID:             job.Id,
		State:             job.State,
		CreatedAt:         job.CreatedAt,
		TranscriptionText: job.TranscriptionText,
	})
}