package model

import (
	"fmt"
	"sort"

	pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"
)

const (
	AllCow                           = "all_cow"
	AdultCow                         = "adult_cow"
	AvgCalvingInterval               = "avg_calving_interval"
	OutputNumber                     = "output_number"
	InputNumber                      = "input_number"
	SalesVolume                      = "sales_volume"
	CalvingNumber                    = "calving_number"
	AdultAbortionRate                = "adult_abortion_rate"
	YouthAbortionRate                = "youth_abortion_rate"
	AllDieNumber                     = "all_die_number"
	DiseaseNumber                    = "disease_number"
	CureNumber                       = "cure_number"
	OutNumber                        = "out_number"
	CalfDieNumber                    = "calf_die_number"
	LactationCow                     = "lactation_cow"
	DryMilkCow                       = "dry_milk_cow"
	ReserveCow                       = "reserve_cow"
	FirstBornSurvivalRate            = "first_born_survival_rate"
	FirstBornDeathRate               = "first_born_death_rate"
	MultiparitySurvivalRate          = "multiparity_survival_rate"
	MultiparityDeathRate             = "multiparity_death_rate"
	AvgAgeFirstMate                  = "avg_age_first_mate"
	YouthPregnantRate                = "youth_pregnant_rate"
	MultiparityPregnantRate          = "multiparity_pregnant_rate"
	MultipartyFirstCheckRate         = "multiparty_first_check_rate"
	MultipartySecondCheckRate        = "multiparty_second_check_rate"
	YouthFirstCheckRate              = "youth_first_check_rate"
	YouthSecondCheckRate             = "youth_second_check_rate"
	ForbiddenCowNumber               = "forbidden_cow_number"
	AvgRegistrationDays              = "avg_registration_days"
	AvgPregnancyDays                 = "avg_pregnancy_days"
	AvgGestationalAge                = "avg_gestational_age"
	Month17UnPregnancyRate           = "month17_un_pregnancy_rate"
	Month20UnPregnancyRate           = "month20_un_pregnancy_rate"
	Multiparty150DaysUnPregnancyRate = "multiparty_150_days_un_pregnancy_rate"
	MultipartyAbortionNumber         = "multiparty_abortion_number"
	MultipartyAbortionRate           = "multiparty_abortion_rate"
	MultipartyPregnancyNumber        = "multiparty_pregnancy_number"
	MultipartyOutNumber              = "multiparty_out_number"
	MultipartyDieNumber              = "multiparty_die_number"
	Calving60DieRate                 = "calving60_die_rate"
	Calving60OutRate                 = "calving60_out_rate"
)

type IndicatorsData struct {
	Id           int64                        `json:"id"`
	PastureId    int64                        `json:"pastureId"`
	CategoryType pasturePb.IndicatorType_Kind `json:"categoryType"`
	CategoryName string                       `json:"categoryName"`
	Date         string                       `json:"date"`
	Kind         string                       `json:"kind"`
	Value        string                       `json:"value"`
	CreatedAt    int64                        `json:"createdAt"`
	UpdatedAt    int64                        `json:"updatedAt"`
}

func (d *IndicatorsData) TableName() string {
	return "indicators_data"
}

type IndicatorsDataSlice []*IndicatorsData

func (i IndicatorsDataSlice) ToPB(header, kinds []string, detailsMap map[string]*IndicatorsDetails) *IndicatorComparison {
	res := &IndicatorComparison{
		Headers: header,
		List:    make([]map[string]string, 0),
	}

	if len(header) <= 0 || len(detailsMap) <= 0 || len(kinds) <= 0 || len(i) <= 0 {
		return res
	}

	sort.Strings(res.Headers) // 确保日期顺序一致
	indicatorsDataMap := make(map[string]*IndicatorsData)
	for _, v := range i {
		indicatorsDataMap[fmt.Sprintf("%s-%s", v.Kind, v.Date)] = v
	}

	dataMap := make(map[string]map[string]string)
	for _, kind := range kinds {
		if dataMap[kind] == nil {
			dataMap[kind] = make(map[string]string)
		}
		details := detailsMap[kind]
		dataMap[kind] = map[string]string{
			"name":         details.Name,
			"kind":         details.Kind,
			"unit":         details.Unit,
			"categoryType": fmt.Sprintf("%d", details.CategoryType),
			"categoryName": details.CategoryName,
			"zh":           details.Zh,
		}

		for _, v := range header {
			pref := fmt.Sprintf("%s-%s", kind, v)
			indicators, ok := indicatorsDataMap[pref]
			if ok {
				dataMap[kind][v] = indicators.Value
			} else {
				dataMap[kind][v] = ""
			}
		}
	}

	for _, data := range dataMap {
		res.List = append(res.List, data)
	}
	return res
}

type IndicatorsComparisonResponse struct {
	Code int32                `json:"code"`
	Msg  string               `json:"msg"`
	Data *IndicatorComparison `json:"data"`
}

type IndicatorComparison struct {
	Headers []string            `json:"headers"`
	List    []map[string]string `json:"list"`
}