package repository

import (
	"context"

	"gitlab.com/mbugroup/lti-api.git/internal/common/repository"
	entity "gitlab.com/mbugroup/lti-api.git/internal/entities"
	"gorm.io/gorm"
)

type ExpenseRealizationRepository interface {
	repository.BaseRepository[entity.ExpenseRealization]
	IdExists(ctx context.Context, id uint64) (bool, error)
	GetByExpenseNonstockID(ctx context.Context, expenseNonstockID uint64) (*entity.ExpenseRealization, error)
	GetByProjectFlockID(ctx context.Context, projectFlockID uint) ([]entity.ExpenseRealization, error)
}

type ExpenseRealizationRepositoryImpl struct {
	*repository.BaseRepositoryImpl[entity.ExpenseRealization]
}

func NewExpenseRealizationRepository(db *gorm.DB) ExpenseRealizationRepository {
	return &ExpenseRealizationRepositoryImpl{
		BaseRepositoryImpl: repository.NewBaseRepository[entity.ExpenseRealization](db),
	}
}

func (r *ExpenseRealizationRepositoryImpl) IdExists(ctx context.Context, id uint64) (bool, error) {
	return repository.Exists[entity.ExpenseRealization](ctx, r.DB(), uint(id))
}

func (r *ExpenseRealizationRepositoryImpl) GetByExpenseNonstockID(ctx context.Context, expenseNonstockID uint64) (*entity.ExpenseRealization, error) {
	var realization entity.ExpenseRealization
	err := r.DB().WithContext(ctx).Where("expense_nonstock_id = ?", expenseNonstockID).First(&realization).Error
	return &realization, err
}

func (r *ExpenseRealizationRepositoryImpl) GetByProjectFlockID(ctx context.Context, projectFlockID uint) ([]entity.ExpenseRealization, error) {
	var realizations []entity.ExpenseRealization
	err := r.DB().WithContext(ctx).
		Preload("ExpenseNonstock").
		Preload("ExpenseNonstock.Nonstock").
		Preload("ExpenseNonstock.Nonstock.Uom").
		Preload("ExpenseNonstock.Expense").
		Joins("JOIN expense_nonstocks ON expense_nonstocks.id = expense_realizations.expense_nonstock_id").
		Joins("JOIN project_flock_kandangs ON project_flock_kandangs.id = expense_nonstocks.project_flock_kandang_id").
		Joins("JOIN expenses ON expenses.id = expense_nonstocks.expense_id").
		Where("project_flock_kandangs.project_flock_id = ?", projectFlockID).
		Where("expenses.category = ?", "BOP").
		Find(&realizations).Error
	return realizations, err
}