mirror of
https://github.com/serty2005/rmser.git
synced 2026-02-04 19:02:33 -06:00
117 lines
3.5 KiB
Go
117 lines
3.5 KiB
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"})
|
||
}
|
||
|
||
// DeleteMatch удаляет связь
|
||
func (h *OCRHandler) DeleteMatch(c *gin.Context) {
|
||
// Получаем raw_name из query параметров, так как в URL path могут быть спецсимволы
|
||
// Пример: DELETE /api/ocr/match?raw_name=Хлеб%20Бородинский
|
||
rawName := c.Query("raw_name")
|
||
if rawName == "" {
|
||
c.JSON(http.StatusBadRequest, gin.H{"error": "raw_name is required"})
|
||
return
|
||
}
|
||
|
||
if err := h.service.DeleteMatch(rawName); 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": "deleted"})
|
||
}
|
||
|
||
// 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)
|
||
}
|