rmser/internal/transport/http/handlers/ocr.go

package handlers

import (
	"net/http"

	"github.com/gin-gonic/gin"
	"github.com/google/uuid"
	"github.com/shopspring/decimal"
	"go.uber.org/zap"

	ocrService "rmser/internal/services/ocr"
	"rmser/pkg/logger"
)

type OCRHandler struct {
	service *ocrService.Service
}

func NewOCRHandler(service *ocrService.Service) *OCRHandler {
	return &OCRHandler{service: service}
}

// GetCatalog возвращает список товаров для OCR сервиса
func (h *OCRHandler) GetCatalog(c *gin.Context) {
	items, err := h.service.GetCatalogForIndexing()
	if err != nil {
		logger.Log.Error("Ошибка получения каталога для OCR", zap.Error(err))
		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
		return
	}
	c.JSON(http.StatusOK, items)
}

type MatchRequest struct {
	RawName     string  `json:"raw_name" binding:"required"`
	ProductID   string  `json:"product_id" binding:"required"`
	Quantity    float64 `json:"quantity"`
	ContainerID *string `json:"container_id"`
}

// SaveMatch сохраняет привязку (обучение)
func (h *OCRHandler) SaveMatch(c *gin.Context) {
	var req MatchRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
		return
	}

	pID, err := uuid.Parse(req.ProductID)
	if err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid product_id format"})
		return
	}

	qty := decimal.NewFromFloat(1.0)
	if req.Quantity > 0 {
		qty = decimal.NewFromFloat(req.Quantity)
	}

	var contID *uuid.UUID
	if req.ContainerID != nil && *req.ContainerID != "" {
		if uid, err := uuid.Parse(*req.ContainerID); err == nil {
			contID = &uid
		}
	}

	if err := h.service.SaveMapping(req.RawName, pID, qty, contID); err != nil {
		logger.Log.Error("Ошибка сохранения матчинга", zap.Error(err))
		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
		return
	}

	c.JSON(http.StatusOK, gin.H{"status": "saved"})
}

// GetMatches возвращает список всех обученных связей
func (h *OCRHandler) GetMatches(c *gin.Context) {
	matches, err := h.service.GetKnownMatches()
	if err != nil {
		logger.Log.Error("Ошибка получения списка матчей", zap.Error(err))
		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
		return
	}

	c.JSON(http.StatusOK, matches)
}

// GetUnmatched возвращает список нераспознанных позиций для подсказок
func (h *OCRHandler) GetUnmatched(c *gin.Context) {
	items, err := h.service.GetUnmatchedItems()
	if err != nil {
		logger.Log.Error("Ошибка получения списка unmatched", zap.Error(err))
		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
		return
	}
	c.JSON(http.StatusOK, items)
}