package orders

import (
	"context"
	"furtuna-be/internal/common/logger"
	"furtuna-be/internal/entity"
	"go.uber.org/zap"
	"gorm.io/gorm"
)

type OrderRepository struct {
	db *gorm.DB
}

func NewOrderRepository(db *gorm.DB) *OrderRepository {
	return &OrderRepository{
		db: db,
	}
}

func (r *OrderRepository) Create(ctx context.Context, order *entity.Order) (*entity.Order, error) {
	err := r.db.WithContext(ctx).Create(order).Error
	if err != nil {
		logger.ContextLogger(ctx).Error("error when creating order", zap.Error(err))
		return nil, err
	}
	return r.FindByID(ctx, order.ID)
}

func (r *OrderRepository) UpdateStatus(ctx context.Context, orderID int64, status string) (*entity.Order, error) {
	order := new(entity.Order)
	if err := r.db.WithContext(ctx).First(order, orderID).Error; err != nil {
		logger.ContextLogger(ctx).Error("error when finding order", zap.Error(err))
		return nil, err
	}
	order.Status = status
	if err := r.db.WithContext(ctx).Save(order).Error; err != nil {
		logger.ContextLogger(ctx).Error("error when updating order status", zap.Error(err))
		return nil, err
	}
	return order, nil
}

func (r *OrderRepository) FindByID(ctx context.Context, id int64) (*entity.Order, error) {
	var order entity.Order

	err := r.db.WithContext(ctx).Preload("OrderItems", func(db *gorm.DB) *gorm.DB {
		return db.Preload("Product")
	}).First(&order, id).Error

	if err != nil {
		logger.ContextLogger(ctx).Error("error when finding order by ID", zap.Error(err))
		return nil, err
	}

	return &order, nil
}

func (r *OrderRepository) SetOrderStatus(ctx context.Context, db *gorm.DB, orderID int64, status string) error {
	var order entity.Order
	if err := db.WithContext(ctx).Preload("OrderItems").First(&order, orderID).Error; err != nil {
		logger.ContextLogger(ctx).Error("error when finding order by ID", zap.Error(err))
		return err
	}

	order.Status = status

	if err := db.WithContext(ctx).Save(&order).Error; err != nil {
		logger.ContextLogger(ctx).Error("error when updating order status", zap.Error(err))
		return err
	}

	return nil
}

func (r *OrderRepository) Update(ctx context.Context, order *entity.Order) (*entity.Order, error) {
	if err := r.db.WithContext(ctx).Save(order).Error; err != nil {
		logger.ContextLogger(ctx).Error("error when updating order", zap.Error(err))
		return nil, err
	}
	return order, nil
}