Добавил черновики накладных и OCR через Яндекс. LLM для расшифровки универсальный

internal/services/ocr/service.go

@@ -10,6 +10,7 @@ import (
 	"go.uber.org/zap"
 
 	"rmser/internal/domain/catalog"
+	"rmser/internal/domain/drafts"
 	"rmser/internal/domain/ocr"
 	"rmser/internal/infrastructure/ocr_client"
 	"rmser/pkg/logger"
@@ -18,57 +19,116 @@ import (
 type Service struct {
 	ocrRepo     ocr.Repository
 	catalogRepo catalog.Repository
+	draftRepo   drafts.Repository
 	pyClient    *ocr_client.Client // Клиент к Python сервису
 }
 
 func NewService(
 	ocrRepo ocr.Repository,
 	catalogRepo catalog.Repository,
+	draftRepo drafts.Repository,
 	pyClient *ocr_client.Client,
 ) *Service {
 	return &Service{
 		ocrRepo:     ocrRepo,
 		catalogRepo: catalogRepo,
+		draftRepo:   draftRepo,
 		pyClient:    pyClient,
 	}
 }
 
-// ProcessReceiptImage - основной метод: Картинка -> Распознавание -> Матчинг
-func (s *Service) ProcessReceiptImage(ctx context.Context, imgData []byte) ([]ProcessedItem, error) {
-	// 1. Отправляем в Python
+// ProcessReceiptImage - Создает черновик, распознает, сохраняет результаты
+func (s *Service) ProcessReceiptImage(ctx context.Context, chatID int64, imgData []byte) (*drafts.DraftInvoice, error) {
+	// 1. Создаем заготовку черновика
+	draft := &drafts.DraftInvoice{
+		ChatID: chatID,
+		Status: drafts.StatusProcessing,
+	}
+	if err := s.draftRepo.Create(draft); err != nil {
+		return nil, fmt.Errorf("failed to create draft: %w", err)
+	}
+	logger.Log.Info("Создан черновик", zap.String("draft_id", draft.ID.String()))
+
+	// 2. Отправляем в Python OCR
 	rawResult, err := s.pyClient.ProcessImage(ctx, imgData, "receipt.jpg")
 	if err != nil {
+		// Ставим статус ошибки
+		draft.Status = drafts.StatusError
+		_ = s.draftRepo.Update(draft)
 		return nil, fmt.Errorf("python ocr error: %w", err)
 	}
 
-	var processed []ProcessedItem
+	// 3. Обрабатываем результаты и создаем Items
+	var draftItems []drafts.DraftInvoiceItem
+
 	for _, rawItem := range rawResult.Items {
-		item := ProcessedItem{
-			RawName: rawItem.RawName,
-			Amount:  decimal.NewFromFloat(rawItem.Amount),
-			Price:   decimal.NewFromFloat(rawItem.Price),
-			Sum:     decimal.NewFromFloat(rawItem.Sum),
+		item := drafts.DraftInvoiceItem{
+			DraftID:   draft.ID,
+			RawName:   rawItem.RawName,
+			RawAmount: decimal.NewFromFloat(rawItem.Amount),
+			RawPrice:  decimal.NewFromFloat(rawItem.Price),
+			// Quantity/Price по умолчанию берем как Raw, если не будет пересчета
+			Quantity: decimal.NewFromFloat(rawItem.Amount),
+			Price:    decimal.NewFromFloat(rawItem.Price),
+			Sum:      decimal.NewFromFloat(rawItem.Sum),
 		}
 
+		// Пытаемся найти матчинг
 		match, err := s.ocrRepo.FindMatch(rawItem.RawName)
 		if err != nil {
 			logger.Log.Error("db error finding match", zap.Error(err))
 		}
 
 		if match != nil {
-			item.ProductID = &match.ProductID
 			item.IsMatched = true
-			item.MatchSource = "learned"
-			// Здесь мы могли бы подтянуть quantity/container из матча,
-			// но пока фронт сам это сделает, запросив /ocr/matches или получив подсказку.
+			item.ProductID = &match.ProductID
+			item.ContainerID = match.ContainerID
+
+			// Важная логика: Если в матче указано ContainerID, то Quantity из чека (например 5 шт)
+			// это 5 коробок. Финальное кол-во (в кг) RMS посчитает сама,
+			// либо мы можем пересчитать тут, если знаем коэффициент.
+			// Пока оставляем Quantity как есть (кол-во упаковок),
+			// так как ContainerID передается в iiko.
 		} else {
+			// Если не нашли - сохраняем в Unmatched для статистики и подсказок
 			if err := s.ocrRepo.UpsertUnmatched(rawItem.RawName); err != nil {
 				logger.Log.Warn("failed to save unmatched", zap.Error(err))
 			}
 		}
-		processed = append(processed, item)
+
+		draftItems = append(draftItems, item)
 	}
-	return processed, nil
+
+	// 4. Сохраняем позиции в БД
+	// Примечание: GORM умеет сохранять вложенные структуры через Update родителя,
+	// но надежнее явно сохранить items, если мы не используем Session FullSaveAssociations.
+	// В данном случае мы уже создали Draft, теперь привяжем к нему items.
+	// Для простоты, так как у нас в Repo нет метода SaveItems,
+	// мы обновим драфт, добавив Items (GORM должен создать их).
+	draft.Status = drafts.StatusReadyToVerify
+
+	if err := s.draftRepo.Update(draft); err != nil {
+		return nil, fmt.Errorf("failed to update draft status: %w", err)
+	}
+	draft.Items = draftItems
+
+	// Используем хак GORM: при обновлении объекта с ассоциациями, он их создаст.
+	// Но надежнее расширить репозиторий. Давай используем Repository Update,
+	// но он у нас обновляет только шапку.
+	// Поэтому лучше расширим draftRepo методом SaveItems или используем прямую запись тут через items?
+	// Сделаем правильно: добавим AddItems в репозиторий прямо сейчас, или воспользуемся тем, что Items сохранятся
+	// если мы сделаем Save через GORM. В нашем Repo метод Create делает Create.
+	// Давайте сделаем SaveItems в репозитории drafts, чтобы было чисто.
+
+	// ВРЕМЕННОЕ РЕШЕНИЕ (чтобы не менять интерфейс снова):
+	// Мы можем создать items через repository, но там нет метода.
+	// Давай я добавлю метод в интерфейс репозитория Drafts в следующем блоке изменений.
+	// Пока предположим, что мы расширили репозиторий.
+	if err := s.draftRepo.CreateItems(draftItems); err != nil {
+		return nil, fmt.Errorf("failed to save items: %w", err)
+	}
+
+	return draft, nil
 }
 
 // ProcessedItem - результат обработки одной строки чека
@@ -137,6 +197,11 @@ func (s *Service) SaveMapping(rawName string, productID uuid.UUID, quantity deci
 	return s.ocrRepo.SaveMatch(rawName, productID, quantity, containerID)
 }
 
+// DeleteMatch удаляет ошибочную привязку
+func (s *Service) DeleteMatch(rawName string) error {
+	return s.ocrRepo.DeleteMatch(rawName)
+}
+
 // GetKnownMatches возвращает список всех обученных связей
 func (s *Service) GetKnownMatches() ([]ocr.ProductMatch, error) {
 	return s.ocrRepo.GetAllMatches()