package backend

import (
	"context"
	"fmt"
	"kpt-pasture/config"
	"kpt-pasture/model"
	"kpt-pasture/util"
	"mime/multipart"
	"os"
	"path/filepath"
	"strconv"
	"time"

	"gorm.io/gorm"

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

	"gitee.com/xuyiping_admin/pkg/xerr"
)

func (s *StoreEntry) Photos(ctx context.Context, files []*multipart.FileHeader) ([]string, error) {
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return nil, err
	}

	workDir := fmt.Sprintf("%s", config.WorkDir)
	pathDir := fmt.Sprintf("/files/photos/%d/%s", userModel.AppPasture.Id, time.Now().Local().Format("20060102"))
	saveDir := filepath.Join(workDir, pathDir)
	if _, err = os.Stat(saveDir); os.IsNotExist(err) {
		if err = os.MkdirAll(saveDir, 0755); err != nil {
			return nil, xerr.Customf("创建目录失败: %s", err.Error())
		}
	}

	// 处理每个文件
	filePaths := make([]string, len(files))
	for i, file := range files {
		contentType := file.Header.Get("Content-Type")
		if contentType != "image/jpeg" && contentType != "image/png" && contentType != "image/gif" {
			return nil, xerr.Customf("图片格式错误: %s", file.Filename)

		}
		if file.Size > 1024*1024*5 {
			return nil, xerr.Customf("单个图片文件不能超过5MB")

		}
		ext := filepath.Ext(file.Filename)
		if ext == "" {
			switch contentType {
			case "image/jpeg":
				ext = ".jpg"
			case "image/png":
				ext = ".png"
			case "image/gif":
				ext = ".gif"
			default:
				ext = ".jpg" // 默认
			}
		}
		randomName := util.GenerateRandomNumberString(32)
		finalFilename := randomName + ext
		fPath := filepath.Join(saveDir, finalFilename)
		urlPath := filepath.Join(pathDir, finalFilename)

		if err = util.SaveUploadedFile(file, fPath); err != nil {
			return nil, xerr.Customf("保存文件失败: %s", err.Error())
		}
		filePaths[i] = urlPath
	}
	return filePaths, nil
}

func (s *StoreEntry) ImportExcel(ctx context.Context, data [][]string) error {
	// 获取当前用户信息
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return err
	}

	penMap := s.PenMap2(ctx, userModel.AppPasture.Id)
	if userModel.AppPasture.Id <= 0 {
		return xerr.Custom("无效的牧场ID")
	}

	// 处理Excel数据
	//headers := data[0]
	eventEnterList := make([]*pasturePb.EventEnterRequest, 0)
	for _, row := range data[1:] {
		if len(row) <= 0 {
			continue
		}
		ts := &pasturePb.EventEnterRequest{
			OperationId:   33,
			OperationName: "kpt_admin",
			MessengerId:   33,
			MessengerName: "kpt_admin",
		}
		for j, d := range row {
			switch j {
			case 0:
				continue
			case 1:
				continue
			case 2:
				ts.EarNumber = d
			case 3:
				if pn, ok := penMap[d]; ok {
					ts.PenId = pn.Id
					ts.PenName = d
				}
			case 4:
				ts.Sex = pasturePb.Genders_Female
				if d == "公" {
					ts.Sex = pasturePb.Genders_Male
				}
			case 5:
				if d == "成母牛" {
					ts.CowType = pasturePb.CowType_Breeding_Calf
				} else if d == "犊牛" {
					ts.CowType = pasturePb.CowType_Lactating_Calf
				} else if d == "青年牛" {
					ts.CowType = pasturePb.CowType_Youth_Calf
				} else if d == "育成牛" {
					ts.CowType = pasturePb.CowType_Reserve_Calf
				}
			case 6:
				continue
			case 7:
				bat, _ := util.TimeParseLocal(model.LayoutTime2, d)
				if !bat.IsZero() {
					ts.BirthAt = int32(bat.Local().Unix())
				}
			case 8:
				lact, _ := strconv.Atoi(d)
				ts.Lact = int32(lact)
			case 9:
				eat, _ := util.TimeParseLocal(model.LayoutTime2, d)
				if !eat.IsZero() {
					ts.EnterAt = int32(eat.Local().Unix())
				}
			case 10:
				ts.FatherNumber = d
			case 11:
				ts.MotherNumber = d
			case 12:
				mat, _ := util.TimeParseLocal(model.LayoutTime2, d)
				if !mat.IsZero() {
					ts.MatingAt = int32(mat.Local().Unix())
				}
			case 13:
				continue
			case 14:
				continue
			case 15:
				mt, _ := strconv.Atoi(d)
				ts.MatingTimes = int32(mt)
			case 16:
				continue
			case 17:
				pat, _ := util.TimeParseLocal(model.LayoutTime2, d)
				if !pat.IsZero() {
					ts.PregnancyCheckAt = int32(pat.Local().Unix())
				}
			case 18:
				continue
			case 19:
				continue
			case 20:
				cat, _ := util.TimeParseLocal(model.LayoutTime2, d)
				if !cat.IsZero() {
					ts.CalvingAt = int32(cat.Local().Unix())
				}
			case 21:
				continue
			case 22:
				continue
			case 23:
				continue
			case 24:
				ts.CowKind = pasturePb.CowKind_AGSN
			}
		}
		eventEnterList = append(eventEnterList, ts)
	}

	if len(eventEnterList) <= 0 {
		return nil
	}

	pastureId := userModel.AppPasture.Id
	if pastureId != 4 {
		pastureId = 4
	}

	return s.ExecExcelData(ctx, pastureId, eventEnterList)
}

