lti-api/internal/common/repository/common.hpp.repository.go

package repository

import (
	"context"
	"time"

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

type HppCostRepository interface {
	GetProjectFlockKandangIDs(ctx context.Context, projectFlockId uint) ([]uint, error)
	GetDocCost(ctx context.Context, kandangIDs []uint) (float64, error)
	GetBudgetCostByProjectFlockId(ctx context.Context, projectFlockId uint) (float64, error)
	GetExpedisionCost(ctx context.Context, kandangIDs []uint) (float64, error)
	GetFeedCost(ctx context.Context, kandangIDs []uint, date *time.Time) (float64, error)
	GetOvkCost(ctx context.Context, kandangIDs []uint, date *time.Time) (float64, error)
	GetTotalPopulation(ctx context.Context, kandangIDs []uint) (float64, error)
	GetPulletCost(ctx context.Context, projectFlockKandangId uint) (float64, error)
	GetTransferSourceSummary(ctx context.Context, projectFlockKandangId uint) (uint, float64, error)
}

type HppRepositoryImpl struct {
	db *gorm.DB
}

func NewHppCostRepository(db *gorm.DB) HppCostRepository {
	return &HppRepositoryImpl{db: db}
}

func (r *HppRepositoryImpl) GetProjectFlockKandangIDs(ctx context.Context, projectFlockId uint) ([]uint, error) {
	var ids []uint
	err := r.db.WithContext(ctx).
		Table("project_flock_kandangs").
		Select("id").
		Where("project_flock_id = ?", projectFlockId).
		Scan(&ids).Error
	if err != nil {
		return nil, err
	}

	return ids, nil
}

func (r *HppRepositoryImpl) GetDocCost(ctx context.Context, kandangIDs []uint) (float64, error) {
	var total float64
	err := r.db.WithContext(ctx).
		Table("project_chickins AS pc").
		Select("COALESCE(SUM(pc.usage_qty * COALESCE(pi.price, 0)), 0)").
		Joins("JOIN stock_allocations AS sa ON sa.usable_type = ? AND sa.usable_id = pc.id AND sa.stockable_type = ?", fifo.UsableKeyProjectChickin.String(), fifo.StockableKeyPurchaseItems.String()).
		Joins("JOIN purchase_items AS pi ON pi.id = sa.stockable_id").
		Where("pc.project_flock_kandang_id IN (?)", kandangIDs).
		Scan(&total).Error
	if err != nil {
		return 0, err
	}

	return total, nil
}

func (r *HppRepositoryImpl) GetBudgetCostByProjectFlockId(ctx context.Context, projectFlockId uint) (float64, error) {
	var total float64
	err := r.db.WithContext(ctx).
		Table("project_chickin_details AS pcd").
		Select("COALESCE(SUM(pcd.qty * pcd.price), 0)").
		Where("pcd.project_flock_id = ?", projectFlockId).
		Scan(&total).Error
	if err != nil {
		return 0, err
	}

	return total, nil
}

func (r *HppRepositoryImpl) GetExpedisionCost(ctx context.Context, kandangIDs []uint) (float64, error) {
	var total float64
	err := r.db.WithContext(ctx).
		Table("expense_nonstocks AS en").
		Select("COALESCE(SUM(er.qty * er.price), 0)").
		Joins("JOIN expense_realizations AS er ON er.expense_nonstock_id = en.id").
		Joins("JOIN flags AS f ON f.flagable_id = en.nonstock_id AND f.flagable_type = ?", entity.FlagableTypeNonstock).
		Where("en.project_flock_kandang_id IN (?)", kandangIDs).
		Where("f.name = ?", utils.FlagEkspedisi).
		Scan(&total).Error
	if err != nil {
		return 0, err
	}

	return total, nil
}

func (r *HppRepositoryImpl) GetFeedCost(ctx context.Context, kandangIDs []uint, date *time.Time) (float64, error) {
	if date == nil {
		now := time.Now()
		date = &now
	}

	var total float64
	err := r.db.WithContext(ctx).
		Table("recordings AS r").
		Select("COALESCE(SUM(rs.usage_qty * COALESCE(pi.price, 0)), 0)").
		Joins("JOIN recording_stocks AS rs ON rs.recording_id = r.id").
		Joins("JOIN product_warehouses AS pw ON pw.id = rs.product_warehouse_id").
		Joins("JOIN flags AS f ON f.flagable_id = pw.product_id AND f.flagable_type = ?", entity.FlagableTypeProduct).
		Joins("JOIN stock_allocations AS sa ON sa.usable_type = ? AND sa.usable_id = rs.id AND sa.stockable_type = ?", fifo.UsableKeyRecordingStock.String(), fifo.StockableKeyPurchaseItems.String()).
		Joins("JOIN purchase_items AS pi ON pi.id = sa.stockable_id").
		Where("r.project_flock_kandangs_id IN (?)", kandangIDs).
		Where("r.record_datetime <= ?", *date).
		Where("f.name = ?", utils.FlagPakan).
		Scan(&total).Error
	if err != nil {
		return 0, err
	}

	return total, nil
}

func (r *HppRepositoryImpl) GetOvkCost(ctx context.Context, kandangIDs []uint, date *time.Time) (float64, error) {
	if date == nil {
		now := time.Now()
		date = &now
	}

	flags := []utils.FlagType{
		utils.FlagOVK,
		utils.FlagObat,
		utils.FlagVitamin,
		utils.FlagKimia,
	}

	var total float64
	err := r.db.WithContext(ctx).
		Table("recordings AS r").
		Select("COALESCE(SUM(rs.usage_qty * COALESCE(pi.price, 0)), 0)").
		Joins("JOIN recording_stocks AS rs ON rs.recording_id = r.id").
		Joins("JOIN product_warehouses AS pw ON pw.id = rs.product_warehouse_id").
		Joins("JOIN flags AS f ON f.flagable_id = pw.product_id AND f.flagable_type = ?", entity.FlagableTypeProduct).
		Joins("JOIN stock_allocations AS sa ON sa.usable_type = ? AND sa.usable_id = rs.id AND sa.stockable_type = ?", fifo.UsableKeyRecordingStock.String(), fifo.StockableKeyPurchaseItems.String()).
		Joins("JOIN purchase_items AS pi ON pi.id = sa.stockable_id").
		Where("r.project_flock_kandangs_id IN (?)", kandangIDs).
		Where("r.record_datetime <= ?", *date).
		Where("f.name IN ?", flags).
		Scan(&total).Error
	if err != nil {
		return 0, err
	}

	return total, nil
}

func (r *HppRepositoryImpl) GetTotalPopulation(ctx context.Context, kandangIDs []uint) (float64, error) {
	var total float64
	err := r.db.WithContext(ctx).
		Table("project_chickins AS pc").
		Select("COALESCE(SUM(pc.usage_qty), 0)").
		Where("pc.project_flock_kandang_id IN (?)", kandangIDs).
		Scan(&total).Error
	if err != nil {
		return 0, err
	}

	return total, nil
}

func (r *HppRepositoryImpl) GetPulletCost(ctx context.Context, projectFlockKandangId uint) (float64, error) {
	stockablePurchase := fifo.StockableKeyPurchaseItems.String()
	stockableTransferIn := fifo.StockableKeyStockTransferIn.String()
	usableProjectChickin := fifo.UsableKeyProjectChickin.String()

	var total float64
	err := r.db.WithContext(ctx).
		Table("project_chickins AS pc").
		Select(`
			COALESCE(SUM(pc.usage_qty * CASE
				WHEN sa.stockable_type = ? THEN COALESCE(pi.price, 0)
				WHEN sa.stockable_type = ? THEN COALESCE(tpi.price, 0)
				ELSE 0
			END), 0)`,
			stockablePurchase, stockableTransferIn).
		Joins("JOIN stock_allocations AS sa ON sa.usable_type = ? AND sa.usable_id = pc.id", usableProjectChickin).
		Joins("LEFT JOIN purchase_items AS pi ON pi.id = sa.stockable_id AND sa.stockable_type = ?", stockablePurchase).
		Joins("LEFT JOIN stock_allocations AS tsa ON tsa.usable_type = ? AND tsa.usable_id = sa.stockable_id AND sa.stockable_type = ? AND tsa.stockable_type = ?", stockableTransferIn, stockableTransferIn, stockablePurchase).
		Joins("LEFT JOIN purchase_items AS tpi ON tpi.id = tsa.stockable_id").
		Where("pc.project_flock_kandang_id = ?", projectFlockKandangId).
		Scan(&total).Error
	if err != nil {
		return 0, err
	}

	return total, nil
}

func (r *HppRepositoryImpl) GetTransferSourceSummary(ctx context.Context, projectFlockKandangId uint) (uint, float64, error) {
	var summary struct {
		ProjectFlockID uint
		TotalQty       float64
	}
	err := r.db.WithContext(ctx).
		Table("laying_transfer_targets AS ltt").
		Select("lt.from_project_flock_id AS project_flock_id, COALESCE(SUM(ltt.total_qty), 0) AS total_qty").
		Joins("JOIN laying_transfers AS lt ON lt.id = ltt.laying_transfer_id").
		Where("ltt.target_project_flock_kandang_id = ?", projectFlockKandangId).
		Scan(&summary).Error
	if err != nil {
		return 0, 0, err
	}

	return summary.ProjectFlockID, summary.TotalQty, nil
}