package repositories

import (
	"fmt"
	"rijig/model"

	"gorm.io/gorm"
)

type RequestPickupRepository interface {
	CreateRequestPickup(request *model.RequestPickup) error
	CreateRequestPickupItem(item *model.RequestPickupItem) error
	FindRequestPickupByID(id string) (*model.RequestPickup, error)
	FindAllRequestPickups(userId string) ([]model.RequestPickup, error)
	FindAllAutomaticMethodRequest(requestMethod, statuspickup string) ([]model.RequestPickup, error)
	FindRequestPickupByAddressAndStatus(userId, status string) (*model.RequestPickup, error)
	GetRequestPickupItems(requestPickupId string) ([]model.RequestPickupItem, error)
	GetAutomaticRequestPickupsForCollector(collectorId string) ([]model.RequestPickup, error)
	UpdateRequestPickup(id string, request *model.RequestPickup) error
	DeleteRequestPickup(id string) error
}

type requestPickupRepository struct {
	DB *gorm.DB
}

func NewRequestPickupRepository(db *gorm.DB) RequestPickupRepository {
	return &requestPickupRepository{DB: db}
}

func (r *requestPickupRepository) CreateRequestPickup(request *model.RequestPickup) error {
	if err := r.DB.Create(request).Error; err != nil {
		return fmt.Errorf("failed to create request pickup: %v", err)
	}

	for _, item := range request.RequestItems {
		item.RequestPickupId = request.ID
		if err := r.DB.Create(&item).Error; err != nil {
			return fmt.Errorf("failed to create request pickup item: %v", err)
		}
	}

	return nil
}

func (r *requestPickupRepository) CreateRequestPickupItem(item *model.RequestPickupItem) error {
	if err := r.DB.Create(item).Error; err != nil {
		return fmt.Errorf("failed to create request pickup item: %v", err)
	}
	return nil
}

func (r *requestPickupRepository) FindRequestPickupByID(id string) (*model.RequestPickup, error) {
	var request model.RequestPickup
	err := r.DB.Preload("RequestItems").First(&request, "id = ?", id).Error
	if err != nil {
		return nil, fmt.Errorf("request pickup with ID %s not found: %v", id, err)
	}
	return &request, nil
}

func (r *requestPickupRepository) FindAllRequestPickups(userId string) ([]model.RequestPickup, error) {
	var requests []model.RequestPickup
	err := r.DB.Preload("RequestItems").Where("user_id = ?", userId).Find(&requests).Error
	if err != nil {
		return nil, fmt.Errorf("failed to fetch all request pickups: %v", err)
	}
	return requests, nil
}

func (r *requestPickupRepository) FindAllAutomaticMethodRequest(requestMethod, statuspickup string) ([]model.RequestPickup, error) {
	var requests []model.RequestPickup
	err := r.DB.Preload("RequestItems").Where("request_method = ? AND status_pickup = ?", requestMethod, statuspickup).Find(&requests).Error
	if err != nil {
		return nil, fmt.Errorf("error fetching request pickups with request_method %s: %v", requestMethod, err)
	}

	return requests, nil
}

func (r *requestPickupRepository) FindRequestPickupByAddressAndStatus(userId, status string) (*model.RequestPickup, error) {
	var request model.RequestPickup
	err := r.DB.Where("user_id = ? AND status_pickup = ?", userId, status).First(&request).Error
	if err != nil {
		if err == gorm.ErrRecordNotFound {
			return nil, nil
		}
		return nil, fmt.Errorf("failed to check existing request pickup: %v", err)
	}
	return &request, nil
}

func (r *requestPickupRepository) UpdateRequestPickup(id string, request *model.RequestPickup) error {
	err := r.DB.Model(&model.RequestPickup{}).Where("id = ?", id).Updates(request).Error
	if err != nil {
		return fmt.Errorf("failed to update request pickup: %v", err)
	}

	return nil
}

func (r *requestPickupRepository) DeleteRequestPickup(id string) error {

	if err := r.DB.Where("request_pickup_id = ?", id).Delete(&model.RequestPickupItem{}).Error; err != nil {
		return fmt.Errorf("failed to delete request pickup items: %v", err)
	}

	err := r.DB.Delete(&model.RequestPickup{}, "id = ?", id).Error
	if err != nil {
		return fmt.Errorf("failed to delete request pickup: %v", err)
	}
	return nil
}

func (r *requestPickupRepository) GetAutomaticRequestPickupsForCollector(collectorId string) ([]model.RequestPickup, error) {
	var requests []model.RequestPickup

	err := r.DB.Preload("Address").
		Where("request_method = ? AND status_pickup = ?", "otomatis", "waiting_collector").
		Find(&requests).Error
	if err != nil {
		return nil, fmt.Errorf("error fetching pickup requests: %v", err)
	}

	return requests, nil
}

func (r *requestPickupRepository) GetRequestPickupItems(requestPickupId string) ([]model.RequestPickupItem, error) {
	var items []model.RequestPickupItem

	err := r.DB.Preload("TrashCategory").Where("request_pickup_id = ?", requestPickupId).Find(&items).Error
	if err != nil {
		return nil, fmt.Errorf("error fetching request pickup items: %v", err)
	}
	return items, nil
}