package recipe

import (
	"context"
	"errors"
	"fmt"

	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
	"git.huangwc.com/pig/pig-farm-controller/internal/infra/repository"

	"gorm.io/gorm"
)

// 定义领域特定的错误
var (
	ErrNutrientNameConflict = fmt.Errorf("营养种类名称已存在")
	ErrNutrientNotFound     = fmt.Errorf("营养种类不存在")
	ErrNutrientInUse        = fmt.Errorf("营养种类正在被原料使用，无法删除")

	ErrRawMaterialNameConflict = fmt.Errorf("原料名称已存在")
	ErrRawMaterialNotFound     = fmt.Errorf("原料不存在")
)

// Service 定义了配方与原料领域的核心业务服务接口
type Service interface {
	// 营养种类相关接口
	CreateNutrient(ctx context.Context, name, description string) (*models.Nutrient, error)
	UpdateNutrient(ctx context.Context, id uint32, name, description string) (*models.Nutrient, error)
	DeleteNutrient(ctx context.Context, id uint32) error
	GetNutrient(ctx context.Context, id uint32) (*models.Nutrient, error)
	ListNutrients(ctx context.Context, page, pageSize int) ([]models.Nutrient, int64, error)

	// 原料相关接口
	CreateRawMaterial(ctx context.Context, name, description string) (*models.RawMaterial, error)
	UpdateRawMaterial(ctx context.Context, id uint32, name, description string) (*models.RawMaterial, error)
	DeleteRawMaterial(ctx context.Context, id uint32) error
	GetRawMaterial(ctx context.Context, id uint32) (*models.RawMaterial, error)
	ListRawMaterials(ctx context.Context, page, pageSize int) ([]models.RawMaterial, int64, error)
}

// recipeServiceImpl 是 RecipeService 的实现
type recipeServiceImpl struct {
	ctx             context.Context
	nutrientRepo    repository.NutrientRepository
	rawMaterialRepo repository.RawMaterialRepository
}

// NewRecipeService 创建一个新的 RecipeService 实例
func NewRecipeService(ctx context.Context, nutrientRepo repository.NutrientRepository, rawMaterialRepo repository.RawMaterialRepository) Service {
	return &recipeServiceImpl{
		ctx:             ctx,
		nutrientRepo:    nutrientRepo,
		rawMaterialRepo: rawMaterialRepo,
	}
}

// CreateNutrient 实现了创建营养种类的核心业务逻辑
func (s *recipeServiceImpl) CreateNutrient(ctx context.Context, name, description string) (*models.Nutrient, error) {
	serviceCtx := logs.AddFuncName(ctx, s.ctx, "CreateNutrient")

	// 检查名称是否已存在
	existing, err := s.nutrientRepo.GetNutrientByName(serviceCtx, name)
	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) { // 只有不是记录未找到的错误才返回
		return nil, fmt.Errorf("检查营养种类名称失败: %w", err)
	}
	if existing != nil {
		return nil, ErrNutrientNameConflict
	}

	nutrient := &models.Nutrient{
		Name:        name,
		Description: description,
	}

	if err := s.nutrientRepo.CreateNutrient(serviceCtx, nutrient); err != nil {
		return nil, fmt.Errorf("创建营养种类失败: %w", err)
	}

	return nutrient, nil
}

// UpdateNutrient 实现了更新营养种类的核心业务逻辑
func (s *recipeServiceImpl) UpdateNutrient(ctx context.Context, id uint32, name, description string) (*models.Nutrient, error) {
	serviceCtx := logs.AddFuncName(ctx, s.ctx, "UpdateNutrient")

	// 检查要更新的实体是否存在
	nutrient, err := s.nutrientRepo.GetNutrientByID(serviceCtx, id)
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) { // 如果是记录未找到错误，则返回领域错误
			return nil, ErrNutrientNotFound
		}
		return nil, fmt.Errorf("获取待更新的营养种类失败: %w", err)
	}

	// 如果名称有变动，检查新名称是否与其它记录冲突
	if nutrient.Name != name {
		existing, err := s.nutrientRepo.GetNutrientByName(serviceCtx, name)
		if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, fmt.Errorf("检查新的营养种类名称失败: %w", err)
		}
		if existing != nil && existing.ID != id {
			return nil, ErrNutrientNameConflict
		}
	}

	nutrient.Name = name
	nutrient.Description = description

	if err := s.nutrientRepo.UpdateNutrient(serviceCtx, nutrient); err != nil {
		return nil, fmt.Errorf("更新营养种类失败: %w", err)
	}

	return nutrient, nil
}

// DeleteNutrient 实现了删除营养种类的核心业务逻辑
func (s *recipeServiceImpl) DeleteNutrient(ctx context.Context, id uint32) error {
	serviceCtx := logs.AddFuncName(ctx, s.ctx, "DeleteNutrient")

	// 检查实体是否存在
	_, err := s.nutrientRepo.GetNutrientByID(serviceCtx, id)
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return ErrNutrientNotFound
		}
		return fmt.Errorf("获取待删除的营养种类失败: %w", err)
	}

	if err := s.nutrientRepo.DeleteNutrient(serviceCtx, id); err != nil {
		return fmt.Errorf("删除营养种类失败: %w", err)
	}

	return nil
}