func (s *StoreEntry) ImportExcel2(ctx context.Context, data [][]string, excelHeader []string, headOfField []string) error {
	// 获取当前用户信息
	userModel, err := s.GetUserModel(ctx)
	if err != nil {
		return err
	}
	// excelHeader ["序号","牛号","性别","出生日期","栏舍名称","牛只类型","牛只品种","牛只来源","入场日期","当前体重(KG)","用途","胎次","配种日期","孕检日期","孕检结果","产犊日期","流产日期","是否禁配",

	pastureId := userModel.AppPasture.Id
	if userModel.AppPasture.Id <= 0 {
		return xerr.Custom("无效的牧场ID")
	}
	penMap := s.PenMap2(ctx, pastureId)
	if len(penMap) <= 0 {
		return xerr.Custom("当前牧场没有配置栏舍信息")
	}

	cowTypeMap := s.CowTypeMap2(userModel)
	cowKindMap := s.CowKindMap2(userModel)
	cowSourceMap := s.CowSourceMap2(userModel)
	purposeMap := s.PurposeMap2(userModel)

	// 处理Excel数据
	eventEnterList := make([]*pasturePb.EventEnterRequest, 0)
	// 遍历数据行（跳过表头）
	for _, rows := range data[1:] {
		if len(rows) == 0 { //  [1 HS10034 母 08-12-22 东区1号 成母牛 西门塔尔牛 产犊 09-12-23 500 繁育 1 12-05-24 01-15-25 有胎   否 2  789]
			continue
		}

		ts := &pasturePb.EventEnterRequest{
			OperationId:   int32(userModel.SystemUser.Id),
			OperationName: userModel.SystemUser.Name,
			MessengerId:   int32(userModel.SystemUser.Id),
			MessengerName: userModel.SystemUser.Name,
		}

		for j, _ := range rows {
			if j == 0 {
				continue
			}
			value := rows[j]
			switch j {
			case 1:
				ts.EarNumber = value
			case 2:
				ts.Sex = pasturePb.Genders_Female
				if value == "公" {
					ts.Sex = pasturePb.Genders_Male
				}
			case 3:
				bat, _ := util.TimeParseLocal(model.LayoutTime2, value+" 00:00:00")
				if !bat.IsZero() {
					ts.BirthAt = int32(bat.Unix())
				} else {
					ts.BirthAt = 0
				}
			case 4:
				ts.PenName = value
			case 5:
				ts.CowTypeName = value
			case 6:
				ts.CowKindName = value
			case 7:
				if cs, ok := cowSourceMap[value]; ok {
					ts.CowSource = cs
				}
			case 8:
				bat, _ := util.TimeParseLocal(model.LayoutTime2, value+" 00:00:00")
				if !bat.IsZero() {
					ts.EnterAt = int32(bat.Unix())
				} else {
					ts.EnterAt = 0
				}
			case 9:
				ct, _ := strconv.ParseFloat(value, 64)
				ts.Weight = float32(ct)
			case 10:
				ts.PurposeName = value
			case 11:
				lact, _ := strconv.Atoi(value)
				ts.Lact = int32(lact)
			case 12:
				bat, _ := util.TimeParseLocal(model.LayoutTime2, value+" 00:00:00")
				if !bat.IsZero() {
					ts.MatingAt = int32(bat.Unix())
				} else {
					ts.MatingAt = 0
				}
			case 13:
				bat, _ := util.TimeParseLocal(model.LayoutTime2, value+" 00:00:00")
				if !bat.IsZero() {
					ts.PregnancyCheckAt = int32(bat.Unix())
				} else {
					ts.PregnancyCheckAt = 0
				}
			case 14:
				ts.PregnantCheckResult = pasturePb.PregnantCheckResult_Pregnant
				if value == "无胎" {
					ts.PregnantCheckResult = pasturePb.PregnantCheckResult_UnPregnant
				}
			case 15:
				bat, _ := util.TimeParseLocal(model.LayoutTime2, value+" 00:00:00")
				if !bat.IsZero() {
					ts.CalvingAt = int32(bat.Unix())
				} else {
					ts.CalvingAt = 0
				}
			case 16:
				bat, _ := util.TimeParseLocal(model.LayoutTime2, value+" 00:00:00")
				if !bat.IsZero() {
					ts.AbortionAt = int32(bat.Unix())
				} else {
					ts.AbortionAt = 0
				}
			case 17:
				if value == "是" {
					ts.BreedStatus = pasturePb.BreedStatus_No_Mating
				}
			case 18:
				mts, _ := strconv.Atoi(value)
				ts.MatingTimes = int32(mts)

			case 19:
				ts.BullNumber = value
			case 20:
				ts.NeckRingNumber = value
			case 21:
				ts.MotherNumber = value
			case 22:
				ts.FatherNumber = value
			case 23:
				ts.BatchNumber = value
			case 24:
				price, _ := strconv.ParseFloat(value, 64)
				ts.Price = float32(price)
			}
		}

		// 特殊处理 PenId 和 PenName（依赖 penMap）
		if ts.PenName != "" {
			if pen, ok := penMap[ts.PenName]; ok {
				ts.PenId = pen.Id
			}
		}

		if ts.CowTypeName != "" {
			if ct, ok := cowTypeMap[ts.CowTypeName]; ok {
				ts.CowType = ct
			}
		}

		if ts.CowKindName != "" {
			if ck, ok := cowKindMap[ts.CowKindName]; ok {
				ts.CowKind = ck
			}
		}

		if ts.PurposeName != "" {
			if p, ok := purposeMap[ts.PurposeName]; ok {
				ts.PurposeKind = p
			}
		}

		eventEnterList = append(eventEnterList, ts)
	}
	if len(eventEnterList) <= 0 {
		return nil
	}
	return s.ExecExcelData(ctx, pastureId, eventEnterList)
}

func (s *StoreEntry) ExecExcelData(ctx context.Context, pastureId int64, dataList []*pasturePb.EventEnterRequest) error {
	if len(dataList) <= 0 {
		return nil
	}

	return s.DB.Transaction(func(tx *gorm.DB) error {
		for _, data := range dataList {
			var count int64
			if err := tx.Model(new(model.Cow)).
				Where("pasture_id = ?", pastureId).
				Where("ear_number = ?", data.EarNumber).
				Count(&count).Error; err != nil {
				return err
			}
			if count > 0 {
				continue
			}

			if err := s.CreateEnter(ctx, data); err != nil {
				return err
			}
		}
		return nil
	})
}