Merge branch 'feat/BE/Sprint-5' of https://gitlab.com/mbugroup/lti-api into dev/teguh

internal/modules/production/project_flocks/repositories/projectflock.repository.go

@@ -11,17 +11,24 @@ import (
 	"gorm.io/gorm"
 )
 
-const baseNameExpression = "LOWER(TRIM(regexp_replace(flock_name, '\\\\s+\\\\d+(\\\\s+\\\\d+)*$', '', 'g')))"
 
 type ProjectflockRepository interface {
 	repository.BaseRepository[entity.ProjectFlock]
 	GetAllWithFilters(ctx context.Context, offset, limit int, params *validation.Query) ([]entity.ProjectFlock, int64, error)
 	WithDefaultRelations() func(*gorm.DB) *gorm.DB
 	ExistsByFlockName(ctx context.Context, flockName string, excludeID *uint) (bool, error)
+	GetNextPeriodsForKandangs(ctx context.Context, kandangIDs []uint) (map[uint]int, error)
+	GetCurrentProjectPeriod(ctx context.Context, projectFlockID uint) (int, error)
+	GetKandangPeriodSummaryRows(ctx context.Context, locationID uint) ([]KandangPeriodRow, error)
 	AreaExists(ctx context.Context, id uint) (bool, error)
 	FcrExists(ctx context.Context, id uint) (bool, error)
 	LocationExists(ctx context.Context, id uint) (bool, error)
 }
+type KandangPeriodRow struct {
+	Id           uint
+	Name         string
+	LatestPeriod int
+}
 
 type ProjectflockRepositoryImpl struct {
 	*repository.BaseRepositoryImpl[entity.ProjectFlock]
@@ -35,19 +42,13 @@ func NewProjectflockRepository(db *gorm.DB) ProjectflockRepository {
 
 func (r *ProjectflockRepositoryImpl) GetAllWithFilters(ctx context.Context, offset, limit int, params *validation.Query) ([]entity.ProjectFlock, int64, error) {
 	return r.GetAll(ctx, offset, limit, func(db *gorm.DB) *gorm.DB {
-		db = r.withDefaultRelations(db)
 		return r.applyQueryFilters(db, params)
 	})
 }
 
 func (r *ProjectflockRepositoryImpl) WithDefaultRelations() func(*gorm.DB) *gorm.DB {
 	return func(db *gorm.DB) *gorm.DB {
-		return r.withDefaultRelations(db)
-	}
-}
-
-func (r *ProjectflockRepositoryImpl) withDefaultRelations(db *gorm.DB) *gorm.DB {
-	return db.
+		return db.
 		Preload("CreatedUser").
 		Preload("Area").
 		Preload("Fcr").
@@ -55,8 +56,10 @@ func (r *ProjectflockRepositoryImpl) withDefaultRelations(db *gorm.DB) *gorm.DB
 		Preload("Kandangs").
 		Preload("KandangHistory").
 		Preload("KandangHistory.Kandang")
+	}
 }
 
+
 func (r *ProjectflockRepositoryImpl) applyQueryFilters(db *gorm.DB, params *validation.Query) *gorm.DB {
 	if params == nil {
 		return db
@@ -163,6 +166,88 @@ func (r *ProjectflockRepositoryImpl) LocationExists(ctx context.Context, id uint
 	return repository.Exists[entity.Location](ctx, r.DB(), id)
 }
 
+func (r *ProjectflockRepositoryImpl) GetNextPeriodsForKandangs(ctx context.Context, kandangIDs []uint) (map[uint]int, error) {
+	result := make(map[uint]int)
+	if len(kandangIDs) == 0 {
+		return result, nil
+	}
+
+	unique := uniqueUintSlice(kandangIDs)
+	for _, id := range unique {
+		result[id] = 1
+	}
+
+	type periodRow struct {
+		KandangID uint
+		MaxPeriod int
+	}
+
+	var rows []periodRow
+	if err := r.DB().WithContext(ctx).
+		Table("project_flock_kandangs").
+		Select("kandang_id, COALESCE(MAX(period), 0) AS max_period").
+		Where("kandang_id IN ?", unique).
+		Group("kandang_id").
+		Scan(&rows).Error; err != nil {
+		return nil, err
+	}
+
+	for _, row := range rows {
+		if row.MaxPeriod > 0 {
+			result[row.KandangID] = row.MaxPeriod + 1
+		}
+	}
+
+	return result, nil
+}
+
+func (r *ProjectflockRepositoryImpl) GetCurrentProjectPeriod(ctx context.Context, projectFlockID uint) (int, error) {
+	if projectFlockID == 0 {
+		return 0, nil
+	}
+
+	var currentPeriod int
+	if err := r.DB().WithContext(ctx).
+		Table("project_flock_kandangs").
+		Where("project_flock_id = ?", projectFlockID).
+		Select("COALESCE(MAX(period), 0)").
+		Scan(&currentPeriod).Error; err != nil {
+		return 0, err
+	}
+
+	return currentPeriod, nil
+}
+
+func (r *ProjectflockRepositoryImpl) GetKandangPeriodSummaryRows(ctx context.Context, locationID uint) ([]KandangPeriodRow, error) {
+	rows := make([]KandangPeriodRow, 0)
+	if err := r.DB().WithContext(ctx).
+		Table("kandangs AS k").
+		Select("k.id, k.name, COALESCE(MAX(pfk.period), 0) AS latest_period").
+		Joins("LEFT JOIN project_flock_kandangs AS pfk ON pfk.kandang_id = k.id").
+		Where("k.location_id = ?", locationID).
+		Where("k.deleted_at IS NULL").
+		Group("k.id, k.name").
+		Order("k.id ASC").
+		Scan(&rows).Error; err != nil {
+		return nil, err
+	}
+
+	return rows, nil
+}
+
+func uniqueUintSlice(values []uint) []uint {
+	seen := make(map[uint]struct{}, len(values))
+	result := make([]uint, 0, len(values))
+	for _, v := range values {
+		if _, ok := seen[v]; ok {
+			continue
+		}
+		seen[v] = struct{}{}
+		result = append(result, v)
+	}
+	return result
+}
+
 func (r *ProjectflockRepositoryImpl) buildOrderExpressions(sortBy, sortOrder string) []string {
 	direction := "ASC"
 	if strings.ToLower(sortOrder) == "desc" {