// GetNutrient 实现了获取单个营养种类的逻辑
func (s *recipeServiceImpl) GetNutrient(ctx context.Context, id uint32) (*models.Nutrient, error) {
	serviceCtx := logs.AddFuncName(ctx, s.ctx, "GetNutrient")

	nutrient, err := s.nutrientRepo.GetNutrientByID(serviceCtx, id)
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, ErrNutrientNotFound
		}
		return nil, fmt.Errorf("获取营养种类失败: %w", err)
	}
	return nutrient, nil
}

// ListNutrients 实现了列出营养种类的逻辑
func (s *recipeServiceImpl) ListNutrients(ctx context.Context, page, pageSize int) ([]models.Nutrient, int64, error) {
	serviceCtx := logs.AddFuncName(ctx, s.ctx, "ListNutrients")

	nutrients, total, err := s.nutrientRepo.ListNutrients(serviceCtx, page, pageSize)
	if err != nil {
		return nil, 0, fmt.Errorf("获取营养种类列表失败: %w", err)
	}
	return nutrients, total, nil
}

// CreateRawMaterial 实现了创建原料的核心业务逻辑
func (s *recipeServiceImpl) CreateRawMaterial(ctx context.Context, name, description string) (*models.RawMaterial, error) {
	serviceCtx := logs.AddFuncName(ctx, s.ctx, "CreateRawMaterial")

	// 检查名称是否已存在
	existing, err := s.rawMaterialRepo.GetRawMaterialByName(serviceCtx, name)
	if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
		return nil, fmt.Errorf("检查原料名称失败: %w", err)
	}
	if existing != nil {
		return nil, ErrRawMaterialNameConflict
	}

	rawMaterial := &models.RawMaterial{
		Name:        name,
		Description: description,
	}

	if err := s.rawMaterialRepo.CreateRawMaterial(serviceCtx, rawMaterial); err != nil {
		return nil, fmt.Errorf("创建原料失败: %w", err)
	}

	return rawMaterial, nil
}

// UpdateRawMaterial 实现了更新原料的核心业务逻辑
func (s *recipeServiceImpl) UpdateRawMaterial(ctx context.Context, id uint32, name, description string) (*models.RawMaterial, error) {
	serviceCtx := logs.AddFuncName(ctx, s.ctx, "UpdateRawMaterial")

	// 检查要更新的实体是否存在
	rawMaterial, err := s.rawMaterialRepo.GetRawMaterialByID(serviceCtx, id)
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, ErrRawMaterialNotFound
		}
		return nil, fmt.Errorf("获取待更新的原料失败: %w", err)
	}

	// 如果名称有变动，检查新名称是否与其它记录冲突
	if rawMaterial.Name != name {
		existing, err := s.rawMaterialRepo.GetRawMaterialByName(serviceCtx, name)
		if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, fmt.Errorf("检查新的原料名称失败: %w", err)
		}
		if existing != nil && existing.ID != id {
			return nil, ErrRawMaterialNameConflict
		}
	}

	rawMaterial.Name = name
	rawMaterial.Description = description

	if err := s.rawMaterialRepo.UpdateRawMaterial(serviceCtx, rawMaterial); err != nil {
		return nil, fmt.Errorf("更新原料失败: %w", err)
	}

	return rawMaterial, nil
}

// DeleteRawMaterial 实现了删除原料的核心业务逻辑
func (s *recipeServiceImpl) DeleteRawMaterial(ctx context.Context, id uint32) error {
	serviceCtx := logs.AddFuncName(ctx, s.ctx, "DeleteRawMaterial")

	// 检查实体是否存在
	_, err := s.rawMaterialRepo.GetRawMaterialByID(serviceCtx, id)
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return ErrRawMaterialNotFound
		}
		return fmt.Errorf("获取待删除的原料失败: %w", err)
	}

	if err := s.rawMaterialRepo.DeleteRawMaterial(serviceCtx, id); err != nil {
		return fmt.Errorf("删除原料失败: %w", err)
	}

	return nil
}

// GetRawMaterial 实现了获取单个原料的逻辑
func (s *recipeServiceImpl) GetRawMaterial(ctx context.Context, id uint32) (*models.RawMaterial, error) {
	serviceCtx := logs.AddFuncName(ctx, s.ctx, "GetRawMaterial")

	rawMaterial, err := s.rawMaterialRepo.GetRawMaterialByID(serviceCtx, id)
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, ErrRawMaterialNotFound
		}
		return nil, fmt.Errorf("获取原料失败: %w", err)
	}
	return rawMaterial, nil
}

// ListRawMaterials 实现了列出原料的逻辑
func (s *recipeServiceImpl) ListRawMaterials(ctx context.Context, page, pageSize int) ([]models.RawMaterial, int64, error) {
	serviceCtx := logs.AddFuncName(ctx, s.ctx, "ListRawMaterials")

	rawMaterials, total, err := s.rawMaterialRepo.ListRawMaterials(serviceCtx, page, pageSize)
	if err != nil {
		return nil, 0, fmt.Errorf("获取原料列表失败: %w", err)
	}
	return rawMaterials, total, nil
}