feat: rename ingredients→products, products→user_products; add barcode/OFF import

- Rename catalog: ingredient/* → product/* (canonical_name, barcode, nutrition per 100g)
- Rename pantry: product/* → userproduct/* (user-owned items with expiry)
- Squash migrations into single 001_initial_schema.sql (clean-db baseline)
- product_categories: add English canonical name column; fix COALESCE in queries
- Remove product_translations: product names are stored in their original language
- Add default_unit_name to product API responses via unit_translations JOIN
- Add cmd/importoff: bulk import from OpenFoodFacts JSONL dump (COPY + ON CONFLICT)
- Diary: support product_id entries alongside dish_id (CHECK num_nonnulls = 1)
- Home: getLoggedCalories joins both recipes and catalog products
- Flutter: rename models/providers/services to match backend rename
- Flutter: add barcode scan flow for diary (mobile_scanner, product_portion_sheet)
- Flutter: localise 6 new keys across 12 languages (barcode scan, portion weight)
- Routes: GET /products/search, GET /products/barcode/{barcode}, /user-products

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
dbastrikin
2026-03-21 12:45:48 +02:00
parent 6861e5e754
commit 205edbdade
72 changed files with 2588 additions and 1444 deletions

View File

@@ -0,0 +1,359 @@
package main
import (
"bufio"
"compress/gzip"
"context"
"encoding/json"
"flag"
"fmt"
"log/slog"
"os"
"os/signal"
"strings"
"syscall"
"github.com/jackc/pgx/v5"
)
// offRecord is the JSON shape of one line in the OpenFoodFacts JSONL dump.
type offRecord struct {
Code string `json:"code"`
ProductName string `json:"product_name"`
ProductNameEN string `json:"product_name_en"`
CategoriesTags []string `json:"categories_tags"`
Nutriments offNutriments `json:"nutriments"`
UniqueScansN int `json:"unique_scans_n"`
}
type offNutriments struct {
EnergyKcal100g *float64 `json:"energy-kcal_100g"`
Proteins100g *float64 `json:"proteins_100g"`
Fat100g *float64 `json:"fat_100g"`
Carbohydrates100g *float64 `json:"carbohydrates_100g"`
Fiber100g *float64 `json:"fiber_100g"`
}
type productImportRow struct {
canonicalName string
barcode string
category *string
calories float64
protein *float64
fat *float64
carbs *float64
fiber *float64
}
// categoryPrefixes maps OpenFoodFacts category tag prefixes to our product_categories slugs.
// Entries are checked in order; the first match wins.
var categoryPrefixes = []struct {
prefix string
slug string
}{
{"en:dairies", "dairy"},
{"en:dairy-products", "dairy"},
{"en:meats", "meat"},
{"en:poultry", "meat"},
{"en:fish-and-seafood", "meat"},
{"en:fruits", "produce"},
{"en:vegetables", "produce"},
{"en:plant-based-foods", "produce"},
{"en:breads", "bakery"},
{"en:pastries", "bakery"},
{"en:baked-goods", "bakery"},
{"en:frozen-foods", "frozen"},
{"en:beverages", "beverages"},
{"en:drinks", "beverages"},
{"en:sodas", "beverages"},
{"en:waters", "beverages"},
}
func resolveCategory(categoriesTags []string) *string {
for _, tag := range categoriesTags {
for _, mapping := range categoryPrefixes {
if strings.HasPrefix(tag, mapping.prefix) {
slug := mapping.slug
return &slug
}
}
}
other := "other"
return &other
}
func main() {
logger := slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelInfo}))
slog.SetDefault(logger)
if runError := run(); runError != nil {
slog.Error("import failed", "error", runError)
os.Exit(1)
}
}
func run() error {
filePathFlag := flag.String("file", "", "path to OFF JSONL or JSONL.GZ file (required)")
dsnFlag := flag.String("dsn", os.Getenv("DATABASE_URL"), "postgres connection DSN")
limitFlag := flag.Int("limit", 0, "stop after N accepted products (0 = no limit)")
batchSizeFlag := flag.Int("batch", 500, "products per upsert batch")
minScansFlag := flag.Int("min-scans", 1, "minimum unique_scans_n to include a product (0 = no filter)")
flag.Parse()
if *filePathFlag == "" {
return fmt.Errorf("-file is required")
}
if *dsnFlag == "" {
return fmt.Errorf("-dsn or DATABASE_URL is required")
}
importContext, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
defer stop()
pgxConn, connectError := pgx.Connect(importContext, *dsnFlag)
if connectError != nil {
return fmt.Errorf("connect to database: %w", connectError)
}
defer pgxConn.Close(importContext)
slog.Info("connected to database")
// Create a temporary staging table for the COPY step.
// It has no unique constraints so COPY never fails on duplicate data.
if _, createError := pgxConn.Exec(importContext, `
CREATE TEMP TABLE IF NOT EXISTS products_import (
canonical_name TEXT,
barcode TEXT,
category TEXT,
calories_per_100g DOUBLE PRECISION,
protein_per_100g DOUBLE PRECISION,
fat_per_100g DOUBLE PRECISION,
carbs_per_100g DOUBLE PRECISION,
fiber_per_100g DOUBLE PRECISION
)`); createError != nil {
return fmt.Errorf("create staging table: %w", createError)
}
dataFile, openError := os.Open(*filePathFlag)
if openError != nil {
return fmt.Errorf("open file: %w", openError)
}
defer dataFile.Close()
var lineScanner *bufio.Scanner
if strings.HasSuffix(*filePathFlag, ".gz") {
gzipReader, gzipError := gzip.NewReader(dataFile)
if gzipError != nil {
return fmt.Errorf("open gzip reader: %w", gzipError)
}
defer gzipReader.Close()
lineScanner = bufio.NewScanner(gzipReader)
} else {
lineScanner = bufio.NewScanner(dataFile)
}
// OFF lines can exceed the default 64 KB scanner buffer.
lineScanner.Buffer(make([]byte, 16*1024*1024), 16*1024*1024)
var (
productBatch []productImportRow
seenInBatch = make(map[string]bool, *batchSizeFlag)
totalAccepted int
totalInserted int64
totalSkipped int
)
flushCurrent := func() error {
inserted, flushError := flushBatch(importContext, pgxConn, productBatch)
if flushError != nil {
return flushError
}
totalInserted += inserted
productBatch = productBatch[:0]
for key := range seenInBatch {
delete(seenInBatch, key)
}
slog.Info("progress",
"accepted", totalAccepted,
"inserted", totalInserted,
"skipped", totalSkipped,
)
return nil
}
for lineScanner.Scan() {
if importContext.Err() != nil {
break
}
var record offRecord
if decodeError := json.Unmarshal(lineScanner.Bytes(), &record); decodeError != nil {
totalSkipped++
continue
}
// Resolve canonical name: prefer English, fall back to any language.
canonicalName := record.ProductNameEN
if canonicalName == "" {
canonicalName = record.ProductName
}
// Apply filter rules.
if record.Code == "" || canonicalName == "" {
totalSkipped++
continue
}
if record.Nutriments.EnergyKcal100g == nil || *record.Nutriments.EnergyKcal100g <= 0 {
totalSkipped++
continue
}
if *minScansFlag > 0 && record.UniqueScansN < *minScansFlag {
totalSkipped++
continue
}
// Deduplicate by canonical_name within the current batch.
if seenInBatch[canonicalName] {
totalSkipped++
continue
}
seenInBatch[canonicalName] = true
totalAccepted++
productBatch = append(productBatch, productImportRow{
canonicalName: canonicalName,
barcode: record.Code,
category: resolveCategory(record.CategoriesTags),
calories: *record.Nutriments.EnergyKcal100g,
protein: record.Nutriments.Proteins100g,
fat: record.Nutriments.Fat100g,
carbs: record.Nutriments.Carbohydrates100g,
fiber: record.Nutriments.Fiber100g,
})
if len(productBatch) >= *batchSizeFlag {
if flushError := flushCurrent(); flushError != nil {
return flushError
}
}
if *limitFlag > 0 && totalAccepted >= *limitFlag {
break
}
}
if scanError := lineScanner.Err(); scanError != nil {
return fmt.Errorf("read file: %w", scanError)
}
// Flush any remaining rows.
if len(productBatch) > 0 {
if flushError := flushCurrent(); flushError != nil {
return flushError
}
}
slog.Info("import complete",
"accepted", totalAccepted,
"inserted", totalInserted,
"skipped", totalSkipped,
)
return nil
}
// flushBatch upserts productBatch into the catalog.
// It uses a staging table + COPY for fast bulk loading, then INSERT … SELECT with
// ON CONFLICT to safely merge into products.
func flushBatch(
requestContext context.Context,
pgxConn *pgx.Conn,
productBatch []productImportRow,
) (int64, error) {
if len(productBatch) == 0 {
return 0, nil
}
// Truncate staging table so it is empty before each batch.
if _, truncateError := pgxConn.Exec(requestContext, `TRUNCATE products_import`); truncateError != nil {
return 0, fmt.Errorf("truncate staging table: %w", truncateError)
}
// COPY product rows into the staging table.
// No unique constraints here, so the COPY never errors on duplicate data.
_, copyError := pgxConn.CopyFrom(
requestContext,
pgx.Identifier{"products_import"},
[]string{
"canonical_name", "barcode", "category",
"calories_per_100g", "protein_per_100g", "fat_per_100g",
"carbs_per_100g", "fiber_per_100g",
},
pgx.CopyFromSlice(len(productBatch), func(rowIndex int) ([]any, error) {
row := productBatch[rowIndex]
return []any{
row.canonicalName, row.barcode, row.category,
row.calories, row.protein, row.fat,
row.carbs, row.fiber,
}, nil
}),
)
if copyError != nil {
return 0, fmt.Errorf("copy products to staging table: %w", copyError)
}
// Nullify any barcode in the staging table that is already assigned to a
// different product in the catalog. This prevents unique-constraint violations
// on the barcode column during the INSERT below.
if _, updateError := pgxConn.Exec(requestContext, `
UPDATE products_import pi
SET barcode = NULL
WHERE pi.barcode IS NOT NULL
AND EXISTS (
SELECT 1 FROM products p
WHERE p.barcode = pi.barcode
)`); updateError != nil {
return 0, fmt.Errorf("nullify conflicting barcodes: %w", updateError)
}
// Insert from staging into the catalog.
// ON CONFLICT (canonical_name): update nutritional columns but leave barcode
// unchanged — we do not want to reassign a barcode that already belongs to this
// canonical entry, nor steal one from another entry.
upsertRows, upsertError := pgxConn.Query(requestContext, `
INSERT INTO products (
canonical_name, barcode, category,
calories_per_100g, protein_per_100g, fat_per_100g,
carbs_per_100g, fiber_per_100g
)
SELECT
canonical_name, barcode, category,
calories_per_100g, protein_per_100g, fat_per_100g,
carbs_per_100g, fiber_per_100g
FROM products_import
ON CONFLICT (canonical_name) DO UPDATE SET
category = COALESCE(EXCLUDED.category, products.category),
calories_per_100g = COALESCE(EXCLUDED.calories_per_100g, products.calories_per_100g),
protein_per_100g = COALESCE(EXCLUDED.protein_per_100g, products.protein_per_100g),
fat_per_100g = COALESCE(EXCLUDED.fat_per_100g, products.fat_per_100g),
carbs_per_100g = COALESCE(EXCLUDED.carbs_per_100g, products.carbs_per_100g),
fiber_per_100g = COALESCE(EXCLUDED.fiber_per_100g, products.fiber_per_100g),
updated_at = now()
RETURNING id`)
if upsertError != nil {
return 0, fmt.Errorf("upsert products: %w", upsertError)
}
var insertedCount int64
for upsertRows.Next() {
var productID string
if scanError := upsertRows.Scan(&productID); scanError != nil {
upsertRows.Close()
return 0, fmt.Errorf("scan upserted product id: %w", scanError)
}
insertedCount++
}
upsertRows.Close()
if rowsError := upsertRows.Err(); rowsError != nil {
return 0, fmt.Errorf("iterate upsert results: %w", rowsError)
}
return insertedCount, nil
}

View File

@@ -6,7 +6,6 @@ import (
"github.com/food-ai/backend/internal/domain/diary"
"github.com/food-ai/backend/internal/domain/dish"
"github.com/food-ai/backend/internal/domain/home"
"github.com/food-ai/backend/internal/domain/ingredient"
"github.com/food-ai/backend/internal/domain/menu"
"github.com/food-ai/backend/internal/domain/product"
"github.com/food-ai/backend/internal/domain/recipe"
@@ -14,6 +13,7 @@ import (
"github.com/food-ai/backend/internal/domain/recommendation"
"github.com/food-ai/backend/internal/domain/savedrecipe"
"github.com/food-ai/backend/internal/domain/user"
"github.com/food-ai/backend/internal/domain/userproduct"
"github.com/food-ai/backend/internal/infra/config"
"github.com/jackc/pgx/v5/pgxpool"
)
@@ -37,14 +37,15 @@ func initApp(appConfig *config.Config, pool *pgxpool.Pool) (*App, error) {
openaiClient := newOpenAIClient(mainGeminiAPIKey)
mainPexelsAPIKey := newPexelsAPIKey(appConfig)
pexelsClient := newPexelsClient(mainPexelsAPIKey)
productRepository := product.NewRepository(pool)
recommendationHandler := recommendation.NewHandler(openaiClient, pexelsClient, userRepository, productRepository)
userProductRepository := userproduct.NewRepository(pool)
recommendationHandler := recommendation.NewHandler(openaiClient, pexelsClient, userRepository, userProductRepository)
dishRepository := dish.NewRepository(pool)
savedrecipeRepository := savedrecipe.NewRepository(pool, dishRepository)
savedrecipeHandler := savedrecipe.NewHandler(savedrecipeRepository)
ingredientRepository := ingredient.NewRepository(pool)
ingredientHandler := ingredient.NewHandler(ingredientRepository)
productHandler := product.NewHandler(productRepository)
productRepository := product.NewRepository(pool)
openFoodFactsClient := product.NewOpenFoodFacts()
productHandler := product.NewHandler(productRepository, openFoodFactsClient)
userProductHandler := userproduct.NewHandler(userProductRepository)
// Kafka producer
kafkaProducer, kafkaProducerError := newKafkaProducer(appConfig)
@@ -55,10 +56,10 @@ func initApp(appConfig *config.Config, pool *pgxpool.Pool) (*App, error) {
// Recognition pipeline
jobRepository := recognition.NewJobRepository(pool)
sseBroker := recognition.NewSSEBroker(pool, jobRepository)
recognitionHandler := recognition.NewHandler(openaiClient, ingredientRepository, jobRepository, kafkaProducer, sseBroker)
recognitionHandler := recognition.NewHandler(openaiClient, productRepository, jobRepository, kafkaProducer, sseBroker)
menuRepository := menu.NewRepository(pool)
menuHandler := menu.NewHandler(menuRepository, openaiClient, pexelsClient, userRepository, productRepository, dishRepository)
menuHandler := menu.NewHandler(menuRepository, openaiClient, pexelsClient, userRepository, userProductRepository, dishRepository)
diaryRepository := diary.NewRepository(pool)
diaryHandler := diary.NewHandler(diaryRepository, dishRepository, dishRepository)
homeHandler := home.NewHandler(pool)
@@ -77,8 +78,8 @@ func initApp(appConfig *config.Config, pool *pgxpool.Pool) (*App, error) {
userHandler,
recommendationHandler,
savedrecipeHandler,
ingredientHandler,
productHandler,
userProductHandler,
recognitionHandler,
menuHandler,
diaryHandler,

View File

@@ -11,7 +11,6 @@ import (
"github.com/food-ai/backend/internal/adapters/kafka"
"github.com/food-ai/backend/internal/adapters/openai"
"github.com/food-ai/backend/internal/domain/home"
"github.com/food-ai/backend/internal/domain/ingredient"
"github.com/food-ai/backend/internal/domain/menu"
"github.com/food-ai/backend/internal/infra/middleware"
"github.com/food-ai/backend/internal/adapters/pexels"
@@ -25,6 +24,7 @@ import (
"github.com/food-ai/backend/internal/domain/tag"
"github.com/food-ai/backend/internal/domain/units"
"github.com/food-ai/backend/internal/domain/user"
"github.com/food-ai/backend/internal/domain/userproduct"
"github.com/jackc/pgx/v5/pgxpool"
)
@@ -125,8 +125,8 @@ func newRouter(
userHandler *user.Handler,
recommendationHandler *recommendation.Handler,
savedRecipeHandler *savedrecipe.Handler,
ingredientHandler *ingredient.Handler,
productHandler *product.Handler,
userProductHandler *userproduct.Handler,
recognitionHandler *recognition.Handler,
menuHandler *menu.Handler,
diaryHandler *diary.Handler,
@@ -145,8 +145,8 @@ func newRouter(
userHandler,
recommendationHandler,
savedRecipeHandler,
ingredientHandler,
productHandler,
userProductHandler,
recognitionHandler,
menuHandler,
diaryHandler,
@@ -204,12 +204,12 @@ func newKafkaProducer(appConfig *config.Config) (*kafka.Producer, error) {
var _ middleware.AccessTokenValidator = (*jwtAdapter)(nil)
var _ menu.PhotoSearcher = (*pexels.Client)(nil)
var _ menu.UserLoader = (*user.Repository)(nil)
var _ menu.ProductLister = (*product.Repository)(nil)
var _ menu.ProductLister = (*userproduct.Repository)(nil)
var _ menu.RecipeSaver = (*dish.Repository)(nil)
var _ recommendation.PhotoSearcher = (*pexels.Client)(nil)
var _ recommendation.UserLoader = (*user.Repository)(nil)
var _ recommendation.ProductLister = (*product.Repository)(nil)
var _ recognition.IngredientRepository = (*ingredient.Repository)(nil)
var _ recommendation.ProductLister = (*userproduct.Repository)(nil)
var _ recognition.ProductRepository = (*product.Repository)(nil)
var _ recognition.KafkaPublisher = (*kafka.Producer)(nil)
var _ recognition.JobRepository = (*recognition.PostgresJobRepository)(nil)
var _ user.UserRepository = (*user.Repository)(nil)

View File

@@ -7,14 +7,15 @@ type Entry struct {
ID string `json:"id"`
Date string `json:"date"` // YYYY-MM-DD
MealType string `json:"meal_type"`
Name string `json:"name"` // from dishes JOIN
Name string `json:"name"` // from dishes or products JOIN
Portions float64 `json:"portions"`
Calories *float64 `json:"calories,omitempty"` // recipe.calories_per_serving * portions
Calories *float64 `json:"calories,omitempty"` // recipe.calories_per_serving * portions, or product * portion_g
ProteinG *float64 `json:"protein_g,omitempty"`
FatG *float64 `json:"fat_g,omitempty"`
CarbsG *float64 `json:"carbs_g,omitempty"`
Source string `json:"source"`
DishID string `json:"dish_id"`
DishID *string `json:"dish_id,omitempty"`
ProductID *string `json:"product_id,omitempty"`
RecipeID *string `json:"recipe_id,omitempty"`
PortionG *float64 `json:"portion_g,omitempty"`
JobID *string `json:"job_id,omitempty"`
@@ -23,13 +24,14 @@ type Entry struct {
// CreateRequest is the body for POST /diary.
type CreateRequest struct {
Date string `json:"date"`
MealType string `json:"meal_type"`
Name string `json:"name"` // input-only; used if DishID is nil
Portions float64 `json:"portions"`
Source string `json:"source"`
DishID *string `json:"dish_id"`
RecipeID *string `json:"recipe_id"`
PortionG *float64 `json:"portion_g"`
JobID *string `json:"job_id"`
Date string `json:"date"`
MealType string `json:"meal_type"`
Name string `json:"name"` // input-only; used if DishID is nil and ProductID is nil
Portions float64 `json:"portions"`
Source string `json:"source"`
DishID *string `json:"dish_id"`
ProductID *string `json:"product_id"`
RecipeID *string `json:"recipe_id"`
PortionG *float64 `json:"portion_g"`
JobID *string `json:"job_id"`
}

View File

@@ -82,29 +82,32 @@ func (h *Handler) Create(w http.ResponseWriter, r *http.Request) {
writeError(w, http.StatusBadRequest, "date and meal_type are required")
return
}
if req.DishID == nil && req.Name == "" {
writeError(w, http.StatusBadRequest, "dish_id or name is required")
if req.DishID == nil && req.ProductID == nil && req.Name == "" {
writeError(w, http.StatusBadRequest, "dish_id, product_id, or name is required")
return
}
if req.DishID == nil {
dishID, _, resolveError := h.dishRepo.FindOrCreate(r.Context(), req.Name)
if resolveError != nil {
slog.Error("resolve dish for diary entry", "name", req.Name, "err", resolveError)
writeError(w, http.StatusInternalServerError, "failed to resolve dish")
return
// Product-based entry: skip dish/recipe resolution entirely.
if req.ProductID == nil {
if req.DishID == nil {
dishID, _, resolveError := h.dishRepo.FindOrCreate(r.Context(), req.Name)
if resolveError != nil {
slog.Error("resolve dish for diary entry", "name", req.Name, "err", resolveError)
writeError(w, http.StatusInternalServerError, "failed to resolve dish")
return
}
req.DishID = &dishID
}
req.DishID = &dishID
}
if req.RecipeID == nil {
recipeID, _, recipeError := h.recipeRepo.FindOrCreateRecipe(r.Context(), *req.DishID, 0, 0, 0, 0)
if recipeError != nil {
slog.Error("find or create recipe for diary entry", "dish_id", *req.DishID, "err", recipeError)
writeError(w, http.StatusInternalServerError, "failed to resolve recipe")
return
if req.RecipeID == nil {
recipeID, _, recipeError := h.recipeRepo.FindOrCreateRecipe(r.Context(), *req.DishID, 0, 0, 0, 0)
if recipeError != nil {
slog.Error("find or create recipe for diary entry", "dish_id", *req.DishID, "err", recipeError)
writeError(w, http.StatusInternalServerError, "failed to resolve recipe")
return
}
req.RecipeID = &recipeID
}
req.RecipeID = &recipeID
}
entry, createError := h.repo.Create(r.Context(), userID, req)

View File

@@ -24,22 +24,35 @@ func NewRepository(pool *pgxpool.Pool) *Repository {
}
// ListByDate returns all diary entries for a user on a given date (YYYY-MM-DD).
// Dish name and macros are computed via JOIN with dishes and recipes.
// Supports both dish-based and catalog product-based entries.
func (r *Repository) ListByDate(ctx context.Context, userID, date string) ([]*Entry, error) {
lang := locale.FromContext(ctx)
rows, err := r.pool.Query(ctx, `
SELECT
md.id, md.date::text, md.meal_type, md.portions,
md.source, md.dish_id::text, md.recipe_id::text, md.portion_g, md.job_id::text, md.created_at,
COALESCE(dt.name, d.name) AS dish_name,
r.calories_per_serving * md.portions,
r.protein_per_serving * md.portions,
r.fat_per_serving * md.portions,
r.carbs_per_serving * md.portions
md.source, md.dish_id::text, md.product_id::text, md.recipe_id::text, md.portion_g, md.job_id::text, md.created_at,
COALESCE(dt.name, d.name, p.canonical_name) AS entry_name,
COALESCE(
r.calories_per_serving * md.portions,
p.calories_per_100g * md.portion_g / 100
),
COALESCE(
r.protein_per_serving * md.portions,
p.protein_per_100g * md.portion_g / 100
),
COALESCE(
r.fat_per_serving * md.portions,
p.fat_per_100g * md.portion_g / 100
),
COALESCE(
r.carbs_per_serving * md.portions,
p.carbs_per_100g * md.portion_g / 100
)
FROM meal_diary md
JOIN dishes d ON d.id = md.dish_id
LEFT JOIN dishes d ON d.id = md.dish_id
LEFT JOIN dish_translations dt ON dt.dish_id = d.id AND dt.lang = $3
LEFT JOIN recipes r ON r.id = md.recipe_id
LEFT JOIN products p ON p.id = md.product_id
WHERE md.user_id = $1 AND md.date = $2::date
ORDER BY md.created_at ASC`, userID, date, lang)
if err != nil {
@@ -72,10 +85,10 @@ func (r *Repository) Create(ctx context.Context, userID string, req CreateReques
var entryID string
insertError := r.pool.QueryRow(ctx, `
INSERT INTO meal_diary (user_id, date, meal_type, portions, source, dish_id, recipe_id, portion_g, job_id)
VALUES ($1, $2::date, $3, $4, $5, $6, $7, $8, $9)
INSERT INTO meal_diary (user_id, date, meal_type, portions, source, dish_id, product_id, recipe_id, portion_g, job_id)
VALUES ($1, $2::date, $3, $4, $5, $6, $7, $8, $9, $10)
RETURNING id`,
userID, req.Date, req.MealType, portions, source, req.DishID, req.RecipeID, req.PortionG, req.JobID,
userID, req.Date, req.MealType, portions, source, req.DishID, req.ProductID, req.RecipeID, req.PortionG, req.JobID,
).Scan(&entryID)
if insertError != nil {
return nil, fmt.Errorf("insert diary entry: %w", insertError)
@@ -84,16 +97,29 @@ func (r *Repository) Create(ctx context.Context, userID string, req CreateReques
row := r.pool.QueryRow(ctx, `
SELECT
md.id, md.date::text, md.meal_type, md.portions,
md.source, md.dish_id::text, md.recipe_id::text, md.portion_g, md.job_id::text, md.created_at,
COALESCE(dt.name, d.name) AS dish_name,
r.calories_per_serving * md.portions,
r.protein_per_serving * md.portions,
r.fat_per_serving * md.portions,
r.carbs_per_serving * md.portions
md.source, md.dish_id::text, md.product_id::text, md.recipe_id::text, md.portion_g, md.job_id::text, md.created_at,
COALESCE(dt.name, d.name, p.canonical_name) AS entry_name,
COALESCE(
r.calories_per_serving * md.portions,
p.calories_per_100g * md.portion_g / 100
),
COALESCE(
r.protein_per_serving * md.portions,
p.protein_per_100g * md.portion_g / 100
),
COALESCE(
r.fat_per_serving * md.portions,
p.fat_per_100g * md.portion_g / 100
),
COALESCE(
r.carbs_per_serving * md.portions,
p.carbs_per_100g * md.portion_g / 100
)
FROM meal_diary md
JOIN dishes d ON d.id = md.dish_id
LEFT JOIN dishes d ON d.id = md.dish_id
LEFT JOIN dish_translations dt ON dt.dish_id = d.id AND dt.lang = $2
LEFT JOIN recipes r ON r.id = md.recipe_id
LEFT JOIN products p ON p.id = md.product_id
WHERE md.id = $1`, entryID, lang)
return scanEntry(row)
}
@@ -121,7 +147,7 @@ func scanEntry(s scannable) (*Entry, error) {
var entry Entry
scanError := s.Scan(
&entry.ID, &entry.Date, &entry.MealType, &entry.Portions,
&entry.Source, &entry.DishID, &entry.RecipeID, &entry.PortionG, &entry.JobID, &entry.CreatedAt,
&entry.Source, &entry.DishID, &entry.ProductID, &entry.RecipeID, &entry.PortionG, &entry.JobID, &entry.CreatedAt,
&entry.Name,
&entry.Calories, &entry.ProteinG, &entry.FatG, &entry.CarbsG,
)

View File

@@ -76,12 +76,21 @@ func (h *Handler) getDailyGoal(ctx context.Context, userID string) int {
}
// getLoggedCalories returns total calories logged in meal_diary for today.
// Supports both recipe-based entries (via recipes JOIN) and catalog product entries (via products JOIN).
func (h *Handler) getLoggedCalories(ctx context.Context, userID, date string) float64 {
var total float64
_ = h.pool.QueryRow(ctx,
`SELECT COALESCE(SUM(calories * portions), 0)
FROM meal_diary
WHERE user_id = $1 AND date::text = $2`,
`SELECT COALESCE(SUM(
COALESCE(
r.calories_per_serving * md.portions,
p.calories_per_100g * md.portion_g / 100,
0
)
), 0)
FROM meal_diary md
LEFT JOIN recipes r ON r.id = md.recipe_id
LEFT JOIN products p ON p.id = md.product_id
WHERE md.user_id = $1 AND md.date::text = $2`,
userID, date,
).Scan(&total)
return total
@@ -153,7 +162,7 @@ func (h *Handler) getExpiringSoon(ctx context.Context, userID string) []Expiring
WITH p AS (
SELECT name, quantity, unit,
(added_at + storage_days * INTERVAL '1 day') AS expires_at
FROM products
FROM user_products
WHERE user_id = $1
)
SELECT name, quantity, unit,

View File

@@ -1,29 +0,0 @@
package ingredient
import (
"encoding/json"
"time"
)
// IngredientMapping is the canonical ingredient record used to link
// user products and recipe ingredients.
// CanonicalName holds the content for the language resolved at query time
// (English by default, or from ingredient_translations when available).
type IngredientMapping struct {
ID string `json:"id"`
CanonicalName string `json:"canonical_name"`
Aliases json.RawMessage `json:"aliases"` // []string, populated by read queries
Category *string `json:"category"`
CategoryName *string `json:"category_name"` // localized category display name
DefaultUnit *string `json:"default_unit"`
CaloriesPer100g *float64 `json:"calories_per_100g"`
ProteinPer100g *float64 `json:"protein_per_100g"`
FatPer100g *float64 `json:"fat_per_100g"`
CarbsPer100g *float64 `json:"carbs_per_100g"`
FiberPer100g *float64 `json:"fiber_per_100g"`
StorageDays *int `json:"storage_days"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}

View File

@@ -1,57 +0,0 @@
package ingredient
import (
"context"
"encoding/json"
"log/slog"
"net/http"
"strconv"
)
// IngredientSearcher is the data layer interface used by Handler.
type IngredientSearcher interface {
Search(ctx context.Context, query string, limit int) ([]*IngredientMapping, error)
}
// Handler handles ingredient HTTP requests.
type Handler struct {
repo IngredientSearcher
}
// NewHandler creates a new Handler.
func NewHandler(repo IngredientSearcher) *Handler {
return &Handler{repo: repo}
}
// Search handles GET /ingredients/search?q=&limit=10.
func (h *Handler) Search(w http.ResponseWriter, r *http.Request) {
q := r.URL.Query().Get("q")
if q == "" {
w.Header().Set("Content-Type", "application/json")
_, _ = w.Write([]byte("[]"))
return
}
limit := 10
if s := r.URL.Query().Get("limit"); s != "" {
if n, err := strconv.Atoi(s); err == nil && n > 0 && n <= 50 {
limit = n
}
}
mappings, err := h.repo.Search(r.Context(), q, limit)
if err != nil {
slog.Error("search ingredients", "q", q, "err", err)
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(http.StatusInternalServerError)
_, _ = w.Write([]byte(`{"error":"search failed"}`))
return
}
if mappings == nil {
mappings = []*IngredientMapping{}
}
w.Header().Set("Content-Type", "application/json")
_ = json.NewEncoder(w).Encode(mappings)
}

View File

@@ -1,301 +0,0 @@
package ingredient
import (
"context"
"encoding/json"
"errors"
"fmt"
"github.com/food-ai/backend/internal/infra/locale"
"github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgxpool"
)
// Repository handles persistence for ingredients and their translations.
type Repository struct {
pool *pgxpool.Pool
}
// NewRepository creates a new Repository.
func NewRepository(pool *pgxpool.Pool) *Repository {
return &Repository{pool: pool}
}
// Upsert inserts or updates an ingredient (English canonical content).
// Conflict is resolved on canonical_name.
func (r *Repository) Upsert(ctx context.Context, m *IngredientMapping) (*IngredientMapping, error) {
query := `
INSERT INTO ingredients (
canonical_name,
category, default_unit,
calories_per_100g, protein_per_100g, fat_per_100g, carbs_per_100g, fiber_per_100g,
storage_days
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)
ON CONFLICT (canonical_name) DO UPDATE SET
category = EXCLUDED.category,
default_unit = EXCLUDED.default_unit,
calories_per_100g = EXCLUDED.calories_per_100g,
protein_per_100g = EXCLUDED.protein_per_100g,
fat_per_100g = EXCLUDED.fat_per_100g,
carbs_per_100g = EXCLUDED.carbs_per_100g,
fiber_per_100g = EXCLUDED.fiber_per_100g,
storage_days = EXCLUDED.storage_days,
updated_at = now()
RETURNING id, canonical_name, category, default_unit,
calories_per_100g, protein_per_100g, fat_per_100g, carbs_per_100g, fiber_per_100g,
storage_days, created_at, updated_at`
row := r.pool.QueryRow(ctx, query,
m.CanonicalName,
m.Category, m.DefaultUnit,
m.CaloriesPer100g, m.ProteinPer100g, m.FatPer100g, m.CarbsPer100g, m.FiberPer100g,
m.StorageDays,
)
return scanMappingWrite(row)
}
// GetByID returns an ingredient by UUID.
// CanonicalName and aliases are resolved for the language stored in ctx.
// Returns nil, nil if not found.
func (r *Repository) GetByID(ctx context.Context, id string) (*IngredientMapping, error) {
lang := locale.FromContext(ctx)
query := `
SELECT im.id,
COALESCE(it.name, im.canonical_name) AS canonical_name,
im.category,
COALESCE(ict.name, im.category) AS category_name,
im.default_unit,
im.calories_per_100g, im.protein_per_100g, im.fat_per_100g, im.carbs_per_100g, im.fiber_per_100g,
im.storage_days, im.created_at, im.updated_at,
COALESCE(al.aliases, '[]'::json) AS aliases
FROM ingredients im
LEFT JOIN ingredient_translations it
ON it.ingredient_id = im.id AND it.lang = $2
LEFT JOIN ingredient_category_translations ict
ON ict.category_slug = im.category AND ict.lang = $2
LEFT JOIN LATERAL (
SELECT json_agg(ia.alias ORDER BY ia.alias) AS aliases
FROM ingredient_aliases ia
WHERE ia.ingredient_id = im.id AND ia.lang = $2
) al ON true
WHERE im.id = $1`
row := r.pool.QueryRow(ctx, query, id, lang)
m, err := scanMappingRead(row)
if errors.Is(err, pgx.ErrNoRows) {
return nil, nil
}
return m, err
}
// FuzzyMatch finds the single best matching ingredient for a given name.
// Searches both English and translated names for the language in ctx.
// Returns nil, nil when no match is found.
func (r *Repository) FuzzyMatch(ctx context.Context, name string) (*IngredientMapping, error) {
results, err := r.Search(ctx, name, 1)
if err != nil {
return nil, err
}
if len(results) == 0 {
return nil, nil
}
return results[0], nil
}
// Search finds ingredients matching the query string.
// Searches aliases table and translated names for the language in ctx.
func (r *Repository) Search(ctx context.Context, query string, limit int) ([]*IngredientMapping, error) {
if limit <= 0 {
limit = 10
}
lang := locale.FromContext(ctx)
q := `
SELECT im.id,
COALESCE(it.name, im.canonical_name) AS canonical_name,
im.category,
COALESCE(ict.name, im.category) AS category_name,
im.default_unit,
im.calories_per_100g, im.protein_per_100g, im.fat_per_100g, im.carbs_per_100g, im.fiber_per_100g,
im.storage_days, im.created_at, im.updated_at,
COALESCE(al.aliases, '[]'::json) AS aliases
FROM ingredients im
LEFT JOIN ingredient_translations it
ON it.ingredient_id = im.id AND it.lang = $3
LEFT JOIN ingredient_category_translations ict
ON ict.category_slug = im.category AND ict.lang = $3
LEFT JOIN LATERAL (
SELECT json_agg(ia.alias ORDER BY ia.alias) AS aliases
FROM ingredient_aliases ia
WHERE ia.ingredient_id = im.id AND ia.lang = $3
) al ON true
WHERE EXISTS (
SELECT 1 FROM ingredient_aliases ia
WHERE ia.ingredient_id = im.id
AND (ia.lang = $3 OR ia.lang = 'en')
AND ia.alias ILIKE '%' || $1 || '%'
)
OR im.canonical_name ILIKE '%' || $1 || '%'
OR it.name ILIKE '%' || $1 || '%'
OR similarity(COALESCE(it.name, im.canonical_name), $1) > 0.3
ORDER BY similarity(COALESCE(it.name, im.canonical_name), $1) DESC
LIMIT $2`
rows, err := r.pool.Query(ctx, q, query, limit, lang)
if err != nil {
return nil, fmt.Errorf("search ingredients: %w", err)
}
defer rows.Close()
return collectMappingsRead(rows)
}
// Count returns the total number of ingredients.
func (r *Repository) Count(ctx context.Context) (int, error) {
var n int
if err := r.pool.QueryRow(ctx, `SELECT count(*) FROM ingredients`).Scan(&n); err != nil {
return 0, fmt.Errorf("count ingredients: %w", err)
}
return n, nil
}
// ListMissingTranslation returns ingredients that have no translation for the
// given language, ordered by id.
func (r *Repository) ListMissingTranslation(ctx context.Context, lang string, limit, offset int) ([]*IngredientMapping, error) {
query := `
SELECT im.id, im.canonical_name,
im.category, im.default_unit,
im.calories_per_100g, im.protein_per_100g, im.fat_per_100g, im.carbs_per_100g, im.fiber_per_100g,
im.storage_days, im.created_at, im.updated_at
FROM ingredients im
WHERE NOT EXISTS (
SELECT 1 FROM ingredient_translations it
WHERE it.ingredient_id = im.id AND it.lang = $3
)
ORDER BY im.id
LIMIT $1 OFFSET $2`
rows, err := r.pool.Query(ctx, query, limit, offset, lang)
if err != nil {
return nil, fmt.Errorf("list missing translation (%s): %w", lang, err)
}
defer rows.Close()
return collectMappingsWrite(rows)
}
// UpsertTranslation inserts or replaces a name translation for an ingredient.
func (r *Repository) UpsertTranslation(ctx context.Context, id, lang, name string) error {
query := `
INSERT INTO ingredient_translations (ingredient_id, lang, name)
VALUES ($1, $2, $3)
ON CONFLICT (ingredient_id, lang) DO UPDATE SET name = EXCLUDED.name`
if _, err := r.pool.Exec(ctx, query, id, lang, name); err != nil {
return fmt.Errorf("upsert ingredient translation %s/%s: %w", id, lang, err)
}
return nil
}
// UpsertAliases inserts aliases for a given ingredient and language.
// Each alias is inserted with ON CONFLICT DO NOTHING, so duplicates are skipped.
func (r *Repository) UpsertAliases(ctx context.Context, id, lang string, aliases []string) error {
if len(aliases) == 0 {
return nil
}
batch := &pgx.Batch{}
for _, alias := range aliases {
batch.Queue(
`INSERT INTO ingredient_aliases (ingredient_id, lang, alias) VALUES ($1, $2, $3) ON CONFLICT DO NOTHING`,
id, lang, alias,
)
}
results := r.pool.SendBatch(ctx, batch)
defer results.Close()
for range aliases {
if _, err := results.Exec(); err != nil {
return fmt.Errorf("upsert ingredient alias %s/%s: %w", id, lang, err)
}
}
return nil
}
// UpsertCategoryTranslation inserts or replaces a localized category name.
func (r *Repository) UpsertCategoryTranslation(ctx context.Context, slug, lang, name string) error {
query := `
INSERT INTO ingredient_category_translations (category_slug, lang, name)
VALUES ($1, $2, $3)
ON CONFLICT (category_slug, lang) DO UPDATE SET name = EXCLUDED.name`
if _, err := r.pool.Exec(ctx, query, slug, lang, name); err != nil {
return fmt.Errorf("upsert category translation %s/%s: %w", slug, lang, err)
}
return nil
}
// --- scan helpers ---
// scanMappingWrite scans rows from Upsert / ListMissingTranslation queries
// (no aliases lateral join, no category_name).
func scanMappingWrite(row pgx.Row) (*IngredientMapping, error) {
var m IngredientMapping
err := row.Scan(
&m.ID, &m.CanonicalName, &m.Category, &m.DefaultUnit,
&m.CaloriesPer100g, &m.ProteinPer100g, &m.FatPer100g, &m.CarbsPer100g, &m.FiberPer100g,
&m.StorageDays, &m.CreatedAt, &m.UpdatedAt,
)
if err != nil {
return nil, err
}
m.Aliases = json.RawMessage("[]")
return &m, nil
}
// scanMappingRead scans rows from GetByID / Search queries
// (includes category_name and aliases lateral join).
func scanMappingRead(row pgx.Row) (*IngredientMapping, error) {
var m IngredientMapping
var aliases []byte
err := row.Scan(
&m.ID, &m.CanonicalName, &m.Category, &m.CategoryName, &m.DefaultUnit,
&m.CaloriesPer100g, &m.ProteinPer100g, &m.FatPer100g, &m.CarbsPer100g, &m.FiberPer100g,
&m.StorageDays, &m.CreatedAt, &m.UpdatedAt, &aliases,
)
if err != nil {
return nil, err
}
m.Aliases = json.RawMessage(aliases)
return &m, nil
}
func collectMappingsWrite(rows pgx.Rows) ([]*IngredientMapping, error) {
var result []*IngredientMapping
for rows.Next() {
var m IngredientMapping
if err := rows.Scan(
&m.ID, &m.CanonicalName, &m.Category, &m.DefaultUnit,
&m.CaloriesPer100g, &m.ProteinPer100g, &m.FatPer100g, &m.CarbsPer100g, &m.FiberPer100g,
&m.StorageDays, &m.CreatedAt, &m.UpdatedAt,
); err != nil {
return nil, fmt.Errorf("scan mapping: %w", err)
}
m.Aliases = json.RawMessage("[]")
result = append(result, &m)
}
return result, rows.Err()
}
func collectMappingsRead(rows pgx.Rows) ([]*IngredientMapping, error) {
var result []*IngredientMapping
for rows.Next() {
var m IngredientMapping
var aliases []byte
if err := rows.Scan(
&m.ID, &m.CanonicalName, &m.Category, &m.CategoryName, &m.DefaultUnit,
&m.CaloriesPer100g, &m.ProteinPer100g, &m.FatPer100g, &m.CarbsPer100g, &m.FiberPer100g,
&m.StorageDays, &m.CreatedAt, &m.UpdatedAt, &aliases,
); err != nil {
return nil, fmt.Errorf("scan mapping: %w", err)
}
m.Aliases = json.RawMessage(aliases)
result = append(result, &m)
}
return result, rows.Err()
}

View File

@@ -268,15 +268,15 @@ func (r *Repository) GetPlanIDByWeek(ctx context.Context, userID, weekStart stri
return id, nil
}
// GetIngredientsByPlan returns all ingredients from all recipes in the plan.
// GetIngredientsByPlan returns all products from all recipes in the plan.
func (r *Repository) GetIngredientsByPlan(ctx context.Context, planID string) ([]ingredientRow, error) {
rows, err := r.pool.Query(ctx, `
SELECT ri.name, ri.amount, ri.unit_code, mi.meal_type
SELECT rp.name, rp.amount, rp.unit_code, mi.meal_type
FROM menu_items mi
JOIN recipes rec ON rec.id = mi.recipe_id
JOIN recipe_ingredients ri ON ri.recipe_id = rec.id
JOIN recipe_products rp ON rp.recipe_id = rec.id
WHERE mi.menu_plan_id = $1
ORDER BY ri.sort_order`, planID)
ORDER BY rp.sort_order`, planID)
if err != nil {
return nil, fmt.Errorf("get ingredients by plan: %w", err)
}

View File

@@ -1,41 +1,29 @@
package product
import "time"
import (
"encoding/json"
"time"
)
// Product is a user's food item in their pantry.
// Product is a catalog entry representing a food ingredient or packaged product.
// CanonicalName holds the English name; localized names are resolved at query time.
type Product struct {
ID string `json:"id"`
UserID string `json:"user_id"`
PrimaryIngredientID *string `json:"primary_ingredient_id"`
Name string `json:"name"`
Quantity float64 `json:"quantity"`
Unit string `json:"unit"`
Category *string `json:"category"`
StorageDays int `json:"storage_days"`
AddedAt time.Time `json:"added_at"`
ExpiresAt time.Time `json:"expires_at"`
DaysLeft int `json:"days_left"`
ExpiringSoon bool `json:"expiring_soon"`
}
ID string `json:"id"`
CanonicalName string `json:"canonical_name"`
Aliases json.RawMessage `json:"aliases"` // []string, populated by read queries
Category *string `json:"category"`
CategoryName *string `json:"category_name"` // localized category display name
DefaultUnit *string `json:"default_unit"`
DefaultUnitName *string `json:"default_unit_name,omitempty"`
Barcode *string `json:"barcode,omitempty"`
// CreateRequest is the body for POST /products.
type CreateRequest struct {
PrimaryIngredientID *string `json:"primary_ingredient_id"`
// Accept both "primary_ingredient_id" (new) and "mapping_id" (legacy client) fields.
MappingID *string `json:"mapping_id"`
Name string `json:"name"`
Quantity float64 `json:"quantity"`
Unit string `json:"unit"`
Category *string `json:"category"`
StorageDays int `json:"storage_days"`
}
CaloriesPer100g *float64 `json:"calories_per_100g"`
ProteinPer100g *float64 `json:"protein_per_100g"`
FatPer100g *float64 `json:"fat_per_100g"`
CarbsPer100g *float64 `json:"carbs_per_100g"`
FiberPer100g *float64 `json:"fiber_per_100g"`
// UpdateRequest is the body for PUT /products/{id}.
// All fields are optional (nil = keep existing value).
type UpdateRequest struct {
Name *string `json:"name"`
Quantity *float64 `json:"quantity"`
Unit *string `json:"unit"`
Category *string `json:"category"`
StorageDays *int `json:"storage_days"`
StorageDays *int `json:"storage_days"`
CreatedAt time.Time `json:"created_at"`
UpdatedAt time.Time `json:"updated_at"`
}

View File

@@ -3,145 +3,121 @@ package product
import (
"context"
"encoding/json"
"errors"
"log/slog"
"net/http"
"strconv"
"github.com/food-ai/backend/internal/infra/middleware"
"github.com/go-chi/chi/v5"
)
// ProductRepository is the data layer interface used by Handler.
type ProductRepository interface {
List(ctx context.Context, userID string) ([]*Product, error)
Create(ctx context.Context, userID string, req CreateRequest) (*Product, error)
BatchCreate(ctx context.Context, userID string, items []CreateRequest) ([]*Product, error)
Update(ctx context.Context, id, userID string, req UpdateRequest) (*Product, error)
Delete(ctx context.Context, id, userID string) error
// ProductSearcher is the data layer interface used by Handler for search.
type ProductSearcher interface {
Search(ctx context.Context, query string, limit int) ([]*Product, error)
GetByBarcode(ctx context.Context, barcode string) (*Product, error)
UpsertByBarcode(ctx context.Context, catalogProduct *Product) (*Product, error)
}
// Handler handles /products HTTP requests.
// OpenFoodFactsClient fetches product data from Open Food Facts.
type OpenFoodFactsClient interface {
Fetch(requestContext context.Context, barcode string) (*Product, error)
}
// Handler handles catalog product HTTP requests.
type Handler struct {
repo ProductRepository
repo ProductSearcher
openFoodFacts OpenFoodFactsClient
}
// NewHandler creates a new Handler.
func NewHandler(repo ProductRepository) *Handler {
return &Handler{repo: repo}
func NewHandler(repo ProductSearcher, openFoodFacts OpenFoodFactsClient) *Handler {
return &Handler{repo: repo, openFoodFacts: openFoodFacts}
}
// List handles GET /products.
func (h *Handler) List(w http.ResponseWriter, r *http.Request) {
userID := middleware.UserIDFromCtx(r.Context())
products, err := h.repo.List(r.Context(), userID)
if err != nil {
slog.Error("list products", "user_id", userID, "err", err)
writeErrorJSON(w, http.StatusInternalServerError, "failed to list products")
// Search handles GET /products/search?q=&limit=10.
func (handler *Handler) Search(responseWriter http.ResponseWriter, request *http.Request) {
query := request.URL.Query().Get("q")
if query == "" {
responseWriter.Header().Set("Content-Type", "application/json")
_, _ = responseWriter.Write([]byte("[]"))
return
}
limit := 10
if limitStr := request.URL.Query().Get("limit"); limitStr != "" {
if parsedLimit, parseError := strconv.Atoi(limitStr); parseError == nil && parsedLimit > 0 && parsedLimit <= 50 {
limit = parsedLimit
}
}
products, searchError := handler.repo.Search(request.Context(), query, limit)
if searchError != nil {
slog.Error("search catalog products", "q", query, "err", searchError)
responseWriter.Header().Set("Content-Type", "application/json")
responseWriter.WriteHeader(http.StatusInternalServerError)
_, _ = responseWriter.Write([]byte(`{"error":"search failed"}`))
return
}
if products == nil {
products = []*Product{}
}
writeJSON(w, http.StatusOK, products)
responseWriter.Header().Set("Content-Type", "application/json")
_ = json.NewEncoder(responseWriter).Encode(products)
}
// Create handles POST /products.
func (h *Handler) Create(w http.ResponseWriter, r *http.Request) {
userID := middleware.UserIDFromCtx(r.Context())
var req CreateRequest
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
writeErrorJSON(w, http.StatusBadRequest, "invalid request body")
return
}
if req.Name == "" {
writeErrorJSON(w, http.StatusBadRequest, "name is required")
// GetByBarcode handles GET /products/barcode/{barcode}.
// Checks the database first; on miss, fetches from Open Food Facts and caches the result.
func (handler *Handler) GetByBarcode(responseWriter http.ResponseWriter, request *http.Request) {
barcode := chi.URLParam(request, "barcode")
if barcode == "" {
writeErrorJSON(responseWriter, http.StatusBadRequest, "barcode is required")
return
}
p, err := h.repo.Create(r.Context(), userID, req)
if err != nil {
slog.Error("create product", "user_id", userID, "err", err)
writeErrorJSON(w, http.StatusInternalServerError, "failed to create product")
// Check the local catalog first.
catalogProduct, lookupError := handler.repo.GetByBarcode(request.Context(), barcode)
if lookupError != nil {
slog.Error("lookup product by barcode", "barcode", barcode, "err", lookupError)
writeErrorJSON(responseWriter, http.StatusInternalServerError, "lookup failed")
return
}
writeJSON(w, http.StatusCreated, p)
}
// BatchCreate handles POST /products/batch.
func (h *Handler) BatchCreate(w http.ResponseWriter, r *http.Request) {
userID := middleware.UserIDFromCtx(r.Context())
var items []CreateRequest
if err := json.NewDecoder(r.Body).Decode(&items); err != nil {
writeErrorJSON(w, http.StatusBadRequest, "invalid request body")
return
}
if len(items) == 0 {
writeJSON(w, http.StatusCreated, []*Product{})
if catalogProduct != nil {
writeJSON(responseWriter, http.StatusOK, catalogProduct)
return
}
products, err := h.repo.BatchCreate(r.Context(), userID, items)
if err != nil {
slog.Error("batch create products", "user_id", userID, "err", err)
writeErrorJSON(w, http.StatusInternalServerError, "failed to create products")
return
}
writeJSON(w, http.StatusCreated, products)
}
// Update handles PUT /products/{id}.
func (h *Handler) Update(w http.ResponseWriter, r *http.Request) {
userID := middleware.UserIDFromCtx(r.Context())
id := chi.URLParam(r, "id")
var req UpdateRequest
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
writeErrorJSON(w, http.StatusBadRequest, "invalid request body")
// Not in catalog — fetch from Open Food Facts.
fetchedProduct, fetchError := handler.openFoodFacts.Fetch(request.Context(), barcode)
if fetchError != nil {
slog.Warn("open food facts fetch failed", "barcode", barcode, "err", fetchError)
writeErrorJSON(responseWriter, http.StatusNotFound, "product not found")
return
}
p, err := h.repo.Update(r.Context(), id, userID, req)
if errors.Is(err, ErrNotFound) {
writeErrorJSON(w, http.StatusNotFound, "product not found")
// Persist the fetched product so subsequent lookups are served from the DB.
savedProduct, upsertError := handler.repo.UpsertByBarcode(request.Context(), fetchedProduct)
if upsertError != nil {
slog.Warn("upsert product from open food facts", "barcode", barcode, "err", upsertError)
// Return the fetched data even if we could not cache it.
writeJSON(responseWriter, http.StatusOK, fetchedProduct)
return
}
if err != nil {
slog.Error("update product", "id", id, "err", err)
writeErrorJSON(w, http.StatusInternalServerError, "failed to update product")
return
}
writeJSON(w, http.StatusOK, p)
}
// Delete handles DELETE /products/{id}.
func (h *Handler) Delete(w http.ResponseWriter, r *http.Request) {
userID := middleware.UserIDFromCtx(r.Context())
id := chi.URLParam(r, "id")
if err := h.repo.Delete(r.Context(), id, userID); err != nil {
if errors.Is(err, ErrNotFound) {
writeErrorJSON(w, http.StatusNotFound, "product not found")
return
}
slog.Error("delete product", "id", id, "err", err)
writeErrorJSON(w, http.StatusInternalServerError, "failed to delete product")
return
}
w.WriteHeader(http.StatusNoContent)
writeJSON(responseWriter, http.StatusOK, savedProduct)
}
type errorResponse struct {
Error string `json:"error"`
}
func writeErrorJSON(w http.ResponseWriter, status int, msg string) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(status)
_ = json.NewEncoder(w).Encode(errorResponse{Error: msg})
func writeErrorJSON(responseWriter http.ResponseWriter, status int, msg string) {
responseWriter.Header().Set("Content-Type", "application/json")
responseWriter.WriteHeader(status)
_ = json.NewEncoder(responseWriter).Encode(errorResponse{Error: msg})
}
func writeJSON(w http.ResponseWriter, status int, v any) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(status)
_ = json.NewEncoder(w).Encode(v)
func writeJSON(responseWriter http.ResponseWriter, status int, value any) {
responseWriter.Header().Set("Content-Type", "application/json")
responseWriter.WriteHeader(status)
_ = json.NewEncoder(responseWriter).Encode(value)
}

View File

@@ -1,36 +0,0 @@
package mocks
import (
"context"
"github.com/food-ai/backend/internal/domain/product"
)
// MockProductRepository is a test double implementing product.ProductRepository.
type MockProductRepository struct {
ListFn func(ctx context.Context, userID string) ([]*product.Product, error)
CreateFn func(ctx context.Context, userID string, req product.CreateRequest) (*product.Product, error)
BatchCreateFn func(ctx context.Context, userID string, items []product.CreateRequest) ([]*product.Product, error)
UpdateFn func(ctx context.Context, id, userID string, req product.UpdateRequest) (*product.Product, error)
DeleteFn func(ctx context.Context, id, userID string) error
}
func (m *MockProductRepository) List(ctx context.Context, userID string) ([]*product.Product, error) {
return m.ListFn(ctx, userID)
}
func (m *MockProductRepository) Create(ctx context.Context, userID string, req product.CreateRequest) (*product.Product, error) {
return m.CreateFn(ctx, userID, req)
}
func (m *MockProductRepository) BatchCreate(ctx context.Context, userID string, items []product.CreateRequest) ([]*product.Product, error) {
return m.BatchCreateFn(ctx, userID, items)
}
func (m *MockProductRepository) Update(ctx context.Context, id, userID string, req product.UpdateRequest) (*product.Product, error) {
return m.UpdateFn(ctx, id, userID, req)
}
func (m *MockProductRepository) Delete(ctx context.Context, id, userID string) error {
return m.DeleteFn(ctx, id, userID)
}

View File

@@ -0,0 +1,84 @@
package product
import (
"context"
"encoding/json"
"fmt"
"net/http"
)
// OpenFoodFacts is the client for the Open Food Facts public API.
type OpenFoodFacts struct {
httpClient *http.Client
}
// NewOpenFoodFacts creates a new OpenFoodFacts client with the default HTTP client.
func NewOpenFoodFacts() *OpenFoodFacts {
return &OpenFoodFacts{httpClient: &http.Client{}}
}
// offProduct is the JSON shape returned by the Open Food Facts v2 API.
type offProduct struct {
ProductName string `json:"product_name"`
Brands string `json:"brands"`
Nutriments offNutriments `json:"nutriments"`
}
type offNutriments struct {
EnergyKcal100g *float64 `json:"energy-kcal_100g"`
Proteins100g *float64 `json:"proteins_100g"`
Fat100g *float64 `json:"fat_100g"`
Carbohydrates100g *float64 `json:"carbohydrates_100g"`
Fiber100g *float64 `json:"fiber_100g"`
}
type offResponse struct {
Status int `json:"status"`
Product offProduct `json:"product"`
}
// Fetch retrieves a product from Open Food Facts by barcode.
// Returns an error if the product is not found or the API call fails.
func (client *OpenFoodFacts) Fetch(requestContext context.Context, barcode string) (*Product, error) {
url := fmt.Sprintf("https://world.openfoodfacts.org/api/v2/product/%s.json", barcode)
httpRequest, requestError := http.NewRequestWithContext(requestContext, http.MethodGet, url, nil)
if requestError != nil {
return nil, fmt.Errorf("build open food facts request: %w", requestError)
}
httpRequest.Header.Set("User-Agent", "FoodAI/1.0")
httpResponse, fetchError := client.httpClient.Do(httpRequest)
if fetchError != nil {
return nil, fmt.Errorf("open food facts request: %w", fetchError)
}
defer httpResponse.Body.Close()
if httpResponse.StatusCode != http.StatusOK {
return nil, fmt.Errorf("open food facts returned status %d for barcode %s", httpResponse.StatusCode, barcode)
}
var offResp offResponse
if decodeError := json.NewDecoder(httpResponse.Body).Decode(&offResp); decodeError != nil {
return nil, fmt.Errorf("decode open food facts response: %w", decodeError)
}
if offResp.Status == 0 {
return nil, fmt.Errorf("product %s not found in open food facts", barcode)
}
canonicalName := offResp.Product.ProductName
if canonicalName == "" {
canonicalName = barcode // Fall back to barcode as canonical name
}
barcodeValue := barcode
catalogProduct := &Product{
CanonicalName: canonicalName,
Barcode: &barcodeValue,
CaloriesPer100g: offResp.Product.Nutriments.EnergyKcal100g,
ProteinPer100g: offResp.Product.Nutriments.Proteins100g,
FatPer100g: offResp.Product.Nutriments.Fat100g,
CarbsPer100g: offResp.Product.Nutriments.Carbohydrates100g,
FiberPer100g: offResp.Product.Nutriments.Fiber100g,
}
return catalogProduct, nil
}

View File

@@ -2,18 +2,16 @@ package product
import (
"context"
"encoding/json"
"errors"
"fmt"
"time"
"github.com/food-ai/backend/internal/infra/locale"
"github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgxpool"
)
// ErrNotFound is returned when a product is not found or does not belong to the user.
var ErrNotFound = errors.New("product not found")
// Repository handles product persistence.
// Repository handles persistence for catalog products and their translations.
type Repository struct {
pool *pgxpool.Pool
}
@@ -23,180 +21,314 @@ func NewRepository(pool *pgxpool.Pool) *Repository {
return &Repository{pool: pool}
}
// expires_at is computed in SQL because TIMESTAMPTZ + INTERVAL is STABLE (not IMMUTABLE),
// which prevents it from being used as a stored generated column.
const selectCols = `id, user_id, primary_ingredient_id, name, quantity, unit, category, storage_days, added_at,
(added_at + storage_days * INTERVAL '1 day') AS expires_at`
// Upsert inserts or updates a catalog product (English canonical content).
// Conflict is resolved on canonical_name.
func (r *Repository) Upsert(requestContext context.Context, catalogProduct *Product) (*Product, error) {
query := `
INSERT INTO products (
canonical_name,
category, default_unit,
calories_per_100g, protein_per_100g, fat_per_100g, carbs_per_100g, fiber_per_100g,
storage_days
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)
ON CONFLICT (canonical_name) DO UPDATE SET
category = EXCLUDED.category,
default_unit = EXCLUDED.default_unit,
calories_per_100g = EXCLUDED.calories_per_100g,
protein_per_100g = EXCLUDED.protein_per_100g,
fat_per_100g = EXCLUDED.fat_per_100g,
carbs_per_100g = EXCLUDED.carbs_per_100g,
fiber_per_100g = EXCLUDED.fiber_per_100g,
storage_days = EXCLUDED.storage_days,
updated_at = now()
RETURNING id, canonical_name, category, default_unit,
calories_per_100g, protein_per_100g, fat_per_100g, carbs_per_100g, fiber_per_100g,
storage_days, created_at, updated_at`
// List returns all products for a user, sorted by expires_at ASC.
func (r *Repository) List(ctx context.Context, userID string) ([]*Product, error) {
rows, err := r.pool.Query(ctx, `
SELECT `+selectCols+`
FROM products
WHERE user_id = $1
ORDER BY expires_at ASC`, userID)
if err != nil {
return nil, fmt.Errorf("list products: %w", err)
row := r.pool.QueryRow(requestContext, query,
catalogProduct.CanonicalName,
catalogProduct.Category, catalogProduct.DefaultUnit,
catalogProduct.CaloriesPer100g, catalogProduct.ProteinPer100g, catalogProduct.FatPer100g, catalogProduct.CarbsPer100g, catalogProduct.FiberPer100g,
catalogProduct.StorageDays,
)
return scanProductWrite(row)
}
// GetByID returns a catalog product by UUID.
// CanonicalName and aliases are resolved for the language stored in requestContext.
// Returns nil, nil if not found.
func (r *Repository) GetByID(requestContext context.Context, id string) (*Product, error) {
lang := locale.FromContext(requestContext)
query := `
SELECT p.id,
p.canonical_name,
p.category,
COALESCE(pct.name, pc.name) AS category_name,
p.default_unit,
COALESCE(ut.name, p.default_unit) AS unit_name,
p.calories_per_100g, p.protein_per_100g, p.fat_per_100g, p.carbs_per_100g, p.fiber_per_100g,
p.storage_days, p.created_at, p.updated_at,
COALESCE(al.aliases, '[]'::json) AS aliases
FROM products p
LEFT JOIN product_categories pc
ON pc.slug = p.category
LEFT JOIN product_category_translations pct
ON pct.product_category_slug = p.category AND pct.lang = $2
LEFT JOIN unit_translations ut
ON ut.unit_code = p.default_unit AND ut.lang = $2
LEFT JOIN LATERAL (
SELECT json_agg(pa.alias ORDER BY pa.alias) AS aliases
FROM product_aliases pa
WHERE pa.product_id = p.id AND pa.lang = $2
) al ON true
WHERE p.id = $1`
row := r.pool.QueryRow(requestContext, query, id, lang)
catalogProduct, queryError := scanProductRead(row)
if errors.Is(queryError, pgx.ErrNoRows) {
return nil, nil
}
return catalogProduct, queryError
}
// GetByBarcode returns a catalog product by barcode value.
// Returns nil, nil if not found.
func (r *Repository) GetByBarcode(requestContext context.Context, barcode string) (*Product, error) {
lang := locale.FromContext(requestContext)
query := `
SELECT p.id,
p.canonical_name,
p.category,
COALESCE(pct.name, pc.name) AS category_name,
p.default_unit,
COALESCE(ut.name, p.default_unit) AS unit_name,
p.calories_per_100g, p.protein_per_100g, p.fat_per_100g, p.carbs_per_100g, p.fiber_per_100g,
p.storage_days, p.created_at, p.updated_at,
COALESCE(al.aliases, '[]'::json) AS aliases
FROM products p
LEFT JOIN product_categories pc
ON pc.slug = p.category
LEFT JOIN product_category_translations pct
ON pct.product_category_slug = p.category AND pct.lang = $2
LEFT JOIN unit_translations ut
ON ut.unit_code = p.default_unit AND ut.lang = $2
LEFT JOIN LATERAL (
SELECT json_agg(pa.alias ORDER BY pa.alias) AS aliases
FROM product_aliases pa
WHERE pa.product_id = p.id AND pa.lang = $2
) al ON true
WHERE p.barcode = $1`
row := r.pool.QueryRow(requestContext, query, barcode, lang)
catalogProduct, queryError := scanProductRead(row)
if errors.Is(queryError, pgx.ErrNoRows) {
return nil, nil
}
return catalogProduct, queryError
}
// UpsertByBarcode inserts or updates a catalog product including its barcode.
func (r *Repository) UpsertByBarcode(requestContext context.Context, catalogProduct *Product) (*Product, error) {
query := `
INSERT INTO products (
canonical_name, barcode,
category, default_unit,
calories_per_100g, protein_per_100g, fat_per_100g, carbs_per_100g, fiber_per_100g,
storage_days
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
ON CONFLICT (canonical_name) DO UPDATE SET
barcode = EXCLUDED.barcode,
category = EXCLUDED.category,
default_unit = EXCLUDED.default_unit,
calories_per_100g = EXCLUDED.calories_per_100g,
protein_per_100g = EXCLUDED.protein_per_100g,
fat_per_100g = EXCLUDED.fat_per_100g,
carbs_per_100g = EXCLUDED.carbs_per_100g,
fiber_per_100g = EXCLUDED.fiber_per_100g,
storage_days = EXCLUDED.storage_days,
updated_at = now()
RETURNING id, canonical_name, category, default_unit,
calories_per_100g, protein_per_100g, fat_per_100g, carbs_per_100g, fiber_per_100g,
storage_days, created_at, updated_at`
row := r.pool.QueryRow(requestContext, query,
catalogProduct.CanonicalName, catalogProduct.Barcode,
catalogProduct.Category, catalogProduct.DefaultUnit,
catalogProduct.CaloriesPer100g, catalogProduct.ProteinPer100g, catalogProduct.FatPer100g, catalogProduct.CarbsPer100g, catalogProduct.FiberPer100g,
catalogProduct.StorageDays,
)
return scanProductWrite(row)
}
// FuzzyMatch finds the single best matching catalog product for a given name.
// Returns nil, nil when no match is found.
func (r *Repository) FuzzyMatch(requestContext context.Context, name string) (*Product, error) {
results, searchError := r.Search(requestContext, name, 1)
if searchError != nil {
return nil, searchError
}
if len(results) == 0 {
return nil, nil
}
return results[0], nil
}
// Search finds catalog products matching the query string.
// Searches aliases table and translated names for the language in requestContext.
func (r *Repository) Search(requestContext context.Context, query string, limit int) ([]*Product, error) {
if limit <= 0 {
limit = 10
}
lang := locale.FromContext(requestContext)
searchQuery := `
SELECT p.id,
p.canonical_name,
p.category,
COALESCE(pct.name, pc.name) AS category_name,
p.default_unit,
COALESCE(ut.name, p.default_unit) AS unit_name,
p.calories_per_100g, p.protein_per_100g, p.fat_per_100g, p.carbs_per_100g, p.fiber_per_100g,
p.storage_days, p.created_at, p.updated_at,
COALESCE(al.aliases, '[]'::json) AS aliases
FROM products p
LEFT JOIN product_categories pc
ON pc.slug = p.category
LEFT JOIN product_category_translations pct
ON pct.product_category_slug = p.category AND pct.lang = $3
LEFT JOIN unit_translations ut
ON ut.unit_code = p.default_unit AND ut.lang = $3
LEFT JOIN LATERAL (
SELECT json_agg(pa.alias ORDER BY pa.alias) AS aliases
FROM product_aliases pa
WHERE pa.product_id = p.id AND pa.lang = $3
) al ON true
WHERE EXISTS (
SELECT 1 FROM product_aliases pa
WHERE pa.product_id = p.id
AND (pa.lang = $3 OR pa.lang = 'en')
AND pa.alias ILIKE '%' || $1 || '%'
)
OR p.canonical_name ILIKE '%' || $1 || '%'
OR similarity(p.canonical_name, $1) > 0.3
ORDER BY similarity(p.canonical_name, $1) DESC
LIMIT $2`
rows, queryError := r.pool.Query(requestContext, searchQuery, query, limit, lang)
if queryError != nil {
return nil, fmt.Errorf("search products: %w", queryError)
}
defer rows.Close()
return collectProducts(rows)
return collectProductsRead(rows)
}
// Create inserts a new product and returns the created record.
func (r *Repository) Create(ctx context.Context, userID string, req CreateRequest) (*Product, error) {
storageDays := req.StorageDays
if storageDays <= 0 {
storageDays = 7
// Count returns the total number of catalog products.
func (r *Repository) Count(requestContext context.Context) (int, error) {
var count int
if queryError := r.pool.QueryRow(requestContext, `SELECT count(*) FROM products`).Scan(&count); queryError != nil {
return 0, fmt.Errorf("count products: %w", queryError)
}
unit := req.Unit
if unit == "" {
unit = "pcs"
}
qty := req.Quantity
if qty <= 0 {
qty = 1
}
// Accept both new and legacy field names.
primaryID := req.PrimaryIngredientID
if primaryID == nil {
primaryID = req.MappingID
}
row := r.pool.QueryRow(ctx, `
INSERT INTO products (user_id, primary_ingredient_id, name, quantity, unit, category, storage_days)
VALUES ($1, $2, $3, $4, $5, $6, $7)
RETURNING `+selectCols,
userID, primaryID, req.Name, qty, unit, req.Category, storageDays,
)
return scanProduct(row)
return count, nil
}
// BatchCreate inserts multiple products sequentially and returns all created records.
func (r *Repository) BatchCreate(ctx context.Context, userID string, items []CreateRequest) ([]*Product, error) {
var result []*Product
for _, req := range items {
p, err := r.Create(ctx, userID, req)
if err != nil {
return nil, fmt.Errorf("batch create product %q: %w", req.Name, err)
// UpsertAliases inserts aliases for a given catalog product and language.
func (r *Repository) UpsertAliases(requestContext context.Context, id, lang string, aliases []string) error {
if len(aliases) == 0 {
return nil
}
batch := &pgx.Batch{}
for _, alias := range aliases {
batch.Queue(
`INSERT INTO product_aliases (product_id, lang, alias) VALUES ($1, $2, $3) ON CONFLICT DO NOTHING`,
id, lang, alias,
)
}
results := r.pool.SendBatch(requestContext, batch)
defer results.Close()
for range aliases {
if _, execError := results.Exec(); execError != nil {
return fmt.Errorf("upsert product alias %s/%s: %w", id, lang, execError)
}
result = append(result, p)
}
return result, nil
}
// Update modifies an existing product. Only non-nil fields are changed.
// Returns ErrNotFound if the product does not exist or belongs to a different user.
func (r *Repository) Update(ctx context.Context, id, userID string, req UpdateRequest) (*Product, error) {
row := r.pool.QueryRow(ctx, `
UPDATE products SET
name = COALESCE($3, name),
quantity = COALESCE($4, quantity),
unit = COALESCE($5, unit),
category = COALESCE($6, category),
storage_days = COALESCE($7, storage_days)
WHERE id = $1 AND user_id = $2
RETURNING `+selectCols,
id, userID, req.Name, req.Quantity, req.Unit, req.Category, req.StorageDays,
)
p, err := scanProduct(row)
if errors.Is(err, pgx.ErrNoRows) {
return nil, ErrNotFound
}
return p, err
}
// Delete removes a product. Returns ErrNotFound if it does not exist or belongs to a different user.
func (r *Repository) Delete(ctx context.Context, id, userID string) error {
tag, err := r.pool.Exec(ctx,
`DELETE FROM products WHERE id = $1 AND user_id = $2`, id, userID)
if err != nil {
return fmt.Errorf("delete product: %w", err)
}
if tag.RowsAffected() == 0 {
return ErrNotFound
}
return nil
}
// ListForPrompt returns a human-readable list of user's products for the AI prompt.
// Expiring soon items are marked with ⚠.
func (r *Repository) ListForPrompt(ctx context.Context, userID string) ([]string, error) {
rows, err := r.pool.Query(ctx, `
WITH p AS (
SELECT name, quantity, unit,
(added_at + storage_days * INTERVAL '1 day') AS expires_at
FROM products
WHERE user_id = $1
)
SELECT name, quantity, unit, expires_at
FROM p
ORDER BY expires_at ASC`, userID)
if err != nil {
return nil, fmt.Errorf("list products for prompt: %w", err)
}
defer rows.Close()
// UpsertCategoryTranslation inserts or replaces a localized category name.
func (r *Repository) UpsertCategoryTranslation(requestContext context.Context, slug, lang, name string) error {
query := `
INSERT INTO product_category_translations (product_category_slug, lang, name)
VALUES ($1, $2, $3)
ON CONFLICT (product_category_slug, lang) DO UPDATE SET name = EXCLUDED.name`
var lines []string
now := time.Now()
for rows.Next() {
var name, unit string
var qty float64
var expiresAt time.Time
if err := rows.Scan(&name, &qty, &unit, &expiresAt); err != nil {
return nil, fmt.Errorf("scan product for prompt: %w", err)
}
daysLeft := int(expiresAt.Sub(now).Hours() / 24)
line := fmt.Sprintf("- %s %.0f %s", name, qty, unit)
switch {
case daysLeft <= 0:
line += " (expires today ⚠)"
case daysLeft == 1:
line += " (expires tomorrow ⚠)"
case daysLeft <= 3:
line += fmt.Sprintf(" (expires in %d days ⚠)", daysLeft)
}
lines = append(lines, line)
if _, execError := r.pool.Exec(requestContext, query, slug, lang, name); execError != nil {
return fmt.Errorf("upsert category translation %s/%s: %w", slug, lang, execError)
}
return lines, rows.Err()
return nil
}
// --- helpers ---
// --- scan helpers ---
func scanProduct(row pgx.Row) (*Product, error) {
var p Product
err := row.Scan(
&p.ID, &p.UserID, &p.PrimaryIngredientID, &p.Name, &p.Quantity, &p.Unit,
&p.Category, &p.StorageDays, &p.AddedAt, &p.ExpiresAt,
func scanProductWrite(row pgx.Row) (*Product, error) {
var catalogProduct Product
scanError := row.Scan(
&catalogProduct.ID, &catalogProduct.CanonicalName, &catalogProduct.Category, &catalogProduct.DefaultUnit,
&catalogProduct.CaloriesPer100g, &catalogProduct.ProteinPer100g, &catalogProduct.FatPer100g, &catalogProduct.CarbsPer100g, &catalogProduct.FiberPer100g,
&catalogProduct.StorageDays, &catalogProduct.CreatedAt, &catalogProduct.UpdatedAt,
)
if err != nil {
return nil, err
if scanError != nil {
return nil, scanError
}
computeDaysLeft(&p)
return &p, nil
catalogProduct.Aliases = json.RawMessage("[]")
return &catalogProduct, nil
}
func collectProducts(rows pgx.Rows) ([]*Product, error) {
func scanProductRead(row pgx.Row) (*Product, error) {
var catalogProduct Product
var aliases []byte
scanError := row.Scan(
&catalogProduct.ID, &catalogProduct.CanonicalName, &catalogProduct.Category, &catalogProduct.CategoryName,
&catalogProduct.DefaultUnit, &catalogProduct.DefaultUnitName,
&catalogProduct.CaloriesPer100g, &catalogProduct.ProteinPer100g, &catalogProduct.FatPer100g, &catalogProduct.CarbsPer100g, &catalogProduct.FiberPer100g,
&catalogProduct.StorageDays, &catalogProduct.CreatedAt, &catalogProduct.UpdatedAt, &aliases,
)
if scanError != nil {
return nil, scanError
}
catalogProduct.Aliases = json.RawMessage(aliases)
return &catalogProduct, nil
}
func collectProductsWrite(rows pgx.Rows) ([]*Product, error) {
var result []*Product
for rows.Next() {
var p Product
if err := rows.Scan(
&p.ID, &p.UserID, &p.PrimaryIngredientID, &p.Name, &p.Quantity, &p.Unit,
&p.Category, &p.StorageDays, &p.AddedAt, &p.ExpiresAt,
); err != nil {
return nil, fmt.Errorf("scan product: %w", err)
var catalogProduct Product
if scanError := rows.Scan(
&catalogProduct.ID, &catalogProduct.CanonicalName, &catalogProduct.Category, &catalogProduct.DefaultUnit,
&catalogProduct.CaloriesPer100g, &catalogProduct.ProteinPer100g, &catalogProduct.FatPer100g, &catalogProduct.CarbsPer100g, &catalogProduct.FiberPer100g,
&catalogProduct.StorageDays, &catalogProduct.CreatedAt, &catalogProduct.UpdatedAt,
); scanError != nil {
return nil, fmt.Errorf("scan product: %w", scanError)
}
computeDaysLeft(&p)
result = append(result, &p)
catalogProduct.Aliases = json.RawMessage("[]")
result = append(result, &catalogProduct)
}
return result, rows.Err()
}
func computeDaysLeft(p *Product) {
d := int(time.Until(p.ExpiresAt).Hours() / 24)
if d < 0 {
d = 0
func collectProductsRead(rows pgx.Rows) ([]*Product, error) {
var result []*Product
for rows.Next() {
var catalogProduct Product
var aliases []byte
if scanError := rows.Scan(
&catalogProduct.ID, &catalogProduct.CanonicalName, &catalogProduct.Category, &catalogProduct.CategoryName,
&catalogProduct.DefaultUnit, &catalogProduct.DefaultUnitName,
&catalogProduct.CaloriesPer100g, &catalogProduct.ProteinPer100g, &catalogProduct.FatPer100g, &catalogProduct.CarbsPer100g, &catalogProduct.FiberPer100g,
&catalogProduct.StorageDays, &catalogProduct.CreatedAt, &catalogProduct.UpdatedAt, &aliases,
); scanError != nil {
return nil, fmt.Errorf("scan product: %w", scanError)
}
catalogProduct.Aliases = json.RawMessage(aliases)
result = append(result, &catalogProduct)
}
p.DaysLeft = d
p.ExpiringSoon = d <= 3
return result, rows.Err()
}

View File

@@ -12,7 +12,7 @@ import (
"github.com/food-ai/backend/internal/adapters/ai"
"github.com/food-ai/backend/internal/domain/dish"
"github.com/food-ai/backend/internal/domain/ingredient"
"github.com/food-ai/backend/internal/domain/product"
"github.com/food-ai/backend/internal/infra/locale"
"github.com/food-ai/backend/internal/infra/middleware"
)
@@ -26,12 +26,10 @@ type DishRepository interface {
AddRecipe(ctx context.Context, dishID string, req dish.CreateRequest) (string, error)
}
// IngredientRepository is the subset of ingredient.Repository used by this handler.
type IngredientRepository interface {
FuzzyMatch(ctx context.Context, name string) (*ingredient.IngredientMapping, error)
Upsert(ctx context.Context, m *ingredient.IngredientMapping) (*ingredient.IngredientMapping, error)
UpsertTranslation(ctx context.Context, id, lang, name string) error
UpsertAliases(ctx context.Context, id, lang string, aliases []string) error
// ProductRepository is the subset of product.Repository used by this handler.
type ProductRepository interface {
FuzzyMatch(ctx context.Context, name string) (*product.Product, error)
Upsert(ctx context.Context, catalogProduct *product.Product) (*product.Product, error)
}
// Recognizer is the AI provider interface for image-based food recognition.
@@ -51,27 +49,27 @@ type KafkaPublisher interface {
// Handler handles POST /ai/* recognition endpoints.
type Handler struct {
recognizer Recognizer
ingredientRepo IngredientRepository
jobRepo JobRepository
kafkaProducer KafkaPublisher
sseBroker *SSEBroker
recognizer Recognizer
productRepo ProductRepository
jobRepo JobRepository
kafkaProducer KafkaPublisher
sseBroker *SSEBroker
}
// NewHandler creates a new Handler with async dish recognition support.
func NewHandler(
recognizer Recognizer,
ingredientRepo IngredientRepository,
productRepo ProductRepository,
jobRepo JobRepository,
kafkaProducer KafkaPublisher,
sseBroker *SSEBroker,
) *Handler {
return &Handler{
recognizer: recognizer,
ingredientRepo: ingredientRepo,
jobRepo: jobRepo,
kafkaProducer: kafkaProducer,
sseBroker: sseBroker,
recognizer: recognizer,
productRepo: productRepo,
jobRepo: jobRepo,
kafkaProducer: kafkaProducer,
sseBroker: sseBroker,
}
}
@@ -293,7 +291,7 @@ func (handler *Handler) GetJob(responseWriter http.ResponseWriter, request *http
// Helpers
// ---------------------------------------------------------------------------
// enrichItems matches each recognized item against ingredient_mappings.
// enrichItems matches each recognized item against the product catalog.
// Items without a match trigger a classification call and upsert into the DB.
func (handler *Handler) enrichItems(ctx context.Context, items []ai.RecognizedItem) []EnrichedItem {
result := make([]EnrichedItem, 0, len(items))
@@ -307,27 +305,27 @@ func (handler *Handler) enrichItems(ctx context.Context, items []ai.RecognizedIt
StorageDays: 7, // sensible default
}
mapping, matchError := handler.ingredientRepo.FuzzyMatch(ctx, item.Name)
catalogProduct, matchError := handler.productRepo.FuzzyMatch(ctx, item.Name)
if matchError != nil {
slog.Warn("fuzzy match ingredient", "name", item.Name, "err", matchError)
slog.Warn("fuzzy match product", "name", item.Name, "err", matchError)
}
if mapping != nil {
id := mapping.ID
if catalogProduct != nil {
id := catalogProduct.ID
enriched.MappingID = &id
if mapping.DefaultUnit != nil {
enriched.Unit = *mapping.DefaultUnit
if catalogProduct.DefaultUnit != nil {
enriched.Unit = *catalogProduct.DefaultUnit
}
if mapping.StorageDays != nil {
enriched.StorageDays = *mapping.StorageDays
if catalogProduct.StorageDays != nil {
enriched.StorageDays = *catalogProduct.StorageDays
}
if mapping.Category != nil {
enriched.Category = *mapping.Category
if catalogProduct.Category != nil {
enriched.Category = *catalogProduct.Category
}
} else {
classification, classifyError := handler.recognizer.ClassifyIngredient(ctx, item.Name)
if classifyError != nil {
slog.Warn("classify unknown ingredient", "name", item.Name, "err", classifyError)
slog.Warn("classify unknown product", "name", item.Name, "err", classifyError)
} else {
saved := handler.saveClassification(ctx, classification)
if saved != nil {
@@ -344,13 +342,13 @@ func (handler *Handler) enrichItems(ctx context.Context, items []ai.RecognizedIt
return result
}
// saveClassification upserts an AI-produced ingredient classification into the DB.
func (handler *Handler) saveClassification(ctx context.Context, classification *ai.IngredientClassification) *ingredient.IngredientMapping {
// saveClassification upserts an AI-produced classification into the product catalog.
func (handler *Handler) saveClassification(ctx context.Context, classification *ai.IngredientClassification) *product.Product {
if classification == nil || classification.CanonicalName == "" {
return nil
}
mapping := &ingredient.IngredientMapping{
catalogProduct := &product.Product{
CanonicalName: classification.CanonicalName,
Category: strPtr(classification.Category),
DefaultUnit: strPtr(classification.DefaultUnit),
@@ -361,29 +359,12 @@ func (handler *Handler) saveClassification(ctx context.Context, classification *
StorageDays: intPtr(classification.StorageDays),
}
saved, upsertError := handler.ingredientRepo.Upsert(ctx, mapping)
saved, upsertError := handler.productRepo.Upsert(ctx, catalogProduct)
if upsertError != nil {
slog.Warn("upsert classified ingredient", "name", classification.CanonicalName, "err", upsertError)
slog.Warn("upsert classified product", "name", classification.CanonicalName, "err", upsertError)
return nil
}
if len(classification.Aliases) > 0 {
if aliasError := handler.ingredientRepo.UpsertAliases(ctx, saved.ID, "en", classification.Aliases); aliasError != nil {
slog.Warn("upsert ingredient aliases", "id", saved.ID, "err", aliasError)
}
}
for _, translation := range classification.Translations {
if translationError := handler.ingredientRepo.UpsertTranslation(ctx, saved.ID, translation.Lang, translation.Name); translationError != nil {
slog.Warn("upsert ingredient translation", "id", saved.ID, "lang", translation.Lang, "err", translationError)
}
if len(translation.Aliases) > 0 {
if aliasError := handler.ingredientRepo.UpsertAliases(ctx, saved.ID, translation.Lang, translation.Aliases); aliasError != nil {
slog.Warn("upsert ingredient translation aliases", "id", saved.ID, "lang", translation.Lang, "err", aliasError)
}
}
}
return saved
}

View File

@@ -0,0 +1,41 @@
package userproduct
import "time"
// UserProduct is a user's food item in their pantry.
type UserProduct struct {
ID string `json:"id"`
UserID string `json:"user_id"`
PrimaryProductID *string `json:"primary_product_id"`
Name string `json:"name"`
Quantity float64 `json:"quantity"`
Unit string `json:"unit"`
Category *string `json:"category"`
StorageDays int `json:"storage_days"`
AddedAt time.Time `json:"added_at"`
ExpiresAt time.Time `json:"expires_at"`
DaysLeft int `json:"days_left"`
ExpiringSoon bool `json:"expiring_soon"`
}
// CreateRequest is the body for POST /user-products.
type CreateRequest struct {
PrimaryProductID *string `json:"primary_product_id"`
// Accept "mapping_id" as a legacy alias for backward compatibility.
MappingID *string `json:"mapping_id"`
Name string `json:"name"`
Quantity float64 `json:"quantity"`
Unit string `json:"unit"`
Category *string `json:"category"`
StorageDays int `json:"storage_days"`
}
// UpdateRequest is the body for PUT /user-products/{id}.
// All fields are optional (nil = keep existing value).
type UpdateRequest struct {
Name *string `json:"name"`
Quantity *float64 `json:"quantity"`
Unit *string `json:"unit"`
Category *string `json:"category"`
StorageDays *int `json:"storage_days"`
}

View File

@@ -0,0 +1,147 @@
package userproduct
import (
"context"
"encoding/json"
"errors"
"log/slog"
"net/http"
"github.com/food-ai/backend/internal/infra/middleware"
"github.com/go-chi/chi/v5"
)
// UserProductRepository is the data layer interface used by Handler.
type UserProductRepository interface {
List(ctx context.Context, userID string) ([]*UserProduct, error)
Create(ctx context.Context, userID string, req CreateRequest) (*UserProduct, error)
BatchCreate(ctx context.Context, userID string, items []CreateRequest) ([]*UserProduct, error)
Update(ctx context.Context, id, userID string, req UpdateRequest) (*UserProduct, error)
Delete(ctx context.Context, id, userID string) error
}
// Handler handles /user-products HTTP requests.
type Handler struct {
repo UserProductRepository
}
// NewHandler creates a new Handler.
func NewHandler(repo UserProductRepository) *Handler {
return &Handler{repo: repo}
}
// List handles GET /user-products.
func (handler *Handler) List(responseWriter http.ResponseWriter, request *http.Request) {
userID := middleware.UserIDFromCtx(request.Context())
userProducts, listError := handler.repo.List(request.Context(), userID)
if listError != nil {
slog.Error("list user products", "user_id", userID, "err", listError)
writeErrorJSON(responseWriter, http.StatusInternalServerError, "failed to list user products")
return
}
if userProducts == nil {
userProducts = []*UserProduct{}
}
writeJSON(responseWriter, http.StatusOK, userProducts)
}
// Create handles POST /user-products.
func (handler *Handler) Create(responseWriter http.ResponseWriter, request *http.Request) {
userID := middleware.UserIDFromCtx(request.Context())
var req CreateRequest
if decodeError := json.NewDecoder(request.Body).Decode(&req); decodeError != nil {
writeErrorJSON(responseWriter, http.StatusBadRequest, "invalid request body")
return
}
if req.Name == "" {
writeErrorJSON(responseWriter, http.StatusBadRequest, "name is required")
return
}
userProduct, createError := handler.repo.Create(request.Context(), userID, req)
if createError != nil {
slog.Error("create user product", "user_id", userID, "err", createError)
writeErrorJSON(responseWriter, http.StatusInternalServerError, "failed to create user product")
return
}
writeJSON(responseWriter, http.StatusCreated, userProduct)
}
// BatchCreate handles POST /user-products/batch.
func (handler *Handler) BatchCreate(responseWriter http.ResponseWriter, request *http.Request) {
userID := middleware.UserIDFromCtx(request.Context())
var items []CreateRequest
if decodeError := json.NewDecoder(request.Body).Decode(&items); decodeError != nil {
writeErrorJSON(responseWriter, http.StatusBadRequest, "invalid request body")
return
}
if len(items) == 0 {
writeJSON(responseWriter, http.StatusCreated, []*UserProduct{})
return
}
userProducts, batchError := handler.repo.BatchCreate(request.Context(), userID, items)
if batchError != nil {
slog.Error("batch create user products", "user_id", userID, "err", batchError)
writeErrorJSON(responseWriter, http.StatusInternalServerError, "failed to create user products")
return
}
writeJSON(responseWriter, http.StatusCreated, userProducts)
}
// Update handles PUT /user-products/{id}.
func (handler *Handler) Update(responseWriter http.ResponseWriter, request *http.Request) {
userID := middleware.UserIDFromCtx(request.Context())
id := chi.URLParam(request, "id")
var req UpdateRequest
if decodeError := json.NewDecoder(request.Body).Decode(&req); decodeError != nil {
writeErrorJSON(responseWriter, http.StatusBadRequest, "invalid request body")
return
}
userProduct, updateError := handler.repo.Update(request.Context(), id, userID, req)
if errors.Is(updateError, ErrNotFound) {
writeErrorJSON(responseWriter, http.StatusNotFound, "user product not found")
return
}
if updateError != nil {
slog.Error("update user product", "id", id, "err", updateError)
writeErrorJSON(responseWriter, http.StatusInternalServerError, "failed to update user product")
return
}
writeJSON(responseWriter, http.StatusOK, userProduct)
}
// Delete handles DELETE /user-products/{id}.
func (handler *Handler) Delete(responseWriter http.ResponseWriter, request *http.Request) {
userID := middleware.UserIDFromCtx(request.Context())
id := chi.URLParam(request, "id")
if deleteError := handler.repo.Delete(request.Context(), id, userID); deleteError != nil {
if errors.Is(deleteError, ErrNotFound) {
writeErrorJSON(responseWriter, http.StatusNotFound, "user product not found")
return
}
slog.Error("delete user product", "id", id, "err", deleteError)
writeErrorJSON(responseWriter, http.StatusInternalServerError, "failed to delete user product")
return
}
responseWriter.WriteHeader(http.StatusNoContent)
}
type errorResponse struct {
Error string `json:"error"`
}
func writeErrorJSON(responseWriter http.ResponseWriter, status int, msg string) {
responseWriter.Header().Set("Content-Type", "application/json")
responseWriter.WriteHeader(status)
_ = json.NewEncoder(responseWriter).Encode(errorResponse{Error: msg})
}
func writeJSON(responseWriter http.ResponseWriter, status int, value any) {
responseWriter.Header().Set("Content-Type", "application/json")
responseWriter.WriteHeader(status)
_ = json.NewEncoder(responseWriter).Encode(value)
}

View File

@@ -0,0 +1,36 @@
package mocks
import (
"context"
"github.com/food-ai/backend/internal/domain/userproduct"
)
// MockUserProductRepository is a test double implementing userproduct.UserProductRepository.
type MockUserProductRepository struct {
ListFn func(ctx context.Context, userID string) ([]*userproduct.UserProduct, error)
CreateFn func(ctx context.Context, userID string, req userproduct.CreateRequest) (*userproduct.UserProduct, error)
BatchCreateFn func(ctx context.Context, userID string, items []userproduct.CreateRequest) ([]*userproduct.UserProduct, error)
UpdateFn func(ctx context.Context, id, userID string, req userproduct.UpdateRequest) (*userproduct.UserProduct, error)
DeleteFn func(ctx context.Context, id, userID string) error
}
func (m *MockUserProductRepository) List(ctx context.Context, userID string) ([]*userproduct.UserProduct, error) {
return m.ListFn(ctx, userID)
}
func (m *MockUserProductRepository) Create(ctx context.Context, userID string, req userproduct.CreateRequest) (*userproduct.UserProduct, error) {
return m.CreateFn(ctx, userID, req)
}
func (m *MockUserProductRepository) BatchCreate(ctx context.Context, userID string, items []userproduct.CreateRequest) ([]*userproduct.UserProduct, error) {
return m.BatchCreateFn(ctx, userID, items)
}
func (m *MockUserProductRepository) Update(ctx context.Context, id, userID string, req userproduct.UpdateRequest) (*userproduct.UserProduct, error) {
return m.UpdateFn(ctx, id, userID, req)
}
func (m *MockUserProductRepository) Delete(ctx context.Context, id, userID string) error {
return m.DeleteFn(ctx, id, userID)
}

View File

@@ -0,0 +1,202 @@
package userproduct
import (
"context"
"errors"
"fmt"
"time"
"github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgxpool"
)
// ErrNotFound is returned when a user product is not found or does not belong to the user.
var ErrNotFound = errors.New("user product not found")
// Repository handles user product persistence.
type Repository struct {
pool *pgxpool.Pool
}
// NewRepository creates a new Repository.
func NewRepository(pool *pgxpool.Pool) *Repository {
return &Repository{pool: pool}
}
// expires_at is computed in SQL because TIMESTAMPTZ + INTERVAL is STABLE (not IMMUTABLE),
// which prevents it from being used as a stored generated column.
const selectCols = `id, user_id, primary_product_id, name, quantity, unit, category, storage_days, added_at,
(added_at + storage_days * INTERVAL '1 day') AS expires_at`
// List returns all user products sorted by expires_at ASC.
func (r *Repository) List(requestContext context.Context, userID string) ([]*UserProduct, error) {
rows, queryError := r.pool.Query(requestContext, `
SELECT `+selectCols+`
FROM user_products
WHERE user_id = $1
ORDER BY expires_at ASC`, userID)
if queryError != nil {
return nil, fmt.Errorf("list user products: %w", queryError)
}
defer rows.Close()
return collectUserProducts(rows)
}
// Create inserts a new user product and returns the created record.
func (r *Repository) Create(requestContext context.Context, userID string, req CreateRequest) (*UserProduct, error) {
storageDays := req.StorageDays
if storageDays <= 0 {
storageDays = 7
}
unit := req.Unit
if unit == "" {
unit = "pcs"
}
qty := req.Quantity
if qty <= 0 {
qty = 1
}
// Accept both new and legacy field names.
primaryID := req.PrimaryProductID
if primaryID == nil {
primaryID = req.MappingID
}
row := r.pool.QueryRow(requestContext, `
INSERT INTO user_products (user_id, primary_product_id, name, quantity, unit, category, storage_days)
VALUES ($1, $2, $3, $4, $5, $6, $7)
RETURNING `+selectCols,
userID, primaryID, req.Name, qty, unit, req.Category, storageDays,
)
return scanUserProduct(row)
}
// BatchCreate inserts multiple user products sequentially and returns all created records.
func (r *Repository) BatchCreate(requestContext context.Context, userID string, items []CreateRequest) ([]*UserProduct, error) {
var result []*UserProduct
for _, req := range items {
userProduct, createError := r.Create(requestContext, userID, req)
if createError != nil {
return nil, fmt.Errorf("batch create user product %q: %w", req.Name, createError)
}
result = append(result, userProduct)
}
return result, nil
}
// Update modifies an existing user product. Only non-nil fields are changed.
// Returns ErrNotFound if the product does not exist or belongs to a different user.
func (r *Repository) Update(requestContext context.Context, id, userID string, req UpdateRequest) (*UserProduct, error) {
row := r.pool.QueryRow(requestContext, `
UPDATE user_products SET
name = COALESCE($3, name),
quantity = COALESCE($4, quantity),
unit = COALESCE($5, unit),
category = COALESCE($6, category),
storage_days = COALESCE($7, storage_days)
WHERE id = $1 AND user_id = $2
RETURNING `+selectCols,
id, userID, req.Name, req.Quantity, req.Unit, req.Category, req.StorageDays,
)
userProduct, scanError := scanUserProduct(row)
if errors.Is(scanError, pgx.ErrNoRows) {
return nil, ErrNotFound
}
return userProduct, scanError
}
// Delete removes a user product. Returns ErrNotFound if it does not exist or belongs to a different user.
func (r *Repository) Delete(requestContext context.Context, id, userID string) error {
tag, execError := r.pool.Exec(requestContext,
`DELETE FROM user_products WHERE id = $1 AND user_id = $2`, id, userID)
if execError != nil {
return fmt.Errorf("delete user product: %w", execError)
}
if tag.RowsAffected() == 0 {
return ErrNotFound
}
return nil
}
// ListForPrompt returns a human-readable list of user's products for the AI prompt.
// Expiring soon items are marked with ⚠.
func (r *Repository) ListForPrompt(requestContext context.Context, userID string) ([]string, error) {
rows, queryError := r.pool.Query(requestContext, `
WITH up AS (
SELECT name, quantity, unit,
(added_at + storage_days * INTERVAL '1 day') AS expires_at
FROM user_products
WHERE user_id = $1
)
SELECT name, quantity, unit, expires_at
FROM up
ORDER BY expires_at ASC`, userID)
if queryError != nil {
return nil, fmt.Errorf("list user products for prompt: %w", queryError)
}
defer rows.Close()
var lines []string
now := time.Now()
for rows.Next() {
var name, unit string
var qty float64
var expiresAt time.Time
if scanError := rows.Scan(&name, &qty, &unit, &expiresAt); scanError != nil {
return nil, fmt.Errorf("scan user product for prompt: %w", scanError)
}
daysLeft := int(expiresAt.Sub(now).Hours() / 24)
line := fmt.Sprintf("- %s %.0f %s", name, qty, unit)
switch {
case daysLeft <= 0:
line += " (expires today ⚠)"
case daysLeft == 1:
line += " (expires tomorrow ⚠)"
case daysLeft <= 3:
line += fmt.Sprintf(" (expires in %d days ⚠)", daysLeft)
}
lines = append(lines, line)
}
return lines, rows.Err()
}
// --- helpers ---
func scanUserProduct(row pgx.Row) (*UserProduct, error) {
var userProduct UserProduct
scanError := row.Scan(
&userProduct.ID, &userProduct.UserID, &userProduct.PrimaryProductID, &userProduct.Name, &userProduct.Quantity, &userProduct.Unit,
&userProduct.Category, &userProduct.StorageDays, &userProduct.AddedAt, &userProduct.ExpiresAt,
)
if scanError != nil {
return nil, scanError
}
computeDaysLeft(&userProduct)
return &userProduct, nil
}
func collectUserProducts(rows pgx.Rows) ([]*UserProduct, error) {
var result []*UserProduct
for rows.Next() {
var userProduct UserProduct
if scanError := rows.Scan(
&userProduct.ID, &userProduct.UserID, &userProduct.PrimaryProductID, &userProduct.Name, &userProduct.Quantity, &userProduct.Unit,
&userProduct.Category, &userProduct.StorageDays, &userProduct.AddedAt, &userProduct.ExpiresAt,
); scanError != nil {
return nil, fmt.Errorf("scan user product: %w", scanError)
}
computeDaysLeft(&userProduct)
result = append(result, &userProduct)
}
return result, rows.Err()
}
func computeDaysLeft(userProduct *UserProduct) {
days := int(time.Until(userProduct.ExpiresAt).Hours() / 24)
if days < 0 {
days = 0
}
userProduct.DaysLeft = days
userProduct.ExpiringSoon = days <= 3
}

View File

@@ -8,16 +8,16 @@ import (
"github.com/food-ai/backend/internal/domain/diary"
"github.com/food-ai/backend/internal/domain/dish"
"github.com/food-ai/backend/internal/domain/home"
"github.com/food-ai/backend/internal/domain/ingredient"
"github.com/food-ai/backend/internal/domain/language"
"github.com/food-ai/backend/internal/domain/menu"
"github.com/food-ai/backend/internal/infra/middleware"
"github.com/food-ai/backend/internal/domain/recipe"
"github.com/food-ai/backend/internal/domain/product"
"github.com/food-ai/backend/internal/domain/recipe"
"github.com/food-ai/backend/internal/domain/recognition"
"github.com/food-ai/backend/internal/domain/recommendation"
"github.com/food-ai/backend/internal/domain/savedrecipe"
"github.com/food-ai/backend/internal/domain/user"
"github.com/food-ai/backend/internal/domain/userproduct"
"github.com/go-chi/chi/v5"
"github.com/jackc/pgx/v5/pgxpool"
)
@@ -28,8 +28,8 @@ func NewRouter(
userHandler *user.Handler,
recommendationHandler *recommendation.Handler,
savedRecipeHandler *savedrecipe.Handler,
ingredientHandler *ingredient.Handler,
productHandler *product.Handler,
userProductHandler *userproduct.Handler,
recognitionHandler *recognition.Handler,
menuHandler *menu.Handler,
diaryHandler *diary.Handler,
@@ -67,7 +67,8 @@ func NewRouter(
r.Group(func(r chi.Router) {
r.Use(authMiddleware)
r.Get("/ingredients/search", ingredientHandler.Search)
r.Get("/products/search", productHandler.Search)
r.Get("/products/barcode/{barcode}", productHandler.GetByBarcode)
r.Get("/profile", userHandler.Get)
r.Put("/profile", userHandler.Update)
@@ -81,12 +82,12 @@ func NewRouter(
r.Delete("/{id}", savedRecipeHandler.Delete)
})
r.Route("/products", func(r chi.Router) {
r.Get("/", productHandler.List)
r.Post("/", productHandler.Create)
r.Post("/batch", productHandler.BatchCreate)
r.Put("/{id}", productHandler.Update)
r.Delete("/{id}", productHandler.Delete)
r.Route("/user-products", func(r chi.Router) {
r.Get("/", userProductHandler.List)
r.Post("/", userProductHandler.Create)
r.Post("/batch", userProductHandler.BatchCreate)
r.Put("/{id}", userProductHandler.Update)
r.Delete("/{id}", userProductHandler.Delete)
})
r.Route("/dishes", func(r chi.Router) {

View File

@@ -44,11 +44,11 @@ $$ LANGUAGE plpgsql VOLATILE;
-- ---------------------------------------------------------------------------
-- Enums
-- ---------------------------------------------------------------------------
CREATE TYPE user_plan AS ENUM ('free', 'paid');
CREATE TYPE user_gender AS ENUM ('male', 'female');
CREATE TYPE user_goal AS ENUM ('lose', 'maintain', 'gain');
CREATE TYPE activity_level AS ENUM ('low', 'moderate', 'high');
CREATE TYPE recipe_source AS ENUM ('spoonacular', 'ai', 'user');
CREATE TYPE user_plan AS ENUM ('free', 'paid');
CREATE TYPE user_gender AS ENUM ('male', 'female');
CREATE TYPE user_goal AS ENUM ('lose', 'maintain', 'gain');
CREATE TYPE activity_level AS ENUM ('low', 'moderate', 'high');
CREATE TYPE recipe_source AS ENUM ('spoonacular', 'ai', 'user');
CREATE TYPE recipe_difficulty AS ENUM ('easy', 'medium', 'hard');
-- ---------------------------------------------------------------------------
@@ -104,28 +104,30 @@ CREATE TABLE unit_translations (
);
-- ---------------------------------------------------------------------------
-- ingredient_categories + translations
-- product_categories + product_category_translations
-- ---------------------------------------------------------------------------
CREATE TABLE ingredient_categories (
CREATE TABLE product_categories (
slug VARCHAR(50) PRIMARY KEY,
name TEXT NOT NULL,
sort_order SMALLINT NOT NULL DEFAULT 0
);
CREATE TABLE ingredient_category_translations (
category_slug VARCHAR(50) NOT NULL REFERENCES ingredient_categories(slug) ON DELETE CASCADE,
lang VARCHAR(10) NOT NULL,
name TEXT NOT NULL,
PRIMARY KEY (category_slug, lang)
CREATE TABLE product_category_translations (
product_category_slug VARCHAR(50) NOT NULL REFERENCES product_categories(slug) ON DELETE CASCADE,
lang VARCHAR(10) NOT NULL,
name TEXT NOT NULL,
PRIMARY KEY (product_category_slug, lang)
);
-- ---------------------------------------------------------------------------
-- ingredients (canonical catalog)
-- products (canonical catalog)
-- ---------------------------------------------------------------------------
CREATE TABLE ingredients (
CREATE TABLE products (
id UUID PRIMARY KEY DEFAULT uuid_generate_v7(),
canonical_name VARCHAR(255) NOT NULL,
category VARCHAR(50) REFERENCES ingredient_categories(slug),
category VARCHAR(50) REFERENCES product_categories(slug),
default_unit VARCHAR(20) REFERENCES units(code),
barcode TEXT UNIQUE,
calories_per_100g DECIMAL(8,2),
protein_per_100g DECIMAL(8,2),
fat_per_100g DECIMAL(8,2),
@@ -134,33 +136,23 @@ CREATE TABLE ingredients (
storage_days INTEGER,
created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
updated_at TIMESTAMPTZ NOT NULL DEFAULT now(),
CONSTRAINT uq_ingredient_canonical_name UNIQUE (canonical_name)
CONSTRAINT uq_product_canonical_name UNIQUE (canonical_name)
);
CREATE INDEX idx_ingredients_canonical_name ON ingredients (canonical_name);
CREATE INDEX idx_ingredients_category ON ingredients (category);
CREATE INDEX idx_products_canonical_name ON products (canonical_name);
CREATE INDEX idx_products_category ON products (category);
CREATE INDEX idx_products_barcode ON products (barcode) WHERE barcode IS NOT NULL;
-- ---------------------------------------------------------------------------
-- ingredient_translations
-- product_aliases
-- ---------------------------------------------------------------------------
CREATE TABLE ingredient_translations (
ingredient_id UUID NOT NULL REFERENCES ingredients(id) ON DELETE CASCADE,
lang VARCHAR(10) NOT NULL,
name VARCHAR(255) NOT NULL,
PRIMARY KEY (ingredient_id, lang)
CREATE TABLE product_aliases (
product_id UUID NOT NULL REFERENCES products(id) ON DELETE CASCADE,
lang VARCHAR(10) NOT NULL,
alias TEXT NOT NULL,
PRIMARY KEY (product_id, lang, alias)
);
CREATE INDEX idx_ingredient_translations_ingredient_id ON ingredient_translations (ingredient_id);
-- ---------------------------------------------------------------------------
-- ingredient_aliases
-- ---------------------------------------------------------------------------
CREATE TABLE ingredient_aliases (
ingredient_id UUID NOT NULL REFERENCES ingredients(id) ON DELETE CASCADE,
lang VARCHAR(10) NOT NULL,
alias TEXT NOT NULL,
PRIMARY KEY (ingredient_id, lang, alias)
);
CREATE INDEX idx_ingredient_aliases_lookup ON ingredient_aliases (ingredient_id, lang);
CREATE INDEX idx_ingredient_aliases_trgm ON ingredient_aliases USING GIN (alias gin_trgm_ops);
CREATE INDEX idx_product_aliases_lookup ON product_aliases (product_id, lang);
CREATE INDEX idx_product_aliases_trgm ON product_aliases USING GIN (alias gin_trgm_ops);
-- ---------------------------------------------------------------------------
-- cuisines + cuisine_translations
@@ -279,22 +271,22 @@ CREATE TABLE recipe_translations (
);
-- ---------------------------------------------------------------------------
-- recipe_ingredients + recipe_ingredient_translations
-- recipe_products + recipe_product_translations
-- ---------------------------------------------------------------------------
CREATE TABLE recipe_ingredients (
id UUID PRIMARY KEY DEFAULT uuid_generate_v7(),
recipe_id UUID NOT NULL REFERENCES recipes(id) ON DELETE CASCADE,
ingredient_id UUID REFERENCES ingredients(id) ON DELETE SET NULL,
name TEXT NOT NULL,
amount DECIMAL(10,2) NOT NULL DEFAULT 0,
unit_code VARCHAR(20),
is_optional BOOLEAN NOT NULL DEFAULT false,
sort_order SMALLINT NOT NULL DEFAULT 0
CREATE TABLE recipe_products (
id UUID PRIMARY KEY DEFAULT uuid_generate_v7(),
recipe_id UUID NOT NULL REFERENCES recipes(id) ON DELETE CASCADE,
product_id UUID REFERENCES products(id) ON DELETE SET NULL,
name TEXT NOT NULL,
amount DECIMAL(10,2) NOT NULL DEFAULT 0,
unit_code VARCHAR(20),
is_optional BOOLEAN NOT NULL DEFAULT false,
sort_order SMALLINT NOT NULL DEFAULT 0
);
CREATE INDEX idx_recipe_ingredients_recipe_id ON recipe_ingredients (recipe_id);
CREATE INDEX idx_recipe_products_recipe_id ON recipe_products (recipe_id);
CREATE TABLE recipe_ingredient_translations (
ri_id UUID NOT NULL REFERENCES recipe_ingredients(id) ON DELETE CASCADE,
CREATE TABLE recipe_product_translations (
ri_id UUID NOT NULL REFERENCES recipe_products(id) ON DELETE CASCADE,
lang VARCHAR(10) NOT NULL,
name TEXT NOT NULL,
PRIMARY KEY (ri_id, lang)
@@ -322,29 +314,29 @@ CREATE TABLE recipe_step_translations (
);
-- ---------------------------------------------------------------------------
-- products (user fridge / pantry items)
-- user_products (user fridge / pantry items)
-- ---------------------------------------------------------------------------
CREATE TABLE products (
id UUID PRIMARY KEY DEFAULT uuid_generate_v7(),
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
primary_ingredient_id UUID REFERENCES ingredients(id),
name TEXT NOT NULL,
quantity DECIMAL(10,2) NOT NULL DEFAULT 1,
unit TEXT NOT NULL DEFAULT 'pcs' REFERENCES units(code),
category TEXT,
storage_days INT NOT NULL DEFAULT 7,
added_at TIMESTAMPTZ NOT NULL DEFAULT now()
CREATE TABLE user_products (
id UUID PRIMARY KEY DEFAULT uuid_generate_v7(),
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
primary_product_id UUID REFERENCES products(id),
name TEXT NOT NULL,
quantity DECIMAL(10,2) NOT NULL DEFAULT 1,
unit TEXT NOT NULL DEFAULT 'pcs' REFERENCES units(code),
category TEXT,
storage_days INT NOT NULL DEFAULT 7,
added_at TIMESTAMPTZ NOT NULL DEFAULT now()
);
CREATE INDEX idx_products_user_id ON products (user_id);
CREATE INDEX idx_user_products_user_id ON user_products (user_id);
-- ---------------------------------------------------------------------------
-- product_ingredients (M2M: composite product ↔ ingredients)
-- user_product_components (M2M: composite user product ↔ catalog products)
-- ---------------------------------------------------------------------------
CREATE TABLE product_ingredients (
product_id UUID NOT NULL REFERENCES products(id) ON DELETE CASCADE,
ingredient_id UUID NOT NULL REFERENCES ingredients(id) ON DELETE CASCADE,
CREATE TABLE user_product_components (
user_product_id UUID NOT NULL REFERENCES user_products(id) ON DELETE CASCADE,
product_id UUID NOT NULL REFERENCES products(id) ON DELETE CASCADE,
amount_per_100g DECIMAL(10,2),
PRIMARY KEY (product_id, ingredient_id)
PRIMARY KEY (user_product_id, product_id)
);
-- ---------------------------------------------------------------------------
@@ -394,21 +386,21 @@ CREATE TABLE shopping_lists (
-- dish_recognition_jobs
-- ---------------------------------------------------------------------------
CREATE TABLE dish_recognition_jobs (
id UUID PRIMARY KEY DEFAULT uuid_generate_v7(),
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
user_plan TEXT NOT NULL,
image_base64 TEXT NOT NULL,
mime_type TEXT NOT NULL DEFAULT 'image/jpeg',
lang TEXT NOT NULL DEFAULT 'en',
target_date DATE,
id UUID PRIMARY KEY DEFAULT uuid_generate_v7(),
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
user_plan TEXT NOT NULL,
image_base64 TEXT NOT NULL,
mime_type TEXT NOT NULL DEFAULT 'image/jpeg',
lang TEXT NOT NULL DEFAULT 'en',
target_date DATE,
target_meal_type TEXT,
status TEXT NOT NULL DEFAULT 'pending',
status TEXT NOT NULL DEFAULT 'pending',
-- pending | processing | done | failed
result JSONB,
error TEXT,
created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
started_at TIMESTAMPTZ,
completed_at TIMESTAMPTZ
result JSONB,
error TEXT,
created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
started_at TIMESTAMPTZ,
completed_at TIMESTAMPTZ
);
CREATE INDEX idx_dish_recognition_jobs_user
ON dish_recognition_jobs (user_id, created_at DESC);
@@ -420,16 +412,18 @@ CREATE INDEX idx_dish_recognition_jobs_pending
-- ---------------------------------------------------------------------------
CREATE TABLE meal_diary (
id UUID PRIMARY KEY DEFAULT uuid_generate_v7(),
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
date DATE NOT NULL,
meal_type TEXT NOT NULL,
portions DECIMAL(5,2) NOT NULL DEFAULT 1,
source TEXT NOT NULL DEFAULT 'manual',
dish_id UUID NOT NULL REFERENCES dishes(id) ON DELETE RESTRICT,
recipe_id UUID REFERENCES recipes(id) ON DELETE SET NULL,
dish_id UUID REFERENCES dishes(id) ON DELETE RESTRICT,
product_id UUID REFERENCES products(id) ON DELETE RESTRICT,
recipe_id UUID REFERENCES recipes(id) ON DELETE SET NULL,
portion_g DECIMAL(10,2),
job_id UUID REFERENCES dish_recognition_jobs(id) ON DELETE SET NULL,
created_at TIMESTAMPTZ NOT NULL DEFAULT now()
job_id UUID REFERENCES dish_recognition_jobs(id) ON DELETE SET NULL,
created_at TIMESTAMPTZ NOT NULL DEFAULT now(),
CONSTRAINT chk_meal_entry_source CHECK (num_nonnulls(dish_id, product_id) = 1)
);
CREATE INDEX idx_meal_diary_user_date ON meal_diary (user_id, date);
CREATE INDEX idx_meal_diary_job_id ON meal_diary (job_id) WHERE job_id IS NOT NULL;
@@ -471,18 +465,18 @@ INSERT INTO unit_translations (unit_code, lang, name) VALUES
('pack', 'ru', 'уп');
-- ---------------------------------------------------------------------------
-- Seed data: ingredient_categories + ingredient_category_translations
-- Seed data: product_categories + product_category_translations
-- ---------------------------------------------------------------------------
INSERT INTO ingredient_categories (slug, sort_order) VALUES
('dairy', 1),
('meat', 2),
('produce', 3),
('bakery', 4),
('frozen', 5),
('beverages', 6),
('other', 7);
INSERT INTO product_categories (slug, name, sort_order) VALUES
('dairy', 'Dairy', 1),
('meat', 'Meat', 2),
('produce', 'Produce', 3),
('bakery', 'Bakery', 4),
('frozen', 'Frozen', 5),
('beverages', 'Beverages', 6),
('other', 'Other', 7);
INSERT INTO ingredient_category_translations (category_slug, lang, name) VALUES
INSERT INTO product_category_translations (product_category_slug, lang, name) VALUES
('dairy', 'ru', 'Молочные продукты'),
('meat', 'ru', 'Мясо и птица'),
('produce', 'ru', 'Овощи и фрукты'),
@@ -617,12 +611,12 @@ DROP TABLE IF EXISTS shopping_lists;
DROP TABLE IF EXISTS menu_items;
DROP TABLE IF EXISTS menu_plans;
DROP TABLE IF EXISTS user_saved_recipes;
DROP TABLE IF EXISTS product_ingredients;
DROP TABLE IF EXISTS products;
DROP TABLE IF EXISTS user_product_components;
DROP TABLE IF EXISTS user_products;
DROP TABLE IF EXISTS recipe_step_translations;
DROP TABLE IF EXISTS recipe_steps;
DROP TABLE IF EXISTS recipe_ingredient_translations;
DROP TABLE IF EXISTS recipe_ingredients;
DROP TABLE IF EXISTS recipe_product_translations;
DROP TABLE IF EXISTS recipe_products;
DROP TABLE IF EXISTS recipe_translations;
DROP TABLE IF EXISTS recipes;
DROP TABLE IF EXISTS dish_tags;
@@ -634,11 +628,10 @@ DROP TABLE IF EXISTS tag_translations;
DROP TABLE IF EXISTS tags;
DROP TABLE IF EXISTS cuisine_translations;
DROP TABLE IF EXISTS cuisines;
DROP TABLE IF EXISTS ingredient_aliases;
DROP TABLE IF EXISTS ingredient_translations;
DROP TABLE IF EXISTS ingredients;
DROP TABLE IF EXISTS ingredient_category_translations;
DROP TABLE IF EXISTS ingredient_categories;
DROP TABLE IF EXISTS product_aliases;
DROP TABLE IF EXISTS products;
DROP TABLE IF EXISTS product_category_translations;
DROP TABLE IF EXISTS product_categories;
DROP TABLE IF EXISTS unit_translations;
DROP TABLE IF EXISTS units;
DROP TABLE IF EXISTS languages;

View File

@@ -140,7 +140,7 @@ func TestCreate_Success(t *testing.T) {
Name: "Oatmeal",
Portions: 1,
Source: "manual",
DishID: "dish-1",
DishID: strPtr("dish-1"),
CreatedAt: time.Now(),
}, nil
},
@@ -207,3 +207,5 @@ func TestDelete_Success(t *testing.T) {
t.Errorf("expected 204, got %d", recorder.Code)
}
}
func strPtr(s string) *string { return &s }

View File

@@ -1,25 +1,44 @@
package ingredient_test
package product_catalog_test
import (
"bytes"
"context"
"encoding/json"
"errors"
"net/http"
"net/http/httptest"
"testing"
"github.com/food-ai/backend/internal/domain/ingredient"
"github.com/food-ai/backend/internal/domain/product"
"github.com/food-ai/backend/internal/infra/middleware"
"github.com/go-chi/chi/v5"
)
// mockIngredientSearcher is an inline mock for ingredient.IngredientSearcher.
type mockIngredientSearcher struct {
searchFn func(ctx context.Context, query string, limit int) ([]*ingredient.IngredientMapping, error)
// mockProductSearcher is an inline mock for product.ProductSearcher (catalog search only).
type mockProductSearcher struct {
searchFn func(ctx context.Context, query string, limit int) ([]*product.Product, error)
}
func (m *mockIngredientSearcher) Search(ctx context.Context, query string, limit int) ([]*ingredient.IngredientMapping, error) {
return m.searchFn(ctx, query, limit)
func (m *mockProductSearcher) Search(ctx context.Context, query string, limit int) ([]*product.Product, error) {
if m.searchFn != nil {
return m.searchFn(ctx, query, limit)
}
return []*product.Product{}, nil
}
func (m *mockProductSearcher) GetByBarcode(_ context.Context, _ string) (*product.Product, error) {
return nil, errors.New("not implemented")
}
func (m *mockProductSearcher) UpsertByBarcode(_ context.Context, catalogProduct *product.Product) (*product.Product, error) {
return catalogProduct, nil
}
// mockOpenFoodFacts is an inline mock for product.OpenFoodFactsClient (unused in search tests).
type mockOpenFoodFacts struct{}
func (m *mockOpenFoodFacts) Fetch(_ context.Context, _ string) (*product.Product, error) {
return nil, errors.New("not implemented")
}
type alwaysAuthValidator struct{ userID string }
@@ -28,10 +47,10 @@ func (v *alwaysAuthValidator) ValidateAccessToken(_ string) (*middleware.TokenCl
return &middleware.TokenClaims{UserID: v.userID}, nil
}
func buildRouter(handler *ingredient.Handler) *chi.Mux {
func buildRouter(handler *product.Handler) *chi.Mux {
router := chi.NewRouter()
router.Use(middleware.Auth(&alwaysAuthValidator{userID: "user-1"}))
router.Get("/ingredients/search", handler.Search)
router.Get("/products/search", handler.Search)
return router
}
@@ -43,11 +62,11 @@ func authorizedRequest(target string) *http.Request {
func TestSearch_EmptyQuery_ReturnsEmptyArray(t *testing.T) {
// When q is empty, the handler returns [] without calling the repository.
handler := ingredient.NewHandler(&mockIngredientSearcher{})
handler := product.NewHandler(&mockProductSearcher{}, &mockOpenFoodFacts{})
router := buildRouter(handler)
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest("/ingredients/search"))
router.ServeHTTP(recorder, authorizedRequest("/products/search"))
if recorder.Code != http.StatusOK {
t.Errorf("expected 200, got %d", recorder.Code)
@@ -60,17 +79,17 @@ func TestSearch_EmptyQuery_ReturnsEmptyArray(t *testing.T) {
func TestSearch_LimitTooLarge_UsesDefault(t *testing.T) {
// When limit > 50, the handler ignores it and uses default 10.
calledLimit := 0
mockRepo := &mockIngredientSearcher{
searchFn: func(ctx context.Context, query string, limit int) ([]*ingredient.IngredientMapping, error) {
mockRepo := &mockProductSearcher{
searchFn: func(ctx context.Context, query string, limit int) ([]*product.Product, error) {
calledLimit = limit
return []*ingredient.IngredientMapping{}, nil
return []*product.Product{}, nil
},
}
handler := ingredient.NewHandler(mockRepo)
handler := product.NewHandler(mockRepo, &mockOpenFoodFacts{})
router := buildRouter(handler)
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest("/ingredients/search?q=apple&limit=100"))
router.ServeHTTP(recorder, authorizedRequest("/products/search?q=apple&limit=100"))
if recorder.Code != http.StatusOK {
t.Errorf("expected 200, got %d", recorder.Code)
@@ -83,17 +102,17 @@ func TestSearch_LimitTooLarge_UsesDefault(t *testing.T) {
func TestSearch_DefaultLimit(t *testing.T) {
// When no limit is supplied, the handler uses default 10.
calledLimit := 0
mockRepo := &mockIngredientSearcher{
searchFn: func(ctx context.Context, query string, limit int) ([]*ingredient.IngredientMapping, error) {
mockRepo := &mockProductSearcher{
searchFn: func(ctx context.Context, query string, limit int) ([]*product.Product, error) {
calledLimit = limit
return []*ingredient.IngredientMapping{}, nil
return []*product.Product{}, nil
},
}
handler := ingredient.NewHandler(mockRepo)
handler := product.NewHandler(mockRepo, &mockOpenFoodFacts{})
router := buildRouter(handler)
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest("/ingredients/search?q=apple"))
router.ServeHTTP(recorder, authorizedRequest("/products/search?q=apple"))
if recorder.Code != http.StatusOK {
t.Errorf("expected 200, got %d", recorder.Code)
@@ -106,17 +125,17 @@ func TestSearch_DefaultLimit(t *testing.T) {
func TestSearch_ValidLimit(t *testing.T) {
// limit=25 is within range and should be forwarded as-is.
calledLimit := 0
mockRepo := &mockIngredientSearcher{
searchFn: func(ctx context.Context, query string, limit int) ([]*ingredient.IngredientMapping, error) {
mockRepo := &mockProductSearcher{
searchFn: func(ctx context.Context, query string, limit int) ([]*product.Product, error) {
calledLimit = limit
return []*ingredient.IngredientMapping{}, nil
return []*product.Product{}, nil
},
}
handler := ingredient.NewHandler(mockRepo)
handler := product.NewHandler(mockRepo, &mockOpenFoodFacts{})
router := buildRouter(handler)
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest("/ingredients/search?q=apple&limit=25"))
router.ServeHTTP(recorder, authorizedRequest("/products/search?q=apple&limit=25"))
if recorder.Code != http.StatusOK {
t.Errorf("expected 200, got %d", recorder.Code)
@@ -127,31 +146,31 @@ func TestSearch_ValidLimit(t *testing.T) {
}
func TestSearch_Success(t *testing.T) {
mockRepo := &mockIngredientSearcher{
searchFn: func(ctx context.Context, query string, limit int) ([]*ingredient.IngredientMapping, error) {
return []*ingredient.IngredientMapping{
{ID: "ing-1", CanonicalName: "apple"},
mockRepo := &mockProductSearcher{
searchFn: func(ctx context.Context, query string, limit int) ([]*product.Product, error) {
return []*product.Product{
{ID: "prod-1", CanonicalName: "apple"},
}, nil
},
}
handler := ingredient.NewHandler(mockRepo)
handler := product.NewHandler(mockRepo, &mockOpenFoodFacts{})
router := buildRouter(handler)
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest("/ingredients/search?q=apple"))
router.ServeHTTP(recorder, authorizedRequest("/products/search?q=apple"))
if recorder.Code != http.StatusOK {
t.Errorf("expected 200, got %d", recorder.Code)
}
var mappings []ingredient.IngredientMapping
if decodeError := json.NewDecoder(recorder.Body).Decode(&mappings); decodeError != nil {
var products []product.Product
if decodeError := json.NewDecoder(recorder.Body).Decode(&products); decodeError != nil {
t.Fatalf("decode response: %v", decodeError)
}
if len(mappings) != 1 {
t.Errorf("expected 1 result, got %d", len(mappings))
if len(products) != 1 {
t.Errorf("expected 1 result, got %d", len(products))
}
if mappings[0].CanonicalName != "apple" {
t.Errorf("expected canonical_name=apple, got %q", mappings[0].CanonicalName)
if products[0].CanonicalName != "apple" {
t.Errorf("expected canonical_name=apple, got %q", products[0].CanonicalName)
}
}

View File

@@ -1,33 +1,33 @@
//go:build integration
package ingredient_test
package product_catalog_test
import (
"context"
"testing"
"github.com/food-ai/backend/internal/domain/ingredient"
"github.com/food-ai/backend/internal/domain/product"
"github.com/food-ai/backend/internal/infra/locale"
"github.com/food-ai/backend/internal/testutil"
)
func TestIngredientRepository_Upsert_Insert(t *testing.T) {
func TestProductRepository_Upsert_Insert(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := ingredient.NewRepository(pool)
repo := product.NewRepository(pool)
ctx := context.Background()
cat := "produce"
unit := "g"
cal := 52.0
mapping := &ingredient.IngredientMapping{
catalogProduct := &product.Product{
CanonicalName: "apple",
Category: &cat,
DefaultUnit: &unit,
CaloriesPer100g: &cal,
}
got, upsertError := repo.Upsert(ctx, mapping)
got, upsertError := repo.Upsert(ctx, catalogProduct)
if upsertError != nil {
t.Fatalf("upsert: %v", upsertError)
}
@@ -42,15 +42,15 @@ func TestIngredientRepository_Upsert_Insert(t *testing.T) {
}
}
func TestIngredientRepository_Upsert_ConflictUpdates(t *testing.T) {
func TestProductRepository_Upsert_ConflictUpdates(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := ingredient.NewRepository(pool)
repo := product.NewRepository(pool)
ctx := context.Background()
cat := "produce"
unit := "g"
first := &ingredient.IngredientMapping{
first := &product.Product{
CanonicalName: "banana",
Category: &cat,
DefaultUnit: &unit,
@@ -61,7 +61,7 @@ func TestIngredientRepository_Upsert_ConflictUpdates(t *testing.T) {
}
cal := 89.0
second := &ingredient.IngredientMapping{
second := &product.Product{
CanonicalName: "banana",
Category: &cat,
DefaultUnit: &unit,
@@ -83,15 +83,15 @@ func TestIngredientRepository_Upsert_ConflictUpdates(t *testing.T) {
}
}
func TestIngredientRepository_GetByID_Found(t *testing.T) {
func TestProductRepository_GetByID_Found(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := ingredient.NewRepository(pool)
repo := product.NewRepository(pool)
ctx := context.Background()
cat := "dairy"
unit := "g"
saved, upsertError := repo.Upsert(ctx, &ingredient.IngredientMapping{
saved, upsertError := repo.Upsert(ctx, &product.Product{
CanonicalName: "cheese",
Category: &cat,
DefaultUnit: &unit,
@@ -112,9 +112,9 @@ func TestIngredientRepository_GetByID_Found(t *testing.T) {
}
}
func TestIngredientRepository_GetByID_NotFound(t *testing.T) {
func TestProductRepository_GetByID_NotFound(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := ingredient.NewRepository(pool)
repo := product.NewRepository(pool)
ctx := context.Background()
got, getError := repo.GetByID(ctx, "00000000-0000-0000-0000-000000000000")
@@ -126,105 +126,17 @@ func TestIngredientRepository_GetByID_NotFound(t *testing.T) {
}
}
func TestIngredientRepository_ListMissingTranslation(t *testing.T) {
func TestProductRepository_UpsertAliases(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := ingredient.NewRepository(pool)
repo := product.NewRepository(pool)
ctx := context.Background()
cat := "produce"
unit := "g"
// Insert 3 without any translation.
for _, name := range []string{"carrot", "onion", "garlic"} {
_, upsertError := repo.Upsert(ctx, &ingredient.IngredientMapping{
CanonicalName: name,
Category: &cat,
DefaultUnit: &unit,
})
if upsertError != nil {
t.Fatalf("upsert %s: %v", name, upsertError)
}
}
// Insert 1 and add a translation — should not appear in the result.
saved, upsertError := repo.Upsert(ctx, &ingredient.IngredientMapping{
CanonicalName: "tomato",
Category: &cat,
DefaultUnit: &unit,
})
if upsertError != nil {
t.Fatalf("upsert tomato: %v", upsertError)
}
if translationError := repo.UpsertTranslation(ctx, saved.ID, "ru", "помидор"); translationError != nil {
t.Fatalf("upsert translation: %v", translationError)
}
missing, listError := repo.ListMissingTranslation(ctx, "ru", 10, 0)
if listError != nil {
t.Fatalf("list missing translation: %v", listError)
}
for _, mapping := range missing {
if mapping.CanonicalName == "tomato" {
t.Error("translated ingredient should not appear in ListMissingTranslation")
}
}
if len(missing) < 3 {
t.Errorf("expected at least 3 untranslated, got %d", len(missing))
}
}
func TestIngredientRepository_UpsertTranslation(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := ingredient.NewRepository(pool)
ctx := context.Background()
cat := "meat"
unit := "g"
saved, upsertError := repo.Upsert(ctx, &ingredient.IngredientMapping{
CanonicalName: "chicken_breast",
Category: &cat,
DefaultUnit: &unit,
})
if upsertError != nil {
t.Fatalf("upsert: %v", upsertError)
}
if translationError := repo.UpsertTranslation(ctx, saved.ID, "ru", "куриная грудка"); translationError != nil {
t.Fatalf("upsert translation: %v", translationError)
}
// Retrieve with Russian context — CanonicalName should be the Russian name.
russianContext := locale.WithLang(ctx, "ru")
got, getError := repo.GetByID(russianContext, saved.ID)
if getError != nil {
t.Fatalf("get by id: %v", getError)
}
if got.CanonicalName != "куриная грудка" {
t.Errorf("expected CanonicalName='куриная грудка', got %q", got.CanonicalName)
}
// Retrieve with English context (default) — CanonicalName should be the English name.
englishContext := locale.WithLang(ctx, "en")
gotEnglish, getEnglishError := repo.GetByID(englishContext, saved.ID)
if getEnglishError != nil {
t.Fatalf("get by id (en): %v", getEnglishError)
}
if gotEnglish.CanonicalName != "chicken_breast" {
t.Errorf("expected English CanonicalName='chicken_breast', got %q", gotEnglish.CanonicalName)
}
}
func TestIngredientRepository_UpsertAliases(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := ingredient.NewRepository(pool)
ctx := context.Background()
cat := "produce"
unit := "g"
saved, upsertError := repo.Upsert(ctx, &ingredient.IngredientMapping{
saved, upsertError := repo.Upsert(ctx, &product.Product{
CanonicalName: "apple_test_aliases",
Category: &cat,
DefaultUnit: &unit,
@@ -238,12 +150,10 @@ func TestIngredientRepository_UpsertAliases(t *testing.T) {
t.Fatalf("upsert aliases: %v", aliasError)
}
// Idempotent — second call should not fail.
if aliasError := repo.UpsertAliases(ctx, saved.ID, "en", aliases); aliasError != nil {
t.Fatalf("second upsert aliases: %v", aliasError)
}
// Retrieve with English context — aliases should appear.
englishContext := locale.WithLang(ctx, "en")
got, getError := repo.GetByID(englishContext, saved.ID)
if getError != nil {
@@ -254,15 +164,14 @@ func TestIngredientRepository_UpsertAliases(t *testing.T) {
}
}
func TestIngredientRepository_UpsertCategoryTranslation(t *testing.T) {
func TestProductRepository_UpsertCategoryTranslation(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := ingredient.NewRepository(pool)
repo := product.NewRepository(pool)
ctx := context.Background()
// Upsert an ingredient with a known category.
cat := "dairy"
unit := "g"
saved, upsertError := repo.Upsert(ctx, &ingredient.IngredientMapping{
saved, upsertError := repo.Upsert(ctx, &product.Product{
CanonicalName: "milk_test_category",
Category: &cat,
DefaultUnit: &unit,
@@ -271,8 +180,6 @@ func TestIngredientRepository_UpsertCategoryTranslation(t *testing.T) {
t.Fatalf("upsert: %v", upsertError)
}
// The migration already inserted Russian translations for known categories.
// Retrieve with Russian context — CategoryName should be set.
russianContext := locale.WithLang(ctx, "ru")
got, getError := repo.GetByID(russianContext, saved.ID)
if getError != nil {
@@ -282,7 +189,6 @@ func TestIngredientRepository_UpsertCategoryTranslation(t *testing.T) {
t.Error("expected non-empty CategoryName for 'dairy' in Russian")
}
// Upsert a new translation and verify it is returned.
if translationError := repo.UpsertCategoryTranslation(ctx, "dairy", "de", "Milchprodukte"); translationError != nil {
t.Fatalf("upsert category translation: %v", translationError)
}

View File

@@ -1,4 +1,4 @@
package product_test
package userproduct_test
import (
"bytes"
@@ -9,8 +9,8 @@ import (
"testing"
"time"
"github.com/food-ai/backend/internal/domain/product"
productmocks "github.com/food-ai/backend/internal/domain/product/mocks"
"github.com/food-ai/backend/internal/domain/userproduct"
userproductmocks "github.com/food-ai/backend/internal/domain/userproduct/mocks"
"github.com/food-ai/backend/internal/infra/middleware"
"github.com/go-chi/chi/v5"
)
@@ -21,14 +21,14 @@ func (v *alwaysAuthValidator) ValidateAccessToken(_ string) (*middleware.TokenCl
return &middleware.TokenClaims{UserID: v.userID}, nil
}
func buildRouter(handler *product.Handler, userID string) *chi.Mux {
func buildRouter(handler *userproduct.Handler, userID string) *chi.Mux {
router := chi.NewRouter()
router.Use(middleware.Auth(&alwaysAuthValidator{userID: userID}))
router.Get("/products", handler.List)
router.Post("/products", handler.Create)
router.Post("/products/batch", handler.BatchCreate)
router.Put("/products/{id}", handler.Update)
router.Delete("/products/{id}", handler.Delete)
router.Get("/user-products", handler.List)
router.Post("/user-products", handler.Create)
router.Post("/user-products/batch", handler.BatchCreate)
router.Put("/user-products/{id}", handler.Update)
router.Delete("/user-products/{id}", handler.Delete)
return router
}
@@ -39,8 +39,8 @@ func authorizedRequest(method, target string, body []byte) *http.Request {
return request
}
func makeProduct(name string) *product.Product {
return &product.Product{
func makeUserProduct(name string) *userproduct.UserProduct {
return &userproduct.UserProduct{
ID: "prod-1",
UserID: "user-1",
Name: name,
@@ -54,16 +54,16 @@ func makeProduct(name string) *product.Product {
}
func TestList_Success(t *testing.T) {
mockRepo := &productmocks.MockProductRepository{
ListFn: func(ctx context.Context, userID string) ([]*product.Product, error) {
return []*product.Product{makeProduct("Milk")}, nil
mockRepo := &userproductmocks.MockUserProductRepository{
ListFn: func(ctx context.Context, userID string) ([]*userproduct.UserProduct, error) {
return []*userproduct.UserProduct{makeUserProduct("Milk")}, nil
},
}
handler := product.NewHandler(mockRepo)
handler := userproduct.NewHandler(mockRepo)
router := buildRouter(handler, "user-1")
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest(http.MethodGet, "/products", nil))
router.ServeHTTP(recorder, authorizedRequest(http.MethodGet, "/user-products", nil))
if recorder.Code != http.StatusOK {
t.Errorf("expected 200, got %d", recorder.Code)
@@ -71,12 +71,12 @@ func TestList_Success(t *testing.T) {
}
func TestCreate_MissingName(t *testing.T) {
handler := product.NewHandler(&productmocks.MockProductRepository{})
handler := userproduct.NewHandler(&userproductmocks.MockUserProductRepository{})
router := buildRouter(handler, "user-1")
body, _ := json.Marshal(map[string]any{"quantity": 1})
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest(http.MethodPost, "/products", body))
router.ServeHTTP(recorder, authorizedRequest(http.MethodPost, "/user-products", body))
if recorder.Code != http.StatusBadRequest {
t.Errorf("expected 400, got %d", recorder.Code)
@@ -84,17 +84,17 @@ func TestCreate_MissingName(t *testing.T) {
}
func TestCreate_Success(t *testing.T) {
mockRepo := &productmocks.MockProductRepository{
CreateFn: func(ctx context.Context, userID string, req product.CreateRequest) (*product.Product, error) {
return makeProduct(req.Name), nil
mockRepo := &userproductmocks.MockUserProductRepository{
CreateFn: func(ctx context.Context, userID string, req userproduct.CreateRequest) (*userproduct.UserProduct, error) {
return makeUserProduct(req.Name), nil
},
}
handler := product.NewHandler(mockRepo)
handler := userproduct.NewHandler(mockRepo)
router := buildRouter(handler, "user-1")
body, _ := json.Marshal(product.CreateRequest{Name: "Milk", Quantity: 1, Unit: "L"})
body, _ := json.Marshal(userproduct.CreateRequest{Name: "Milk", Quantity: 1, Unit: "L"})
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest(http.MethodPost, "/products", body))
router.ServeHTTP(recorder, authorizedRequest(http.MethodPost, "/user-products", body))
if recorder.Code != http.StatusCreated {
t.Errorf("expected 201, got %d", recorder.Code)
@@ -102,24 +102,24 @@ func TestCreate_Success(t *testing.T) {
}
func TestBatchCreate_Success(t *testing.T) {
mockRepo := &productmocks.MockProductRepository{
BatchCreateFn: func(ctx context.Context, userID string, items []product.CreateRequest) ([]*product.Product, error) {
result := make([]*product.Product, len(items))
mockRepo := &userproductmocks.MockUserProductRepository{
BatchCreateFn: func(ctx context.Context, userID string, items []userproduct.CreateRequest) ([]*userproduct.UserProduct, error) {
result := make([]*userproduct.UserProduct, len(items))
for index, item := range items {
result[index] = makeProduct(item.Name)
result[index] = makeUserProduct(item.Name)
}
return result, nil
},
}
handler := product.NewHandler(mockRepo)
handler := userproduct.NewHandler(mockRepo)
router := buildRouter(handler, "user-1")
body, _ := json.Marshal([]product.CreateRequest{
body, _ := json.Marshal([]userproduct.CreateRequest{
{Name: "Milk", Quantity: 1, Unit: "L"},
{Name: "Eggs", Quantity: 12, Unit: "pcs"},
})
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest(http.MethodPost, "/products/batch", body))
router.ServeHTTP(recorder, authorizedRequest(http.MethodPost, "/user-products/batch", body))
if recorder.Code != http.StatusCreated {
t.Errorf("expected 201, got %d", recorder.Code)
@@ -127,18 +127,18 @@ func TestBatchCreate_Success(t *testing.T) {
}
func TestUpdate_NotFound(t *testing.T) {
mockRepo := &productmocks.MockProductRepository{
UpdateFn: func(ctx context.Context, id, userID string, req product.UpdateRequest) (*product.Product, error) {
return nil, product.ErrNotFound
mockRepo := &userproductmocks.MockUserProductRepository{
UpdateFn: func(ctx context.Context, id, userID string, req userproduct.UpdateRequest) (*userproduct.UserProduct, error) {
return nil, userproduct.ErrNotFound
},
}
handler := product.NewHandler(mockRepo)
handler := userproduct.NewHandler(mockRepo)
router := buildRouter(handler, "user-1")
namePtr := "NewName"
body, _ := json.Marshal(product.UpdateRequest{Name: &namePtr})
body, _ := json.Marshal(userproduct.UpdateRequest{Name: &namePtr})
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest(http.MethodPut, "/products/nonexistent", body))
router.ServeHTTP(recorder, authorizedRequest(http.MethodPut, "/user-products/nonexistent", body))
if recorder.Code != http.StatusNotFound {
t.Errorf("expected 404, got %d", recorder.Code)
@@ -146,19 +146,19 @@ func TestUpdate_NotFound(t *testing.T) {
}
func TestUpdate_Success(t *testing.T) {
mockRepo := &productmocks.MockProductRepository{
UpdateFn: func(ctx context.Context, id, userID string, req product.UpdateRequest) (*product.Product, error) {
updated := makeProduct(*req.Name)
mockRepo := &userproductmocks.MockUserProductRepository{
UpdateFn: func(ctx context.Context, id, userID string, req userproduct.UpdateRequest) (*userproduct.UserProduct, error) {
updated := makeUserProduct(*req.Name)
return updated, nil
},
}
handler := product.NewHandler(mockRepo)
handler := userproduct.NewHandler(mockRepo)
router := buildRouter(handler, "user-1")
namePtr := "Oat Milk"
body, _ := json.Marshal(product.UpdateRequest{Name: &namePtr})
body, _ := json.Marshal(userproduct.UpdateRequest{Name: &namePtr})
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest(http.MethodPut, "/products/prod-1", body))
router.ServeHTTP(recorder, authorizedRequest(http.MethodPut, "/user-products/prod-1", body))
if recorder.Code != http.StatusOK {
t.Errorf("expected 200, got %d", recorder.Code)
@@ -166,16 +166,16 @@ func TestUpdate_Success(t *testing.T) {
}
func TestDelete_NotFound(t *testing.T) {
mockRepo := &productmocks.MockProductRepository{
mockRepo := &userproductmocks.MockUserProductRepository{
DeleteFn: func(ctx context.Context, id, userID string) error {
return product.ErrNotFound
return userproduct.ErrNotFound
},
}
handler := product.NewHandler(mockRepo)
handler := userproduct.NewHandler(mockRepo)
router := buildRouter(handler, "user-1")
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest(http.MethodDelete, "/products/nonexistent", nil))
router.ServeHTTP(recorder, authorizedRequest(http.MethodDelete, "/user-products/nonexistent", nil))
if recorder.Code != http.StatusNotFound {
t.Errorf("expected 404, got %d", recorder.Code)
@@ -183,16 +183,16 @@ func TestDelete_NotFound(t *testing.T) {
}
func TestDelete_Success(t *testing.T) {
mockRepo := &productmocks.MockProductRepository{
mockRepo := &userproductmocks.MockUserProductRepository{
DeleteFn: func(ctx context.Context, id, userID string) error {
return nil
},
}
handler := product.NewHandler(mockRepo)
handler := userproduct.NewHandler(mockRepo)
router := buildRouter(handler, "user-1")
recorder := httptest.NewRecorder()
router.ServeHTTP(recorder, authorizedRequest(http.MethodDelete, "/products/prod-1", nil))
router.ServeHTTP(recorder, authorizedRequest(http.MethodDelete, "/user-products/prod-1", nil))
if recorder.Code != http.StatusNoContent {
t.Errorf("expected 204, got %d", recorder.Code)

View File

@@ -1,29 +1,29 @@
//go:build integration
package product_test
package userproduct_test
import (
"context"
"testing"
"github.com/food-ai/backend/internal/domain/product"
"github.com/food-ai/backend/internal/domain/userproduct"
"github.com/food-ai/backend/internal/testutil"
)
func TestProductRepository_Create_Defaults(t *testing.T) {
func TestUserProductRepository_Create_Defaults(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := product.NewRepository(pool)
repo := userproduct.NewRepository(pool)
requestContext := context.Background()
// storage_days=0 → 7; unit="" → "pcs"; quantity=0 → 1
created, createError := repo.Create(requestContext, "test-user", product.CreateRequest{
created, createError := repo.Create(requestContext, "test-user", userproduct.CreateRequest{
Name: "Milk",
StorageDays: 0,
Unit: "",
Quantity: 0,
})
if createError != nil {
t.Fatalf("create product: %v", createError)
t.Fatalf("create user product: %v", createError)
}
if created.StorageDays != 7 {
t.Errorf("expected storage_days=7, got %d", created.StorageDays)
@@ -36,25 +36,24 @@ func TestProductRepository_Create_Defaults(t *testing.T) {
}
}
func TestProductRepository_List_OrderByExpiry(t *testing.T) {
func TestUserProductRepository_List_OrderByExpiry(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := product.NewRepository(pool)
repo := userproduct.NewRepository(pool)
requestContext := context.Background()
userID := "list-order-user"
// Create product with longer expiry first, shorter expiry second.
_, createError := repo.Create(requestContext, userID, product.CreateRequest{Name: "Milk", StorageDays: 14})
_, createError := repo.Create(requestContext, userID, userproduct.CreateRequest{Name: "Milk", StorageDays: 14})
if createError != nil {
t.Fatalf("create Milk: %v", createError)
}
_, createError = repo.Create(requestContext, userID, product.CreateRequest{Name: "Butter", StorageDays: 3})
_, createError = repo.Create(requestContext, userID, userproduct.CreateRequest{Name: "Butter", StorageDays: 3})
if createError != nil {
t.Fatalf("create Butter: %v", createError)
}
products, listError := repo.List(requestContext, userID)
if listError != nil {
t.Fatalf("list products: %v", listError)
t.Fatalf("list user products: %v", listError)
}
if len(products) != 2 {
t.Fatalf("expected 2 products, got %d", len(products))
@@ -65,12 +64,12 @@ func TestProductRepository_List_OrderByExpiry(t *testing.T) {
}
}
func TestProductRepository_BatchCreate(t *testing.T) {
func TestUserProductRepository_BatchCreate(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := product.NewRepository(pool)
repo := userproduct.NewRepository(pool)
requestContext := context.Background()
products, batchError := repo.BatchCreate(requestContext, "batch-user", []product.CreateRequest{
products, batchError := repo.BatchCreate(requestContext, "batch-user", []userproduct.CreateRequest{
{Name: "Eggs", Quantity: 12, Unit: "pcs"},
{Name: "Flour", Quantity: 500, Unit: "g"},
})
@@ -82,46 +81,46 @@ func TestProductRepository_BatchCreate(t *testing.T) {
}
}
func TestProductRepository_Update_NotFound(t *testing.T) {
func TestUserProductRepository_Update_NotFound(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := product.NewRepository(pool)
repo := userproduct.NewRepository(pool)
requestContext := context.Background()
newName := "Ghost"
_, updateError := repo.Update(requestContext, "00000000-0000-0000-0000-000000000000", "any-user",
product.UpdateRequest{Name: &newName})
if updateError != product.ErrNotFound {
userproduct.UpdateRequest{Name: &newName})
if updateError != userproduct.ErrNotFound {
t.Errorf("expected ErrNotFound, got %v", updateError)
}
}
func TestProductRepository_Delete_WrongUser(t *testing.T) {
func TestUserProductRepository_Delete_WrongUser(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := product.NewRepository(pool)
repo := userproduct.NewRepository(pool)
requestContext := context.Background()
created, createError := repo.Create(requestContext, "owner-user", product.CreateRequest{Name: "Cheese"})
created, createError := repo.Create(requestContext, "owner-user", userproduct.CreateRequest{Name: "Cheese"})
if createError != nil {
t.Fatalf("create product: %v", createError)
t.Fatalf("create user product: %v", createError)
}
deleteError := repo.Delete(requestContext, created.ID, "other-user")
if deleteError != product.ErrNotFound {
if deleteError != userproduct.ErrNotFound {
t.Errorf("expected ErrNotFound when deleting another user's product, got %v", deleteError)
}
}
func TestProductRepository_ListForPrompt(t *testing.T) {
func TestUserProductRepository_ListForPrompt(t *testing.T) {
pool := testutil.SetupTestDB(t)
repo := product.NewRepository(pool)
repo := userproduct.NewRepository(pool)
requestContext := context.Background()
userID := "prompt-user"
_, createError := repo.Create(requestContext, userID, product.CreateRequest{
_, createError := repo.Create(requestContext, userID, userproduct.CreateRequest{
Name: "Tomatoes", Quantity: 4, Unit: "pcs", StorageDays: 5,
})
if createError != nil {
t.Fatalf("create product: %v", createError)
t.Fatalf("create user product: %v", createError)
}
lines, listError := repo.ListForPrompt(requestContext, userID)