package helper

import (
	"users_management/m/model/dto/res"
	"users_management/m/model/entity"

	"github.com/google/uuid"
)

func ConvertToBackboneResponses(backbone []entity.Backbone, totalFishbone map[uuid.UUID]int) ([]res.BackboneResponse, error) {
	var responses []res.BackboneResponse
	for _, backbone := range backbone {
		count := 0
		if fishboneCount, exist := totalFishbone[backbone.ID]; exist {
			count = fishboneCount
		}
		backboneResp := res.BackboneResponse {
			ID: backbone.ID,
			BackboneCode: backbone.BackboneCode,
			DevStart: backbone.DeviceStart.DeviceCode,
			DevEnd: backbone.DeviceEnd.DeviceCode,
			CoreAmount: backbone.CoreAmount,
			TotalFishbone: count,
			CreatedAt: backbone.CreatedAt,
			UpdatedAt: backbone.UpdatedAt,
		}
		responses = append(responses, backboneResp)
	}

	return responses, nil
}

func ConvertToBackboneRespId(backbone entity.Backbone, fishboneCount int) (res.BackboneResponse, error) {
    backboneResp := res.BackboneResponse {
        ID: backbone.ID,
        DevStart: backbone.DeviceStart.DeviceCode,
        DevEnd: backbone.DeviceEnd.DeviceCode,
        CoreAmount: backbone.CoreAmount,
        TotalFishbone: fishboneCount, 
        CreatedAt: backbone.CreatedAt,
        UpdatedAt: backbone.UpdatedAt,
    }

    return backboneResp, nil
}